vpuhelper.c 75 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246
  1. //------------------------------------------------------------------------------
  2. // File: $Id$
  3. //
  4. // Copyright (c) 2006, Chips & Media. All rights reserved.
  5. //------------------------------------------------------------------------------
  6. #include <stdio.h>
  7. #include <string.h>
  8. #include "main_helper.h"
  9. #ifdef PLATFORM_WIN32
  10. #pragma warning(disable : 4996) //!<< disable waring C4996: The POSIX name for this item is deprecated.
  11. #endif
  12. #ifndef min
  13. #define min(a,b) (((a) < (b)) ? (a) : (b))
  14. #endif
  15. /*******************************************************************************
  16. * REPORT *
  17. *******************************************************************************/
  18. #define USER_DATA_INFO_OFFSET (8*17)
  19. #define FN_PIC_INFO "dec_pic_disp_info.log"
  20. #define FN_SEQ_INFO "dec_seq_disp_info.log"
  21. #define FN_PIC_TYPE "dec_pic_type.log"
  22. #define FN_USER_DATA "dec_user_data.log"
  23. #define FN_SEQ_USER_DATA "dec_seq_user_data.log"
  24. // VC1 specific
  25. enum {
  26. BDU_SEQUENCE_END = 0x0A,
  27. BDU_SLICE = 0x0B,
  28. BDU_FIELD = 0x0C,
  29. BDU_FRAME = 0x0D,
  30. BDU_ENTRYPOINT_HEADER = 0x0E,
  31. BDU_SEQUENCE_HEADER = 0x0F,
  32. BDU_SLICE_LEVEL_USER_DATA = 0x1B,
  33. BDU_FIELD_LEVEL_USER_DATA = 0x1C,
  34. BDU_FRAME_LEVEL_USER_DATA = 0x1D,
  35. BDU_ENTRYPOINT_LEVEL_USER_DATA = 0x1E,
  36. BDU_SEQUENCE_LEVEL_USER_DATA = 0x1F
  37. };
  38. // AVC specific - SEI
  39. enum {
  40. SEI_REGISTERED_ITUTT35_USERDATA = 0x04,
  41. SEI_UNREGISTERED_USERDATA = 0x05,
  42. SEI_MVC_SCALABLE_NESTING = 0x25
  43. };
  44. static vpu_rpt_info_t s_rpt_info[MAX_VPU_CORE_NUM];
  45. void OpenDecReport(
  46. Uint32 core_idx,
  47. VpuReportConfig_t* cfg
  48. )
  49. {
  50. vpu_rpt_info_t *rpt = &s_rpt_info[core_idx];
  51. rpt->fpPicDispInfoLogfile = NULL;
  52. rpt->fpPicTypeLogfile = NULL;
  53. rpt->fpSeqDispInfoLogfile = NULL;
  54. rpt->fpUserDataLogfile = NULL;
  55. rpt->fpSeqUserDataLogfile = NULL;
  56. rpt->decIndex = 0;
  57. rpt->userDataEnable = cfg->userDataEnable;
  58. rpt->userDataReportMode = cfg->userDataReportMode;
  59. rpt->reportOpened = TRUE;
  60. return;
  61. }
  62. void CloseDecReport(
  63. Uint32 core_idx
  64. )
  65. {
  66. vpu_rpt_info_t *rpt = &s_rpt_info[core_idx];
  67. if (rpt->reportOpened == FALSE) {
  68. return;
  69. }
  70. if (rpt->fpPicDispInfoLogfile) {
  71. osal_fclose(rpt->fpPicDispInfoLogfile);
  72. rpt->fpPicDispInfoLogfile = NULL;
  73. }
  74. if (rpt->fpPicTypeLogfile) {
  75. osal_fclose(rpt->fpPicTypeLogfile);
  76. rpt->fpPicTypeLogfile = NULL;
  77. }
  78. if (rpt->fpSeqDispInfoLogfile) {
  79. osal_fclose(rpt->fpSeqDispInfoLogfile);
  80. rpt->fpSeqDispInfoLogfile = NULL;
  81. }
  82. if (rpt->fpUserDataLogfile) {
  83. osal_fclose(rpt->fpUserDataLogfile);
  84. rpt->fpUserDataLogfile= NULL;
  85. }
  86. if (rpt->fpSeqUserDataLogfile) {
  87. osal_fclose(rpt->fpSeqUserDataLogfile);
  88. rpt->fpSeqUserDataLogfile = NULL;
  89. }
  90. if (rpt->vb_rpt.base) {
  91. vdi_free_dma_memory(core_idx, &rpt->vb_rpt);
  92. }
  93. rpt->decIndex = 0;
  94. return;
  95. }
  96. static void SaveUserData(
  97. Uint32 core_idx,
  98. BYTE* userDataBuf
  99. )
  100. {
  101. vpu_rpt_info_t *rpt = &s_rpt_info[core_idx];
  102. Uint32 i;
  103. Uint32 UserDataType;
  104. Uint32 UserDataSize;
  105. Uint32 userDataNum;
  106. Uint32 TotalSize;
  107. BYTE* tmpBuf;
  108. if (rpt->reportOpened == FALSE) {
  109. return;
  110. }
  111. if(rpt->fpUserDataLogfile == 0) {
  112. rpt->fpUserDataLogfile = osal_fopen(FN_USER_DATA, "w+");
  113. }
  114. tmpBuf = userDataBuf;
  115. userDataNum = (short)((tmpBuf[0]<<8) | (tmpBuf[1]<<0));
  116. TotalSize = (short)((tmpBuf[2]<<8) | (tmpBuf[3]<<0));
  117. tmpBuf = userDataBuf + 8;
  118. for(i=0; i<userDataNum; i++) {
  119. UserDataType = (short)((tmpBuf[0]<<8) | (tmpBuf[1]<<0));
  120. UserDataSize = (short)((tmpBuf[2]<<8) | (tmpBuf[3]<<0));
  121. osal_fprintf(rpt->fpUserDataLogfile, "\n[Idx Type Size] : [%4d %4d %4d]",i, UserDataType, UserDataSize);
  122. tmpBuf += 8;
  123. }
  124. osal_fprintf(rpt->fpUserDataLogfile, "\n");
  125. tmpBuf = userDataBuf + USER_DATA_INFO_OFFSET;
  126. for(i=0; i<TotalSize; i++) {
  127. osal_fprintf(rpt->fpUserDataLogfile, "%02x", tmpBuf[i]);
  128. if ((i&7) == 7) {
  129. osal_fprintf(rpt->fpUserDataLogfile, "\n");
  130. }
  131. }
  132. osal_fprintf(rpt->fpUserDataLogfile, "\n");
  133. osal_fflush(rpt->fpUserDataLogfile);
  134. }
  135. static void SaveUserDataINT(
  136. Uint32 core_idx,
  137. BYTE* userDataBuf,
  138. Int32 size,
  139. Int32 intIssued,
  140. Int32 decIdx,
  141. CodStd bitstreamFormat
  142. )
  143. {
  144. vpu_rpt_info_t *rpt = &s_rpt_info[core_idx];
  145. Int32 i;
  146. Int32 UserDataType = 0;
  147. Int32 UserDataSize = 0;
  148. Int32 userDataNum = 0;
  149. Int32 TotalSize;
  150. BYTE* tmpBuf;
  151. BYTE* backupBufTmp;
  152. static Int32 backupSize = 0;
  153. static BYTE* backupBuf = NULL;
  154. if (rpt->reportOpened == FALSE) {
  155. return;
  156. }
  157. if(rpt->fpUserDataLogfile == NULL) {
  158. rpt->fpUserDataLogfile = osal_fopen(FN_USER_DATA, "w+");
  159. }
  160. backupBufTmp = (BYTE *)osal_malloc(backupSize + size);
  161. if (backupBufTmp == 0) {
  162. VLOG( ERR, "Can't mem allock\n");
  163. return;
  164. }
  165. for (i=0; i<backupSize; i++) {
  166. backupBufTmp[i] = backupBuf[i];
  167. }
  168. if (backupBuf != NULL) {
  169. osal_free(backupBuf);
  170. }
  171. backupBuf = backupBufTmp;
  172. tmpBuf = userDataBuf + USER_DATA_INFO_OFFSET;
  173. size -= USER_DATA_INFO_OFFSET;
  174. for(i=0; i<size; i++) {
  175. backupBuf[backupSize + i] = tmpBuf[i];
  176. }
  177. backupSize += size;
  178. if (intIssued) {
  179. return;
  180. }
  181. tmpBuf = userDataBuf;
  182. userDataNum = (short)((tmpBuf[0]<<8) | (tmpBuf[1]<<0));
  183. if(userDataNum == 0) {
  184. return;
  185. }
  186. tmpBuf = userDataBuf + 8;
  187. UserDataSize = (short)((tmpBuf[2]<<8) | (tmpBuf[3]<<0));
  188. UserDataSize = (UserDataSize+7)/8*8;
  189. osal_fprintf(rpt->fpUserDataLogfile, "FRAME [%1d]\n", decIdx);
  190. for(i=0; i<backupSize; i++) {
  191. osal_fprintf(rpt->fpUserDataLogfile, "%02x", backupBuf[i]);
  192. if ((i&7) == 7) {
  193. osal_fprintf(rpt->fpUserDataLogfile, "\n");
  194. }
  195. if( (i%8==7) && (i==UserDataSize-1) && (UserDataSize != backupSize)) {
  196. osal_fprintf(rpt->fpUserDataLogfile, "\n");
  197. tmpBuf+=8;
  198. UserDataSize += (short)((tmpBuf[2]<<8) | (tmpBuf[3]<<0));
  199. UserDataSize = (UserDataSize+7)/8*8;
  200. }
  201. }
  202. if (backupSize > 0) {
  203. osal_fprintf(rpt->fpUserDataLogfile, "\n");
  204. }
  205. tmpBuf = userDataBuf;
  206. userDataNum = (short)((tmpBuf[0]<<8) | (tmpBuf[1]<<0));
  207. TotalSize = (short)((tmpBuf[2]<<8) | (tmpBuf[3]<<0));
  208. osal_fprintf(rpt->fpUserDataLogfile, "User Data Num: [%d]\n", userDataNum);
  209. osal_fprintf(rpt->fpUserDataLogfile, "User Data Total Size: [%d]\n", TotalSize);
  210. tmpBuf = userDataBuf + 8;
  211. for(i=0; i<userDataNum; i++) {
  212. UserDataType = (short)((tmpBuf[0]<<8) | (tmpBuf[1]<<0));
  213. UserDataSize = (short)((tmpBuf[2]<<8) | (tmpBuf[3]<<0));
  214. if(bitstreamFormat == STD_VC1) {
  215. switch (UserDataType) {
  216. case BDU_SLICE_LEVEL_USER_DATA:
  217. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "BDU_SLICE_LEVEL_USER_DATA");
  218. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  219. break;
  220. case BDU_FIELD_LEVEL_USER_DATA:
  221. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "BDU_FIELD_LEVEL_USER_DATA");
  222. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  223. break;
  224. case BDU_FRAME_LEVEL_USER_DATA:
  225. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "BDU_FRAME_LEVEL_USER_DATA");
  226. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  227. break;
  228. case BDU_ENTRYPOINT_LEVEL_USER_DATA:
  229. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "BDU_ENTRYPOINT_LEVEL_USER_DATA");
  230. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  231. break;
  232. case BDU_SEQUENCE_LEVEL_USER_DATA:
  233. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "BDU_SEQUENCE_LEVEL_USER_DATA");
  234. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  235. break;
  236. }
  237. }
  238. else if(bitstreamFormat == STD_AVC) {
  239. switch (UserDataType) {
  240. case SEI_REGISTERED_ITUTT35_USERDATA:
  241. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "registered_itu_t_t35");
  242. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  243. break;
  244. case SEI_UNREGISTERED_USERDATA:
  245. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "unregistered");
  246. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  247. break;
  248. case SEI_MVC_SCALABLE_NESTING:
  249. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "mvc_scalable_nesting");
  250. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  251. break;
  252. }
  253. }
  254. else if(bitstreamFormat == STD_MPEG2) {
  255. switch (UserDataType) {
  256. case 0:
  257. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Seq]\n", i);
  258. break;
  259. case 1:
  260. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Gop]\n", i);
  261. break;
  262. case 2:
  263. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Pic]\n", i);
  264. break;
  265. default:
  266. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Error]\n", i);
  267. break;
  268. }
  269. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  270. }
  271. else if(bitstreamFormat == STD_AVS) {
  272. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:%s]\n", i, "User Data");
  273. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  274. }
  275. else {
  276. switch (UserDataType) {
  277. case 0:
  278. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Vos]\n", i);
  279. break;
  280. case 1:
  281. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Vis]\n", i);
  282. break;
  283. case 2:
  284. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Vol]\n", i);
  285. break;
  286. case 3:
  287. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Gov]\n", i);
  288. break;
  289. default:
  290. osal_fprintf(rpt->fpUserDataLogfile, "User Data Type: [%d:Error]\n", i);
  291. break;
  292. }
  293. osal_fprintf(rpt->fpUserDataLogfile, "User Data Size: [%d]\n", UserDataSize);
  294. }
  295. tmpBuf += 8;
  296. }
  297. osal_fprintf(rpt->fpUserDataLogfile, "\n");
  298. osal_fflush(rpt->fpUserDataLogfile);
  299. backupSize = 0;
  300. if (backupBuf != NULL) {
  301. osal_free(backupBuf);
  302. }
  303. backupBuf = 0;
  304. }
  305. void CheckUserDataInterrupt(
  306. Uint32 core_idx,
  307. DecHandle handle,
  308. Int32 decodeIdx,
  309. CodStd bitstreamFormat,
  310. Int32 int_reason
  311. )
  312. {
  313. vpu_rpt_info_t *rpt = &s_rpt_info[core_idx];
  314. if (int_reason & (1<<INT_BIT_USERDATA)) {
  315. // USER DATA INTERRUPT Issued
  316. // User Data save
  317. if (rpt->userDataEnable == TRUE) {
  318. int size;
  319. BYTE *userDataBuf;
  320. size = rpt->vb_rpt.size + USER_DATA_INFO_OFFSET;
  321. userDataBuf = osal_malloc(size);
  322. osal_memset(userDataBuf, 0, size);
  323. vdi_read_memory(core_idx, rpt->vb_rpt.phys_addr, userDataBuf, size, VDI_BIG_ENDIAN);
  324. if (decodeIdx >= 0)
  325. SaveUserDataINT(core_idx, userDataBuf, size, 1, rpt->decIndex, bitstreamFormat);
  326. osal_free(userDataBuf);
  327. } else {
  328. VLOG(ERR, "Unexpected Interrupt issued");
  329. }
  330. }
  331. }
  332. void ConfigDecReport(
  333. Uint32 core_idx,
  334. DecHandle handle,
  335. CodStd bitstreamFormat
  336. )
  337. {
  338. vpu_rpt_info_t *rpt = &s_rpt_info[core_idx];
  339. if (rpt->reportOpened == FALSE) {
  340. return;
  341. }
  342. // Report Information
  343. if (!rpt->vb_rpt.base) {
  344. rpt->vb_rpt.size = SIZE_REPORT_BUF;
  345. if (vdi_allocate_dma_memory(core_idx, &rpt->vb_rpt) < 0) {
  346. VLOG(ERR, "fail to allocate report buffer\n" );
  347. return;
  348. }
  349. }
  350. VPU_DecGiveCommand(handle, SET_ADDR_REP_USERDATA, &rpt->vb_rpt.phys_addr );
  351. VPU_DecGiveCommand(handle, SET_SIZE_REP_USERDATA, &rpt->vb_rpt.size );
  352. VPU_DecGiveCommand(handle, SET_USERDATA_REPORT_MODE, &rpt->userDataReportMode );
  353. if (rpt->userDataEnable == TRUE) {
  354. VPU_DecGiveCommand( handle, ENABLE_REP_USERDATA, 0 );
  355. }
  356. else {
  357. VPU_DecGiveCommand( handle, DISABLE_REP_USERDATA, 0 );
  358. }
  359. }
  360. void SaveDecReport(
  361. Uint32 core_idx,
  362. DecHandle handle,
  363. DecOutputInfo* pDecInfo,
  364. CodStd bitstreamFormat,
  365. Uint32 mbNumX,
  366. Uint32 mbNumY
  367. )
  368. {
  369. vpu_rpt_info_t *rpt = &s_rpt_info[core_idx];
  370. if (rpt->reportOpened == FALSE) {
  371. return ;
  372. }
  373. // Report Information
  374. // User Data
  375. if ((pDecInfo->indexFrameDecoded >= 0 || (bitstreamFormat == STD_VC1)) &&
  376. rpt->userDataEnable == TRUE &&
  377. pDecInfo->decOutputExtData.userDataSize > 0) {
  378. // Vc1 Frame user data follow picture. After last frame decoding, user data should be reported.
  379. Uint32 size = 0;
  380. BYTE* userDataBuf = NULL;
  381. if (pDecInfo->decOutputExtData.userDataBufFull == TRUE) {
  382. VLOG(ERR, "User Data Buffer is Full\n");
  383. }
  384. size = (pDecInfo->decOutputExtData.userDataSize+7)/8*8 + USER_DATA_INFO_OFFSET;
  385. userDataBuf = (BYTE*)osal_malloc(size);
  386. osal_memset(userDataBuf, 0, size);
  387. vdi_read_memory(core_idx, rpt->vb_rpt.phys_addr, userDataBuf, size, HOST_ENDIAN);
  388. if (pDecInfo->indexFrameDecoded >= 0) {
  389. SaveUserData(core_idx, userDataBuf);
  390. }
  391. osal_free(userDataBuf);
  392. }
  393. if (((pDecInfo->indexFrameDecoded >= 0 || (bitstreamFormat == STD_VC1 )) && rpt->userDataEnable) || // Vc1 Frame user data follow picture. After last frame decoding, user data should be reported.
  394. (pDecInfo->indexFrameDisplay >= 0 && rpt->userDataEnable) ) {
  395. Uint32 size = 0;
  396. Uint32 dataSize = 0;
  397. BYTE* userDataBuf = NULL;
  398. if (pDecInfo->decOutputExtData.userDataBufFull) {
  399. VLOG(ERR, "User Data Buffer is Full\n");
  400. }
  401. dataSize = pDecInfo->decOutputExtData.userDataSize % rpt->vb_rpt.size;
  402. if (dataSize == 0 && pDecInfo->decOutputExtData.userDataSize != 0) {
  403. dataSize = rpt->vb_rpt.size;
  404. }
  405. size = (dataSize+7)/8*8 + USER_DATA_INFO_OFFSET;
  406. userDataBuf = (BYTE*)osal_malloc(size);
  407. osal_memset(userDataBuf, 0, size);
  408. vdi_read_memory(core_idx, rpt->vb_rpt.phys_addr, userDataBuf, size, HOST_ENDIAN);
  409. if (pDecInfo->indexFrameDecoded >= 0 || (bitstreamFormat == STD_VC1)) {
  410. SaveUserDataINT(core_idx, userDataBuf, size, 0, rpt->decIndex, bitstreamFormat);
  411. }
  412. osal_free(userDataBuf);
  413. }
  414. if (pDecInfo->indexFrameDecoded >= 0) {
  415. if (rpt->fpPicTypeLogfile == NULL) {
  416. rpt->fpPicTypeLogfile = osal_fopen(FN_PIC_TYPE, "w+");
  417. }
  418. osal_fprintf(rpt->fpPicTypeLogfile, "FRAME [%1d]\n", rpt->decIndex);
  419. switch (bitstreamFormat) {
  420. case STD_AVC:
  421. if(pDecInfo->pictureStructure == 3) { // FIELD_INTERLACED
  422. osal_fprintf(rpt->fpPicTypeLogfile, "Top Field Type: [%s]\n", pDecInfo->picTypeFirst == 0 ? "I_TYPE" :
  423. (pDecInfo->picTypeFirst) == 1 ? "P_TYPE" :
  424. (pDecInfo->picTypeFirst) == 2 ? "BI_TYPE" :
  425. (pDecInfo->picTypeFirst) == 3 ? "B_TYPE" :
  426. (pDecInfo->picTypeFirst) == 4 ? "SKIP_TYPE" :
  427. (pDecInfo->picTypeFirst) == 5 ? "IDR_TYPE" :
  428. "FORBIDDEN");
  429. osal_fprintf(rpt->fpPicTypeLogfile, "Bottom Field Type: [%s]\n", pDecInfo->picType == 0 ? "I_TYPE" :
  430. (pDecInfo->picType) == 1 ? "P_TYPE" :
  431. (pDecInfo->picType) == 2 ? "BI_TYPE" :
  432. (pDecInfo->picType) == 3 ? "B_TYPE" :
  433. (pDecInfo->picType) == 4 ? "SKIP_TYPE" :
  434. (pDecInfo->picType) == 5 ? "IDR_TYPE" :
  435. "FORBIDDEN");
  436. }
  437. else {
  438. osal_fprintf(rpt->fpPicTypeLogfile, "Picture Type: [%s]\n", pDecInfo->picType == 0 ? "I_TYPE" :
  439. (pDecInfo->picType) == 1 ? "P_TYPE" :
  440. (pDecInfo->picType) == 2 ? "BI_TYPE" :
  441. (pDecInfo->picType) == 3 ? "B_TYPE" :
  442. (pDecInfo->picType) == 4 ? "SKIP_TYPE" :
  443. (pDecInfo->picType) == 5 ? "IDR_TYPE" :
  444. "FORBIDDEN");
  445. }
  446. break;
  447. case STD_MPEG2 :
  448. osal_fprintf(rpt->fpPicTypeLogfile, "Picture Type: [%s]\n", pDecInfo->picType == 0 ? "I_TYPE" :
  449. pDecInfo->picType == 1 ? "P_TYPE" :
  450. pDecInfo->picType == 2 ? "B_TYPE" :
  451. "D_TYPE");
  452. break;
  453. case STD_MPEG4 :
  454. osal_fprintf(rpt->fpPicTypeLogfile, "Picture Type: [%s]\n", pDecInfo->picType == 0 ? "I_TYPE" :
  455. pDecInfo->picType == 1 ? "P_TYPE" :
  456. pDecInfo->picType == 2 ? "B_TYPE" :
  457. "S_TYPE");
  458. break;
  459. case STD_VC1:
  460. if(pDecInfo->pictureStructure == 3) { // FIELD_INTERLACED
  461. osal_fprintf(rpt->fpPicTypeLogfile, "Top Field Type: [%s]\n", pDecInfo->picTypeFirst == 0 ? "I_TYPE" :
  462. (pDecInfo->picTypeFirst) == 1 ? "P_TYPE" :
  463. (pDecInfo->picTypeFirst) == 2 ? "BI_TYPE" :
  464. (pDecInfo->picTypeFirst) == 3 ? "B_TYPE" :
  465. (pDecInfo->picTypeFirst) == 4 ? "SKIP_TYPE" :
  466. "FORBIDDEN");
  467. osal_fprintf(rpt->fpPicTypeLogfile, "Bottom Field Type: [%s]\n", pDecInfo->picType == 0 ? "I_TYPE" :
  468. (pDecInfo->picType) == 1 ? "P_TYPE" :
  469. (pDecInfo->picType) == 2 ? "BI_TYPE" :
  470. (pDecInfo->picType) == 3 ? "B_TYPE" :
  471. (pDecInfo->picType) == 4 ? "SKIP_TYPE" :
  472. "FORBIDDEN");
  473. }
  474. else {
  475. osal_fprintf(rpt->fpPicTypeLogfile, "Picture Type: [%s]\n", pDecInfo->picType == 0 ? "I_TYPE" :
  476. (pDecInfo->picTypeFirst) == 1 ? "P_TYPE" :
  477. (pDecInfo->picTypeFirst) == 2 ? "BI_TYPE" :
  478. (pDecInfo->picTypeFirst) == 3 ? "B_TYPE" :
  479. (pDecInfo->picTypeFirst) == 4 ? "SKIP_TYPE" :
  480. "FORBIDDEN");
  481. }
  482. break;
  483. default:
  484. osal_fprintf(rpt->fpPicTypeLogfile, "Picture Type: [%s]\n", pDecInfo->picType == 0 ? "I_TYPE" :
  485. pDecInfo->picType == 1 ? "P_TYPE" :
  486. "B_TYPE");
  487. break;
  488. }
  489. }
  490. if (pDecInfo->indexFrameDecoded >= 0) {
  491. if (rpt->fpPicDispInfoLogfile == NULL) {
  492. rpt->fpPicDispInfoLogfile = osal_fopen(FN_PIC_INFO, "w+");
  493. }
  494. osal_fprintf(rpt->fpPicDispInfoLogfile, "FRAME [%1d]\n", rpt->decIndex);
  495. switch (bitstreamFormat) {
  496. case STD_MPEG2:
  497. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n",
  498. pDecInfo->picType == 0 ? "I_TYPE" :
  499. pDecInfo->picType == 1 ? "P_TYPE" :
  500. pDecInfo->picType == 2 ? "B_TYPE" :
  501. "D_TYPE");
  502. break;
  503. case STD_MPEG4:
  504. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n",
  505. pDecInfo->picType == 0 ? "I_TYPE" :
  506. pDecInfo->picType == 1 ? "P_TYPE" :
  507. pDecInfo->picType == 2 ? "B_TYPE" :
  508. "S_TYPE");
  509. break;
  510. case STD_VC1 :
  511. if(pDecInfo->pictureStructure == 3) { // FIELD_INTERLACED
  512. osal_fprintf(rpt->fpPicDispInfoLogfile, "Top : %s\n", (pDecInfo->picType>>3) == 0 ? "I_TYPE" :
  513. (pDecInfo->picType>>3) == 1 ? "P_TYPE" :
  514. (pDecInfo->picType>>3) == 2 ? "BI_TYPE" :
  515. (pDecInfo->picType>>3) == 3 ? "B_TYPE" :
  516. (pDecInfo->picType>>3) == 4 ? "SKIP_TYPE" :
  517. "FORBIDDEN");
  518. osal_fprintf(rpt->fpPicDispInfoLogfile, "Bottom : %s\n", (pDecInfo->picType&0x7) == 0 ? "I_TYPE" :
  519. (pDecInfo->picType&0x7) == 1 ? "P_TYPE" :
  520. (pDecInfo->picType&0x7) == 2 ? "BI_TYPE" :
  521. (pDecInfo->picType&0x7) == 3 ? "B_TYPE" :
  522. (pDecInfo->picType&0x7) == 4 ? "SKIP_TYPE" :
  523. "FORBIDDEN");
  524. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Interlaced Picture");
  525. }
  526. else {
  527. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", (pDecInfo->picType>>3) == 0 ? "I_TYPE" :
  528. (pDecInfo->picType>>3) == 1 ? "P_TYPE" :
  529. (pDecInfo->picType>>3) == 2 ? "BI_TYPE" :
  530. (pDecInfo->picType>>3) == 3 ? "B_TYPE" :
  531. (pDecInfo->picType>>3) == 4 ? "SKIP_TYPE" :
  532. "FORBIDDEN");
  533. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Frame Picture");
  534. }
  535. break;
  536. default:
  537. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n",
  538. pDecInfo->picType == 0 ? "I_TYPE" :
  539. pDecInfo->picType == 1 ? "P_TYPE" :
  540. "B_TYPE");
  541. break;
  542. }
  543. if(bitstreamFormat != STD_VC1) {
  544. if (pDecInfo->interlacedFrame) {
  545. if(bitstreamFormat == STD_AVS) {
  546. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Frame Picture");
  547. }
  548. else {
  549. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Interlaced Picture");
  550. }
  551. }
  552. else {
  553. if(bitstreamFormat == STD_AVS) {
  554. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Interlaced Picture");
  555. }
  556. else {
  557. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Frame Picture");
  558. }
  559. }
  560. }
  561. if (bitstreamFormat != STD_RV) {
  562. if(bitstreamFormat == STD_VC1) {
  563. switch(pDecInfo->pictureStructure) {
  564. case 0: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "PROGRESSIVE"); break;
  565. case 2: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "FRAME_INTERLACE"); break;
  566. case 3: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "FIELD_INTERLACE"); break;
  567. default: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "FORBIDDEN"); break;
  568. }
  569. }
  570. else if(bitstreamFormat == STD_AVC) {
  571. if(!pDecInfo->interlacedFrame) {
  572. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "FRAME_PICTURE");
  573. }
  574. else {
  575. if(pDecInfo->topFieldFirst) {
  576. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Top Field First");
  577. }
  578. else {
  579. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Bottom Field First");
  580. }
  581. }
  582. }
  583. else if (bitstreamFormat != STD_MPEG4 && bitstreamFormat != STD_AVS) {
  584. switch (pDecInfo->pictureStructure) {
  585. case 1: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "TOP_FIELD"); break;
  586. case 2: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "BOTTOM_FIELD"); break;
  587. case 3: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "FRAME_PICTURE"); break;
  588. default: osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "FORBIDDEN"); break;
  589. }
  590. }
  591. if(bitstreamFormat != STD_AVC) {
  592. if (pDecInfo->topFieldFirst) {
  593. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Top Field First");
  594. }
  595. else {
  596. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Bottom Field First");
  597. }
  598. if (bitstreamFormat != STD_MPEG4) {
  599. if (pDecInfo->repeatFirstField) {
  600. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Repeat First Field");
  601. }
  602. else {
  603. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Not Repeat First Field");
  604. }
  605. if (bitstreamFormat == STD_VC1) {
  606. osal_fprintf(rpt->fpPicDispInfoLogfile, "VC1 RPTFRM [%1d]\n", pDecInfo->progressiveFrame);
  607. }
  608. else if (pDecInfo->progressiveFrame) {
  609. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Progressive Frame");
  610. }
  611. else {
  612. osal_fprintf(rpt->fpPicDispInfoLogfile, "%s\n", "Interlaced Frame");
  613. }
  614. }
  615. }
  616. }
  617. if (bitstreamFormat == STD_MPEG2) {
  618. osal_fprintf(rpt->fpPicDispInfoLogfile, "Field Sequence [%d]\n\n", pDecInfo->fieldSequence);
  619. }
  620. else {
  621. osal_fprintf(rpt->fpPicDispInfoLogfile, "\n");
  622. }
  623. osal_fflush(rpt->fpPicDispInfoLogfile);
  624. }
  625. if(pDecInfo->indexFrameDecoded >= 0) {
  626. rpt->decIndex ++;
  627. }
  628. return;
  629. }
  630. #define DEFAULT_ENC_OUTPUT_NUM 30
  631. /*******************************************************************************
  632. * FUNCTIONS RELATED TO CPB *
  633. *******************************************************************************/
  634. int FillBsResetBufHelper(Uint32 core_idx,
  635. BYTE *buf,
  636. PhysicalAddress paBsBufAddr,
  637. int bsBufsize,
  638. int endian)
  639. {
  640. if( !bsBufsize )
  641. return -1;
  642. VpuReadMem(core_idx, paBsBufAddr, buf, bsBufsize, endian);
  643. return bsBufsize;
  644. }
  645. RetCode ReadBsRingBufHelper(Uint32 core_idx,
  646. EncHandle handle,
  647. osal_file_t bsFp,
  648. PhysicalAddress bitstreamBuffer,
  649. Uint32 bitstreamBufferSize,
  650. int defaultsize,
  651. int endian)
  652. {
  653. RetCode ret = RETCODE_SUCCESS;
  654. int loadSize = 0;
  655. PhysicalAddress paRdPtr, paWrPtr;
  656. int size = 0;
  657. PhysicalAddress paBsBufStart = bitstreamBuffer;
  658. PhysicalAddress paBsBufEnd = bitstreamBuffer+bitstreamBufferSize;
  659. ret = VPU_EncGetBitstreamBuffer(handle, &paRdPtr, &paWrPtr, &size);
  660. if( ret != RETCODE_SUCCESS )
  661. {
  662. VLOG(ERR, "VPU_EncGetBitstreamBuffer failed Error code is 0x%x \n", ret );
  663. goto LOAD_BS_ERROR;
  664. }
  665. if( size > 0 )
  666. {
  667. if( defaultsize > 0 )
  668. {
  669. if( size < defaultsize )
  670. loadSize = ( ( size >> 9 ) << 9 );
  671. else
  672. loadSize = defaultsize;
  673. }
  674. else
  675. {
  676. loadSize = size;
  677. }
  678. if( loadSize > 0 )
  679. {
  680. ProcessEncodedBitstreamBurst(core_idx, bsFp, paRdPtr, paBsBufStart, paBsBufEnd, loadSize, endian, NULL);
  681. ret = VPU_EncUpdateBitstreamBuffer(handle, loadSize);
  682. if( ret != RETCODE_SUCCESS )
  683. {
  684. VLOG(ERR, "VPU_EncUpdateBitstreamBuffer failed Error code is 0x%x \n", ret );
  685. goto LOAD_BS_ERROR;
  686. }
  687. }
  688. }
  689. LOAD_BS_ERROR:
  690. return ret;
  691. }
  692. /******************************************************************************
  693. DPB Image Data Control
  694. ******************************************************************************/
  695. /* THIS FUNCTION WILL BE REMOVED */
  696. /*lint -save -e438 */
  697. int LoadYuvImageHelperFormat(Uint32 core_idx,
  698. osal_file_t yuvFp,
  699. Uint8 *pYuv,
  700. FrameBuffer *fb,
  701. TiledMapConfig mapCfg,
  702. int picWidth,
  703. int picHeight,
  704. int stride,
  705. int interleave,
  706. int format,
  707. int endian,
  708. BOOL converter)
  709. {
  710. int frameSize;
  711. switch (format)
  712. {
  713. case FORMAT_420:
  714. frameSize = picWidth * picHeight * 3 / 2;
  715. break;
  716. case FORMAT_224:
  717. frameSize = picWidth * picHeight * 4 / 2;
  718. break;
  719. case FORMAT_422:
  720. frameSize = picWidth * picHeight * 4 / 2;
  721. break;
  722. case FORMAT_444:
  723. frameSize = picWidth * picHeight * 6 / 2;
  724. break;
  725. case FORMAT_400:
  726. frameSize = picWidth * picHeight;
  727. break;
  728. case FORMAT_YUYV:
  729. frameSize = picWidth * picHeight * 4 / 2;
  730. picWidth *= 2;
  731. stride*= 2;
  732. break;
  733. default:
  734. frameSize = picWidth * picHeight * 3 / 2;
  735. break;
  736. }
  737. // Load source one picture image to encode to SDRAM frame buffer.
  738. if( !osal_fread(pYuv, 1, frameSize, yuvFp) )
  739. {
  740. if( !osal_feof( yuvFp ) )
  741. VLOG(ERR, "Yuv Data osal_fread failed file handle is 0x%x \n", yuvFp );
  742. return 0;
  743. }
  744. if (fb->mapType)
  745. LoadTiledImageYuvBurst(core_idx, pYuv, picWidth, picHeight, fb, mapCfg);
  746. else
  747. {
  748. LoadYuvImageBurstFormat(core_idx, pYuv, picWidth, picHeight, fb, TRUE);
  749. }
  750. return 1;
  751. }
  752. /*lint -restore */
  753. #if 0
  754. int SaveYuvImageHelperFormat(
  755. Uint32 core_idx,
  756. osal_file_t yuvFp,
  757. FrameBuffer* fbSrc,
  758. TiledMapConfig mapCfg,
  759. Uint8* pYuv,
  760. VpuRect rect,
  761. BOOL enableCrop,
  762. int interLeave,
  763. int format,
  764. int endian)
  765. {
  766. int frameSize;
  767. int picWidth;
  768. int picHeight;
  769. picWidth = enableCrop == TRUE ? (int)(rect.right - rect.left) : fbSrc->stride;
  770. picHeight = enableCrop == TRUE ? (int)(rect.bottom - rect.top) : fbSrc->height;
  771. switch (format)
  772. {
  773. case FORMAT_420:
  774. frameSize = picWidth * ((picHeight+1)/2*2) * 3/2;
  775. break;
  776. case FORMAT_224:
  777. frameSize = picWidth * ((picHeight+1)/2*2) * 4/2;
  778. break;
  779. case FORMAT_422:
  780. frameSize = picWidth * picHeight * 4/2;
  781. break;
  782. case FORMAT_444:
  783. frameSize = picWidth * picHeight * 6/2;
  784. break;
  785. case FORMAT_400:
  786. frameSize = picWidth * picHeight;
  787. break;
  788. case FORMAT_420_P10_16BIT_LSB:
  789. case FORMAT_420_P10_16BIT_MSB:
  790. frameSize = picWidth*2 * ((picHeight+1)/2*2) * 3/2;
  791. break;
  792. default:
  793. frameSize = picWidth * ((picHeight+1)/2*2) * 3/2;
  794. break;
  795. }
  796. StoreYuvImageBurstFormat(core_idx, fbSrc, mapCfg, pYuv, rect, enableCrop, format, endian);
  797. if (yuvFp)
  798. {
  799. if( !osal_fwrite(pYuv, sizeof(Uint8), frameSize , yuvFp) )
  800. {
  801. VLOG(ERR, "Frame Data osal_fwrite failed file handle is 0x%x \n", yuvFp );
  802. return 0;
  803. }
  804. }
  805. return 1;
  806. }
  807. #endif
  808. int setWaveEncOpenParam(EncOpenParam *pEncOP, TestEncConfig *pEncConfig, ENC_CFG *pCfg)
  809. {
  810. Int32 i = 0;
  811. Int32 srcWidth;
  812. Int32 srcHeight;
  813. Int32 outputNum;
  814. Int32 bitrate;
  815. EncHevcParam *param = &pEncOP->EncStdParam.hevcParam;
  816. srcWidth = (pEncConfig->picWidth > 0) ? pEncConfig->picWidth : pCfg->hevcCfg.picX;
  817. srcHeight = (pEncConfig->picHeight > 0) ? pEncConfig->picHeight : pCfg->hevcCfg.picY;
  818. outputNum = (pEncConfig->outNum > 0) ? min(pEncConfig->outNum,pCfg->NumFrame) : pCfg->NumFrame;
  819. bitrate = (pEncConfig->kbps > 0) ? pEncConfig->kbps*1024 : pCfg->RcBitRate;
  820. pEncConfig->outNum = outputNum;
  821. pEncOP->picWidth = srcWidth;
  822. pEncOP->picHeight = srcHeight;
  823. pEncOP->frameRateInfo = pCfg->hevcCfg.frameRate;
  824. param->level = 0;
  825. param->tier = 0;
  826. pEncOP->srcBitDepth = pCfg->SrcBitDepth;
  827. if (pCfg->hevcCfg.internalBitDepth == 0)
  828. param->internalBitDepth = pCfg->SrcBitDepth;
  829. else
  830. param->internalBitDepth = pCfg->hevcCfg.internalBitDepth;
  831. if (param->internalBitDepth > 8)
  832. param->profile = HEVC_PROFILE_MAIN10;
  833. else
  834. param->profile = HEVC_PROFILE_MAIN;
  835. param->chromaFormatIdc = 0;
  836. param->losslessEnable = pCfg->hevcCfg.losslessEnable;
  837. param->constIntraPredFlag = pCfg->hevcCfg.constIntraPredFlag;
  838. if (pCfg->hevcCfg.useAsLongtermPeriod > 0 || pCfg->hevcCfg.refLongtermPeriod > 0)
  839. param->useLongTerm = 1;
  840. else
  841. param->useLongTerm = 0;
  842. /* for CMD_ENC_SEQ_GOP_PARAM */
  843. param->gopPresetIdx = pCfg->hevcCfg.gopPresetIdx;
  844. /* for CMD_ENC_SEQ_INTRA_PARAM */
  845. param->decodingRefreshType = pCfg->hevcCfg.decodingRefreshType;
  846. param->intraPeriod = pCfg->hevcCfg.intraPeriod;
  847. param->intraQP = pCfg->hevcCfg.intraQP;
  848. param->forcedIdrHeaderEnable = pCfg->hevcCfg.forcedIdrHeaderEnable;
  849. /* for CMD_ENC_SEQ_CONF_WIN_TOP_BOT/LEFT_RIGHT */
  850. param->confWinTop = pCfg->hevcCfg.confWinTop;
  851. param->confWinBot = pCfg->hevcCfg.confWinBot;
  852. param->confWinLeft = pCfg->hevcCfg.confWinLeft;
  853. param->confWinRight = pCfg->hevcCfg.confWinRight;
  854. /* for CMD_ENC_SEQ_INDEPENDENT_SLICE */
  855. param->independSliceMode = pCfg->hevcCfg.independSliceMode;
  856. param->independSliceModeArg = pCfg->hevcCfg.independSliceModeArg;
  857. /* for CMD_ENC_SEQ_DEPENDENT_SLICE */
  858. param->dependSliceMode = pCfg->hevcCfg.dependSliceMode;
  859. param->dependSliceModeArg = pCfg->hevcCfg.dependSliceModeArg;
  860. /* for CMD_ENC_SEQ_INTRA_REFRESH_PARAM */
  861. param->intraRefreshMode = pCfg->hevcCfg.intraRefreshMode;
  862. param->intraRefreshArg = pCfg->hevcCfg.intraRefreshArg;
  863. param->useRecommendEncParam = pCfg->hevcCfg.useRecommendEncParam;
  864. /* for CMD_ENC_PARAM */
  865. param->scalingListEnable = pCfg->hevcCfg.scalingListEnable;
  866. param->cuSizeMode = pCfg->hevcCfg.cuSizeMode;
  867. param->tmvpEnable = pCfg->hevcCfg.tmvpEnable;
  868. param->wppEnable = pCfg->hevcCfg.wppenable;
  869. param->maxNumMerge = pCfg->hevcCfg.maxNumMerge;
  870. param->dynamicMerge8x8Enable = pCfg->hevcCfg.dynamicMerge8x8Enable;
  871. param->dynamicMerge16x16Enable = pCfg->hevcCfg.dynamicMerge16x16Enable;
  872. param->dynamicMerge32x32Enable = pCfg->hevcCfg.dynamicMerge32x32Enable;
  873. param->disableDeblk = pCfg->hevcCfg.disableDeblk;
  874. param->lfCrossSliceBoundaryEnable = pCfg->hevcCfg.lfCrossSliceBoundaryEnable;
  875. param->betaOffsetDiv2 = pCfg->hevcCfg.betaOffsetDiv2;
  876. param->tcOffsetDiv2 = pCfg->hevcCfg.tcOffsetDiv2;
  877. param->skipIntraTrans = pCfg->hevcCfg.skipIntraTrans;
  878. param->saoEnable = pCfg->hevcCfg.saoEnable;
  879. param->intraInInterSliceEnable = pCfg->hevcCfg.intraInInterSliceEnable;
  880. param->intraNxNEnable = pCfg->hevcCfg.intraNxNEnable;
  881. /* for CMD_ENC_RC_PARAM */
  882. pEncOP->rcEnable = pCfg->RcEnable;
  883. pEncOP->initialDelay = pCfg->RcInitDelay;
  884. param->cuLevelRCEnable = pCfg->hevcCfg.cuLevelRCEnable;
  885. param->hvsQPEnable = pCfg->hevcCfg.hvsQPEnable;
  886. param->hvsQpScaleEnable = pCfg->hevcCfg.hvsQpScaleEnable;
  887. param->hvsQpScale = pCfg->hevcCfg.hvsQpScale;
  888. param->ctuOptParam.roiDeltaQp= pCfg->hevcCfg.ctuOptParam.roiDeltaQp;
  889. param->intraQpOffset = pCfg->hevcCfg.intraQpOffset;
  890. param->initBufLevelx8 = pCfg->hevcCfg.initBufLevelx8;
  891. param->bitAllocMode = pCfg->hevcCfg.bitAllocMode;
  892. for (i = 0; i < MAX_GOP_NUM; i++) {
  893. param->fixedBitRatio[i] = pCfg->hevcCfg.fixedBitRatio[i];
  894. }
  895. /* for CMD_ENC_RC_MIN_MAX_QP */
  896. param->minQp = pCfg->hevcCfg.minQp;
  897. param->maxQp = pCfg->hevcCfg.maxQp;
  898. param->maxDeltaQp = pCfg->hevcCfg.maxDeltaQp;
  899. param->transRate = pCfg->hevcCfg.transRate;
  900. pEncOP->bitRate = bitrate;
  901. param->gopParam.enTemporalLayerQp = pCfg->hevcCfg.gopParam.enTemporalLayerQp;
  902. param->gopParam.tidQp0 = pCfg->hevcCfg.gopParam.tidQp0;
  903. param->gopParam.tidQp1 = pCfg->hevcCfg.gopParam.tidQp1;
  904. param->gopParam.tidQp2 = pCfg->hevcCfg.gopParam.tidQp2;
  905. param->gopParam.tidPeriod0 = pCfg->hevcCfg.gopParam.tidPeriod0;
  906. /* for CMD_ENC_CUSTOM_GOP_PARAM */
  907. param->gopParam.customGopSize = pCfg->hevcCfg.gopParam.customGopSize;
  908. param->gopParam.useDeriveLambdaWeight = pCfg->hevcCfg.gopParam.useDeriveLambdaWeight;
  909. for (i= 0; i<param->gopParam.customGopSize; i++) {
  910. param->gopParam.picParam[i].picType = pCfg->hevcCfg.gopParam.picParam[i].picType;
  911. param->gopParam.picParam[i].pocOffset = pCfg->hevcCfg.gopParam.picParam[i].pocOffset;
  912. param->gopParam.picParam[i].picQp = pCfg->hevcCfg.gopParam.picParam[i].picQp;
  913. param->gopParam.picParam[i].refPocL0 = pCfg->hevcCfg.gopParam.picParam[i].refPocL0;
  914. param->gopParam.picParam[i].refPocL1 = pCfg->hevcCfg.gopParam.picParam[i].refPocL1;
  915. param->gopParam.picParam[i].temporalId = pCfg->hevcCfg.gopParam.picParam[i].temporalId;
  916. param->gopParam.gopPicLambda[i] = pCfg->hevcCfg.gopParam.gopPicLambda[i];
  917. }
  918. param->ctuOptParam.roiEnable = pCfg->hevcCfg.ctuOptParam.roiEnable;
  919. param->ctuOptParam.ctuQpEnable = pCfg->hevcCfg.ctuOptParam.ctuQpEnable;
  920. // VPS & VUI
  921. param->numUnitsInTick = pCfg->hevcCfg.numUnitsInTick;
  922. param->timeScale = pCfg->hevcCfg.timeScale;
  923. param->numTicksPocDiffOne = pCfg->hevcCfg.numTicksPocDiffOne;
  924. if(pCfg->hevcCfg.frameRate) {
  925. param->numUnitsInTick = 1000;
  926. param->timeScale = (pCfg->hevcCfg.frameRate)*1000;
  927. }
  928. param->vuiParam.vuiParamFlags = pCfg->hevcCfg.vuiParam.vuiParamFlags;
  929. param->vuiParam.vuiAspectRatioIdc = pCfg->hevcCfg.vuiParam.vuiAspectRatioIdc;
  930. param->vuiParam.vuiSarSize = pCfg->hevcCfg.vuiParam.vuiSarSize;
  931. param->vuiParam.vuiOverScanAppropriate = pCfg->hevcCfg.vuiParam.vuiOverScanAppropriate;
  932. param->vuiParam.videoSignal = pCfg->hevcCfg.vuiParam.videoSignal;
  933. param->vuiParam.vuiChromaSampleLoc = pCfg->hevcCfg.vuiParam.vuiChromaSampleLoc;
  934. param->vuiParam.vuiDispWinLeftRight = pCfg->hevcCfg.vuiParam.vuiDispWinLeftRight;
  935. param->vuiParam.vuiDispWinTopBottom = pCfg->hevcCfg.vuiParam.vuiDispWinTopBottom;
  936. pEncOP->encodeVuiRbsp = pCfg->hevcCfg.vuiDataEnable;
  937. pEncOP->vuiRbspDataSize = pCfg->hevcCfg.vuiDataSize;
  938. pEncOP->encodeHrdRbspInVPS = pCfg->hevcCfg.hrdInVPS;
  939. pEncOP->hrdRbspDataSize = pCfg->hevcCfg.hrdDataSize;
  940. pEncOP->encodeHrdRbspInVUI = pCfg->hevcCfg.hrdInVUI;
  941. param->chromaCbQpOffset = pCfg->hevcCfg.chromaCbQpOffset;
  942. param->chromaCrQpOffset = pCfg->hevcCfg.chromaCrQpOffset;
  943. param->initialRcQp = pCfg->hevcCfg.initialRcQp;
  944. param->nrYEnable = pCfg->hevcCfg.nrYEnable;
  945. param->nrCbEnable = pCfg->hevcCfg.nrCbEnable;
  946. param->nrCrEnable = pCfg->hevcCfg.nrCrEnable;
  947. param->nrNoiseEstEnable = pCfg->hevcCfg.nrNoiseEstEnable;
  948. param->nrNoiseSigmaY = pCfg->hevcCfg.nrNoiseSigmaY;
  949. param->nrNoiseSigmaCb = pCfg->hevcCfg.nrNoiseSigmaCb;
  950. param->nrNoiseSigmaCr = pCfg->hevcCfg.nrNoiseSigmaCr;
  951. param->nrIntraWeightY = pCfg->hevcCfg.nrIntraWeightY;
  952. param->nrIntraWeightCb = pCfg->hevcCfg.nrIntraWeightCb;
  953. param->nrIntraWeightCr = pCfg->hevcCfg.nrIntraWeightCr;
  954. param->nrInterWeightY = pCfg->hevcCfg.nrInterWeightY;
  955. param->nrInterWeightCb = pCfg->hevcCfg.nrInterWeightCb;
  956. param->nrInterWeightCr = pCfg->hevcCfg.nrInterWeightCr;
  957. param->intraMinQp = pCfg->hevcCfg.intraMinQp;
  958. param->intraMaxQp = pCfg->hevcCfg.intraMaxQp;
  959. return 1;
  960. }
  961. int setCoda9EncOpenParam(EncOpenParam *pEncOP, TestEncConfig *pEncConfig, ENC_CFG *pCfg)
  962. {
  963. Int32 bitFormat;
  964. Int32 srcWidth;
  965. Int32 srcHeight;
  966. Int32 outputNum;
  967. bitFormat = pEncOP->bitstreamFormat;
  968. srcWidth = (pEncConfig->picWidth > 0) ? pEncConfig->picWidth : pCfg->PicX;
  969. srcHeight = (pEncConfig->picHeight > 0) ? pEncConfig->picHeight : pCfg->PicY;
  970. outputNum = (pEncConfig->outNum > 0) ? pEncConfig->outNum : pCfg->NumFrame;
  971. pEncConfig->outNum = outputNum;
  972. osal_memcpy(pEncConfig->skipPicNums, pCfg->skipPicNums, sizeof(pCfg->skipPicNums));
  973. pEncOP->picWidth = srcWidth;
  974. pEncOP->picHeight = srcHeight;
  975. pEncOP->frameRateInfo = pCfg->FrameRate;
  976. pEncOP->bitRate = pCfg->RcBitRate;
  977. pEncOP->rcInitDelay = pCfg->RcInitDelay;
  978. pEncOP->vbvBufferSize = pCfg->RcBufSize;
  979. pEncOP->frameSkipDisable = pCfg->frameSkipDisable; // for compare with C-model ( C-model = only 1 )
  980. pEncOP->meBlkMode = pCfg->MeBlkModeEnable; // for compare with C-model ( C-model = only 0 )
  981. pEncOP->gopSize = pCfg->GopPicNum;
  982. pEncOP->idrInterval = pCfg->IDRInterval;
  983. pEncOP->sliceMode.sliceMode = pCfg->SliceMode;
  984. pEncOP->sliceMode.sliceSizeMode = pCfg->SliceSizeMode;
  985. pEncOP->sliceMode.sliceSize = pCfg->SliceSizeNum;
  986. pEncOP->intraRefreshNum = pCfg->IntraRefreshNum;
  987. pEncOP->ConscIntraRefreshEnable = pCfg->ConscIntraRefreshEnable;
  988. pEncOP->rcIntraQp = pCfg->RCIntraQP;
  989. pEncOP->intraCostWeight = pCfg->intraCostWeight;
  990. pEncOP->MESearchRange = pCfg->SearchRange;
  991. pEncOP->rcEnable = pCfg->RcEnable;
  992. if (!pCfg->RcEnable)
  993. pEncOP->bitRate = 0;
  994. if (!pCfg->GammaSetEnable)
  995. pEncOP->userGamma = -1;
  996. else
  997. pEncOP->userGamma = pCfg->Gamma;
  998. pEncOP->MEUseZeroPmv = pCfg->MeUseZeroPmv;
  999. /* It was agreed that the statements below would be used. but Cmodel at r25518 is not changed yet according to the statements below
  1000. if (bitFormat == STD_MPEG4)
  1001. pEncOP->MEUseZeroPmv = 1;
  1002. else
  1003. pEncOP->MEUseZeroPmv = 0;
  1004. */
  1005. // MP4 263 Only
  1006. if (!pCfg->ConstantIntraQPEnable)
  1007. pEncOP->rcIntraQp = -1;
  1008. if (pCfg->MaxQpSetEnable)
  1009. pEncOP->userQpMax = pCfg->MaxQp;
  1010. else
  1011. pEncOP->userQpMax = -1;
  1012. // H.264 Only
  1013. if (bitFormat == STD_AVC)
  1014. {
  1015. if(pCfg->MinQpSetEnable)
  1016. pEncOP->userQpMin = pCfg->MinQp;
  1017. else
  1018. pEncOP->userQpMin = 12;
  1019. if (pCfg->MaxQpSetEnable)
  1020. pEncOP->userQpMax = pCfg->MaxQp;
  1021. else
  1022. pEncOP->userQpMax = 51;
  1023. if(pCfg->MaxDeltaQpSetEnable)
  1024. pEncOP->userMaxDeltaQp = pCfg->MaxDeltaQp;
  1025. else
  1026. pEncOP->userMaxDeltaQp = -1;
  1027. if(pCfg->MinDeltaQpSetEnable)
  1028. pEncOP->userMinDeltaQp = pCfg->MinDeltaQp;
  1029. else
  1030. pEncOP->userMinDeltaQp = -1;
  1031. }
  1032. pEncOP->rcIntervalMode = pCfg->rcIntervalMode; // 0:normal, 1:frame_level, 2:slice_level, 3: user defined Mb_level
  1033. pEncOP->mbInterval = pCfg->RcMBInterval; // FIXME
  1034. // Standard specific
  1035. if( bitFormat == STD_MPEG4 ) {
  1036. pEncOP->EncStdParam.mp4Param.mp4DataPartitionEnable = pCfg->DataPartEn;
  1037. pEncOP->EncStdParam.mp4Param.mp4ReversibleVlcEnable = pCfg->RevVlcEn;
  1038. pEncOP->EncStdParam.mp4Param.mp4IntraDcVlcThr = pCfg->IntraDcVlcThr;
  1039. pEncOP->EncStdParam.mp4Param.mp4HecEnable = pCfg->HecEnable;
  1040. pEncOP->EncStdParam.mp4Param.mp4Verid = pCfg->VerId;
  1041. }
  1042. else if( bitFormat == STD_H263 ) {
  1043. pEncOP->EncStdParam.h263Param.h263AnnexIEnable = pCfg->AnnexI;
  1044. pEncOP->EncStdParam.h263Param.h263AnnexJEnable = pCfg->AnnexJ;
  1045. pEncOP->EncStdParam.h263Param.h263AnnexKEnable = pCfg->AnnexK;
  1046. pEncOP->EncStdParam.h263Param.h263AnnexTEnable = pCfg->AnnexT;
  1047. }
  1048. else if( bitFormat == STD_AVC ) {
  1049. pEncOP->EncStdParam.avcParam.constrainedIntraPredFlag = pCfg->ConstIntraPredFlag;
  1050. pEncOP->EncStdParam.avcParam.disableDeblk = pCfg->DisableDeblk;
  1051. pEncOP->EncStdParam.avcParam.deblkFilterOffsetAlpha = pCfg->DeblkOffsetA;
  1052. pEncOP->EncStdParam.avcParam.deblkFilterOffsetBeta = pCfg->DeblkOffsetB;
  1053. pEncOP->EncStdParam.avcParam.chromaQpOffset = pCfg->ChromaQpOffset;
  1054. pEncOP->EncStdParam.avcParam.audEnable = pCfg->aud_en;
  1055. pEncOP->EncStdParam.avcParam.frameCroppingFlag = pCfg->frameCroppingFlag;
  1056. pEncOP->EncStdParam.avcParam.frameCropLeft = pCfg->frameCropLeft;
  1057. pEncOP->EncStdParam.avcParam.frameCropRight = pCfg->frameCropRight;
  1058. pEncOP->EncStdParam.avcParam.frameCropTop = pCfg->frameCropTop;
  1059. pEncOP->EncStdParam.avcParam.frameCropBottom = pCfg->frameCropBottom;
  1060. pEncOP->EncStdParam.avcParam.level = pCfg->level;
  1061. // Update cropping information : Usage example for H.264 frame_cropping_flag
  1062. if (pEncOP->picHeight == 1080)
  1063. {
  1064. // In case of AVC encoder, when we want to use unaligned display width(For example, 1080),
  1065. // frameCroppingFlag parameters should be adjusted to displayable rectangle
  1066. if (pEncConfig->rotAngle != 90 && pEncConfig->rotAngle != 270) // except rotation
  1067. {
  1068. if (pEncOP->EncStdParam.avcParam.frameCroppingFlag == 0)
  1069. {
  1070. pEncOP->EncStdParam.avcParam.frameCroppingFlag = 1;
  1071. // frameCropBottomOffset = picHeight(MB-aligned) - displayable rectangle height
  1072. pEncOP->EncStdParam.avcParam.frameCropBottom = 8;
  1073. }
  1074. }
  1075. }
  1076. }
  1077. else {
  1078. VLOG(ERR, "Invalid codec standard mode \n" );
  1079. return 0;
  1080. }
  1081. return 1;
  1082. }
  1083. #if 0
  1084. void changeRcParaTest(Uint32 core_idx, EncHandle handle,
  1085. osal_file_t bsFp,
  1086. EncParam *pEncParam,
  1087. EncHeaderParam *pEncHeaderParam,
  1088. TestEncConfig *pEncConfig,
  1089. EncOpenParam *pEncOP)
  1090. {
  1091. int optionNum;
  1092. while(1)
  1093. {
  1094. printf("\n 0: GOP number change\n");
  1095. printf(" 1: Intra Qp change\n");
  1096. printf(" 2: Bit Rate change\n");
  1097. printf(" 3: Frame Rate change\n");
  1098. printf(" 4: Intra Refresh Number change\n");
  1099. printf(" 5: Slice Mode change\n");
  1100. if(pEncOP->bitstreamFormat == STD_MPEG4)
  1101. {
  1102. printf(" 6: HEC Mode change\n");
  1103. }
  1104. printf(" 9: go encoding\n");
  1105. scanf("%d", &optionNum);
  1106. switch(optionNum){
  1107. case 0:
  1108. {
  1109. int newGopNum=0;
  1110. printf("\n New Gop Number=");
  1111. scanf("%d", &newGopNum);
  1112. VPU_EncGiveCommand(handle, ENC_SET_GOP_NUMBER, &newGopNum);
  1113. }
  1114. break;
  1115. case 1:
  1116. {
  1117. int newIntraQp=0;
  1118. printf("\n New Intra Qp value=");
  1119. scanf("%d", &newIntraQp);
  1120. VPU_EncGiveCommand(handle, ENC_SET_INTRA_QP, &newIntraQp);
  1121. }
  1122. break;
  1123. case 2:
  1124. {
  1125. int newBitrate=0;
  1126. printf("\n New Bit Rate=");
  1127. scanf("%d", &newBitrate);
  1128. VPU_EncGiveCommand(handle, ENC_SET_BITRATE, &newBitrate);
  1129. }
  1130. break;
  1131. case 3:
  1132. {
  1133. int newFramerate=0;
  1134. printf("\n New Frame Rate=");
  1135. scanf("%d", &newFramerate);
  1136. VPU_EncGiveCommand(handle, ENC_SET_FRAME_RATE, &newFramerate);
  1137. if( pEncOP->bitstreamFormat == STD_MPEG4 )
  1138. {
  1139. pEncHeaderParam->headerType = VOL_HEADER;
  1140. VPU_EncGiveCommand(handle, ENC_PUT_VIDEO_HEADER, pEncHeaderParam);
  1141. #ifdef SUPPORT_FFMPEG_DEMUX
  1142. if( pEncOP->ringBufferEnable == 0 && !pEncConfig->en_container )
  1143. #else
  1144. if( pEncOP->ringBufferEnable == 0 )
  1145. #endif
  1146. {
  1147. #if 0
  1148. if (!ReadBsResetBufHelper(core_idx, bsFp, pEncHeaderParam->buf, pEncHeaderParam->size, pEncOP->streamEndian))
  1149. break;
  1150. #else
  1151. BitstreamReader_Act(bsReader, core_idx, pEncHeaderParam->buf, pEncHeaderParam->size, pEncOP->streamEndian, comparator);
  1152. #endif
  1153. }
  1154. }
  1155. }
  1156. break;
  1157. case 4:
  1158. {
  1159. int newIntraRefreshNum=0;
  1160. printf("\n New Intra Refresh Number=");
  1161. scanf("%d", &newIntraRefreshNum);
  1162. VPU_EncGiveCommand(handle, ENC_SET_INTRA_MB_REFRESH_NUMBER, &newIntraRefreshNum);
  1163. }
  1164. break;
  1165. case 5:
  1166. {
  1167. EncSliceMode newSlice;
  1168. printf("\n New Slice Mode[0:one slice, 1:muliple slice]=");
  1169. scanf("%d", &newSlice.sliceMode);
  1170. if(!newSlice.sliceMode)
  1171. {
  1172. newSlice.sliceSizeMode = 0;
  1173. newSlice.sliceSize = 0;
  1174. }
  1175. else
  1176. {
  1177. printf("\n New Slice Size Mode[0:bit number, 1:mb number]=");
  1178. scanf("%d", &newSlice.sliceSizeMode);
  1179. if(!newSlice.sliceSizeMode)
  1180. {
  1181. printf("\n New Slice bit number=");
  1182. }
  1183. else
  1184. {
  1185. printf("\n New Slice MB number=");
  1186. }
  1187. scanf("%d", &newSlice.sliceSize);
  1188. }
  1189. VPU_EncGiveCommand(handle, ENC_SET_SLICE_INFO, &newSlice);
  1190. }
  1191. break;
  1192. case 6:
  1193. {
  1194. int newHecMode=0;
  1195. printf("\n New Hec Mode Enable[0:disable, 1:enable]=");
  1196. scanf("%d", &newHecMode);
  1197. if(newHecMode > 0)
  1198. {
  1199. VPU_EncGiveCommand(handle, ENC_ENABLE_HEC, &newHecMode);
  1200. }
  1201. else
  1202. {
  1203. VPU_EncGiveCommand(handle, ENC_DISABLE_HEC, &newHecMode);
  1204. }
  1205. }
  1206. break;
  1207. default:
  1208. break;
  1209. }
  1210. if(optionNum == 9)
  1211. break;
  1212. }
  1213. }
  1214. #endif
  1215. /******************************************************************************
  1216. EncOpenParam Initialization
  1217. ******************************************************************************/
  1218. /**
  1219. * To init EncOpenParam by runtime evaluation
  1220. * IN
  1221. * EncConfigParam *pEncConfig
  1222. * OUT
  1223. * EncOpenParam *pEncOP
  1224. */
  1225. Int32 GetEncOpenParamDefault(EncOpenParam *pEncOP, TestEncConfig *pEncConfig)
  1226. {
  1227. int bitFormat;
  1228. pEncConfig->outNum = pEncConfig->outNum == 0 ? DEFAULT_ENC_OUTPUT_NUM : pEncConfig->outNum;
  1229. bitFormat = pEncOP->bitstreamFormat;
  1230. pEncOP->picWidth = pEncConfig->picWidth;
  1231. pEncOP->picHeight = pEncConfig->picHeight;
  1232. pEncOP->frameRateInfo = 30;
  1233. pEncOP->MESearchRange = 3;
  1234. pEncOP->bitRate = pEncConfig->kbps;
  1235. pEncOP->rcInitDelay = 0;
  1236. pEncOP->vbvBufferSize = 0; // 0 = ignore
  1237. pEncOP->meBlkMode = 0; // for compare with C-model ( C-model = only 0 )
  1238. pEncOP->frameSkipDisable = 1; // for compare with C-model ( C-model = only 1 )
  1239. pEncOP->gopSize = 30; // only first picture is I
  1240. pEncOP->sliceMode.sliceMode = 1; // 1 slice per picture
  1241. pEncOP->sliceMode.sliceSizeMode = 1;
  1242. pEncOP->sliceMode.sliceSize = 115;
  1243. pEncOP->intraRefreshNum = 0;
  1244. pEncOP->rcIntraQp = -1; // disable == -1
  1245. pEncOP->userQpMax = -1; // disable == -1
  1246. pEncOP->userGamma = (Uint32)(0.75*32768); // (0*32768 < gamma < 1*32768)
  1247. pEncOP->rcIntervalMode = 1; // 0:normal, 1:frame_level, 2:slice_level, 3: user defined Mb_level
  1248. pEncOP->mbInterval = 0;
  1249. pEncConfig->picQpY = 23;
  1250. if (bitFormat == STD_MPEG4)
  1251. pEncOP->MEUseZeroPmv = 1;
  1252. else
  1253. pEncOP->MEUseZeroPmv = 0;
  1254. pEncOP->intraCostWeight = 400;
  1255. // Standard specific
  1256. if( bitFormat == STD_MPEG4 ) {
  1257. pEncOP->EncStdParam.mp4Param.mp4DataPartitionEnable = 0;
  1258. pEncOP->EncStdParam.mp4Param.mp4ReversibleVlcEnable = 0;
  1259. pEncOP->EncStdParam.mp4Param.mp4IntraDcVlcThr = 0;
  1260. pEncOP->EncStdParam.mp4Param.mp4HecEnable = 0;
  1261. pEncOP->EncStdParam.mp4Param.mp4Verid = 2;
  1262. }
  1263. else if( bitFormat == STD_H263 ) {
  1264. pEncOP->EncStdParam.h263Param.h263AnnexIEnable = 0;
  1265. pEncOP->EncStdParam.h263Param.h263AnnexJEnable = 0;
  1266. pEncOP->EncStdParam.h263Param.h263AnnexKEnable = 0;
  1267. pEncOP->EncStdParam.h263Param.h263AnnexTEnable = 0;
  1268. }
  1269. else if( bitFormat == STD_AVC ) {
  1270. pEncOP->EncStdParam.avcParam.constrainedIntraPredFlag = 0;
  1271. pEncOP->EncStdParam.avcParam.disableDeblk = 1;
  1272. pEncOP->EncStdParam.avcParam.deblkFilterOffsetAlpha = 6;
  1273. pEncOP->EncStdParam.avcParam.deblkFilterOffsetBeta = 0;
  1274. pEncOP->EncStdParam.avcParam.chromaQpOffset = 10;
  1275. pEncOP->EncStdParam.avcParam.audEnable = 0;
  1276. pEncOP->EncStdParam.avcParam.frameCroppingFlag = 0;
  1277. pEncOP->EncStdParam.avcParam.frameCropLeft = 0;
  1278. pEncOP->EncStdParam.avcParam.frameCropRight = 0;
  1279. pEncOP->EncStdParam.avcParam.frameCropTop = 0;
  1280. pEncOP->EncStdParam.avcParam.frameCropBottom = 0;
  1281. pEncOP->EncStdParam.avcParam.level = 0;
  1282. // Update cropping information : Usage example for H.264 frame_cropping_flag
  1283. if (pEncOP->picHeight == 1080)
  1284. {
  1285. // In case of AVC encoder, when we want to use unaligned display width(For example, 1080),
  1286. // frameCroppingFlag parameters should be adjusted to displayable rectangle
  1287. if (pEncConfig->rotAngle != 90 && pEncConfig->rotAngle != 270) // except rotation
  1288. {
  1289. if (pEncOP->EncStdParam.avcParam.frameCroppingFlag == 0)
  1290. {
  1291. pEncOP->EncStdParam.avcParam.frameCroppingFlag = 1;
  1292. // frameCropBottomOffset = picHeight(MB-aligned) - displayable rectangle height
  1293. pEncOP->EncStdParam.avcParam.frameCropBottom = 8;
  1294. }
  1295. }
  1296. }
  1297. }
  1298. else if( bitFormat == STD_HEVC ) {
  1299. EncHevcParam *param = &pEncOP->EncStdParam.hevcParam;
  1300. Int32 rcBitrate = pEncConfig->kbps * 1000;
  1301. Int32 i=0;
  1302. pEncOP->bitRate = rcBitrate;
  1303. param->profile = HEVC_PROFILE_MAIN;
  1304. param->level = 0;
  1305. param->tier = 0;
  1306. param->internalBitDepth = 8;
  1307. pEncOP->srcBitDepth = 8;
  1308. param->chromaFormatIdc = 0;
  1309. param->losslessEnable = 0;
  1310. param->constIntraPredFlag = 0;
  1311. param->enableAFBCD = 0;
  1312. param->useLongTerm = 0;
  1313. /* for CMD_ENC_SEQ_GOP_PARAM */
  1314. param->gopPresetIdx = PRESET_IDX_IBBBP;
  1315. /* for CMD_ENC_SEQ_INTRA_PARAM */
  1316. param->decodingRefreshType = 1;
  1317. param->intraPeriod = 28;
  1318. param->intraQP = 0;
  1319. /* for CMD_ENC_SEQ_CONF_WIN_TOP_BOT/LEFT_RIGHT */
  1320. param->confWinTop = 0;
  1321. param->confWinBot = 0;
  1322. param->confWinLeft = 0;
  1323. param->confWinRight = 0;
  1324. /* for CMD_ENC_SEQ_INDEPENDENT_SLICE */
  1325. param->independSliceMode = 0;
  1326. param->independSliceModeArg = 0;
  1327. /* for CMD_ENC_SEQ_DEPENDENT_SLICE */
  1328. param->dependSliceMode = 0;
  1329. param->dependSliceModeArg = 0;
  1330. /* for CMD_ENC_SEQ_INTRA_REFRESH_PARAM */
  1331. param->intraRefreshMode = 0;
  1332. param->intraRefreshArg = 0;
  1333. param->useRecommendEncParam = 1;
  1334. pEncConfig->seiDataEnc.prefixSeiNalEnable = 0;
  1335. pEncConfig->seiDataEnc.suffixSeiNalEnable = 0;
  1336. pEncOP->encodeHrdRbspInVPS = 0;
  1337. pEncOP->encodeHrdRbspInVUI = 0;
  1338. pEncOP->encodeVuiRbsp = 0;
  1339. pEncConfig->roi_enable = 0;
  1340. pEncConfig->ctu_qp_enable = 0;
  1341. /* for CMD_ENC_PARAM */
  1342. if (param->useRecommendEncParam != 1) { // 0 : Custom, 2 : Boost mode (normal encoding speed, normal picture quality), 3 : Fast mode (high encoding speed, low picture quality)
  1343. param->scalingListEnable = 0;
  1344. param->cuSizeMode = 0x7;
  1345. param->tmvpEnable = 1;
  1346. param->wppEnable = 0;
  1347. param->maxNumMerge = 2;
  1348. param->dynamicMerge8x8Enable = 1;
  1349. param->dynamicMerge16x16Enable = 1;
  1350. param->dynamicMerge32x32Enable = 1;
  1351. param->disableDeblk = 0;
  1352. param->lfCrossSliceBoundaryEnable = 1;
  1353. param->betaOffsetDiv2 = 0;
  1354. param->tcOffsetDiv2 = 0;
  1355. param->skipIntraTrans = 1;
  1356. param->saoEnable = 1;
  1357. param->intraInInterSliceEnable = 1;
  1358. param->intraNxNEnable = 1;
  1359. }
  1360. /* for CMD_ENC_RC_PARAM */
  1361. pEncOP->rcEnable = rcBitrate == 0 ? FALSE : TRUE;
  1362. pEncOP->initialDelay = 3000;
  1363. param->ctuOptParam.roiEnable = 0;
  1364. param->ctuOptParam.roiDeltaQp = 3;
  1365. param->ctuOptParam.ctuQpEnable = 0;
  1366. param->intraQpOffset = 0;
  1367. param->initBufLevelx8 = 1;
  1368. param->bitAllocMode = 0;
  1369. for (i = 0; i < MAX_GOP_NUM; i++) {
  1370. param->fixedBitRatio[i] = 1;
  1371. }
  1372. param->cuLevelRCEnable = 0;
  1373. param->hvsQPEnable = 1;
  1374. param->hvsQpScale = 2;
  1375. param->hvsQpScaleEnable = (param->hvsQpScale > 0) ? 1: 0;
  1376. /* for CMD_ENC_RC_MIN_MAX_QP */
  1377. param->minQp = 8;
  1378. param->maxQp = 51;
  1379. param->maxDeltaQp = 10;
  1380. param->minQpI = 8;
  1381. param->maxQpI = 51;
  1382. param->minQpP = 8;
  1383. param->maxQpP = 51;
  1384. param->minQpB = 8;
  1385. param->maxQpB = 51;
  1386. param->hvsMaxDeltaQp = 10;
  1387. param->gopParam.enTemporalLayerQp = 0;
  1388. param->gopParam.tidQp0 = 30;
  1389. param->gopParam.tidQp1 = 33;
  1390. param->gopParam.tidQp2 = 36;
  1391. param->gopParam.tidPeriod0 = 60;
  1392. /* for CMD_ENC_CUSTOM_GOP_PARAM */
  1393. param->gopParam.customGopSize = 0;
  1394. param->gopParam.useDeriveLambdaWeight = 0;
  1395. for (i= 0; i<param->gopParam.customGopSize; i++) {
  1396. param->gopParam.picParam[i].picType = PIC_TYPE_I;
  1397. param->gopParam.picParam[i].pocOffset = 1;
  1398. param->gopParam.picParam[i].picQp = 30;
  1399. param->gopParam.picParam[i].refPocL0 = 0;
  1400. param->gopParam.picParam[i].refPocL1 = 0;
  1401. param->gopParam.picParam[i].temporalId = 0;
  1402. param->gopParam.gopPicLambda[i] = 0;
  1403. }
  1404. param->transRate = 0;
  1405. // for VUI / time information.
  1406. param->numTicksPocDiffOne = 0;
  1407. param->timeScale = pEncOP->frameRateInfo * 1000;
  1408. param->numUnitsInTick = 1000;
  1409. param->vuiParam.vuiParamFlags = 0; // when vuiParamFlags == 0, VPU doesn't encode VUI
  1410. param->chromaCbQpOffset = 0;
  1411. param->chromaCrQpOffset = 0;
  1412. param->initialRcQp = 63; // 63 is meaningless.
  1413. param->nrYEnable = 0;
  1414. param->nrCbEnable = 0;
  1415. param->nrCrEnable = 0;
  1416. param->nrNoiseEstEnable = 0;
  1417. param->intraMinQp = 8;
  1418. param->intraMaxQp = 51;
  1419. param->rcWeightParam = 2;
  1420. param->rcWeightBuf = 128;
  1421. // for H.264 encoder
  1422. param->avcIdrPeriod = 0;
  1423. param->rdoSkip = 1;
  1424. param->lambdaScalingEnable = 1;
  1425. param->transform8x8Enable = 1;
  1426. param->avcSliceMode = 0;
  1427. param->avcSliceArg = 0;
  1428. param->intraMbRefreshMode = 0;
  1429. param->intraMbRefreshArg = 1;
  1430. param->mbLevelRcEnable = 0;
  1431. param->entropyCodingMode = 1;
  1432. if (bitFormat == STD_AVC)
  1433. param->disableDeblk = 2;
  1434. }
  1435. else {
  1436. VLOG(ERR, "Invalid codec standard mode: codec index(%d) \n", bitFormat);
  1437. return 0;
  1438. }
  1439. return 1;
  1440. }
  1441. /**
  1442. * To init EncOpenParam by CFG file
  1443. * IN
  1444. * EncConfigParam *pEncConfig
  1445. * OUT
  1446. * EncOpenParam *pEncOP
  1447. * char *srcYuvFileName
  1448. */
  1449. Int32 GetEncOpenParam(EncOpenParam *pEncOP, TestEncConfig *pEncConfig, ENC_CFG *pEncCfg)
  1450. {
  1451. int bitFormat;
  1452. ENC_CFG encCfgInst;
  1453. ENC_CFG *pCfg;
  1454. char yuvDir[256] = "yuv/";
  1455. // Source YUV Image File to load
  1456. if (pEncCfg) {
  1457. pCfg = pEncCfg;
  1458. }
  1459. else {
  1460. osal_memset( &encCfgInst, 0x00, sizeof(ENC_CFG));
  1461. pCfg = &encCfgInst;
  1462. }
  1463. bitFormat = pEncOP->bitstreamFormat;
  1464. switch(bitFormat) {
  1465. case STD_AVC:
  1466. if (parseAvcCfgFile(pCfg, pEncConfig->cfgFileName) == 0)
  1467. return 0;
  1468. pEncConfig->picQpY = pCfg->PicQpY;
  1469. if (pEncCfg)
  1470. strcpy(pEncConfig->yuvFileName, pCfg->SrcFileName);
  1471. else
  1472. sprintf(pEncConfig->yuvFileName, "%s%s", yuvDir, pCfg->SrcFileName);
  1473. #ifdef SUPPORT_980_RC_LIB
  1474. if(pEncOP->EncStdParam.avcParam.svcExtension == TRUE)
  1475. {
  1476. if(pCfg->RcEnable != 0)
  1477. pCfg->RcEnable = 4; // Other modes are not supported.
  1478. if(pCfg->field_flag == 1) // Field is not supported
  1479. return 0;
  1480. }
  1481. if (pCfg->coda9RoiEnable) {
  1482. strcpy(pEncConfig->roi_file_name, pCfg->RoiFile);
  1483. }
  1484. #endif
  1485. break;
  1486. case STD_MPEG4:
  1487. case STD_H263:
  1488. if (parseMp4CfgFile(pCfg, pEncConfig->cfgFileName) == 0)
  1489. return 0;
  1490. pEncConfig->picQpY = pCfg->VopQuant;
  1491. if (pEncCfg)
  1492. strcpy(pEncConfig->yuvFileName, pCfg->SrcFileName);
  1493. else
  1494. sprintf(pEncConfig->yuvFileName, "%s%s", yuvDir, pCfg->SrcFileName);
  1495. if (pCfg->ShortVideoHeader == 1) {
  1496. pEncOP->bitstreamFormat = STD_H263;
  1497. bitFormat = STD_H263;
  1498. }
  1499. break;
  1500. case STD_HEVC:
  1501. if (parseHevcCfgFile(pCfg, pEncConfig->cfgFileName) == 0)
  1502. return 0;
  1503. if (pEncCfg)
  1504. strcpy(pEncConfig->yuvFileName, pCfg->SrcFileName);
  1505. else
  1506. sprintf(pEncConfig->yuvFileName, "%s%s", yuvDir, pCfg->SrcFileName);
  1507. if (pEncConfig->bitstreamFileName[0] == 0 && pCfg->BitStreamFileName[0] != 0)
  1508. sprintf(pEncConfig->bitstreamFileName, "%s", pCfg->BitStreamFileName);
  1509. if ( pEncConfig->bitstreamFileName[0] == 0 )
  1510. sprintf(pEncConfig->bitstreamFileName, "%s", "output_stream.265");
  1511. if (pCfg->hevcCfg.ctuOptParam.roiEnable)
  1512. strcpy(pEncConfig->roi_file_name, pCfg->hevcCfg.roiFileName);
  1513. pEncConfig->roi_enable = pCfg->hevcCfg.ctuOptParam.roiEnable;
  1514. pEncConfig->roi_delta_qp= pCfg->hevcCfg.ctuOptParam.roiDeltaQp;
  1515. if (pCfg->hevcCfg.prefixSeiEnable)
  1516. strcpy(pEncConfig->prefix_sei_nal_file_name, pCfg->hevcCfg.prefixSeiDataFileName);
  1517. if (pCfg->hevcCfg.suffixSeiEnable)
  1518. strcpy(pEncConfig->suffix_sei_nal_file_name, pCfg->hevcCfg.suffixSeiDataFileName);
  1519. pEncConfig->seiDataEnc.prefixSeiNalEnable = pCfg->hevcCfg.prefixSeiEnable;
  1520. pEncConfig->seiDataEnc.prefixSeiDataSize = pCfg->hevcCfg.prefixSeiDataSize;
  1521. pEncConfig->seiDataEnc.prefixSeiDataEncOrder = pCfg->hevcCfg.prefixSeiTimingFlag;
  1522. pEncConfig->seiDataEnc.suffixSeiNalEnable = pCfg->hevcCfg.suffixSeiEnable;
  1523. pEncConfig->seiDataEnc.suffixSeiDataSize = pCfg->hevcCfg.suffixSeiDataSize;
  1524. pEncConfig->seiDataEnc.suffixSeiDataEncOrder = pCfg->hevcCfg.suffixSeiTimingFlag;
  1525. if (pCfg->hevcCfg.hrdInVPS || pCfg->hevcCfg.hrdInVUI)
  1526. strcpy(pEncConfig->hrd_rbsp_file_name, pCfg->hevcCfg.hrdDataFileName);
  1527. if (pCfg->hevcCfg.vuiDataEnable)
  1528. strcpy(pEncConfig->vui_rbsp_file_name, pCfg->hevcCfg.vuiDataFileName);
  1529. if (pCfg->hevcCfg.ctuOptParam.ctuQpEnable)
  1530. strcpy(pEncConfig->ctuqp_file_name, pCfg->hevcCfg.ctuQpFileName);
  1531. pEncConfig->ctu_qp_enable = pCfg->hevcCfg.ctuOptParam.ctuQpEnable;
  1532. pEncConfig->encAUD = pCfg->hevcCfg.encAUD;
  1533. pEncConfig->encEOS = pCfg->hevcCfg.encEOS;
  1534. pEncConfig->encEOB = pCfg->hevcCfg.encEOB;
  1535. pEncConfig->useAsLongtermPeriod = pCfg->hevcCfg.useAsLongtermPeriod;
  1536. pEncConfig->refLongtermPeriod = pCfg->hevcCfg.refLongtermPeriod;
  1537. break;
  1538. default :
  1539. break;
  1540. }
  1541. if (bitFormat == STD_HEVC) {
  1542. if (setWaveEncOpenParam(pEncOP, pEncConfig, pCfg) == 0)
  1543. return 0;
  1544. }
  1545. else {
  1546. if (setCoda9EncOpenParam(pEncOP, pEncConfig, pCfg) == 0)
  1547. return 0;
  1548. }
  1549. return 1;
  1550. }
  1551. int ReadBsResetBufHelper(Uint32 core_idx,
  1552. osal_file_t streamFp,
  1553. PhysicalAddress bitstream,
  1554. int size,
  1555. int endian)
  1556. {
  1557. unsigned char *buf = osal_malloc(size);
  1558. if (!buf)
  1559. {
  1560. VLOG(ERR, "fail to allocate bitstream buffer\n" );
  1561. return 0;
  1562. }
  1563. vdi_read_memory(core_idx, bitstream, buf, size, endian);
  1564. osal_fwrite((void *)buf, sizeof(Uint8), size, streamFp);
  1565. osal_fflush(streamFp);
  1566. osal_free(buf);
  1567. return 1;
  1568. }
  1569. /*
  1570. * To be compatible with Ref-SW 4.0
  1571. */
  1572. frame_queue_item_t* frame_queue_init(
  1573. Int32 count
  1574. )
  1575. {
  1576. frame_queue_item_t* queue = NULL;
  1577. queue = (frame_queue_item_t *)osal_malloc(sizeof(frame_queue_item_t));
  1578. if (!queue) {
  1579. return NULL;
  1580. }
  1581. queue->size = count;
  1582. queue->count = 0;
  1583. queue->front = 0;
  1584. queue->rear = 0;
  1585. queue->buffer = (DecOutputInfo*)osal_malloc(count*sizeof(DecOutputInfo));
  1586. return queue;
  1587. }
  1588. void frame_queue_deinit(
  1589. frame_queue_item_t* queue
  1590. )
  1591. {
  1592. if (queue == NULL) {
  1593. return;
  1594. }
  1595. if (queue->buffer) {
  1596. osal_free(queue->buffer);
  1597. }
  1598. osal_free(queue);
  1599. }
  1600. /*
  1601. * Return 0 on success.
  1602. * -1 on failure
  1603. */
  1604. Int32 frame_queue_enqueue(
  1605. frame_queue_item_t* queue,
  1606. DecOutputInfo data
  1607. )
  1608. {
  1609. if (queue == NULL) {
  1610. return -1;
  1611. }
  1612. /* Queue is full */
  1613. if (queue->count == queue->size) {
  1614. return -1;
  1615. }
  1616. queue->buffer[queue->rear++] = data;
  1617. queue->rear %= queue->size;
  1618. queue->count++;
  1619. return 0;
  1620. }
  1621. /*
  1622. * Return 0 on success.
  1623. * -1 on failure
  1624. */
  1625. Int32 frame_queue_dequeue(
  1626. frame_queue_item_t* queue,
  1627. DecOutputInfo* data
  1628. )
  1629. {
  1630. if (queue == NULL) {
  1631. return -1;
  1632. }
  1633. /* Queue is empty */
  1634. if (queue->count == 0) {
  1635. return -1;
  1636. }
  1637. *data = queue->buffer[queue->front++];
  1638. queue->front %= queue->size;
  1639. queue->count--;
  1640. return 0;
  1641. }
  1642. Int32 frame_queue_dequeue_all(
  1643. frame_queue_item_t* queue
  1644. )
  1645. {
  1646. Int32 ret;
  1647. DecOutputInfo data;
  1648. if (queue == NULL) {
  1649. return -1;
  1650. }
  1651. do {
  1652. ret = frame_queue_dequeue(queue, &data);
  1653. if (ret >=0) {
  1654. VLOG(INFO, "Empty display Queue for flush display_index=%d\n", data.indexFrameDisplay);
  1655. }
  1656. } while (ret >= 0);
  1657. return 0;
  1658. }
  1659. Int32 frame_queue_peekqueue(
  1660. frame_queue_item_t* queue,
  1661. DecOutputInfo* data
  1662. )
  1663. {
  1664. if (queue == NULL) {
  1665. return -1;
  1666. }
  1667. /* Queue is empty */
  1668. if (queue->count == 0) {
  1669. return -1;
  1670. }
  1671. *data = queue->buffer[queue->front];
  1672. return 0;
  1673. }
  1674. Int32 frame_queue_check_in_queue(
  1675. frame_queue_item_t* queue,
  1676. Int32 index
  1677. )
  1678. {
  1679. DecOutputInfo data;
  1680. Int32 front;
  1681. Int32 count;
  1682. if (queue == NULL) {
  1683. return -1;
  1684. }
  1685. front = queue->front;
  1686. count = queue->count;
  1687. while(count > 0) {
  1688. data = queue->buffer[front++];
  1689. if (data.indexFrameDisplay == index) {
  1690. return 1;
  1691. }
  1692. count--;
  1693. front %= queue->size;
  1694. }
  1695. return 0;
  1696. }
  1697. Int32 frame_queue_count(
  1698. frame_queue_item_t* queue
  1699. )
  1700. {
  1701. if (queue == NULL) {
  1702. return -1;
  1703. }
  1704. return queue->count;
  1705. }
  1706. #ifdef SUPPORT_980_RC_LIB
  1707. #define AVC_ENC_NUT_SLICE 1
  1708. #define AVC_ENC_NUT_IDR 5
  1709. static frm_t *get_current_frame(
  1710. EncInfo *seq)
  1711. {
  1712. int i;
  1713. frm_t *curr_frm = NULL;
  1714. if (seq->openParam.gopSize== 1) //all intra
  1715. {
  1716. curr_frm = &seq->frm[0];
  1717. seq->prev_idx = 0;
  1718. }
  1719. else
  1720. {
  1721. for (i = 0; i < seq->num_total_frames; i++)
  1722. {
  1723. if (seq->frm[i].used_for_ref == 0)
  1724. {
  1725. curr_frm = &seq->frm[i];
  1726. seq->prev_idx = i;
  1727. break;
  1728. }
  1729. }
  1730. }
  1731. return curr_frm;
  1732. }
  1733. static void set_mmco(
  1734. EncInfo *seq)
  1735. {
  1736. int frmidx;
  1737. int next_idx = (seq->enc_idx_modulo == seq->gop_size - 1) ? 0 : seq->enc_idx_modulo + 1;
  1738. rps_t *next_rps = &seq->rps[next_idx];
  1739. int curr_gop_poc = seq->curr_frm->poc - seq->gop_entry[seq->enc_idx_modulo].curr_poc;
  1740. int next_gop_poc = curr_gop_poc + ((next_idx == 0) ? seq->gop_size : 0);
  1741. //mmco1 : set short-term as non-ref
  1742. seq->num_mmco1 = 0;
  1743. for (frmidx = 0; frmidx < seq->num_total_frames; frmidx++)
  1744. {
  1745. frm_t *frm = &seq->frm[frmidx];
  1746. if (frm->used_for_ref && (frm != seq->curr_frm) && (frmidx != seq->longterm_frmidx) )
  1747. {
  1748. int i;
  1749. int found = 0;
  1750. //check if the ref-frame is in next rps
  1751. for (i = 0; i < next_rps->num_poc; i++)
  1752. if (frm->poc == next_gop_poc + next_rps->poc[i])
  1753. found = 1;
  1754. if (!found)
  1755. {
  1756. seq->mmco1_frmidx[seq->num_mmco1++] = frmidx;
  1757. }
  1758. }
  1759. }
  1760. }
  1761. static int search_ref_frame(
  1762. EncInfo *seq,
  1763. int poc)
  1764. {
  1765. int i;
  1766. for (i = 0; i < 31; i++)
  1767. if (seq->frm[i].used_for_ref && seq->frm[i].poc == poc)
  1768. return i;
  1769. return -1;
  1770. }
  1771. static void set_refpic_list(
  1772. EncInfo *seq,
  1773. int ref_long_term)
  1774. {
  1775. int ref, list;
  1776. int num_ref, num_list;
  1777. int enc_idx;
  1778. int gop_poc;
  1779. enc_idx = seq->enc_idx_modulo;
  1780. gop_poc = seq->curr_frm->poc - seq->gop_entry[enc_idx].curr_poc;
  1781. seq->num_ref_idx = 0;
  1782. seq->ref_pic_list =0;
  1783. //fill ref_list
  1784. num_ref = 1;//(seq->curr_frm->slice_type == AVC_ENC_SLICE_TYPE_B || seq->gop_entry[enc_idx].use_multi_ref_p) ? 2 : 1;
  1785. for (ref = 0; ref < num_ref; ref++)
  1786. {
  1787. int frmidx;
  1788. int poc = gop_poc + seq->gop_entry[enc_idx].ref_poc;
  1789. if (ref_long_term && (ref == num_ref - 1) && (seq->longterm_frmidx >= 0)) //long-term
  1790. frmidx = seq->longterm_frmidx;
  1791. else
  1792. frmidx = search_ref_frame(seq, poc);
  1793. if (frmidx != -1)
  1794. {
  1795. seq->ref_pic_list = frmidx;
  1796. seq->num_ref_idx++;
  1797. }
  1798. }
  1799. //when num_ref_idx is 0, fill ref_list with rps
  1800. num_list = 1;
  1801. for (list = 0; list < num_list; list++)
  1802. {
  1803. if (seq->num_ref_idx == 0)
  1804. {
  1805. int i;
  1806. rps_t *rps = &seq->rps[enc_idx];
  1807. for (i = 0; i < rps->num_poc; i++)
  1808. {
  1809. int poc = gop_poc + rps->poc[i];
  1810. int frmidx = search_ref_frame(seq, poc);
  1811. if (frmidx != -1)
  1812. {
  1813. seq->ref_pic_list = frmidx;
  1814. seq->num_ref_idx++;
  1815. break;
  1816. }
  1817. }
  1818. }
  1819. }
  1820. }
  1821. static int get_nal_ref_idc(
  1822. EncInfo *seq,
  1823. int enc_idx)
  1824. {
  1825. if (seq->idr_picture)
  1826. return 3;
  1827. else if (seq->gop_entry[enc_idx].curr_poc == seq->gop_size) //anchor
  1828. return 2;
  1829. else if (seq->curr_frm->used_for_ref)
  1830. return 1;
  1831. else
  1832. return 0;
  1833. }
  1834. #define AVC_ENC_NUT_SLICE 1
  1835. #define AVC_ENC_NUT_IDR 5
  1836. void dpb_pic_init( EncInfo *pAvcInfo )
  1837. {
  1838. int encode_frame;
  1839. int enc_idx;// = pAvcInfo->enc_idx_modulo;
  1840. int i;
  1841. int gop_num = pAvcInfo->openParam.gopSize;
  1842. encode_frame = pAvcInfo->encoded_frames_in_gop;
  1843. enc_idx = pAvcInfo->enc_idx_modulo;
  1844. if (gop_num == 1) // All I
  1845. encode_frame = 0;
  1846. else
  1847. {
  1848. if (encode_frame >= gop_num && gop_num != 0) // I frame
  1849. encode_frame = 0;
  1850. }
  1851. pAvcInfo->idr_picture = 0;
  1852. if(pAvcInfo->frm_cnt == 0)
  1853. {
  1854. pAvcInfo->Idr_cnt = 0;
  1855. pAvcInfo->idr_picture = 1;
  1856. }
  1857. else
  1858. {
  1859. if((pAvcInfo->openParam.idrInterval) > 0 && (pAvcInfo->openParam.gopSize > 0))
  1860. {
  1861. if(encode_frame == 0)
  1862. {
  1863. pAvcInfo->Idr_cnt++;
  1864. }
  1865. if(pAvcInfo->openParam.idrInterval == pAvcInfo->Idr_cnt)
  1866. {
  1867. pAvcInfo->frm_cnt = 0;
  1868. pAvcInfo->Idr_cnt = 0;
  1869. pAvcInfo->idr_picture = 1;
  1870. }
  1871. }
  1872. }
  1873. if (pAvcInfo->idr_picture)
  1874. pAvcInfo->Idr_picId = (pAvcInfo->Idr_picId + 1) & 0xFFFF; // Idr_pic_id: 0 ~ 65535
  1875. pAvcInfo->slice_type = (encode_frame == 0) ? 0 : 1;
  1876. if(pAvcInfo->enc_long_term_cnt > 0)
  1877. {
  1878. if(pAvcInfo->enc_long_term_cnt == pAvcInfo->long_term_period)
  1879. pAvcInfo->enc_long_term_cnt = 0;
  1880. }
  1881. if(pAvcInfo->enc_long_term_cnt == 0 && pAvcInfo->long_term_period != -1 )
  1882. pAvcInfo->curr_long_term = 1;// enc_long_term_cnt % pAvcInfo->long_term_period;
  1883. else
  1884. pAvcInfo->curr_long_term = 0;
  1885. pAvcInfo->ref_long_term = pAvcInfo->curr_long_term || pAvcInfo->gop_entry[pAvcInfo->enc_idx_modulo].ref_long_term;
  1886. if(pAvcInfo->enc_idx_modulo >=0)
  1887. pAvcInfo->src_idx = pAvcInfo->enc_idx_gop + pAvcInfo->gop_entry[pAvcInfo->enc_idx_modulo].curr_poc;
  1888. else
  1889. pAvcInfo->src_idx = pAvcInfo->enc_idx_gop;
  1890. pAvcInfo->num_mmco1 = 0;
  1891. pAvcInfo->curr_frm = get_current_frame(pAvcInfo);
  1892. if(pAvcInfo->idr_picture)
  1893. pAvcInfo->src_idx_last_idr = pAvcInfo->src_idx;
  1894. if (pAvcInfo->idr_picture )
  1895. {
  1896. //pAvcInfo->Idr_picId ^= 1;
  1897. //clear all refs
  1898. for (i = 0; i < 31; i++)
  1899. pAvcInfo->frm[i].used_for_ref = 0;
  1900. pAvcInfo->curr_frm->used_for_ref = 1;
  1901. pAvcInfo->curr_frm->slice_type = 0;//I_SLICE;
  1902. pAvcInfo->curr_frm->poc = 0;
  1903. pAvcInfo->curr_frm->frame_num = 0;
  1904. }
  1905. else
  1906. {
  1907. int enc_idx = pAvcInfo->enc_idx_modulo;
  1908. pAvcInfo->curr_frm->used_for_ref = pAvcInfo->rps[enc_idx].used_for_ref || pAvcInfo->curr_long_term;
  1909. pAvcInfo->curr_frm->slice_type = pAvcInfo->slice_type;
  1910. pAvcInfo->curr_frm->poc = pAvcInfo->src_idx - pAvcInfo->src_idx_last_idr;
  1911. pAvcInfo->curr_frm->frame_num = (pAvcInfo->prev_frame_num + 1) % (1 << 4);
  1912. if (pAvcInfo->curr_frm->used_for_ref)
  1913. set_mmco(pAvcInfo);
  1914. if (pAvcInfo->curr_frm->slice_type != 0) // I_SLICE)
  1915. set_refpic_list(pAvcInfo,pAvcInfo->ref_long_term);
  1916. }
  1917. pAvcInfo->nal_ref_idc = get_nal_ref_idc(pAvcInfo, enc_idx);
  1918. pAvcInfo->nal_unit_type = pAvcInfo->idr_picture ? AVC_ENC_NUT_IDR : AVC_ENC_NUT_SLICE;
  1919. //update prev_frame_num
  1920. if (pAvcInfo->curr_frm->used_for_ref)
  1921. pAvcInfo->prev_frame_num = pAvcInfo->curr_frm->frame_num;
  1922. // pAvcInfo->RefPicList[0] = pAvcInfo->ref_pic_list;
  1923. pAvcInfo->temporal_id = (pAvcInfo->enc_idx_modulo == -1) ? 0 : pAvcInfo->gop_entry[pAvcInfo->enc_idx_modulo].temporal_id;
  1924. if(pAvcInfo->curr_long_term)
  1925. pAvcInfo->temporal_id = 0;
  1926. //Update cnt
  1927. pAvcInfo->enc_long_term_cnt++;
  1928. pAvcInfo->frm_cnt++;
  1929. }
  1930. #endif