libmpcodecs/ve_raw.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4
   5 #include "config.h"
   6 #include "mp_msg.h"
   7 #include "help_mp.h"
   8
   9 #include "codec-cfg.h"
  10 #include "stream/stream.h"
  11 #include "libmpdemux/demuxer.h"
  12 #include "libmpdemux/stheader.h"
  13
  14 #include "stream/stream.h"
  15 #include "libmpdemux/muxer.h"
  16
  17 #include "img_format.h"
  18 #include "mp_image.h"
  19 #include "vf.h"
  20
  21
  22 //===========================================================================//
  23
  24 struct vf_priv_s {
  25     muxer_stream_t* mux;
  26 };
  27 #define mux_v (vf->priv->mux)
  28
  29 static int set_format(struct vf_instance_s *vf, unsigned int fmt) {
  30     mux_v->bih->biCompression = fmt;
  31
  32     mux_v->bih->biPlanes = 1;
  33     if (IMGFMT_IS_RGB(fmt)) {
  34         if (IMGFMT_RGB_DEPTH(fmt) < 8 && !(fmt&128))
  35             mux_v->bih->biBitCount = IMGFMT_RGB_DEPTH(fmt);
  36         else
  37             mux_v->bih->biBitCount = (IMGFMT_RGB_DEPTH(fmt)+7)&(~7);
  38         return 1;
  39     }
  40     if (IMGFMT_IS_BGR(fmt)) {
  41         if (IMGFMT_BGR_DEPTH(fmt) < 8 && !(fmt&128))
  42             mux_v->bih->biBitCount = IMGFMT_BGR_DEPTH(fmt);
  43         else
  44             mux_v->bih->biBitCount = (IMGFMT_BGR_DEPTH(fmt)+7)&(~7);
  45         return 1;
  46     }
  47     switch (fmt) {
  48     case IMGFMT_I420:
  49     case IMGFMT_IYUV:
  50     case IMGFMT_YV12:
  51     case IMGFMT_411P:
  52         mux_v->bih->biPlanes = 3;
  53         mux_v->bih->biBitCount = 12;
  54         break;
  55     case IMGFMT_444P:
  56         mux_v->bih->biPlanes = 3;
  57         mux_v->bih->biBitCount = 24;
  58         break;
  59     case IMGFMT_422P:
  60         mux_v->bih->biPlanes = 3;
  61         mux_v->bih->biBitCount = 16;
  62         break;
  63     case IMGFMT_IF09:
  64         mux_v->bih->biPlanes = 4;
  65     case IMGFMT_YVU9:
  66         mux_v->bih->biBitCount = 9;
  67         break;
  68     case IMGFMT_UYVY:
  69     case IMGFMT_YUY2:
  70         mux_v->bih->biBitCount = 16;
  71         break;
  72     case IMGFMT_Y8:
  73         mux_v->bih->biBitCount = 8;
  74         break;
  75     default:
  76         mp_msg(MSGT_MENCODER, MSGL_INFO, MSGTR_MPCODECS_OutputWithFourccNotSupported, fmt);
  77         mux_v->bih->biCompression = 0;
  78         return 0;
  79     }
  80     return 1;
  81 }
  82
  83
  84 static int config(struct vf_instance_s *vf,
  85         int width, int height, int d_width, int d_height,
  86         unsigned int flags, unsigned int outfmt)
  87 {
  88     int ret;
  89     mux_v->bih->biWidth = width;
  90     mux_v->bih->biHeight = height;
  91     mux_v->aspect = (float)d_width/d_height;
  92     ret = set_format(vf, outfmt);
  93     if (!ret) return 0;
  94
  95     mux_v->bih->biSizeImage = mux_v->bih->biWidth*mux_v->bih->biHeight*mux_v->bih->biBitCount/8;
  96     return 1;
  97 }
  98
  99 static int control(struct vf_instance_s *vf, int request, void *data) {
 100     return CONTROL_UNKNOWN;
 101 }
 102
 103 static int query_format(struct vf_instance_s *vf, unsigned int fmt) {
 104     if (IMGFMT_IS_RGB(fmt) || IMGFMT_IS_BGR(fmt))
 105         return VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW;
 106     switch (fmt) {
 107     case IMGFMT_I420:
 108     case IMGFMT_IYUV:
 109     case IMGFMT_YV12:
 110     case IMGFMT_411P:
 111     case IMGFMT_444P:
 112     case IMGFMT_422P:
 113     case IMGFMT_UYVY:
 114     case IMGFMT_YUY2:
 115     case IMGFMT_YVU9:
 116     case IMGFMT_IF09:
 117     case IMGFMT_Y8:
 118         return VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW;
 119     }
 120
 121     return 0;
 122 }
 123
 124 static int put_image(struct vf_instance_s *vf, mp_image_t *mpi, double pts) {
 125     mux_v->buffer = mpi->planes[0];
 126     muxer_write_chunk(mux_v, mpi->width*mpi->height*mux_v->bih->biBitCount/8, 0x10, pts, pts);
 127     return 1;
 128 }
 129
 130 //===========================================================================//
 131
 132 static int vf_open(vf_instance_t *vf, char* args){
 133     vf->config = config;
 134     vf->default_caps = VFCAP_CONSTANT;
 135     vf->control = control;
 136     vf->query_format = query_format;
 137     vf->put_image = put_image;
 138     vf->default_caps = 0;
 139     vf->priv = malloc(sizeof(struct vf_priv_s));
 140     memset(vf->priv, 0, sizeof(struct vf_priv_s));
 141     vf->priv->mux = (muxer_stream_t*)args;
 142
 143     mux_v->bih = calloc(1, sizeof(BITMAPINFOHEADER));
 144     mux_v->bih->biSize = sizeof(BITMAPINFOHEADER);
 145     mux_v->bih->biWidth = 0;
 146     mux_v->bih->biHeight = 0;
 147
 148     return 1;
 149 }
 150
 151 vf_info_t ve_info_raw = {
 152     "raw encoder",
 153     "raw",
 154     "jwe21@cam.ac.uk",
 155     "Based on rawrgb",
 156     vf_open
 157 };
 158
 159 //===========================================================================//