ffmpeg/FFmpeg
1
0
Fork 0
mirror of https://git.ffmpeg.org/ffmpeg.git synced 2024-09-19 21:06:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

avfilter/vf_bilateral: simplify code a little

Browse Source 
Make alpha_ calculation faster.
This commit is contained in:
Paul B Mahol 2020-07-17 20:29:48 +02:00
parent 36e51c190b
commit fa8345cf05
1 changed files with 11 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
libavfilter/vf_bilateral.c
View File

@ -41,6 +41,7 @@ typedef struct BilateralContext {
int planewidth[4];
int planeheight[4];
float alpha;
float range_table[65536];
float *img_out_f;
@ -99,10 +100,11 @@ static int config_input(AVFilterLink *inlink)
s->depth = desc->comp[0].depth;
inv_sigma_range = 1.0f / (s->sigmaR * ((1 << s->depth) - 1));
s->alpha = expf(-sqrtf(2.f) / s->sigmaS);
//compute a lookup table
for (int i = 0; i < (1 << s->depth); i++)
s->range_table[i] = expf(-i * inv_sigma_range);
s->range_table[i] = s->alpha * expf(-i * inv_sigma_range);
s->planewidth[1] = s->planewidth[2] = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
s->planewidth[0] = s->planewidth[3] = inlink->w;
@ -144,10 +146,10 @@ static void bilateral_##name(BilateralContext *s, const uint8_t *ssrc, uint8_t *
float *map_factor_a = s->map_factor_a, *map_factor_b = s->map_factor_b; \
float *slice_factor_a = s->slice_factor_a, *slice_factor_b = s->slice_factor_b; \
float *line_factor_a = s->line_factor_a, *line_factor_b = s->line_factor_b; \
float *range_table = s->range_table; \
float alpha = expf(-sqrtf(2.f) / sigma_spatial); \
const float *range_table = s->range_table; \
const float alpha = s->alpha; \
float ypr, ycr, *ycy, *ypy, *xcy, fp, fc; \
float inv_alpha_ = 1 - alpha; \
const float inv_alpha_ = 1.f - alpha; \
float *ycf, *ypf, *xcf, *in_factor; \
const type *tcy, *tpy; \
int h1; \
@ -165,14 +167,13 @@ static void bilateral_##name(BilateralContext *s, const uint8_t *ssrc, uint8_t *
*temp_factor_x++ = fp = 1; \
\
for (int x = 1; x < width; x++) { \
float weight, alpha_; \
float alpha_; \
int range_dist; \
type tcr = *texture_x++; \
type dr = abs(tcr - tpr); \
\
range_dist = dr; \
weight = range_table[range_dist]; \
alpha_ = weight*alpha; \
alpha_ = range_table[range_dist]; \
*temp_x++ = ycr = inv_alpha_*(*in_x++) + alpha_*ypr; \
tpr = tcr; \
ypr = ycr; \
@ -190,8 +191,7 @@ static void bilateral_##name(BilateralContext *s, const uint8_t *ssrc, uint8_t *
type tcr = *--texture_x; \
type dr = abs(tcr - tpr); \
int range_dist = dr; \
float weight = range_table[range_dist]; \
float alpha_ = weight * alpha; \
float alpha_ = range_table[range_dist]; \
\
ycr = inv_alpha_ * (*--in_x) + alpha_ * ypr; \
--temp_x; *temp_x = 0.5f*((*temp_x) + ycr); \
@ -206,8 +206,6 @@ static void bilateral_##name(BilateralContext *s, const uint8_t *ssrc, uint8_t *
} \
memcpy(img_out_f, img_temp, sizeof(float) * width); \
\
alpha = expf(-sqrtf(2.f) / sigma_spatial); \
inv_alpha_ = 1 - alpha; \
in_factor = map_factor_a; \
memcpy(map_factor_b, in_factor, sizeof(float) * width); \
for (int y = 1; y < height; y++) { \
@ -223,8 +221,7 @@ static void bilateral_##name(BilateralContext *s, const uint8_t *ssrc, uint8_t *
for (int x = 0; x < width; x++) { \
type dr = abs((*tcy++) - (*tpy++)); \
int range_dist = dr; \
float weight = range_table[range_dist]; \
float alpha_ = weight*alpha; \
float alpha_ = range_table[range_dist]; \
\
*ycy++ = inv_alpha_*(*xcy++) + alpha_*(*ypy++); \
*ycf++ = inv_alpha_*(*xcf++) + alpha_*(*ypf++); \
@ -263,8 +260,7 @@ static void bilateral_##name(BilateralContext *s, const uint8_t *ssrc, uint8_t *
for (int x = 0; x < width; x++) { \
type dr = abs((*tcy++) - (*tpy++)); \
int range_dist = dr; \
float weight = range_table[range_dist]; \
float alpha_ = weight*alpha; \
float alpha_ = range_table[range_dist]; \
float ycc, fcc = inv_alpha_*(*xcf++) + alpha_*(*ypf_++); \
\
*ycf_++ = fcc; \