modules/codec/xwd.c

   1 /*****************************************************************************
   2  * xwd.c: X Window system raster image dump pseudo-decoder
   3  *****************************************************************************
   4  * Copyright (C) 2012 Rémi Denis-Courmont
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU Lesser General Public License as published by
   8  * the Free Software Foundation; either version 2.1 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14  * GNU Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  19  *****************************************************************************/
  20
  21 #ifdef HAVE_CONFIG_H
  22 # include "config.h"
  23 #endif
  24
  25 #include <assert.h>
  26 #include <arpa/inet.h>
  27 #include <X11/XWDFile.h>
  28
  29 #include <vlc_common.h>
  30 #include <vlc_plugin.h>
  31 #include <vlc_codec.h>
  32
  33 static int Open(vlc_object_t *);
  34
  35 vlc_module_begin()
  36     set_description(N_("XWD image decoder"))
  37     set_capability("video decoder", 50)
  38     set_category(CAT_INPUT)
  39     set_subcategory(SUBCAT_INPUT_VCODEC)
  40     set_callbacks(Open, NULL)
  41 vlc_module_end()
  42
  43 static int Decode(decoder_t *, block_t *);
  44
  45 static int Open(vlc_object_t *obj)
  46 {
  47     decoder_t *dec = (decoder_t *)obj;
  48
  49     if (dec->fmt_in.i_codec != VLC_CODEC_XWD)
  50         return VLC_EGENERIC;
  51
  52     dec->pf_decode = Decode;
  53     es_format_Copy(&dec->fmt_out, &dec->fmt_in);
  54     dec->fmt_out.i_codec = VLC_CODEC_RGB32;
  55     return VLC_SUCCESS;
  56 }
  57
  58 static int Decode (decoder_t *dec, block_t *block)
  59 {
  60     picture_t *pic = NULL;
  61
  62     if (block == NULL) /* No Drain */
  63         return VLCDEC_SUCCESS;
  64
  65     if (block->i_pts <= VLC_TS_INVALID)
  66         goto drop; /* undated block, should never happen */
  67     if (block->i_buffer < sz_XWDheader)
  68         goto drop;
  69
  70     /* Skip XWD header */
  71     const XWDFileHeader *hdr = (const void *)block->p_buffer;
  72     uint32_t hdrlen = ntohl(hdr->header_size);
  73     if (hdrlen < sz_XWDheader
  74      || ntohl(hdr->file_version) < XWD_FILE_VERSION
  75      || ntohl(hdr->pixmap_format) != 2 /* ZPixmap */)
  76         goto drop;
  77
  78     hdrlen += ntohl(hdr->ncolors) * sz_XWDColor;
  79     if (hdrlen > block->i_buffer)
  80         goto drop;
  81     block->p_buffer += hdrlen;
  82     block->i_buffer -= hdrlen;
  83
  84     /* Parse XWD header */
  85     vlc_fourcc_t chroma = 0;
  86     switch (ntohl(hdr->pixmap_depth))
  87     {
  88         case 8:
  89             if (ntohl(hdr->bits_per_pixel) == 8)
  90                 chroma = VLC_CODEC_RGB8;
  91             break;
  92         case 15:
  93             if (ntohl(hdr->bits_per_pixel) == 16)
  94                 chroma = VLC_CODEC_RGB15;
  95             break;
  96         case 16:
  97             if (ntohl(hdr->bits_per_pixel) == 16)
  98                 chroma = VLC_CODEC_RGB16;
  99             break;
 100         case 24:
 101             switch (ntohl(hdr->bits_per_pixel))
 102             {
 103                 case 32: chroma = VLC_CODEC_RGB32; break;
 104                 case 24: chroma = VLC_CODEC_RGB24; break;
 105             }
 106             break;
 107         case 32:
 108             if (ntohl(hdr->bits_per_pixel) == 32)
 109                 chroma = VLC_CODEC_ARGB;
 110             break;
 111     }
 112     /* TODO: check image endianess, set RGB mask */
 113     if (!chroma)
 114         goto drop;
 115
 116     video_format_Setup(&dec->fmt_out.video, chroma,
 117                        ntohl(hdr->pixmap_width), ntohl(hdr->pixmap_height),
 118                        ntohl(hdr->pixmap_width), ntohl(hdr->pixmap_height),
 119                        dec->fmt_in.video.i_sar_num,
 120                        dec->fmt_in.video.i_sar_den);
 121
 122     const size_t copy = dec->fmt_out.video.i_width
 123                         * (dec->fmt_out.video.i_bits_per_pixel / 8);
 124     const uint32_t pitch = ntohl(hdr->bytes_per_line);
 125
 126     /* Build picture */
 127     if (pitch < copy
 128      || (block->i_buffer / pitch) < dec->fmt_out.video.i_height)
 129         goto drop;
 130
 131     if (decoder_UpdateVideoFormat(dec))
 132         goto drop;
 133     pic = decoder_NewPicture(dec);
 134     if (pic == NULL)
 135         goto drop;
 136
 137     const uint8_t *in = block->p_buffer;
 138     uint8_t *out = pic->p->p_pixels;
 139     for (unsigned i = 0; i < dec->fmt_out.video.i_height; i++)
 140     {
 141         memcpy(out, in, copy);
 142         in += pitch;
 143         out += pic->p->i_pitch;
 144     }
 145     pic->date = block->i_pts;
 146     pic->b_progressive = true;
 147
 148 drop:
 149     block_Release(block);
 150     decoder_QueueVideo(dec, pic);
 151     return VLCDEC_SUCCESS;
 152 }