mirrors/librempeg
1
0
Fork 0
mirror of https://github.com/librempeg/librempeg synced 2024-11-23 03:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

avfilter/vf_codecview: added new options for block

Browse Source 
Signed-off-by: Limin Wang <lance.lmwang@gmail.com>
This commit is contained in:
Limin Wang 2021-08-17 18:46:41 +08:00
parent 9f40b5badb
commit da5497a1a2
3 changed files with 39 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
doc/filters.texi
View File

@ -8184,6 +8184,9 @@ means. For example, some MPEG based codecs export motion vectors through the
The filter accepts the following option:
@table @option
@item block
Display block partition structure using the luma plane.
@item mv
Set motion vectors to visualize.

2
libavfilter/version.h
View File

@ -31,7 +31,7 @@
#define LIBAVFILTER_VERSION_MAJOR 8
#define LIBAVFILTER_VERSION_MINOR 10
#define LIBAVFILTER_VERSION_MICRO 100
#define LIBAVFILTER_VERSION_MICRO 101
#define LIBAVFILTER_VERSION_INT AV_VERSION_INT(LIBAVFILTER_VERSION_MAJOR, \

35
libavfilter/vf_codecview.c
View File

@ -32,6 +32,7 @@
#include "libavutil/imgutils.h"
#include "libavutil/motion_vector.h"
#include "libavutil/opt.h"
#include "libavutil/video_enc_params.h"
#include "avfilter.h"
#include "qp_table.h"
#include "internal.h"
@ -52,6 +53,7 @@ typedef struct CodecViewContext {
unsigned mv_type;
int hsub, vsub;
int qp;
int block;
} CodecViewContext;
#define OFFSET(x) offsetof(CodecViewContext, x)
@ -73,6 +75,7 @@ static const AVOption codecview_options[] = {
CONST("if", "I-frames", FRAME_TYPE_I, "frame_type"),
CONST("pf", "P-frames", FRAME_TYPE_P, "frame_type"),
CONST("bf", "B-frames", FRAME_TYPE_B, "frame_type"),
{ "block", "set block partitioning structure to visualize", OFFSET(block), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS },
{ NULL }
};
@ -210,6 +213,21 @@ static void draw_arrow(uint8_t *buf, int sx, int sy, int ex,
draw_line(buf, sx, sy, ex, ey, w, h, stride, color);
}
static void draw_block_rectangle(uint8_t *buf, int sx, int sy, int w, int h, int stride, int color)
{
for (int x = sx; x < sx + w; x++)
buf[x] = color;
for (int y = sy; y < sy + h; y++) {
buf[sx] = color;
buf[sx + w - 1] = color;
buf += stride;
}
for (int x = sx; x < sx + w; x++)
buf[x] = color;
}
static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
{
AVFilterContext *ctx = inlink->dst;
@ -247,6 +265,23 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
av_freep(&qp_table);
}
if (s->block) {
AVFrameSideData *sd = av_frame_get_side_data(frame, AV_FRAME_DATA_VIDEO_ENC_PARAMS);
if (sd) {
AVVideoEncParams *par = (AVVideoEncParams*)sd->data;
const int stride = frame->linesize[0];
if (par->nb_blocks) {
for (int block_idx = 0; block_idx < par->nb_blocks; block_idx++) {
AVVideoBlockParams *b = av_video_enc_params_block(par, block_idx);
uint8_t *buf = frame->data[0] + b->src_y * stride;
draw_block_rectangle(buf, b->src_x, b->src_y, b->w, b->h, stride, 100);
}
}
}
}
if (s->mv || s->mv_type) {
AVFrameSideData *sd = av_frame_get_side_data(frame, AV_FRAME_DATA_MOTION_VECTORS);
if (sd) {