|
|
@ -46,6 +46,7 @@ typedef struct HistogramContext { |
|
|
|
int scale_height; |
|
|
|
int scale_height; |
|
|
|
int step; |
|
|
|
int step; |
|
|
|
int waveform_mode; |
|
|
|
int waveform_mode; |
|
|
|
|
|
|
|
int waveform_mirror; |
|
|
|
int display_mode; |
|
|
|
int display_mode; |
|
|
|
int levels_mode; |
|
|
|
int levels_mode; |
|
|
|
const AVPixFmtDescriptor *desc; |
|
|
|
const AVPixFmtDescriptor *desc; |
|
|
@ -66,6 +67,7 @@ static const AVOption histogram_options[] = { |
|
|
|
{ "waveform_mode", "set waveform mode", OFFSET(waveform_mode), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "waveform_mode"}, |
|
|
|
{ "waveform_mode", "set waveform mode", OFFSET(waveform_mode), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "waveform_mode"}, |
|
|
|
{ "row", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "waveform_mode" }, |
|
|
|
{ "row", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "waveform_mode" }, |
|
|
|
{ "column", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "waveform_mode" }, |
|
|
|
{ "column", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "waveform_mode" }, |
|
|
|
|
|
|
|
{ "waveform_mirror", "set waveform mirroring", OFFSET(waveform_mirror), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "waveform_mirror"}, |
|
|
|
{ "display_mode", "set display mode", OFFSET(display_mode), AV_OPT_TYPE_INT, {.i64=1}, 0, 1, FLAGS, "display_mode"}, |
|
|
|
{ "display_mode", "set display mode", OFFSET(display_mode), AV_OPT_TYPE_INT, {.i64=1}, 0, 1, FLAGS, "display_mode"}, |
|
|
|
{ "parade", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "display_mode" }, |
|
|
|
{ "parade", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "display_mode" }, |
|
|
|
{ "overlay", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "display_mode" }, |
|
|
|
{ "overlay", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "display_mode" }, |
|
|
@ -183,29 +185,38 @@ static void gen_waveform(HistogramContext *h, AVFrame *inpicref, AVFrame *outpic |
|
|
|
int component, int intensity, int offset, int col_mode) |
|
|
|
int component, int intensity, int offset, int col_mode) |
|
|
|
{ |
|
|
|
{ |
|
|
|
const int plane = h->desc->comp[component].plane; |
|
|
|
const int plane = h->desc->comp[component].plane; |
|
|
|
|
|
|
|
const int mirror = h->waveform_mirror; |
|
|
|
const int is_chroma = (component == 1 || component == 2); |
|
|
|
const int is_chroma = (component == 1 || component == 2); |
|
|
|
const int shift_w = (is_chroma ? h->desc->log2_chroma_w : 0); |
|
|
|
const int shift_w = (is_chroma ? h->desc->log2_chroma_w : 0); |
|
|
|
const int shift_h = (is_chroma ? h->desc->log2_chroma_h : 0); |
|
|
|
const int shift_h = (is_chroma ? h->desc->log2_chroma_h : 0); |
|
|
|
const int src_linesize = inpicref->linesize[plane]; |
|
|
|
const int src_linesize = inpicref->linesize[plane]; |
|
|
|
const int dst_linesize = outpicref->linesize[plane]; |
|
|
|
const int dst_linesize = outpicref->linesize[plane]; |
|
|
|
|
|
|
|
const int dst_signed_linesize = dst_linesize * (mirror == 1 ? -1 : 1); |
|
|
|
uint8_t *src_data = inpicref->data[plane]; |
|
|
|
uint8_t *src_data = inpicref->data[plane]; |
|
|
|
uint8_t *dst_data = outpicref->data[plane] + (col_mode ? (offset >> shift_h) * dst_linesize : offset >> shift_w); |
|
|
|
uint8_t *dst_data = outpicref->data[plane] + (col_mode ? (offset >> shift_h) * dst_linesize : offset >> shift_w); |
|
|
|
uint8_t * const dst_line = dst_data; |
|
|
|
uint8_t * const dst_bottom_line = dst_data + dst_linesize * ((256 >> shift_h) - 1); |
|
|
|
|
|
|
|
uint8_t * const dst_line = (mirror ? dst_bottom_line : dst_data); |
|
|
|
const uint8_t max = 255 - intensity; |
|
|
|
const uint8_t max = 255 - intensity; |
|
|
|
const int src_h = FF_CEIL_RSHIFT(inpicref->height, shift_h); |
|
|
|
const int src_h = FF_CEIL_RSHIFT(inpicref->height, shift_h); |
|
|
|
const int src_w = FF_CEIL_RSHIFT(inpicref->width, shift_w); |
|
|
|
const int src_w = FF_CEIL_RSHIFT(inpicref->width, shift_w); |
|
|
|
uint8_t *dst, *p; |
|
|
|
uint8_t *dst, *p; |
|
|
|
int y; |
|
|
|
int y; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (!col_mode && mirror) |
|
|
|
|
|
|
|
dst_data += 256 >> shift_w; |
|
|
|
for (y = 0; y < src_h; y++) { |
|
|
|
for (y = 0; y < src_h; y++) { |
|
|
|
const uint8_t *src_data_end = src_data + src_w; |
|
|
|
const uint8_t *src_data_end = src_data + src_w; |
|
|
|
dst = dst_line; |
|
|
|
dst = dst_line; |
|
|
|
for (p = src_data; p < src_data_end; p++) { |
|
|
|
for (p = src_data; p < src_data_end; p++) { |
|
|
|
uint8_t *target; |
|
|
|
uint8_t *target; |
|
|
|
if (col_mode) |
|
|
|
if (col_mode) { |
|
|
|
target = dst++ + dst_linesize * (*p >> shift_h); |
|
|
|
target = dst++ + dst_signed_linesize * (*p >> shift_h); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
if (mirror) |
|
|
|
|
|
|
|
target = dst_data - (*p >> shift_w); |
|
|
|
else |
|
|
|
else |
|
|
|
target = dst_data + (*p >> shift_w); |
|
|
|
target = dst_data + (*p >> shift_w); |
|
|
|
|
|
|
|
} |
|
|
|
if (*target <= max) |
|
|
|
if (*target <= max) |
|
|
|
*target += intensity; |
|
|
|
*target += intensity; |
|
|
|
else |
|
|
|
else |
|
|
|