120 lines · 4.4 KB · C
RawDownload
/* * This file is part of Libav. * * Libav is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * Libav is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with Libav; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include <string.h> #include "libavutil/avutil.h"#include "libavutil/colorspace.h"#include "libavutil/mem.h"#include "libavutil/pixdesc.h"#include "drawutils.h" enum { RED = 0, GREEN, BLUE, ALPHA }; int ff_fill_line_with_color(uint8_t *line[4], int pixel_step[4], int w, uint8_t dst_color[4],                            enum AVPixelFormat pix_fmt, uint8_t rgba_color[4],                            int *is_packed_rgba, uint8_t rgba_map_ptr[4]){    uint8_t rgba_map[4] = {0};    int i;    const AVPixFmtDescriptor *pix_desc = av_pix_fmt_desc_get(pix_fmt);    int hsub = pix_desc->log2_chroma_w;     *is_packed_rgba = 1;    switch (pix_fmt) {    case AV_PIX_FMT_ARGB:  rgba_map[ALPHA] = 0; rgba_map[RED  ] = 1; rgba_map[GREEN] = 2; rgba_map[BLUE ] = 3; break;    case AV_PIX_FMT_ABGR:  rgba_map[ALPHA] = 0; rgba_map[BLUE ] = 1; rgba_map[GREEN] = 2; rgba_map[RED  ] = 3; break;    case AV_PIX_FMT_RGBA:    case AV_PIX_FMT_RGB24: rgba_map[RED  ] = 0; rgba_map[GREEN] = 1; rgba_map[BLUE ] = 2; rgba_map[ALPHA] = 3; break;    case AV_PIX_FMT_BGRA:    case AV_PIX_FMT_BGR24: rgba_map[BLUE ] = 0; rgba_map[GREEN] = 1; rgba_map[RED  ] = 2; rgba_map[ALPHA] = 3; break;    default:        *is_packed_rgba = 0;    }     if (*is_packed_rgba) {        pixel_step[0] = (av_get_bits_per_pixel(pix_desc))>>3;        for (i = 0; i < 4; i++)            dst_color[rgba_map[i]] = rgba_color[i];         line[0] = av_malloc(w * pixel_step[0]);        for (i = 0; i < w; i++)            memcpy(line[0] + i * pixel_step[0], dst_color, pixel_step[0]);        if (rgba_map_ptr)            memcpy(rgba_map_ptr, rgba_map, sizeof(rgba_map[0]) * 4);    } else {        int plane;         dst_color[0] = RGB_TO_Y_CCIR(rgba_color[0], rgba_color[1], rgba_color[2]);        dst_color[1] = RGB_TO_U_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);        dst_color[2] = RGB_TO_V_CCIR(rgba_color[0], rgba_color[1], rgba_color[2], 0);        dst_color[3] = rgba_color[3];         for (plane = 0; plane < 4; plane++) {            int line_size;            int hsub1 = (plane == 1 || plane == 2) ? hsub : 0;             pixel_step[plane] = 1;            line_size = (w >> hsub1) * pixel_step[plane];            line[plane] = av_malloc(line_size);            memset(line[plane], dst_color[plane], line_size);        }    }     return 0;} void ff_draw_rectangle(uint8_t *dst[4], int dst_linesize[4],                       uint8_t *src[4], int pixelstep[4],                       int hsub, int vsub, int x, int y, int w, int h){    int i, plane;    uint8_t *p;     for (plane = 0; plane < 4 && dst[plane]; plane++) {        int hsub1 = plane == 1 || plane == 2 ? hsub : 0;        int vsub1 = plane == 1 || plane == 2 ? vsub : 0;         p = dst[plane] + (y >> vsub1) * dst_linesize[plane];        for (i = 0; i < (h >> vsub1); i++) {            memcpy(p + (x >> hsub1) * pixelstep[plane],                   src[plane], (w >> hsub1) * pixelstep[plane]);            p += dst_linesize[plane];        }    }} void ff_copy_rectangle(uint8_t *dst[4], int dst_linesize[4],                       uint8_t *src[4], int src_linesize[4], int pixelstep[4],                       int hsub, int vsub, int x, int y, int y2, int w, int h){    int i, plane;    uint8_t *p;     for (plane = 0; plane < 4 && dst[plane]; plane++) {        int hsub1 = plane == 1 || plane == 2 ? hsub : 0;        int vsub1 = plane == 1 || plane == 2 ? vsub : 0;         p = dst[plane] + (y >> vsub1) * dst_linesize[plane];        for (i = 0; i < (h >> vsub1); i++) {            memcpy(p + (x >> hsub1) * pixelstep[plane],                   src[plane] + src_linesize[plane]*(i+(y2>>vsub1)), (w >> hsub1) * pixelstep[plane]);            p += dst_linesize[plane];        }    }}

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t