virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-08-10 06:10:52 +02:00
Code Issues Releases Activity

avcodec/mpegvideo: flip motion vector visualization for backward motion vectors

Browse Source 
Also support changing arrow head/tail shape

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer
2014-07-09 03:07:25 +02:00
parent 98eab98159
commit c8b2cf4996
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
libavcodec/mpegvideo.c
View File

@@ -2036,10 +2036,15 @@ static void draw_line(uint8_t *buf, int sx, int sy, int ex, int ey,
* @param color color of the arrow * @param color color of the arrow
*/ */
static void draw_arrow(uint8_t *buf, int sx, int sy, int ex, static void draw_arrow(uint8_t *buf, int sx, int sy, int ex,
int ey, int w, int h, int stride, int color) int ey, int w, int h, int stride, int color, int tail, int direction)
{ {
int dx,dy; int dx,dy;
if (direction) {
FFSWAP(int, sx, ex);
FFSWAP(int, sy, ey);
}
sx = av_clip(sx, -100, w + 100); sx = av_clip(sx, -100, w + 100);
sy = av_clip(sy, -100, h + 100); sy = av_clip(sy, -100, h + 100);
ex = av_clip(ex, -100, w + 100); ex = av_clip(ex, -100, w + 100);
@@ -2057,6 +2062,11 @@ static void draw_arrow(uint8_t *buf, int sx, int sy, int ex,
rx = ROUNDED_DIV(rx * 3 << 4, length); rx = ROUNDED_DIV(rx * 3 << 4, length);
ry = ROUNDED_DIV(ry * 3 << 4, length); ry = ROUNDED_DIV(ry * 3 << 4, length);
if (tail) {
rx = -rx;
ry = -ry;
}
draw_line(buf, sx, sy, sx + rx, sy + ry, w, h, stride, color); draw_line(buf, sx, sy, sx + rx, sy + ry, w, h, stride, color);
draw_line(buf, sx, sy, sx - ry, sy + rx, w, h, stride, color); draw_line(buf, sx, sy, sx - ry, sy + rx, w, h, stride, color);
} }
@@ -2210,7 +2220,7 @@ void ff_print_debug_info2(AVCodecContext *avctx, AVFrame *pict, uint8_t *mbskip_
int mx = (motion_val[direction][xy][0] >> shift) + sx; int mx = (motion_val[direction][xy][0] >> shift) + sx;
int my = (motion_val[direction][xy][1] >> shift) + sy; int my = (motion_val[direction][xy][1] >> shift) + sy;
draw_arrow(ptr, sx, sy, mx, my, width, draw_arrow(ptr, sx, sy, mx, my, width,
height, pict->linesize[0], 100); height, pict->linesize[0], 100, 0, direction);
} }
} else if (IS_16X8(mbtype_table[mb_index])) { } else if (IS_16X8(mbtype_table[mb_index])) {
int i; int i;
@@ -2225,7 +2235,7 @@ void ff_print_debug_info2(AVCodecContext *avctx, AVFrame *pict, uint8_t *mbskip_
my *= 2; my *= 2;
draw_arrow(ptr, sx, sy, mx + sx, my + sy, width, draw_arrow(ptr, sx, sy, mx + sx, my + sy, width,
height, pict->linesize[0], 100); height, pict->linesize[0], 100, 0, direction);
} }
} else if (IS_8X16(mbtype_table[mb_index])) { } else if (IS_8X16(mbtype_table[mb_index])) {
int i; int i;
@@ -2240,7 +2250,7 @@ void ff_print_debug_info2(AVCodecContext *avctx, AVFrame *pict, uint8_t *mbskip_
my *= 2; my *= 2;
draw_arrow(ptr, sx, sy, mx + sx, my + sy, width, draw_arrow(ptr, sx, sy, mx + sx, my + sy, width,
height, pict->linesize[0], 100); height, pict->linesize[0], 100, 0, direction);
} }
} else { } else {
int sx= mb_x * 16 + 8; int sx= mb_x * 16 + 8;
@@ -2248,7 +2258,7 @@ void ff_print_debug_info2(AVCodecContext *avctx, AVFrame *pict, uint8_t *mbskip_
int xy= (mb_x + mb_y * mv_stride) << mv_sample_log2; int xy= (mb_x + mb_y * mv_stride) << mv_sample_log2;
int mx= (motion_val[direction][xy][0]>>shift) + sx; int mx= (motion_val[direction][xy][0]>>shift) + sx;
int my= (motion_val[direction][xy][1]>>shift) + sy; int my= (motion_val[direction][xy][1]>>shift) + sy;
draw_arrow(ptr, sx, sy, mx, my, width, height, pict->linesize[0], 100); draw_arrow(ptr, sx, sy, mx, my, width, height, pict->linesize[0], 100, 0, direction);
} }
} }
} }