avfilter/vf_lut2: also export video input bit depth

Signed-off-by: Paul B Mahol <[email protected]>
ycqian · Sep 16, 2016 · 5b509fa · 5b509fa
1 parent 51000b9
commit 5b509fa
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/doc/filters.texi b/doc/filters.texi
@@ -9352,6 +9352,12 @@ The first input value for the pixel component.
 
 @item y
 The second input value for the pixel component.
+
+@item bdx
+The first input video bit depth.
+
+@item bdy
+The second input video bit depth.
 @end table
 
 All expressions default to "x".

diff --git a/libavfilter/vf_lut2.c b/libavfilter/vf_lut2.c
@@ -35,6 +35,8 @@ static const char *const var_names[] = {
     "h",        ///< height of the input video
     "x",        ///< input value for the pixel from input #1
     "y",        ///< input value for the pixel from input #2
+    "bdx",      ///< input #1 video bitdepth
+    "bdy",      ///< input #2 video bitdepth
     NULL
 };
 
@@ -43,6 +45,8 @@ enum var_name {
     VAR_H,
     VAR_X,
     VAR_Y,
+    VAR_BITDEPTHX,
+    VAR_BITDEPTHY,
     VAR_VARS_NB
 };
 
@@ -127,6 +131,7 @@ static int config_inputx(AVFilterLink *inlink)
     s->var_values[VAR_W] = inlink->w;
     s->var_values[VAR_H] = inlink->h;
     s->depthx = desc->comp[0].depth;
+    s->var_values[VAR_BITDEPTHX] = s->depthx;
 
     return 0;
 }
@@ -138,6 +143,7 @@ static int config_inputy(AVFilterLink *inlink)
     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
 
     s->depthy = desc->comp[0].depth;
+    s->var_values[VAR_BITDEPTHY] = s->depthy;
 
     return 0;
 }