arcodange/mediabunny
1
0
Fork 0
mirror of https://github.com/arcodange-org/mediabunny.git synced 2026-10-03 05:43:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add SampleRetrievalOptions to samples iterator

Browse Source
This commit is contained in:
Vanilagy
2025-01-04 01:11:01 +01:00
parent 1145e0ebbe
commit 2e8d49f900
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/media-drain.ts
+7 -3
View File
@@ -41,7 +41,11 @@ export abstract class BaseSampleDrain<Sample extends EncodedVideoSample | Encode
abstract getKeySample(timestamp: number, options?: SampleRetrievalOptions): Promise<Sample | null>; abstract getKeySample(timestamp: number, options?: SampleRetrievalOptions): Promise<Sample | null>;
abstract getNextKeySample(sample: Sample, options?: SampleRetrievalOptions): Promise<Sample | null>; abstract getNextKeySample(sample: Sample, options?: SampleRetrievalOptions): Promise<Sample | null>;
samples(startSample?: Sample, endTimestamp = Infinity): AsyncGenerator<Sample, void, unknown> { samples(
startSample?: Sample,
endTimestamp = Infinity,
options?: SampleRetrievalOptions,
): AsyncGenerator<Sample, void, unknown> {
const sampleQueue: Sample[] = []; const sampleQueue: Sample[] = [];
let { promise: queueNotEmpty, resolve: onQueueNotEmpty } = promiseWithResolvers(); let { promise: queueNotEmpty, resolve: onQueueNotEmpty } = promiseWithResolvers();
@@ -60,7 +64,7 @@ export abstract class BaseSampleDrain<Sample extends EncodedVideoSample | Encode
// The following is the "pump" process that keeps pumping samples into the queue // The following is the "pump" process that keeps pumping samples into the queue
(async () => { (async () => {
let sample = startSample ?? await this.getFirstSample(); let sample = startSample ?? await this.getFirstSample(options);
while (sample && !terminated) { while (sample && !terminated) {
if (sample.timestamp >= endTimestamp) { if (sample.timestamp >= endTimestamp) {
@@ -78,7 +82,7 @@ export abstract class BaseSampleDrain<Sample extends EncodedVideoSample | Encode
onQueueNotEmpty(); onQueueNotEmpty();
({ promise: queueNotEmpty, resolve: onQueueNotEmpty } = promiseWithResolvers()); ({ promise: queueNotEmpty, resolve: onQueueNotEmpty } = promiseWithResolvers());
sample = await this.getNextSample(sample); sample = await this.getNextSample(sample, options);
} }
ended = true; ended = true;