From 37cf92c07a7d664907ede87e9464680cc36bd5ec Mon Sep 17 00:00:00 2001 From: wm4 Date: Sun, 3 Jul 2016 19:11:33 +0200 Subject: [PATCH] sub: change how libass output is converted to RGBA in some cases This affects VOs (or other code which render OSD) which does not support the LIBASS format, but only RGBA. Instead of having a converter stage in osd.c, make mp_ass_packer_pack() output directly in RGBA. In general, this is work towards refcounted subtitle images. Although we could keep the "converter" design, doing it this way seems simpler, at least considering the current situation with only 2 OSD formats. It also prevents copying & packing the data twice, which will lead to better performance. (Although I guess this case is not important at all.) It also fixes --force-rgba-osd-rendering when used with vo_opengl, vo_vdpau, and vo_direct3d. --- sub/ass_mp.c | 105 +++++++++++++++++++++++++++++++++++++++++++++- sub/img_convert.c | 97 ------------------------------------------ sub/img_convert.h | 3 -- sub/osd.c | 24 ----------- sub/osd_state.h | 6 --- 5 files changed, 103 insertions(+), 132 deletions(-) diff --git a/sub/ass_mp.c b/sub/ass_mp.c index fc2c17d115..84a706b681 100644 --- a/sub/ass_mp.c +++ b/sub/ass_mp.c @@ -32,6 +32,7 @@ #include "common/msg.h" #include "options/path.h" #include "ass_mp.h" +#include "img_convert.h" #include "osd.h" #include "stream/stream.h" #include "options/options.h" @@ -149,11 +150,45 @@ void mp_ass_flush_old_events(ASS_Track *track, long long ts) } } +static void draw_ass_rgba(unsigned char *src, int src_w, int src_h, + int src_stride, unsigned char *dst, size_t dst_stride, + int dst_x, int dst_y, uint32_t color) +{ + const unsigned int r = (color >> 24) & 0xff; + const unsigned int g = (color >> 16) & 0xff; + const unsigned int b = (color >> 8) & 0xff; + const unsigned int a = 0xff - (color & 0xff); + + dst += dst_y * dst_stride + dst_x * 4; + + for (int y = 0; y < src_h; y++, dst += dst_stride, src += src_stride) { + uint32_t *dstrow = (uint32_t *) dst; + for (int x = 0; x < src_w; x++) { + const unsigned int v = src[x]; + int rr = (r * a * v); + int gg = (g * a * v); + int bb = (b * a * v); + int aa = a * v; + uint32_t dstpix = dstrow[x]; + unsigned int dstb = dstpix & 0xFF; + unsigned int dstg = (dstpix >> 8) & 0xFF; + unsigned int dstr = (dstpix >> 16) & 0xFF; + unsigned int dsta = (dstpix >> 24) & 0xFF; + dstb = (bb + dstb * (255 * 255 - aa)) / (255 * 255); + dstg = (gg + dstg * (255 * 255 - aa)) / (255 * 255); + dstr = (rr + dstr * (255 * 255 - aa)) / (255 * 255); + dsta = (aa * 255 + dsta * (255 * 255 - aa)) / (255 * 255); + dstrow[x] = dstb | (dstg << 8) | (dstr << 16) | (dsta << 24); + } + } +} + struct mp_ass_packer { struct sub_bitmap *cached_parts; // only for the array memory struct mp_image *cached_img; struct sub_bitmaps cached_subs; bool cached_subs_valid; + struct sub_bitmap rgba_imgs[MP_SUB_BB_LIST_MAX]; struct bitmap_packer *packer; }; @@ -224,6 +259,60 @@ static bool pack_libass(struct mp_ass_packer *p, struct sub_bitmaps *res) return true; } +static bool pack_rgba(struct mp_ass_packer *p, struct sub_bitmaps *res) +{ + struct mp_rect bb_list[MP_SUB_BB_LIST_MAX]; + int num_bb = mp_get_sub_bb_list(res, bb_list, MP_SUB_BB_LIST_MAX); + + struct sub_bitmaps imgs = { + .change_id = res->change_id, + .format = SUBBITMAP_RGBA, + .parts = p->rgba_imgs, + .num_parts = num_bb, + }; + + for (int n = 0; n < imgs.num_parts; n++) { + imgs.parts[n].w = bb_list[n].x1 - bb_list[n].x0; + imgs.parts[n].h = bb_list[n].y1 - bb_list[n].y0; + } + + if (!pack(p, &imgs, IMGFMT_BGRA)) + return false; + + for (int n = 0; n < num_bb; n++) { + struct mp_rect bb = bb_list[n]; + struct sub_bitmap *b = &imgs.parts[n]; + + b->x = bb.x0; + b->y = bb.y0; + b->w = b->dw = bb.x1 - bb.x0; + b->h = b->dh = bb.y1 - bb.y0; + b->stride = imgs.packed->stride[0]; + b->bitmap = (uint8_t *)imgs.packed->planes[0] + + b->stride * b->src_y + b->src_x * 4; + + memset_pic(b->bitmap, 0, b->w * 4, b->h, b->stride); + + for (int i = 0; i < res->num_parts; i++) { + struct sub_bitmap *s = &res->parts[i]; + + // Assume mp_get_sub_bb_list() never splits sub bitmaps + // So we don't clip/adjust the size of the sub bitmap + if (s->x > bb.x1 || s->x + s->w < bb.x0 || + s->y > bb.y1 || s->y + s->h < bb.y0) + continue; + + draw_ass_rgba(s->bitmap, s->w, s->h, s->stride, + b->bitmap, b->stride, + s->x - bb.x0, s->y - bb.y0, + s->libass.color); + } + } + + *res = imgs; + return true; +} + // Pack the contents of image_lists[0] to image_lists[num_image_lists-1] into // a single image, and make *out point to it. *out is completely overwritten. // If libass reported any change, image_lists_changed must be set (it then @@ -233,7 +322,12 @@ void mp_ass_packer_pack(struct mp_ass_packer *p, ASS_Image **image_lists, int num_image_lists, bool image_lists_changed, int preferred_osd_format, struct sub_bitmaps *out) { - if (p->cached_subs_valid && !image_lists_changed) { + int format = preferred_osd_format == SUBBITMAP_RGBA ? SUBBITMAP_RGBA + : SUBBITMAP_LIBASS; + + if (p->cached_subs_valid && !image_lists_changed && + p->cached_subs.format == format) + { *out = p->cached_subs; return; } @@ -265,7 +359,14 @@ void mp_ass_packer_pack(struct mp_ass_packer *p, ASS_Image **image_lists, } } - if (!pack_libass(p, &res)) + bool r = false; + if (format == SUBBITMAP_RGBA) { + r = pack_rgba(p, &res); + } else { + r = pack_libass(p, &res); + } + + if (!r) return; *out = res; diff --git a/sub/img_convert.c b/sub/img_convert.c index 78be881fdf..e49b7785af 100644 --- a/sub/img_convert.c +++ b/sub/img_convert.c @@ -109,103 +109,6 @@ bool osd_scale_rgba(struct osd_conv_cache *c, struct sub_bitmaps *imgs) return true; } -static void draw_ass_rgba(unsigned char *src, int src_w, int src_h, - int src_stride, unsigned char *dst, size_t dst_stride, - int dst_x, int dst_y, uint32_t color) -{ - const unsigned int r = (color >> 24) & 0xff; - const unsigned int g = (color >> 16) & 0xff; - const unsigned int b = (color >> 8) & 0xff; - const unsigned int a = 0xff - (color & 0xff); - - dst += dst_y * dst_stride + dst_x * 4; - - for (int y = 0; y < src_h; y++, dst += dst_stride, src += src_stride) { - uint32_t *dstrow = (uint32_t *) dst; - for (int x = 0; x < src_w; x++) { - const unsigned int v = src[x]; - int rr = (r * a * v); - int gg = (g * a * v); - int bb = (b * a * v); - int aa = a * v; - uint32_t dstpix = dstrow[x]; - unsigned int dstb = dstpix & 0xFF; - unsigned int dstg = (dstpix >> 8) & 0xFF; - unsigned int dstr = (dstpix >> 16) & 0xFF; - unsigned int dsta = (dstpix >> 24) & 0xFF; - dstb = (bb + dstb * (255 * 255 - aa)) / (255 * 255); - dstg = (gg + dstg * (255 * 255 - aa)) / (255 * 255); - dstr = (rr + dstr * (255 * 255 - aa)) / (255 * 255); - dsta = (aa * 255 + dsta * (255 * 255 - aa)) / (255 * 255); - dstrow[x] = dstb | (dstg << 8) | (dstr << 16) | (dsta << 24); - } - } -} - -bool osd_conv_ass_to_rgba(struct osd_conv_cache *c, struct sub_bitmaps *imgs) -{ - struct sub_bitmaps src = *imgs; - if (src.format != SUBBITMAP_LIBASS) - return false; - assert(!src.scaled); // ASS is always unscaled - - struct mp_rect bb_list[MP_SUB_BB_LIST_MAX]; - int num_bb = mp_get_sub_bb_list(&src, bb_list, MP_SUB_BB_LIST_MAX); - - imgs->format = SUBBITMAP_RGBA; - imgs->parts = c->part; - imgs->num_parts = num_bb; - imgs->packed = NULL; - - size_t newsize = 0; - for (int n = 0; n < num_bb; n++) { - struct mp_rect bb = bb_list[n]; - int w = bb.x1 - bb.x0; - int h = bb.y1 - bb.y0; - int stride = w * 4; - newsize += h * stride; - } - - if (talloc_get_size(c->scratch) < newsize) { - talloc_free(c->scratch); - c->scratch = talloc_array(c, uint8_t, newsize); - } - - uint8_t *data = c->scratch; - - for (int n = 0; n < num_bb; n++) { - struct mp_rect bb = bb_list[n]; - struct sub_bitmap *bmp = &c->part[n]; - - bmp->x = bb.x0; - bmp->y = bb.y0; - bmp->w = bmp->dw = bb.x1 - bb.x0; - bmp->h = bmp->dh = bb.y1 - bb.y0; - bmp->stride = bmp->w * 4; - bmp->bitmap = data; - data += bmp->h * bmp->stride; - - memset_pic(bmp->bitmap, 0, bmp->w * 4, bmp->h, bmp->stride); - - for (int p = 0; p < src.num_parts; p++) { - struct sub_bitmap *s = &src.parts[p]; - - // Assume mp_get_sub_bb_list() never splits sub bitmaps - // So we don't clip/adjust the size of the sub bitmap - if (s->x > bb.x1 || s->x + s->w < bb.x0 || - s->y > bb.y1 || s->y + s->h < bb.y0) - continue; - - draw_ass_rgba(s->bitmap, s->w, s->h, s->stride, - bmp->bitmap, bmp->stride, - s->x - bb.x0, s->y - bb.y0, - s->libass.color); - } - } - - return true; -} - bool mp_sub_bitmaps_bb(struct sub_bitmaps *imgs, struct mp_rect *out_bb) { struct mp_rect bb = {INT_MAX, INT_MAX, INT_MIN, INT_MIN}; diff --git a/sub/img_convert.h b/sub/img_convert.h index a0020df497..bf228a1f65 100644 --- a/sub/img_convert.h +++ b/sub/img_convert.h @@ -10,9 +10,6 @@ struct mp_rect; struct osd_conv_cache *osd_conv_cache_new(void); -// These functions convert from one OSD format to another. On success, they copy -// the converted image data into c, and change imgs to point to the data. -bool osd_conv_ass_to_rgba(struct osd_conv_cache *c, struct sub_bitmaps *imgs); // Sub postprocessing void mp_blur_rgba_sub_bitmap(struct sub_bitmap *d, double gblur); bool osd_scale_rgba(struct osd_conv_cache *c, struct sub_bitmaps *imgs); diff --git a/sub/osd.c b/sub/osd.c index 001666d088..55edf6b123 100644 --- a/sub/osd.c +++ b/sub/osd.c @@ -127,8 +127,6 @@ struct osd_state *osd_create(struct mpv_global *global) .text = talloc_strdup(obj, ""), .progbar_state = {.type = -1}, }; - for (int i = 0; i < OSD_CONV_CACHE_MAX; i++) - obj->cache[i] = talloc_steal(obj, osd_conv_cache_new()); osd->objs[n] = obj; } @@ -251,11 +249,6 @@ static void render_object(struct osd_state *osd, struct osd_object *obj, if (!sub_formats[format] || opts->force_rgba_osd) format = SUBBITMAP_RGBA; - bool formats[SUBBITMAP_COUNT]; - memcpy(formats, sub_formats, sizeof(formats)); - if (opts->force_rgba_osd) - formats[SUBBITMAP_LIBASS] = false; - *out_imgs = (struct sub_bitmaps) {0}; check_obj_resize(osd, res, obj); @@ -285,25 +278,8 @@ static void render_object(struct osd_state *osd, struct osd_object *obj, if (out_imgs->num_parts == 0) return; - if (obj->cached.change_id == obj->vo_change_id && formats[obj->cached.format]) - { - *out_imgs = obj->cached; - return; - } - out_imgs->render_index = obj->type; out_imgs->change_id = obj->vo_change_id; - - if (formats[out_imgs->format]) - return; - - bool cached = false; // do we have a copy of all the image data? - - if (formats[SUBBITMAP_RGBA] && out_imgs->format == SUBBITMAP_LIBASS) - cached |= osd_conv_ass_to_rgba(obj->cache[3], out_imgs); - - if (cached) - obj->cached = *out_imgs; } // draw_flags is a bit field of OSD_DRAW_* constants diff --git a/sub/osd_state.h b/sub/osd_state.h index c2e7911d1b..81bdbd633a 100644 --- a/sub/osd_state.h +++ b/sub/osd_state.h @@ -5,8 +5,6 @@ #include "osd.h" -#define OSD_CONV_CACHE_MAX 4 - enum mp_osdtype { OSDTYPE_SUB, OSDTYPE_SUB2, // IDs must be numerically successive @@ -48,10 +46,6 @@ struct osd_object { // OSDTYPE_EXTERNAL2 struct sub_bitmaps *external2; - // caches for OSD conversion (internal to render_object()) - struct osd_conv_cache *cache[OSD_CONV_CACHE_MAX]; - struct sub_bitmaps cached; - // VO cache state int vo_change_id; struct mp_osd_res vo_res;