bitstreamfeeder.c 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258
  1. // SPDX-License-Identifier: LGPL-2.1 OR BSD-3-Clause
  2. /*
  3. * Copyright (c) 2018, Chips&Media
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions are met:
  8. *
  9. * 1. Redistributions of source code must retain the above copyright notice, this
  10. * list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright notice,
  12. * this list of conditions and the following disclaimer in the documentation
  13. * and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  16. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  17. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18. * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
  19. * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  20. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  21. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  22. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  24. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25. */
  26. #include "jpuapi.h"
  27. #include "main_helper.h"
  28. typedef struct {
  29. FeedingMethod method;
  30. Uint8* remainData;
  31. Uint32 remainDataSize;
  32. Uint32 remainOffset;
  33. void* actualFeeder;
  34. Uint32 room;
  35. BOOL eos;
  36. EndianMode endian;
  37. } BitstreamFeeder;
  38. extern void* BSFeederFixedSize_Create(const char* path);
  39. extern BOOL BSFeederFixedSize_Destroy(void* feeder);
  40. extern Int32 BSFeederFixedSize_Act(void* feeder, BSChunk* chunk);
  41. extern void BSFeederFixedSize_SetFeedingSize(void* feeder, Uint32 feedingSize);
  42. extern void* BSFeederFrameSize_Create(const char* path);
  43. extern BOOL BSFeederFrameSize_Destroy(void* feeder);
  44. extern Int32 BSFeederFrameSize_Act(void* feeder, BSChunk* packet);
  45. extern void* BSFeederBuffer_Create(const char* path);
  46. extern BOOL BSFeederBuffer_Destroy(void* feeder);
  47. extern Int32 BSFeederBuffer_Act(void* feeder,BSChunk* chunk);
  48. extern BOOL BSFeederBuffer_Rewind(void* feeder);
  49. extern void BSFeederBuffer_SetFeedingSize(void* feeder,Uint32 feedingSize);
  50. /**
  51. * Abstract Bitstream Feeader Functions
  52. */
  53. BSFeeder BitstreamFeeder_Create(const char* path, FeedingMethod method, EndianMode endian)
  54. {
  55. BitstreamFeeder* handle = NULL;
  56. void* feeder = NULL;
  57. switch (method) {
  58. case FEEDING_METHOD_FIXED_SIZE:
  59. feeder = BSFeederFixedSize_Create(path);
  60. break;
  61. #ifdef SUPPORT_FFMPEG
  62. case FEEDING_METHOD_FRAME_SIZE:
  63. feeder = BSFeederFrameSize_Create(path);
  64. break;
  65. #endif
  66. #ifdef USE_FEEDING_METHOD_BUFFER
  67. case FEEDING_METHOD_BUFFER:
  68. feeder = BSFeederBuffer_Create(path);
  69. break;
  70. #endif
  71. default:
  72. feeder = NULL;
  73. break;
  74. }
  75. if (feeder != NULL) {
  76. if ((handle=(BitstreamFeeder*)malloc(sizeof(BitstreamFeeder))) == NULL) {
  77. JLOG(ERR, "%s:%d Failed to allocate memory\n", __FUNCTION__, __LINE__);
  78. return NULL;
  79. }
  80. handle->actualFeeder = feeder;
  81. handle->method = method;
  82. handle->remainData = NULL;
  83. handle->remainDataSize = 0;
  84. handle->eos = FALSE;
  85. handle->endian = endian;
  86. }
  87. return (BSFeeder)handle;
  88. }
  89. Uint32 BitstreamFeeder_SetData(BSFeeder feeder, void *data, Uint32 size)
  90. {
  91. BitstreamFeeder* bsf = (BitstreamFeeder*)feeder;
  92. bsf->remainData = data;
  93. bsf->remainDataSize = size;
  94. return 0;
  95. }
  96. Uint32 BitstreamFeeder_Act(BSFeeder feeder, JpgDecHandle handle, jpu_buffer_t* bsBuffer)
  97. {
  98. BitstreamFeeder* bsf = (BitstreamFeeder*)feeder;
  99. Int32 feedingSize = 0;
  100. int room;
  101. BSChunk chunk = {0};
  102. EndianMode endian;
  103. PhysicalAddress wrPtr;
  104. if (bsf == NULL) {
  105. JLOG(ERR, "%s:%d Null handle\n", __FUNCTION__, __LINE__);
  106. return 0;
  107. }
  108. JPU_DecGetBitstreamBuffer(handle, NULL/* rdPtr */, &wrPtr, &room);
  109. JLOG(INFO, "wrptr address = %x\r\n", wrPtr);
  110. endian = bsf->endian;
  111. JLOG(INFO, "endian = %d\r\n", endian);
  112. if (bsf->remainData == NULL) {
  113. chunk.size = bsBuffer->size;
  114. chunk.data = malloc(chunk.size);
  115. chunk.eos = FALSE;
  116. if (chunk.data == NULL) {
  117. JLOG(ERR, "%s:%d failed to allocate memory\n", __FUNCTION__, __LINE__);
  118. return 0;
  119. }
  120. switch (bsf->method) {
  121. case FEEDING_METHOD_FIXED_SIZE:
  122. feedingSize = BSFeederFixedSize_Act(bsf->actualFeeder, &chunk);
  123. break;
  124. #ifdef SUPPORT_FFMPEG
  125. case FEEDING_METHOD_FRAME_SIZE:
  126. feedingSize = BSFeederFrameSize_Act(bsf->actualFeeder, &chunk);
  127. break;
  128. #endif
  129. #ifdef USE_FEEDING_METHOD_BUFFER
  130. case FEEDING_METHOD_BUFFER:
  131. feedingSize = BSFeederBuffer_Act(bsf->actualFeeder, &chunk);
  132. break;
  133. #endif
  134. default:
  135. JLOG(ERR, "%s:%d Invalid method(%d)\n", __FUNCTION__, __LINE__, bsf->method);
  136. free(chunk.data);
  137. return 0;
  138. }
  139. }
  140. else {
  141. JLOG(INFO, "Get data from remain\r\n");
  142. chunk.data = bsf->remainData;
  143. feedingSize = bsf->remainDataSize;
  144. }
  145. if (feedingSize < 0) {
  146. JLOG(ERR, "feeding size is negative value: %d\n", feedingSize);
  147. free(chunk.data);
  148. return 0;
  149. }
  150. if (feedingSize > 0) {
  151. Uint32 rightSize=0, leftSize=0;
  152. PhysicalAddress base = bsBuffer->phys_addr;
  153. Uint32 size = bsBuffer->size;
  154. Uint32 wSize = feedingSize;
  155. Uint8* ptr = chunk.data;;
  156. if ((Int32)room < feedingSize) {
  157. wSize = room;
  158. ptr += bsf->remainOffset;
  159. bsf->remainData = chunk.data;
  160. bsf->remainDataSize = feedingSize - wSize;
  161. bsf->remainOffset += wSize;
  162. }
  163. else {
  164. bsf->remainData = NULL;
  165. bsf->remainDataSize = 0;
  166. bsf->remainOffset = 0;
  167. }
  168. leftSize = wSize;
  169. if ((wrPtr+wSize) >= (base+size)) {
  170. PhysicalAddress endAddr = base+size;
  171. rightSize = endAddr-wrPtr;
  172. leftSize = (wrPtr+wSize) - endAddr;
  173. if (rightSize > 0) {
  174. JLOG(INFO, "1 wrptr address = %x, addr = %p, size = %x\r\n", wrPtr, ptr, rightSize);
  175. JpuWriteMem(wrPtr, ptr, rightSize, (int)endian);
  176. }
  177. wrPtr = base;
  178. }
  179. JLOG(INFO, "2 wrptr address = %x, ptr = %p, rightsize = %x, size = %x\r\n", wrPtr, ptr, rightSize, leftSize);
  180. JpuWriteMem(wrPtr, ptr+rightSize, leftSize, (int)endian);
  181. JPU_DecUpdateBitstreamBuffer(handle, wSize);
  182. }
  183. if (TRUE == chunk.eos) {
  184. JPU_DecUpdateBitstreamBuffer(handle, 0);
  185. }
  186. JLOG(INFO, "bsf->remainData = %p, chunk.data = %p\r\n", bsf->remainData, chunk.data);
  187. bsf->eos = chunk.eos;
  188. // if (NULL == bsf->remainData)
  189. // free(chunk.data);
  190. JLOG(INFO, "%s out\r\n", __FUNCTION__);
  191. return feedingSize;
  192. }
  193. BOOL BitstreamFeeder_IsEos(BSFeeder feeder)
  194. {
  195. BitstreamFeeder* bsf = (BitstreamFeeder*)feeder;
  196. if (bsf == NULL) {
  197. JLOG(ERR, "%s:%d Null handle\n", __FUNCTION__, __LINE__);
  198. return FALSE;
  199. }
  200. return bsf->eos;
  201. }
  202. BOOL BitstreamFeeder_Destroy(BSFeeder feeder)
  203. {
  204. BitstreamFeeder* bsf = (BitstreamFeeder*)feeder;
  205. if (bsf == NULL) {
  206. return FALSE;
  207. }
  208. switch (bsf->method) {
  209. case FEEDING_METHOD_FIXED_SIZE:
  210. BSFeederFixedSize_Destroy(bsf->actualFeeder);
  211. break;
  212. #ifdef SUPPORT_FFMPEG
  213. case FEEDING_METHOD_FRAME_SIZE:
  214. BSFeederFrameSize_Destroy(bsf->actualFeeder);
  215. break;
  216. #endif
  217. #ifdef USE_FEEDING_METHOD_BUFFER
  218. case FEEDING_METHOD_BUFFER:
  219. BSFeederBuffer_Destroy(bsf->actualFeeder);
  220. break;
  221. #endif
  222. default:
  223. JLOG(ERR, "%s:%d Invalid method(%d)\n", __FUNCTION__, __LINE__, bsf->method);
  224. break;
  225. }
  226. // if (bsf->remainData) {
  227. // free(bsf->remainData);
  228. // }
  229. free(bsf);
  230. return TRUE;
  231. }