Skip to content

Commit

Permalink
avfilter/vsrc_mptestsrc: add options to set the maximum number of frames
Browse files Browse the repository at this point in the history 
Signed-off-by: Limin Wang <[email protected]>
Signed-off-by: Michael Niedermayer <[email protected]>
  • Loading branch information
@lance-lmwang @michaelni
lance-lmwang authored and michaelni committed Oct 21, 2019
1 parent 7a477c4 commit 6079bd5
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 12 deletions.
3 changes: 3 additions & 0 deletions doc/filters.texi
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20940,6 +20940,9 @@ Set the number or the name of the test to perform. Supported tests are:
@item ring2
@item all

@item max_frames, m
Set the maximum number of frames generated for each test, default value is 30.

@end table

Default value is "all", which will cycle through the list of all tests.
Expand Down
29 changes: 17 additions & 12 deletions libavfilter/vsrc_mptestsrc.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@ typedef struct MPTestContext {
const AVClass *class;
AVRational frame_rate;
int64_t pts, max_pts, duration;
int64_t max_frames;
int hsub, vsub;
int test; ///< test_type
} MPTestContext;
Expand All @@ -79,6 +80,10 @@ static const AVOption mptestsrc_options[]= {
{ "ring1", "", 0, AV_OPT_TYPE_CONST, {.i64=TEST_RING1}, INT_MIN, INT_MAX, FLAGS, "test" },
{ "ring2", "", 0, AV_OPT_TYPE_CONST, {.i64=TEST_RING2}, INT_MIN, INT_MAX, FLAGS, "test" },
{ "all", "", 0, AV_OPT_TYPE_CONST, {.i64=TEST_ALL}, INT_MIN, INT_MAX, FLAGS, "test" },
{ "max_frames", "Set the maximum number of frames generated for each test", OFFSET(max_frames),
AV_OPT_TYPE_INT64, {.i64 = 30}, 1, INT64_MAX, FLAGS },
{ "m", "Set the maximum number of frames generated for each test", OFFSET(max_frames),
AV_OPT_TYPE_INT64, {.i64 = 30}, 1, INT64_MAX, FLAGS },
{ NULL }
};

Expand Down Expand Up @@ -322,20 +327,20 @@ static int request_frame(AVFilterLink *outlink)
memset(picref->data[2] + i*picref->linesize[2], 128, cw);
}

if (tt == TEST_ALL && frame%30) /* draw a black frame at the beginning of each test */
tt = (frame/30)%(TEST_NB-1);
if (tt == TEST_ALL && frame%test->max_frames) /* draw a black frame at the beginning of each test */
tt = (frame/test->max_frames)%(TEST_NB-1);

switch (tt) {
case TEST_DC_LUMA: dc_test(picref->data[0], picref->linesize[0], 256, 256, frame%30); break;
case TEST_DC_CHROMA: dc_test(picref->data[1], picref->linesize[1], 256, 256, frame%30); break;
case TEST_FREQ_LUMA: freq_test(picref->data[0], picref->linesize[0], frame%30); break;
case TEST_FREQ_CHROMA: freq_test(picref->data[1], picref->linesize[1], frame%30); break;
case TEST_AMP_LUMA: amp_test(picref->data[0], picref->linesize[0], frame%30); break;
case TEST_AMP_CHROMA: amp_test(picref->data[1], picref->linesize[1], frame%30); break;
case TEST_CBP: cbp_test(picref->data , picref->linesize , frame%30); break;
case TEST_MV: mv_test(picref->data[0], picref->linesize[0], frame%30); break;
case TEST_RING1: ring1_test(picref->data[0], picref->linesize[0], frame%30); break;
case TEST_RING2: ring2_test(picref->data[0], picref->linesize[0], frame%30); break;
case TEST_DC_LUMA: dc_test(picref->data[0], picref->linesize[0], 256, 256, frame%test->max_frames); break;
case TEST_DC_CHROMA: dc_test(picref->data[1], picref->linesize[1], 256, 256, frame%test->max_frames); break;
case TEST_FREQ_LUMA: freq_test(picref->data[0], picref->linesize[0], frame%test->max_frames); break;
case TEST_FREQ_CHROMA: freq_test(picref->data[1], picref->linesize[1], frame%test->max_frames); break;
case TEST_AMP_LUMA: amp_test(picref->data[0], picref->linesize[0], frame%test->max_frames); break;
case TEST_AMP_CHROMA: amp_test(picref->data[1], picref->linesize[1], frame%test->max_frames); break;
case TEST_CBP: cbp_test(picref->data , picref->linesize , frame%test->max_frames); break;
case TEST_MV: mv_test(picref->data[0], picref->linesize[0], frame%test->max_frames); break;
case TEST_RING1: ring1_test(picref->data[0], picref->linesize[0], frame%test->max_frames); break;
case TEST_RING2: ring2_test(picref->data[0], picref->linesize[0], frame%test->max_frames); break;
}

return ff_filter_frame(outlink, picref);
Expand Down

0 comments on commit 6079bd5

Please sign in to comment.