Browse code

vsrc_mandelbrot: set options through the opt framework

Stefano Sabatini authored on 2011/11/14 03:12:00
Showing 2 changed files
... ...
@@ -30,7 +30,7 @@
30 30
 
31 31
 #define LIBAVFILTER_VERSION_MAJOR  2
32 32
 #define LIBAVFILTER_VERSION_MINOR 48
33
-#define LIBAVFILTER_VERSION_MICRO  0
33
+#define LIBAVFILTER_VERSION_MICRO  1
34 34
 
35 35
 #define LIBAVFILTER_VERSION_INT AV_VERSION_INT(LIBAVFILTER_VERSION_MAJOR, \
36 36
                                                LIBAVFILTER_VERSION_MINOR, \
... ...
@@ -28,6 +28,7 @@
28 28
 
29 29
 #include "avfilter.h"
30 30
 #include "libavutil/imgutils.h"
31
+#include "libavutil/opt.h"
31 32
 #include "libavutil/parseutils.h"
32 33
 
33 34
 #define SQR(a) ((a)*(a))
... ...
@@ -48,9 +49,11 @@ typedef struct Point {
48 48
 } Point;
49 49
 
50 50
 typedef struct {
51
+    const AVClass *class;
51 52
     int w, h;
52 53
     AVRational time_base;
53 54
     uint64_t pts;
55
+    char *size, *rate;
54 56
     int maxiter;
55 57
     double start_x;
56 58
     double start_y;
... ...
@@ -67,41 +70,72 @@ typedef struct {
67 67
     double (*zyklus)[2];
68 68
 } MBContext;
69 69
 
70
+#define OFFSET(x) offsetof(MBContext, x)
71
+
72
+static const AVOption mandelbrot_options[] = {
73
+    {"size",        "set frame size",                OFFSET(size),    AV_OPT_TYPE_STRING,     {.str="640x480"},  CHAR_MIN, CHAR_MAX },
74
+    {"s",           "set frame size",                OFFSET(size),    AV_OPT_TYPE_STRING,     {.str="640x480"},  CHAR_MIN, CHAR_MAX },
75
+    {"rate",        "set frame rate",                OFFSET(rate),    AV_OPT_TYPE_STRING,     {.str="25"},  CHAR_MIN, CHAR_MAX },
76
+    {"r",           "set frame rate",                OFFSET(rate),    AV_OPT_TYPE_STRING,     {.str="25"},  CHAR_MIN, CHAR_MAX },
77
+    {"maxiter",     "set max iterations number",     OFFSET(maxiter), AV_OPT_TYPE_INT,        {.dbl=4096},  1,        INT_MAX  },
78
+    {"start_x",     "set the initial x position",    OFFSET(start_x), AV_OPT_TYPE_DOUBLE,     {.dbl=-0.743643887037158704752191506114774}, -INFINITY, INFINITY  },
79
+    {"start_y",     "set the initial y position",    OFFSET(start_y), AV_OPT_TYPE_DOUBLE,     {.dbl=-0.131825904205311970493132056385139}, -INFINITY, INFINITY  },
80
+    {"start_scale", "set the initial scale value",   OFFSET(start_scale), AV_OPT_TYPE_DOUBLE, {.dbl=3.0},  -INFINITY, INFINITY },
81
+    {"end_scale",   "set the terminal scale value",  OFFSET(end_scale), AV_OPT_TYPE_DOUBLE,   {.dbl=0.3},  -INFINITY, INFINITY },
82
+    {"end_pts",     "set the terminal pts value",    OFFSET(end_pts), AV_OPT_TYPE_DOUBLE,     {.dbl=800},  -INFINITY, INFINITY },
83
+    {"bailout",     "set the bailout value",         OFFSET(bailout), AV_OPT_TYPE_DOUBLE,     {.dbl=10},   -INFINITY, INFINITY },
84
+
85
+    {"outer",       "set outer coloring mode",       OFFSET(outer), AV_OPT_TYPE_INT, {.dbl=NORMALIZED_ITERATION_COUNT}, 0, INT_MAX, 0, "outer"},
86
+    {"iteration_count", "set iteration count mode",  0, AV_OPT_TYPE_CONST, {.dbl=ITERATION_COUNT}, INT_MIN, INT_MAX, 0, "outer" },
87
+    {"normalized_iteration_count", "set normalized iteration count mode",   0, AV_OPT_TYPE_CONST, {.dbl=NORMALIZED_ITERATION_COUNT}, INT_MIN, INT_MAX, 0, "outer" },
88
+
89
+    {"inner",       "set inner coloring mode",       OFFSET(inner), AV_OPT_TYPE_INT, {.dbl=BLACK}, 0, INT_MAX, 0, "inner"},
90
+    {"black",       "set black mode",                0, AV_OPT_TYPE_CONST, {.dbl=BLACK}, INT_MIN, INT_MAX, 0, "inner" },
91
+    {"period",      "set period mode",               0, AV_OPT_TYPE_CONST, {.dbl=PERIOD}, INT_MIN, INT_MAX, 0, "inner" },
92
+
93
+    {NULL},
94
+};
95
+
96
+static const char *mandelbrot_get_name(void *ctx)
97
+{
98
+    return "mandelbrot";
99
+}
100
+
101
+static const AVClass mandelbrot_class = {
102
+    "MBContext",
103
+    mandelbrot_get_name,
104
+    mandelbrot_options
105
+};
106
+
70 107
 static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
71 108
 {
72 109
     MBContext *mb = ctx->priv;
73
-    char frame_size  [128] = "640x480";
74
-    char frame_rate  [128] = "25";
75
-    AVRational frame_rate_q;
76
-
77
-    mb->maxiter=4096;
78
-    mb->start_x=-0.743643887037158704752191506114774;
79
-    mb->start_y=-0.131825904205311970493132056385139;
80
-    mb->start_scale=3.0;
81
-    mb->end_scale=0.3;
82
-    mb->end_pts=800;
83
-    mb->bailout=10;
84
-    mb->outer= NORMALIZED_ITERATION_COUNT;
85
-    if (args)
86
-        sscanf(args, "%127[^:]:%127[^:]:%d:%lf:%lf:%lf:%lf:%lf:%lf:%d", frame_size, frame_rate,
87
-               &mb->maxiter, &mb->start_x, &mb->start_y, &mb->start_scale, &mb->end_scale,
88
-               &mb->end_pts, &mb->bailout, &mb->outer);
110
+    AVRational rate_q;
111
+    int err;
112
+
113
+    mb->class = &mandelbrot_class;
114
+    av_opt_set_defaults(mb);
115
+
116
+    if ((err = (av_set_options_string(mb, args, "=", ":"))) < 0) {
117
+        av_log(ctx, AV_LOG_ERROR, "Error parsing options string: '%s'\n", args);
118
+        return err;
119
+    }
89 120
     mb->bailout *= mb->bailout;
90 121
 
91
-    if (av_parse_video_size(&mb->w, &mb->h, frame_size) < 0) {
92
-        av_log(ctx, AV_LOG_ERROR, "Invalid frame size: %s\n", frame_size);
122
+    if (av_parse_video_size(&mb->w, &mb->h, mb->size) < 0) {
123
+        av_log(ctx, AV_LOG_ERROR, "Invalid frame size: %s\n", mb->size);
93 124
         return AVERROR(EINVAL);
94 125
     }
95 126
     mb->start_scale /=mb->h;
96 127
     mb->end_scale /=mb->h;
97 128
 
98
-    if (av_parse_video_rate(&frame_rate_q, frame_rate) < 0 ||
99
-        frame_rate_q.den <= 0 || frame_rate_q.num <= 0) {
100
-        av_log(ctx, AV_LOG_ERROR, "Invalid frame rate: %s\n", frame_rate);
129
+    if (av_parse_video_rate(&rate_q, mb->rate) < 0 ||
130
+        rate_q.den <= 0 || rate_q.num <= 0) {
131
+        av_log(ctx, AV_LOG_ERROR, "Invalid frame rate: %s\n", mb->rate);
101 132
         return AVERROR(EINVAL);
102 133
     }
103
-    mb->time_base.num = frame_rate_q.den;
104
-    mb->time_base.den = frame_rate_q.num;
134
+    mb->time_base.num = rate_q.den;
135
+    mb->time_base.den = rate_q.num;
105 136
 
106 137
     mb->cache_allocated = mb->w * mb->h * 3;
107 138
     mb->cache_used = 0;
... ...
@@ -116,6 +150,8 @@ static av_cold void uninit(AVFilterContext *ctx)
116 116
 {
117 117
     MBContext *mb = ctx->priv;
118 118
 
119
+    av_freep(&mb->size);
120
+    av_freep(&mb->rate);
119 121
     av_freep(&mb->point_cache);
120 122
     av_freep(&mb-> next_cache);
121 123
     av_freep(&mb->zyklus);