summaryrefslogtreecommitdiffstats
path: root/libavcodec/xpmdec.c
diff options
context:
space:
mode:
Diffstat (limited to 'libavcodec/xpmdec.c')
-rw-r--r--libavcodec/xpmdec.c443
1 files changed, 443 insertions, 0 deletions
diff --git a/libavcodec/xpmdec.c b/libavcodec/xpmdec.c
new file mode 100644
index 0000000..03172e4
--- /dev/null
+++ b/libavcodec/xpmdec.c
@@ -0,0 +1,443 @@
+/*
+ * XPM image format
+ *
+ * Copyright (c) 2012 Paul B Mahol
+ * Copyright (c) 2017 Paras Chadha
+ *
+ * This file is part of FFmpeg.
+ *
+ * FFmpeg 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.
+ *
+ * FFmpeg 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 FFmpeg; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#include "libavutil/parseutils.h"
+#include "libavutil/avstring.h"
+#include "avcodec.h"
+#include "internal.h"
+
+typedef struct XPMContext {
+ uint32_t *pixels;
+ int pixels_size;
+ uint8_t *buf;
+ int buf_size;
+} XPMDecContext;
+
+typedef struct ColorEntry {
+ const char *name; ///< a string representing the name of the color
+ uint32_t rgb_color; ///< RGB values for the color
+} ColorEntry;
+
+static int color_table_compare(const void *lhs, const void *rhs)
+{
+ return av_strcasecmp(lhs, ((const ColorEntry *)rhs)->name);
+}
+
+static const ColorEntry color_table[] = {
+ { "AliceBlue", 0xFFF0F8FF },
+ { "AntiqueWhite", 0xFFFAEBD7 },
+ { "Aqua", 0xFF00FFFF },
+ { "Aquamarine", 0xFF7FFFD4 },
+ { "Azure", 0xFFF0FFFF },
+ { "Beige", 0xFFF5F5DC },
+ { "Bisque", 0xFFFFE4C4 },
+ { "Black", 0xFF000000 },
+ { "BlanchedAlmond", 0xFFFFEBCD },
+ { "Blue", 0xFF0000FF },
+ { "BlueViolet", 0xFF8A2BE2 },
+ { "Brown", 0xFFA52A2A },
+ { "BurlyWood", 0xFFDEB887 },
+ { "CadetBlue", 0xFF5F9EA0 },
+ { "Chartreuse", 0xFF7FFF00 },
+ { "Chocolate", 0xFFD2691E },
+ { "Coral", 0xFFFF7F50 },
+ { "CornflowerBlue", 0xFF6495ED },
+ { "Cornsilk", 0xFFFFF8DC },
+ { "Crimson", 0xFFDC143C },
+ { "Cyan", 0xFF00FFFF },
+ { "DarkBlue", 0xFF00008B },
+ { "DarkCyan", 0xFF008B8B },
+ { "DarkGoldenRod", 0xFFB8860B },
+ { "DarkGray", 0xFFA9A9A9 },
+ { "DarkGreen", 0xFF006400 },
+ { "DarkKhaki", 0xFFBDB76B },
+ { "DarkMagenta", 0xFF8B008B },
+ { "DarkOliveGreen", 0xFF556B2F },
+ { "Darkorange", 0xFFFF8C00 },
+ { "DarkOrchid", 0xFF9932CC },
+ { "DarkRed", 0xFF8B0000 },
+ { "DarkSalmon", 0xFFE9967A },
+ { "DarkSeaGreen", 0xFF8FBC8F },
+ { "DarkSlateBlue", 0xFF483D8B },
+ { "DarkSlateGray", 0xFF2F4F4F },
+ { "DarkTurquoise", 0xFF00CED1 },
+ { "DarkViolet", 0xFF9400D3 },
+ { "DeepPink", 0xFFFF1493 },
+ { "DeepSkyBlue", 0xFF00BFFF },
+ { "DimGray", 0xFF696969 },
+ { "DodgerBlue", 0xFF1E90FF },
+ { "FireBrick", 0xFFB22222 },
+ { "FloralWhite", 0xFFFFFAF0 },
+ { "ForestGreen", 0xFF228B22 },
+ { "Fuchsia", 0xFFFF00FF },
+ { "Gainsboro", 0xFFDCDCDC },
+ { "GhostWhite", 0xFFF8F8FF },
+ { "Gold", 0xFFFFD700 },
+ { "GoldenRod", 0xFFDAA520 },
+ { "Gray", 0xFFBEBEBE },
+ { "Green", 0xFF00FF00 },
+ { "GreenYellow", 0xFFADFF2F },
+ { "HoneyDew", 0xFFF0FFF0 },
+ { "HotPink", 0xFFFF69B4 },
+ { "IndianRed", 0xFFCD5C5C },
+ { "Indigo", 0xFF4B0082 },
+ { "Ivory", 0xFFFFFFF0 },
+ { "Khaki", 0xFFF0E68C },
+ { "Lavender", 0xFFE6E6FA },
+ { "LavenderBlush", 0xFFFFF0F5 },
+ { "LawnGreen", 0xFF7CFC00 },
+ { "LemonChiffon", 0xFFFFFACD },
+ { "LightBlue", 0xFFADD8E6 },
+ { "LightCoral", 0xFFF08080 },
+ { "LightCyan", 0xFFE0FFFF },
+ { "LightGoldenRodYellow", 0xFFFAFAD2 },
+ { "LightGreen", 0xFF90EE90 },
+ { "LightGrey", 0xFFD3D3D3 },
+ { "LightPink", 0xFFFFB6C1 },
+ { "LightSalmon", 0xFFFFA07A },
+ { "LightSeaGreen", 0xFF20B2AA },
+ { "LightSkyBlue", 0xFF87CEFA },
+ { "LightSlateGray", 0xFF778899 },
+ { "LightSteelBlue", 0xFFB0C4DE },
+ { "LightYellow", 0xFFFFFFE0 },
+ { "Lime", 0xFF00FF00 },
+ { "LimeGreen", 0xFF32CD32 },
+ { "Linen", 0xFFFAF0E6 },
+ { "Magenta", 0xFFFF00FF },
+ { "Maroon", 0xFFB03060 },
+ { "MediumAquaMarine", 0xFF66CDAA },
+ { "MediumBlue", 0xFF0000CD },
+ { "MediumOrchid", 0xFFBA55D3 },
+ { "MediumPurple", 0xFF9370D8 },
+ { "MediumSeaGreen", 0xFF3CB371 },
+ { "MediumSlateBlue", 0xFF7B68EE },
+ { "MediumSpringGreen", 0xFF00FA9A },
+ { "MediumTurquoise", 0xFF48D1CC },
+ { "MediumVioletRed", 0xFFC71585 },
+ { "MidnightBlue", 0xFF191970 },
+ { "MintCream", 0xFFF5FFFA },
+ { "MistyRose", 0xFFFFE4E1 },
+ { "Moccasin", 0xFFFFE4B5 },
+ { "NavajoWhite", 0xFFFFDEAD },
+ { "Navy", 0xFF000080 },
+ { "None", 0x00000000 },
+ { "OldLace", 0xFFFDF5E6 },
+ { "Olive", 0xFF808000 },
+ { "OliveDrab", 0xFF6B8E23 },
+ { "Orange", 0xFFFFA500 },
+ { "OrangeRed", 0xFFFF4500 },
+ { "Orchid", 0xFFDA70D6 },
+ { "PaleGoldenRod", 0xFFEEE8AA },
+ { "PaleGreen", 0xFF98FB98 },
+ { "PaleTurquoise", 0xFFAFEEEE },
+ { "PaleVioletRed", 0xFFD87093 },
+ { "PapayaWhip", 0xFFFFEFD5 },
+ { "PeachPuff", 0xFFFFDAB9 },
+ { "Peru", 0xFFCD853F },
+ { "Pink", 0xFFFFC0CB },
+ { "Plum", 0xFFDDA0DD },
+ { "PowderBlue", 0xFFB0E0E6 },
+ { "Purple", 0xFFA020F0 },
+ { "Red", 0xFFFF0000 },
+ { "RosyBrown", 0xFFBC8F8F },
+ { "RoyalBlue", 0xFF4169E1 },
+ { "SaddleBrown", 0xFF8B4513 },
+ { "Salmon", 0xFFFA8072 },
+ { "SandyBrown", 0xFFF4A460 },
+ { "SeaGreen", 0xFF2E8B57 },
+ { "SeaShell", 0xFFFFF5EE },
+ { "Sienna", 0xFFA0522D },
+ { "Silver", 0xFFC0C0C0 },
+ { "SkyBlue", 0xFF87CEEB },
+ { "SlateBlue", 0xFF6A5ACD },
+ { "SlateGray", 0xFF708090 },
+ { "Snow", 0xFFFFFAFA },
+ { "SpringGreen", 0xFF00FF7F },
+ { "SteelBlue", 0xFF4682B4 },
+ { "Tan", 0xFFD2B48C },
+ { "Teal", 0xFF008080 },
+ { "Thistle", 0xFFD8BFD8 },
+ { "Tomato", 0xFFFF6347 },
+ { "Turquoise", 0xFF40E0D0 },
+ { "Violet", 0xFFEE82EE },
+ { "Wheat", 0xFFF5DEB3 },
+ { "White", 0xFFFFFFFF },
+ { "WhiteSmoke", 0xFFF5F5F5 },
+ { "Yellow", 0xFFFFFF00 },
+ { "YellowGreen", 0xFF9ACD32 }
+};
+
+static unsigned hex_char_to_number(uint8_t x)
+{
+ if (x >= 'a' && x <= 'f')
+ x -= 'a' - 10;
+ else if (x >= 'A' && x <= 'F')
+ x -= 'A' - 10;
+ else if (x >= '0' && x <= '9')
+ x -= '0';
+ else
+ x = 0;
+ return x;
+}
+
+/*
+ * Function same as strcspn but ignores characters if they are inside a C style comments
+ */
+static size_t mod_strcspn(const char *string, const char *reject)
+{
+ int i, j;
+
+ for (i = 0; string && string[i]; i++) {
+ if (string[i] == '/' && string[i+1] == '*') {
+ i += 2;
+ while ( string && string[i] && (string[i] != '*' || string[i+1] != '/') )
+ i++;
+ i++;
+ } else if (string[i] == '/' && string[i+1] == '/') {
+ i += 2;
+ while ( string && string[i] && string[i] != '\n' )
+ i++;
+ } else {
+ for (j = 0; reject && reject[j]; j++) {
+ if (string[i] == reject[j])
+ break;
+ }
+ if (reject && reject[j])
+ break;
+ }
+ }
+ return i;
+}
+
+static uint32_t color_string_to_rgba(const char *p, int len)
+{
+ uint32_t ret = 0xFF000000;
+ const ColorEntry *entry;
+ char color_name[100];
+
+ len = FFMIN(FFMAX(len, 0), sizeof(color_name) - 1);
+
+ if (*p == '#') {
+ p++;
+ len--;
+ if (len == 3) {
+ ret |= (hex_char_to_number(p[2]) << 4) |
+ (hex_char_to_number(p[1]) << 12) |
+ (hex_char_to_number(p[0]) << 20);
+ } else if (len == 4) {
+ ret = (hex_char_to_number(p[3]) << 4) |
+ (hex_char_to_number(p[2]) << 12) |
+ (hex_char_to_number(p[1]) << 20) |
+ (hex_char_to_number(p[0]) << 28);
+ } else if (len == 6) {
+ ret |= hex_char_to_number(p[5]) |
+ (hex_char_to_number(p[4]) << 4) |
+ (hex_char_to_number(p[3]) << 8) |
+ (hex_char_to_number(p[2]) << 12) |
+ (hex_char_to_number(p[1]) << 16) |
+ (hex_char_to_number(p[0]) << 20);
+ } else if (len == 8) {
+ ret = hex_char_to_number(p[7]) |
+ (hex_char_to_number(p[6]) << 4) |
+ (hex_char_to_number(p[5]) << 8) |
+ (hex_char_to_number(p[4]) << 12) |
+ (hex_char_to_number(p[3]) << 16) |
+ (hex_char_to_number(p[2]) << 20) |
+ (hex_char_to_number(p[1]) << 24) |
+ (hex_char_to_number(p[0]) << 28);
+ }
+ } else {
+ strncpy(color_name, p, len);
+ color_name[len] = '\0';
+
+ entry = bsearch(color_name,
+ color_table,
+ FF_ARRAY_ELEMS(color_table),
+ sizeof(ColorEntry),
+ color_table_compare);
+
+ if (!entry)
+ return ret;
+
+ ret = entry->rgb_color;
+ }
+ return ret;
+}
+
+static int ascii2index(const uint8_t *cpixel, int cpp)
+{
+ const uint8_t *p = cpixel;
+ int n = 0, m = 1, i;
+
+ for (i = 0; i < cpp; i++) {
+ if (*p < ' ' || *p > '~')
+ return AVERROR_INVALIDDATA;
+ n += (*p++ - ' ') * m;
+ m *= 95;
+ }
+ return n;
+}
+
+static int xpm_decode_frame(AVCodecContext *avctx, void *data,
+ int *got_frame, AVPacket *avpkt)
+{
+ XPMDecContext *x = avctx->priv_data;
+ AVFrame *p=data;
+ const uint8_t *end, *ptr;
+ int ncolors, cpp, ret, i, j;
+ int64_t size;
+ uint32_t *dst;
+
+ avctx->pix_fmt = AV_PIX_FMT_BGRA;
+
+ av_fast_padded_malloc(&x->buf, &x->buf_size, avpkt->size);
+ if (!x->buf)
+ return AVERROR(ENOMEM);
+ memcpy(x->buf, avpkt->data, avpkt->size);
+ x->buf[avpkt->size] = 0;
+
+ ptr = x->buf;
+ end = x->buf + avpkt->size;
+ while (end - ptr > 9 && memcmp(ptr, "/* XPM */", 9))
+ ptr++;
+
+ if (end - ptr <= 9) {
+ av_log(avctx, AV_LOG_ERROR, "missing signature\n");
+ return AVERROR_INVALIDDATA;
+ }
+
+ ptr += mod_strcspn(ptr, "\"");
+ if (sscanf(ptr, "\"%u %u %u %u\",",
+ &avctx->width, &avctx->height, &ncolors, &cpp) != 4) {
+ av_log(avctx, AV_LOG_ERROR, "missing image parameters\n");
+ return AVERROR_INVALIDDATA;
+ }
+
+ if ((ret = ff_set_dimensions(avctx, avctx->width, avctx->height)) < 0)
+ return ret;
+
+ if ((ret = ff_get_buffer(avctx, p, 0)) < 0)
+ return ret;
+
+ if (cpp <= 0 || cpp >= 5) {
+ av_log(avctx, AV_LOG_ERROR, "unsupported/invalid number of chars per pixel: %d\n", cpp);
+ return AVERROR_INVALIDDATA;
+ }
+
+ size = 1;
+ for (i = 0; i < cpp; i++)
+ size *= 95;
+
+ if (ncolors <= 0 || ncolors > size) {
+ av_log(avctx, AV_LOG_ERROR, "invalid number of colors: %d\n", ncolors);
+ return AVERROR_INVALIDDATA;
+ }
+
+ size *= 4;
+
+ av_fast_padded_malloc(&x->pixels, &x->pixels_size, size);
+ if (!x->pixels)
+ return AVERROR(ENOMEM);
+
+ ptr += mod_strcspn(ptr, ",") + 1;
+ if (end - ptr < 1)
+ return AVERROR_INVALIDDATA;
+
+ for (i = 0; i < ncolors; i++) {
+ const uint8_t *index;
+ int len;
+
+ ptr += mod_strcspn(ptr, "\"") + 1;
+ if (end - ptr < cpp)
+ return AVERROR_INVALIDDATA;
+ index = ptr;
+ ptr += cpp;
+
+ ptr = strstr(ptr, "c ");
+ if (ptr) {
+ ptr += 2;
+ } else {
+ return AVERROR_INVALIDDATA;
+ }
+
+ len = strcspn(ptr, "\" ");
+
+ if ((ret = ascii2index(index, cpp)) < 0)
+ return ret;
+
+ x->pixels[ret] = color_string_to_rgba(ptr, len);
+ ptr += mod_strcspn(ptr, ",") + 1;
+ if (end - ptr < 1)
+ return AVERROR_INVALIDDATA;
+ }
+
+ for (i = 0; i < avctx->height; i++) {
+ dst = (uint32_t *)(p->data[0] + i * p->linesize[0]);
+ if (end - ptr < 1)
+ return AVERROR_INVALIDDATA;
+ ptr += mod_strcspn(ptr, "\"") + 1;
+ if (end - ptr < 1)
+ return AVERROR_INVALIDDATA;
+
+ for (j = 0; j < avctx->width; j++) {
+ if (end - ptr < cpp)
+ return AVERROR_INVALIDDATA;
+
+ if ((ret = ascii2index(ptr, cpp)) < 0)
+ return ret;
+
+ *dst++ = x->pixels[ret];
+ ptr += cpp;
+ }
+ ptr += mod_strcspn(ptr, ",") + 1;
+ }
+
+ p->key_frame = 1;
+ p->pict_type = AV_PICTURE_TYPE_I;
+
+ *got_frame = 1;
+
+ return avpkt->size;
+}
+
+static av_cold int xpm_decode_close(AVCodecContext *avctx)
+{
+ XPMDecContext *x = avctx->priv_data;
+ av_freep(&x->pixels);
+
+ av_freep(&x->buf);
+ x->buf_size = 0;
+
+ return 0;
+}
+
+AVCodec ff_xpm_decoder = {
+ .name = "xpm",
+ .type = AVMEDIA_TYPE_VIDEO,
+ .id = AV_CODEC_ID_XPM,
+ .priv_data_size = sizeof(XPMDecContext),
+ .close = xpm_decode_close,
+ .decode = xpm_decode_frame,
+ .capabilities = AV_CODEC_CAP_DR1,
+ .long_name = NULL_IF_CONFIG_SMALL("XPM (X PixMap) image")
+};
OpenPOWER on IntegriCloud