mediabunny 1.0.5 → 1.1.0

package/dist/bundles/mediabunny.cjs

@@ -6224,6 +6224,9 @@ ${cue.notes ?? ""}`;
       return null;
     }
     reader.pos += 1;
+    if (firstByte !== 255) {
+      return null;
+    }
     if ((secondByte & 224) !== 224) {
       return null;
     }
@@ -7366,18 +7369,52 @@ ${cue.notes ?? ""}`;
         });
       }
     }
-    /**
-     * Draws the video sample to a 2D canvas context. Rotation metadata will be taken into account.
-     *
-     * @param dx - The x-coordinate in the destination canvas at which to place the top-left corner of the source image.
-     * @param dy - The y-coordinate in the destination canvas at which to place the top-left corner of the source image.
-     * @param dWidth - The width in pixels with which to draw the image in the destination canvas.
-     * @param dHeight - The height in pixels with which to draw the image in the destination canvas.
-     */
-    draw(context, dx, dy, dWidth = this.displayWidth, dHeight = this.displayHeight) {
+    draw(context, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8) {
+      let sx = 0;
+      let sy = 0;
+      let sWidth = this.displayWidth;
+      let sHeight = this.displayHeight;
+      let dx = 0;
+      let dy = 0;
+      let dWidth = this.displayWidth;
+      let dHeight = this.displayHeight;
+      if (arg5 !== void 0) {
+        sx = arg1;
+        sy = arg2;
+        sWidth = arg3;
+        sHeight = arg4;
+        dx = arg5;
+        dy = arg6;
+        if (arg7 !== void 0) {
+          dWidth = arg7;
+          dHeight = arg8;
+        } else {
+          dWidth = sWidth;
+          dHeight = sHeight;
+        }
+      } else {
+        dx = arg1;
+        dy = arg2;
+        if (arg3 !== void 0) {
+          dWidth = arg3;
+          dHeight = arg4;
+        }
+      }
       if (!(typeof CanvasRenderingContext2D !== "undefined" && context instanceof CanvasRenderingContext2D || typeof OffscreenCanvasRenderingContext2D !== "undefined" && context instanceof OffscreenCanvasRenderingContext2D)) {
         throw new TypeError("context must be a CanvasRenderingContext2D or OffscreenCanvasRenderingContext2D.");
       }
+      if (!Number.isFinite(sx)) {
+        throw new TypeError("sx must be a number.");
+      }
+      if (!Number.isFinite(sy)) {
+        throw new TypeError("sy must be a number.");
+      }
+      if (!Number.isFinite(sWidth) || sWidth < 0) {
+        throw new TypeError("sWidth must be a non-negative number.");
+      }
+      if (!Number.isFinite(sHeight) || sHeight < 0) {
+        throw new TypeError("sHeight must be a non-negative number.");
+      }
       if (!Number.isFinite(dx)) {
         throw new TypeError("dx must be a number.");
       }
@@ -7393,6 +7430,26 @@ ${cue.notes ?? ""}`;
       if (this._closed) {
         throw new Error("VideoSample is closed.");
       }
+      if (this.rotation === 90) {
+        [sx, sy, sWidth, sHeight] = [
+          sy,
+          this.codedHeight - sx - sWidth,
+          sHeight,
+          sWidth
+        ];
+      } else if (this.rotation === 180) {
+        [sx, sy] = [
+          this.codedWidth - sx - sWidth,
+          this.codedHeight - sy - sHeight
+        ];
+      } else if (this.rotation === 270) {
+        [sx, sy, sWidth, sHeight] = [
+          this.codedWidth - sy - sHeight,
+          sx,
+          sHeight,
+          sWidth
+        ];
+      }
       const source = this.toCanvasImageSource();
       context.save();
       const centerX = dx + dWidth / 2;
@@ -7403,6 +7460,10 @@ ${cue.notes ?? ""}`;
       context.scale(1 / aspectRatioChange, aspectRatioChange);
       context.drawImage(
         source,
+        sx,
+        sy,
+        sWidth,
+        sHeight,
         -dWidth / 2,
         -dHeight / 2,
         dWidth,
@@ -8367,6 +8428,7 @@ ${cue.notes ?? ""}`;
     finalizeAndEmitSample(sample) {
       sample.setTimestamp(Math.round(sample.timestamp * this.timeResolution) / this.timeResolution);
       sample.setDuration(Math.round(sample.duration * this.timeResolution) / this.timeResolution);
+      sample.setRotation(this.rotation);
       this.onSample(sample);
     }
     getDecodeQueueSize() {
@@ -11823,6 +11885,7 @@ ${cue.notes ?? ""}`;
               fragmentLookupTable: null,
               currentFragmentState: null,
               fragments: [],
+              fragmentsWithKeyFrame: [],
               editListPreviousSegmentDurations: 0,
               editListOffset: 0
             };
@@ -11911,7 +11974,8 @@ ${cue.notes ?? ""}`;
                 continue;
               }
               if (mediaRate !== 1) {
-                throw new Error("Unsupported edit list: media rate must be 1.");
+                console.warn("Unsupported edit list entry: media rate must be 1.");
+                break;
               }
               track.editListPreviousSegmentDurations = previousSegmentDurations;
               track.editListOffset = mediaTime;
@@ -12099,7 +12163,8 @@ ${cue.notes ?? ""}`;
                   } else if (sampleSize === 16) {
                     track.info.codec = "pcm-s16be";
                   } else {
-                    throw new Error(`Unsupported sample size ${sampleSize} for codec 'twos'.`);
+                    console.warn(`Unsupported sample size ${sampleSize} for codec 'twos'.`);
+                    track.info.codec = null;
                   }
                 } else if (lowercaseBoxName === "sowt") {
                   if (sampleSize === 8) {
@@ -12107,7 +12172,8 @@ ${cue.notes ?? ""}`;
                   } else if (sampleSize === 16) {
                     track.info.codec = "pcm-s16";
                   } else {
-                    throw new Error(`Unsupported sample size ${sampleSize} for codec 'sowt'.`);
+                    console.warn(`Unsupported sample size ${sampleSize} for codec 'sowt'.`);
+                    track.info.codec = null;
                   }
                 } else if (lowercaseBoxName === "raw ") {
                   track.info.codec = "pcm-u8";
@@ -12327,7 +12393,8 @@ ${cue.notes ?? ""}`;
                 } else if (pcmSampleSize === 32) {
                   track.info.codec = "pcm-s32";
                 } else {
-                  throw new Error(`Invalid ipcm sample size ${pcmSampleSize}.`);
+                  console.warn(`Invalid ipcm sample size ${pcmSampleSize}.`);
+                  track.info.codec = null;
                 }
               } else {
                 if (pcmSampleSize === 16) {
@@ -12337,7 +12404,8 @@ ${cue.notes ?? ""}`;
                 } else if (pcmSampleSize === 32) {
                   track.info.codec = "pcm-s32be";
                 } else {
-                  throw new Error(`Invalid ipcm sample size ${pcmSampleSize}.`);
+                  console.warn(`Invalid ipcm sample size ${pcmSampleSize}.`);
+                  track.info.codec = null;
                 }
               }
             } else if (track.info.codec === "pcm-f32be") {
@@ -12347,7 +12415,8 @@ ${cue.notes ?? ""}`;
                 } else if (pcmSampleSize === 64) {
                   track.info.codec = "pcm-f64";
                 } else {
-                  throw new Error(`Invalid fpcm sample size ${pcmSampleSize}.`);
+                  console.warn(`Invalid fpcm sample size ${pcmSampleSize}.`);
+                  track.info.codec = null;
                 }
               } else {
                 if (pcmSampleSize === 32) {
@@ -12355,7 +12424,8 @@ ${cue.notes ?? ""}`;
                 } else if (pcmSampleSize === 64) {
                   track.info.codec = "pcm-f64be";
                 } else {
-                  throw new Error(`Invalid fpcm sample size ${pcmSampleSize}.`);
+                  console.warn(`Invalid fpcm sample size ${pcmSampleSize}.`);
+                  track.info.codec = null;
                 }
               }
             }
@@ -12505,9 +12575,24 @@ ${cue.notes ?? ""}`;
           break;
         case "stz2":
           {
-            throw new Error("Unsupported.");
+            const track = this.currentTrack;
+            assert(track);
+            if (!track.sampleTable) {
+              break;
+            }
+            this.metadataReader.pos += 4;
+            this.metadataReader.pos += 3;
+            const fieldSize = this.metadataReader.readU8();
+            const sampleCount = this.metadataReader.readU32();
+            const bytes2 = this.metadataReader.readBytes(Math.ceil(sampleCount * fieldSize / 8));
+            const bitstream = new Bitstream(bytes2);
+            for (let i = 0; i < sampleCount; i++) {
+              const sampleSize = bitstream.readBits(fieldSize);
+              track.sampleTable.sampleSizes.push(sampleSize);
+            }
           }
           ;
+          break;
         case "stss":
           {
             const track = this.currentTrack;
@@ -12705,6 +12790,15 @@ ${cue.notes ?? ""}`;
                   (x) => x.moofOffset
                 );
                 this.currentTrack.fragments.splice(insertionIndex + 1, 0, this.currentFragment);
+                const hasKeyFrame = trackData.firstKeyFrameTimestamp !== null;
+                if (hasKeyFrame) {
+                  const insertionIndex2 = binarySearchLessOrEqual(
+                    this.currentTrack.fragmentsWithKeyFrame,
+                    this.currentFragment.moofOffset,
+                    (x) => x.moofOffset
+                  );
+                  this.currentTrack.fragmentsWithKeyFrame.splice(insertionIndex2 + 1, 0, this.currentFragment);
+                }
                 const { currentFragmentState } = this.currentTrack;
                 assert(currentFragmentState);
                 if (currentFragmentState.startTimestamp !== null) {
@@ -12791,7 +12885,8 @@ ${cue.notes ?? ""}`;
             assert(this.currentFragment);
             assert(track.currentFragmentState);
             if (this.currentFragment.trackData.has(track.id)) {
-              throw new Error("Can't have two trun boxes for the same track in one fragment.");
+              console.warn("Can't have two trun boxes for the same track in one fragment. Ignoring...");
+              break;
             }
             const version = this.metadataReader.readU8();
             const flags = this.metadataReader.readU24();
@@ -12819,6 +12914,7 @@ ${cue.notes ?? ""}`;
             const trackData = {
               startTimestamp: 0,
               endTimestamp: 0,
+              firstKeyFrameTimestamp: null,
               samples: [],
               presentationTimestamps: [],
               startTimestampIsFinal: false
@@ -12869,11 +12965,16 @@ ${cue.notes ?? ""}`;
               currentTimestamp += sampleDuration;
             }
             trackData.presentationTimestamps = trackData.samples.map((x, i) => ({ presentationTimestamp: x.presentationTimestamp, sampleIndex: i })).sort((a, b) => a.presentationTimestamp - b.presentationTimestamp);
-            for (let i = 0; i < trackData.presentationTimestamps.length - 1; i++) {
-              const current = trackData.presentationTimestamps[i];
-              const next = trackData.presentationTimestamps[i + 1];
-              const duration = next.presentationTimestamp - current.presentationTimestamp;
-              trackData.samples[current.sampleIndex].duration = duration;
+            for (let i = 0; i < trackData.presentationTimestamps.length; i++) {
+              const currentEntry = trackData.presentationTimestamps[i];
+              const currentSample = trackData.samples[currentEntry.sampleIndex];
+              if (trackData.firstKeyFrameTimestamp === null && currentSample.isKeyFrame) {
+                trackData.firstKeyFrameTimestamp = currentSample.presentationTimestamp;
+              }
+              if (i < trackData.presentationTimestamps.length - 1) {
+                const nextEntry = trackData.presentationTimestamps[i + 1];
+                currentSample.duration = nextEntry.presentationTimestamp - currentEntry.presentationTimestamp;
+              }
             }
             const firstSample = trackData.samples[trackData.presentationTimestamps[0].sampleIndex];
             const lastSample = trackData.samples[last(trackData.presentationTimestamps).sampleIndex];
@@ -13072,7 +13173,7 @@ ${cue.notes ?? ""}`;
               while (currentFragment.nextFragment) {
                 currentFragment = currentFragment.nextFragment;
                 const trackData2 = currentFragment.trackData.get(this.internalTrack.id);
-                if (trackData2) {
+                if (trackData2 && trackData2.firstKeyFrameTimestamp !== null) {
                   const fragmentIndex2 = binarySearchExact(
                     this.internalTrack.fragments,
                     currentFragment.moofOffset,
@@ -13080,9 +13181,7 @@ ${cue.notes ?? ""}`;
                   );
                   assert(fragmentIndex2 !== -1);
                   const keyFrameIndex = trackData2.samples.findIndex((x) => x.isKeyFrame);
-                  if (keyFrameIndex === -1) {
-                    throw new Error("Not supported: Fragment does not contain key sample.");
-                  }
+                  assert(keyFrameIndex !== -1);
                   return {
                     fragmentIndex: fragmentIndex2,
                     sampleIndex: keyFrameIndex,
@@ -13195,24 +13294,29 @@ ${cue.notes ?? ""}`;
       return { fragmentIndex, sampleIndex, correctSampleFound };
     }
     findKeySampleInFragmentsForTimestamp(timestampInTimescale) {
-      const fragmentIndex = binarySearchLessOrEqual(
+      const indexInKeyFrameFragments = binarySearchLessOrEqual(
         // This array is technically not sorted by start timestamp, but for any reasonable file, it basically is.
-        this.internalTrack.fragments,
+        this.internalTrack.fragmentsWithKeyFrame,
         timestampInTimescale,
         (x) => x.trackData.get(this.internalTrack.id).startTimestamp
       );
+      let fragmentIndex = -1;
       let sampleIndex = -1;
       let correctSampleFound = false;
-      if (fragmentIndex !== -1) {
-        const fragment = this.internalTrack.fragments[fragmentIndex];
+      if (indexInKeyFrameFragments !== -1) {
+        const fragment = this.internalTrack.fragmentsWithKeyFrame[indexInKeyFrameFragments];
+        fragmentIndex = binarySearchExact(
+          this.internalTrack.fragments,
+          fragment.moofOffset,
+          (x) => x.moofOffset
+        );
+        assert(fragmentIndex !== -1);
         const trackData = fragment.trackData.get(this.internalTrack.id);
         const index = findLastIndex(trackData.presentationTimestamps, (x) => {
           const sample = trackData.samples[x.sampleIndex];
           return sample.isKeyFrame && x.presentationTimestamp <= timestampInTimescale;
         });
-        if (index === -1) {
-          throw new Error("Not supported: Fragment does not begin with a key sample.");
-        }
+        assert(index !== -1);
         const entry = trackData.presentationTimestamps[index];
         sampleIndex = entry.sampleIndex;
         correctSampleFound = timestampInTimescale < trackData.endTimestamp;
@@ -13735,20 +13839,15 @@ ${cue.notes ?? ""}`;
           trackData.blocks = sortBlocksByReferences(trackData.blocks);
         }
         trackData.presentationTimestamps = trackData.blocks.map((block, i) => ({ timestamp: block.timestamp, blockIndex: i })).sort((a, b) => a.timestamp - b.timestamp);
-        let hasKeyFrame = false;
         for (let i = 0; i < trackData.presentationTimestamps.length; i++) {
-          const entry = trackData.presentationTimestamps[i];
-          const block = trackData.blocks[entry.blockIndex];
-          if (block.isKeyFrame) {
-            hasKeyFrame = true;
-            if (trackData.firstKeyFrameTimestamp === null && block.isKeyFrame) {
-              trackData.firstKeyFrameTimestamp = block.timestamp;
-            }
+          const currentEntry = trackData.presentationTimestamps[i];
+          const currentBlock = trackData.blocks[currentEntry.blockIndex];
+          if (trackData.firstKeyFrameTimestamp === null && currentBlock.isKeyFrame) {
+            trackData.firstKeyFrameTimestamp = currentBlock.timestamp;
           }
           if (i < trackData.presentationTimestamps.length - 1) {
             const nextEntry = trackData.presentationTimestamps[i + 1];
-            const nextBlock = trackData.blocks[nextEntry.blockIndex];
-            block.duration = nextBlock.timestamp - block.timestamp;
+            currentBlock.duration = nextEntry.timestamp - currentBlock.timestamp;
           }
         }
         const firstBlock = trackData.blocks[trackData.presentationTimestamps[0].blockIndex];
@@ -13763,6 +13862,7 @@ ${cue.notes ?? ""}`;
             (x) => x.elementStartPos
           );
           track.clusters.splice(insertionIndex2 + 1, 0, cluster);
+          const hasKeyFrame = trackData.firstKeyFrameTimestamp !== null;
           if (hasKeyFrame) {
             const insertionIndex3 = binarySearchLessOrEqual(
               track.clustersWithKeyFrame,
@@ -14801,7 +14901,7 @@ ${cue.notes ?? ""}`;
     readNextFrameHeader(until) {
       assert(this.fileSize);
       until ??= this.fileSize;
-      while (this.pos < until - FRAME_HEADER_SIZE) {
+      while (this.pos <= until - FRAME_HEADER_SIZE) {
         const word = this.readU32();
         this.pos -= 4;
         const header = readFrameHeader(word, this);
@@ -15843,7 +15943,8 @@ ${cue.notes ?? ""}`;
         return true;
       }
       mp3Reader.pos = firstHeader.startPos + firstHeader.totalSize;
-      const secondHeader = mp3Reader.readNextFrameHeader(Math.min(framesStartPos + 4096, sourceSize));
+      await mp3Reader.reader.loadRange(mp3Reader.pos, mp3Reader.pos + FRAME_HEADER_SIZE);
+      const secondHeader = mp3Reader.readNextFrameHeader(mp3Reader.pos + FRAME_HEADER_SIZE);
       if (!secondHeader) {
         return false;
       }