decoder.js 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. var isFinish = false;
  2. var isSwitchSharpness = false;
  3. var h264Queue = [];
  4. var ret;
  5. var maxWidth = 1080;
  6. var maxHeight = 1920;
  7. var globalYuvPtr = undefined;
  8. var golbalYuvData;//全局,只分配一次
  9. var renderCount = 0;
  10. function doSomeInit() {
  11. var allocSize = maxWidth * maxHeight * 3;
  12. golbalYuvData = new Uint8Array(allocSize);
  13. }
  14. self.importScripts("ffmpeghelper.js");
  15. self.Module.onRuntimeInitialized = function () {
  16. isFinish = true;
  17. doSomeInit();
  18. ret = Module._openDecoder();
  19. if (!ret) {
  20. console.log("打开编码器成功");
  21. }
  22. }
  23. self.addEventListener('message', function (e) {
  24. var msg = e.data;
  25. if (msg.type == "rawData") {
  26. var buffer = e.data.data;
  27. if (buffer[0] !== 0xff) { // 音频解码
  28. if (isSwitchSharpness) {
  29. var type = buffer[4] & 0x1f;
  30. if (type == 7 || type == 8) {
  31. console.log("视频类型:" + type);
  32. isSwitchSharpness = false;
  33. h264Queue.push(buffer);
  34. }
  35. } else {
  36. h264Queue.push(buffer);
  37. }
  38. }
  39. }
  40. if (msg.type == "switchSharpness") { // 分辨率切换实现
  41. isSwitchSharpness = true;
  42. closeDecoder();
  43. var ret = Module._openDecoder();//再次开启解码器
  44. var timeFlag = setInterval(decodeVideo, 1);
  45. console.log("切换解码器成功");
  46. }
  47. }, false);
  48. function PrintfLog(str) {
  49. var curTime = new Date().getTime();
  50. var objData = {
  51. cmd: 0,
  52. data: str,
  53. time: curTime
  54. };
  55. self.postMessage(objData);
  56. }
  57. function doRequestIFrame() {
  58. var objData = {
  59. cmd: 5,
  60. };
  61. self.postMessage(objData);
  62. }
  63. function decodeVideo() {
  64. if (h264Queue.length > 0 && isFinish) {
  65. decodeH264(h264Queue.shift());
  66. }
  67. }
  68. var timeFlag = setInterval(decodeVideo, 1);
  69. //实现C 端到js yuv数据赋值, 并分配给opengl 渲染
  70. /*yuvData, 存放yuv数据的数组
  71. * inputYuvPtr C++
  72. */
  73. function dispatchYuvData(yuvData, inputYuvPtr, videoWidth, videoHeight, copyLen) {
  74. for (i = 0; i < copyLen; i++) {
  75. yuvData[i] = Module.HEAPU8[(inputYuvPtr) + i];
  76. }
  77. var curTime = new Date().getTime();
  78. var objData = {
  79. cmd: 1,
  80. data: yuvData,
  81. time: curTime,
  82. width: videoWidth,
  83. height: videoHeight
  84. };
  85. self.postMessage(objData);
  86. }
  87. function decodeH264(data) {
  88. var frameWidth = 0;
  89. var frameHeight = 0;
  90. var inputPtr = Module._malloc(data.length); //输入数据
  91. for (i = 0; i < data.length; i++) {
  92. Module.HEAPU8[(inputPtr) + i] = data[i]; //转换为堆数据
  93. }
  94. var time = new Date().getTime();
  95. var allocSize = maxWidth * maxHeight * 3 / 2;
  96. if (globalYuvPtr == undefined) {
  97. globalYuvPtr = Module._malloc(allocSize);
  98. }
  99. var ret = Module._feedData(inputPtr, data.length, globalYuvPtr);
  100. if (ret >= 0)//解码成功才考虑渲染
  101. {
  102. frameWidth = Module._getVideoWidth();//拿到解码器宽、高
  103. frameHeight = Module._getVideoHeight();
  104. var curCost = new Date().getTime() - time;
  105. var copyLen = frameWidth * frameHeight * 3 / 2;//只拷贝必须的长度
  106. if (renderCount > 1)//第一帧因为画面时全绿色的不渲染
  107. {
  108. dispatchYuvData(golbalYuvData, globalYuvPtr, frameWidth, frameHeight, copyLen);
  109. } else {
  110. renderCount++;
  111. }
  112. }
  113. Module._free(inputPtr);
  114. }
  115. function closeDecoder() {
  116. clearInterval(timeFlag);//关闭原有定时器
  117. Module._closeDecoder();
  118. renderCount = 0;
  119. if (globalYuvPtr != undefined) {
  120. Module._free(globalYuvPtr);
  121. globalYuvPtr = undefined;
  122. }
  123. console.log("此时buffer长度: %d", h264Queue.length);
  124. while (h264Queue.lengh > 0) {
  125. h264Queue.shift();
  126. }
  127. }