fix closestFrame

devdanim · Jan 10, 2024 · 5bd35ff · 5bd35ff
1 parent 1f803cd
commit 5bd35ff
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 15 deletions.
diff --git a/framefusion.ts b/framefusion.ts
@@ -58,19 +58,19 @@ export interface Extractor {
 //
 // Example:
 //
-// const TEST_VIDEO = './samples/bbb10m.mp4';
-// export const run = async() => {
-//     // Arrange
-//     const extractor = await BeamcoderExtractor.create({
-//         inputFileOrUrl: TEST_VIDEO,
-//     });
-//
-//     const frame = await extractor.getImageDataAtTime(0.3);
-//     console.log(frame);
-//     extractor.dispose();
-// };
-//
-// run();
-//
+const TEST_VIDEO = './samples/tmp.mp4';
+export const run = async() => {
+    // Arrange
+    const extractor = await BeamcoderExtractor.create({
+        inputFileOrUrl: TEST_VIDEO,
+    });
+
+    const frame = await extractor.getImageDataAtTime(230.63333333333335);
+    console.log(frame);
+    extractor.dispose();
+};
+
+run();
+
 
 export { BeamcoderExtractor };
diff --git a/src/backends/beamcoder.ts b/src/backends/beamcoder.ts
@@ -382,7 +382,7 @@ export class BeamcoderExtractor extends BaseExtractor implements Extractor {
 
                 // The packet contains frames, but all of them have PTS larger than our a targetPTS (we looked too far)
                 if (!closestFrame) {
-                    return outputFrame;
+                    return outputFrame || (filteredFrames.reduce((nearest, current) => (Math.abs(current.pts - targetPTS) < Math.abs(nearest.pts - targetPTS) ? current : nearest), filteredFrames[0]) ?? null);
                 }
 
                 // store the filtered packet frames for later reuse