diff options
Diffstat (limited to 'libhb/chroma_smooth.c')
| -rw-r--r-- | libhb/chroma_smooth.c | 370 |
1 files changed, 370 insertions, 0 deletions
diff --git a/libhb/chroma_smooth.c b/libhb/chroma_smooth.c new file mode 100644 index 000000000..e27ecaa77 --- /dev/null +++ b/libhb/chroma_smooth.c @@ -0,0 +1,370 @@ +/* chroma_smooth.c + + Copyright (c) 2002 Rémi Guyomarch <rguyom at pobox.com> + Copyright (c) 2003-2019 HandBrake Team + This file is part of the HandBrake source code + Homepage: <http://handbrake.fr/>. + It may be used under the terms of the GNU General Public License v2. + For full terms see the file COPYING file or visit http://www.gnu.org/licenses/gpl-2.0.html + */ + +#include "hb.h" + +#define CHROMA_SMOOTH_STRENGTH_DEFAULT 0.25 +#define CHROMA_SMOOTH_SIZE_DEFAULT 7 +#define CHROMA_SMOOTH_SIZE_MIN 3 +#define CHROMA_SMOOTH_SIZE_MAX 63 + +typedef struct +{ + int pix_fmt; // source pixel format + int width; // source video width + double strength; // strength + int size; // pixel context region width (must be odd) + + int steps; + int amount; + int scalebits; + int32_t halfscale; +} chroma_smooth_plane_context_t; + +typedef struct +{ + uint32_t * SC[CHROMA_SMOOTH_SIZE_MAX - 1]; +} chroma_smooth_thread_context_t; + +typedef chroma_smooth_thread_context_t chroma_smooth_thread_context3_t[3]; + +struct hb_filter_private_s +{ + chroma_smooth_plane_context_t plane_ctx[3]; + chroma_smooth_thread_context3_t * thread_ctx; + int threads; +}; + +static int chroma_smooth_init(hb_filter_object_t *filter, + hb_filter_init_t *init); + +static int chroma_smooth_init_thread(hb_filter_object_t *filter, int threads); + +static int chroma_smooth_work(hb_filter_object_t *filter, + hb_buffer_t ** buf_in, + hb_buffer_t ** buf_out); +static int chroma_smooth_work_thread(hb_filter_object_t *filter, + hb_buffer_t ** buf_in, + hb_buffer_t ** buf_out, int thread); + +static void chroma_smooth_close(hb_filter_object_t *filter); + +static const char chroma_smooth_template[] = + "cb-strength=^"HB_FLOAT_REG"$:cb-size=^"HB_INT_REG"$:" + "cr-strength=^"HB_FLOAT_REG"$:cr-size=^"HB_INT_REG"$"; + +hb_filter_object_t hb_filter_chroma_smooth = +{ + .id = HB_FILTER_CHROMA_SMOOTH, + .enforce_order = 1, + .name = "Chroma Smooth", + .settings = NULL, + .init = chroma_smooth_init, + .init_thread = chroma_smooth_init_thread, + .work = chroma_smooth_work, + .work_thread = chroma_smooth_work_thread, + .close = chroma_smooth_close, + .settings_template = chroma_smooth_template, +}; + +static void luma_copy(const uint8_t *src, + uint8_t *dst, + const int width, + const int height, + const int stride, + chroma_smooth_plane_context_t * ctx, + chroma_smooth_thread_context_t * tctx) +{ + for (int y = 0; y < height; y++) + { + memcpy(dst + y * stride, src + y * stride, stride); + } +} + +static void chroma_smooth(const uint8_t *src, + uint8_t *dst, + const int width, + const int height, + const int stride, + chroma_smooth_plane_context_t * ctx, + chroma_smooth_thread_context_t * tctx) +{ + uint32_t **SC = tctx->SC; + uint32_t SR[CHROMA_SMOOTH_SIZE_MAX - 1], + Tmp1, + Tmp2; + const uint8_t *src2 = src; // avoid gcc warning + int32_t res; + int x, y, z; + int amount = ctx->amount; + int steps = ctx->steps; + int scalebits = ctx->scalebits; + int32_t halfscale = ctx->halfscale; + + if (!amount) + { + if (src != dst) + { + memcpy(dst, src, stride*height); + } + + return; + } + + for (y = 0; y < 2 * steps; y++) + { + memset(SC[y], 0, sizeof(SC[y][0]) * (width + 2 * steps)); + } + + for (y = -steps; y < height + steps; y++) + { + if (y < height) + { + src2 = src; + } + + memset(SR, 0, sizeof(SR[0]) * (2 * steps - 1)); + + for (x = -steps; x < width + steps; x++) + { + Tmp1 = x <= 0 ? src2[0] : x >= width ? src2[width - 1] : src2[x]; + + for (z = 0; z < steps * 2; z += 2) + { + Tmp2 = SR[z + 0] + Tmp1; SR[z + 0] = Tmp1; + Tmp1 = SR[z + 1] + Tmp2; SR[z + 1] = Tmp2; + } + + for (z = 0; z < steps * 2; z += 2) + { + Tmp2 = SC[z + 0][x + steps] + Tmp1; SC[z + 0][x + steps] = Tmp1; + Tmp1 = SC[z + 1][x + steps] + Tmp2; SC[z + 1][x + steps] = Tmp2; + } + + if (x >= steps && y >= steps) + { + const uint8_t * srx = src - steps * stride + x - steps; + uint8_t * dsx = dst - steps * stride + x - steps; + + //res = (int32_t)*srx + ((((int32_t)*srx - + // (int32_t)((Tmp1 + halfscale) >> scalebits)) * amount) >> 16); + res = (int32_t)*srx - ((((int32_t)*srx - + (int32_t)((Tmp1 + halfscale) >> scalebits)) * amount) >> 16); + *dsx = res > 255 ? 255 : res < 0 ? 0 : (uint8_t)res; + } + } + + if (y >= 0) + { + dst += stride; + src += stride; + } + } +} + +static int chroma_smooth_init(hb_filter_object_t *filter, + hb_filter_init_t *init) +{ + filter->private_data = calloc(sizeof(struct hb_filter_private_s), 1); + if (filter->private_data == NULL) + { + hb_error("Chroma Smooth calloc failed"); + return -1; + } + hb_filter_private_t * pv = filter->private_data; + + // Mark parameters unset + for (int c = 0; c < 3; c++) + { + pv->plane_ctx[c].strength = -1; + pv->plane_ctx[c].size = -1; + } + + // Read user parameters + if (filter->settings != NULL) + { + hb_dict_t * dict = filter->settings; + hb_dict_extract_double(&pv->plane_ctx[1].strength, dict, "cb-strength"); + hb_dict_extract_int(&pv->plane_ctx[1].size, dict, "cb-size"); + + hb_dict_extract_double(&pv->plane_ctx[2].strength, dict, "cr-strength"); + hb_dict_extract_int(&pv->plane_ctx[2].size, dict, "cr-size"); + } + + // Cascade values + // Cr not set; inherit Cb. Cb not set; defaults. + for (int c = 2; c < 3; c++) + { + chroma_smooth_plane_context_t * prev_ctx = &pv->plane_ctx[c - 1]; + chroma_smooth_plane_context_t * ctx = &pv->plane_ctx[c]; + + if (ctx->strength == -1) ctx->strength = prev_ctx->strength; + if (ctx->size == -1) ctx->size = prev_ctx->size; + } + + for (int c = 0; c < 3; c++) + { + chroma_smooth_plane_context_t * ctx = &pv->plane_ctx[c]; + + ctx->width = init->geometry.width; + + // Replace unset values with defaults + if (ctx->strength == -1) + { + ctx->strength = CHROMA_SMOOTH_STRENGTH_DEFAULT; + } + if (ctx->size == -1) + { + ctx->size = CHROMA_SMOOTH_SIZE_DEFAULT; + } + + // Sanitize + if (ctx->strength < 0) ctx->strength = 0; + if (ctx->strength > 1.5) ctx->strength = 1.5; + if (ctx->size % 2 == 0) ctx->size--; + if (ctx->size < CHROMA_SMOOTH_SIZE_MIN) ctx->size = CHROMA_SMOOTH_SIZE_MIN; + if (ctx->size > CHROMA_SMOOTH_SIZE_MAX) ctx->size = CHROMA_SMOOTH_SIZE_MAX; + + ctx->amount = ctx->strength * 65536.0; + ctx->steps = ctx->size / 2; + ctx->scalebits = ctx->steps * 4; + ctx->halfscale = 1 << (ctx->scalebits - 1); + } + + if (chroma_smooth_init_thread(filter, 1) < 0) + { + chroma_smooth_close(filter); + return -1; + } + + return 0; +} + +static void chroma_smooth_thread_close(hb_filter_private_t *pv) +{ + int c, z; + for (c = 0; c < 3; c++) + { + chroma_smooth_plane_context_t * ctx = &pv->plane_ctx[c]; + for (int t = 0; t < pv->threads; t++) + { + chroma_smooth_thread_context_t * tctx = &pv->thread_ctx[t][c]; + for (z = 0; z < 2 * ctx->steps; z++) + { + free(tctx->SC[z]); + tctx->SC[z] = NULL; + } + } + } + free(pv->thread_ctx); +} + +static int chroma_smooth_init_thread(hb_filter_object_t *filter, int threads) +{ + hb_filter_private_t * pv = filter->private_data; + + chroma_smooth_thread_close(pv); + pv->thread_ctx = calloc(threads, sizeof(chroma_smooth_thread_context3_t)); + pv->threads = threads; + for (int c = 0; c < 3; c++) + { + chroma_smooth_plane_context_t * ctx = &pv->plane_ctx[c]; + int w = hb_image_width(ctx->pix_fmt, ctx->width, c); + + for (int t = 0; t < threads; t++) + { + chroma_smooth_thread_context_t * tctx = &pv->thread_ctx[t][c]; + int z; + for (z = 0; z < 2 * ctx->steps; z++) + { + tctx->SC[z] = malloc(sizeof(*(tctx->SC[z])) * + (w + 2 * ctx->steps)); + if (tctx->SC[z] == NULL) + { + hb_error("Chroma Smooth calloc failed"); + chroma_smooth_close(filter); + return -1; + } + } + } + } + return 0; +} + +static void chroma_smooth_close(hb_filter_object_t * filter) +{ + hb_filter_private_t *pv = filter->private_data; + + if (pv == NULL) + { + return; + } + + chroma_smooth_thread_close(pv); + free(pv); + filter->private_data = NULL; +} + +static int chroma_smooth_work_thread(hb_filter_object_t *filter, + hb_buffer_t ** buf_in, + hb_buffer_t ** buf_out, int thread) +{ + hb_filter_private_t *pv = filter->private_data; + hb_buffer_t *in = *buf_in, *out; + + if (in->s.flags & HB_BUF_FLAG_EOF) + { + *buf_out = in; + *buf_in = NULL; + return HB_FILTER_DONE; + } + + out = hb_frame_buffer_init(in->f.fmt, in->f.width, in->f.height); + + int c; + for (c = 0; c < 3; c++) + { + chroma_smooth_plane_context_t * ctx = &pv->plane_ctx[c]; + chroma_smooth_thread_context_t * tctx = &pv->thread_ctx[thread][c]; + + if (c == 0) + { + // Luma + luma_copy(in->plane[c].data, + out->plane[c].data, + in->plane[c].width, + in->plane[c].height, + in->plane[c].stride, + ctx, tctx); + } + else + { + // Chroma + chroma_smooth(in->plane[c].data, + out->plane[c].data, + in->plane[c].width, + in->plane[c].height, + in->plane[c].stride, + ctx, tctx); + } + } + + out->s = in->s; + *buf_out = out; + + return HB_FILTER_OK; +} + +static int chroma_smooth_work(hb_filter_object_t *filter, + hb_buffer_t ** buf_in, + hb_buffer_t ** buf_out) +{ + return chroma_smooth_work_thread(filter, buf_in, buf_out, 0); +} |