Skip to content

Commit 6d5fedc

Browse files
committed
sdk: remove filter from VideoFrameGenerator
1 parent 7eca7f6 commit 6d5fedc

File tree

5 files changed

+9
-15
lines changed

5 files changed

+9
-15
lines changed

Diff for: plugins/objectdetector/src/ffmpeg-videoframes-no-sharp.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ class RawImage implements Image, RawFrame {
5050
}
5151

5252
export class FFmpegVideoFrameGenerator extends ScryptedDeviceBase implements VideoFrameGenerator {
53-
async *generateVideoFramesInternal(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions, filter?: (videoFrame: VideoFrame) => Promise<boolean>): AsyncGenerator<VideoFrame, any, unknown> {
53+
async *generateVideoFramesInternal(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions): AsyncGenerator<VideoFrame, any, unknown> {
5454
const ffmpegInput = await sdk.mediaManager.convertMediaObjectToJSON<FFmpegInput>(mediaObject, ScryptedMimeTypes.FFmpegInput);
5555
const gray = options?.format === 'gray';
5656
const format = options?.format || 'rgb';
@@ -171,7 +171,7 @@ export class FFmpegVideoFrameGenerator extends ScryptedDeviceBase implements Vid
171171
}
172172

173173

174-
async generateVideoFrames(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions, filter?: (videoFrame: VideoFrame & MediaObject) => Promise<boolean>): Promise<AsyncGenerator<VideoFrame, any, unknown>> {
175-
return this.generateVideoFramesInternal(mediaObject, options, filter);
174+
async generateVideoFrames(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions): Promise<AsyncGenerator<VideoFrame, any, unknown>> {
175+
return this.generateVideoFramesInternal(mediaObject, options);
176176
}
177177
}

Diff for: plugins/objectdetector/src/ffmpeg-videoframes.ts

+3-4
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ class VipsImage implements Image {
132132
}
133133

134134
export class FFmpegVideoFrameGenerator extends ScryptedDeviceBase implements VideoFrameGenerator {
135-
async *generateVideoFramesInternal(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions, filter?: (videoFrame: VideoFrame) => Promise<boolean>): AsyncGenerator<VideoFrame, any, unknown> {
135+
async *generateVideoFramesInternal(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions): AsyncGenerator<VideoFrame, any, unknown> {
136136
const ffmpegInput = await sdk.mediaManager.convertMediaObjectToJSON<FFmpegInput>(mediaObject, ScryptedMimeTypes.FFmpegInput);
137137
const gray = options?.format === 'gray';
138138
const channels = gray ? 1 : 3;
@@ -249,8 +249,7 @@ export class FFmpegVideoFrameGenerator extends ScryptedDeviceBase implements Vid
249249
}
250250
}
251251

252-
253-
async generateVideoFrames(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions, filter?: (videoFrame: VideoFrame) => Promise<boolean>): Promise<AsyncGenerator<VideoFrame, any, unknown>> {
254-
return this.generateVideoFramesInternal(mediaObject, options, filter);
252+
async generateVideoFrames(mediaObject: MediaObject, options?: VideoFrameGeneratorOptions): Promise<AsyncGenerator<VideoFrame, any, unknown>> {
253+
return this.generateVideoFramesInternal(mediaObject, options);
255254
}
256255
}

Diff for: plugins/python-codecs/src/gstreamer.py

-1
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,6 @@ async def createGstMediaObject(image: GstImage):
314314
async def generateVideoFramesGstreamer(
315315
mediaObject: scrypted_sdk.MediaObject,
316316
options: scrypted_sdk.VideoFrameGeneratorOptions = None,
317-
filter: Any = None,
318317
h264Decoder: str = None,
319318
h265Decoder: str = None,
320319
postProcessPipeline: str = None,

Diff for: plugins/python-codecs/src/libav.py

-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
async def generateVideoFramesLibav(
2121
mediaObject: scrypted_sdk.MediaObject,
2222
options: scrypted_sdk.VideoFrameGeneratorOptions = None,
23-
filter: Any = None,
2423
) -> scrypted_sdk.VideoFrame:
2524
ffmpegInput: scrypted_sdk.FFmpegInput = (
2625
await scrypted_sdk.mediaManager.convertMediaObjectToJSON(

Diff for: plugins/python-codecs/src/main.py

+3-6
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ async def generateVideoFrames(
3636
filter: Any = None,
3737
) -> scrypted_sdk.VideoFrame:
3838
forked: CodecFork = await self.zygote().result
39-
return await forked.generateVideoFramesLibav(mediaObject, options, filter)
39+
return await forked.generateVideoFramesLibav(mediaObject, options)
4040

4141

4242
class GstreamerGenerator(
@@ -60,7 +60,6 @@ async def generateVideoFrames(
6060
return await forked.generateVideoFramesGstreamer(
6161
mediaObject,
6262
options,
63-
filter,
6463
self.storage.getItem("h264Decoder"),
6564
self.storage.getItem("h265Decoder"),
6665
self.storage.getItem("postProcessPipeline"),
@@ -242,14 +241,13 @@ async def generateVideoFramesGstreamer(
242241
self,
243242
mediaObject: scrypted_sdk.MediaObject,
244243
options: scrypted_sdk.VideoFrameGeneratorOptions,
245-
filter: Any,
246244
h264Decoder: str,
247245
h265Decoder: str,
248246
postProcessPipeline: str,
249247
) -> scrypted_sdk.VideoFrame:
250248
async for data in self.generateVideoFrames(
251249
gstreamer.generateVideoFramesGstreamer(
252-
mediaObject, options, filter, h264Decoder, h265Decoder, postProcessPipeline
250+
mediaObject, options, h264Decoder, h265Decoder, postProcessPipeline
253251
),
254252
"gstreamer",
255253
options and options.get("firstFrameOnly"),
@@ -260,10 +258,9 @@ async def generateVideoFramesLibav(
260258
self,
261259
mediaObject: scrypted_sdk.MediaObject,
262260
options: scrypted_sdk.VideoFrameGeneratorOptions = None,
263-
filter: Any = None,
264261
) -> scrypted_sdk.VideoFrame:
265262
async for data in self.generateVideoFrames(
266-
libav.generateVideoFramesLibav(mediaObject, options, filter),
263+
libav.generateVideoFramesLibav(mediaObject, options),
267264
"libav",
268265
options and options.get("firstFrameOnly"),
269266
):

0 commit comments

Comments
 (0)