libmpdemux/demux_rawvideo.c

   1
   2 #include "config.h"
   3
   4 #include <stdlib.h>
   5 #include <stdio.h>
   6 #include <unistd.h>
   7 #include <string.h>
   8
   9 #include "m_option.h"
  10
  11 #include "stream/stream.h"
  12 #include "demuxer.h"
  13 #include "stheader.h"
  14
  15 #include "libmpcodecs/img_format.h"
  16
  17 extern int demuxer_type;
  18 static int format = IMGFMT_I420;
  19 static int size_id = 0;
  20 static int width = 0;
  21 static int height = 0;
  22 static float fps = 25;
  23 static int imgsize=0;
  24
  25 const m_option_t demux_rawvideo_opts[] = {
  26   // size:
  27   { "w", &width, CONF_TYPE_INT,CONF_RANGE,1,8192, NULL },
  28   { "h", &height, CONF_TYPE_INT,CONF_RANGE,1,8192, NULL },
  29   { "sqcif", &size_id, CONF_TYPE_FLAG,0,0,1, NULL },
  30   { "qcif", &size_id, CONF_TYPE_FLAG,0,0,2, NULL },
  31   { "cif", &size_id, CONF_TYPE_FLAG,0,0,3, NULL },
  32   { "4cif", &size_id, CONF_TYPE_FLAG,0,0,4, NULL },
  33   { "pal", &size_id, CONF_TYPE_FLAG,0,0,5, NULL },
  34   { "ntsc", &size_id, CONF_TYPE_FLAG,0,0,6, NULL },
  35   { "16cif", &size_id, CONF_TYPE_FLAG,0,0,7, NULL },
  36   { "sif", &size_id, CONF_TYPE_FLAG,0,0,8, NULL },
  37   // format:
  38   { "format", &format, CONF_TYPE_IMGFMT, 0, 0 , 0, NULL },
  39   // below options are obsolete
  40   { "i420", &format, CONF_TYPE_FLAG, 0, 0 , IMGFMT_I420, NULL },
  41   { "yv12", &format, CONF_TYPE_FLAG, 0, 0 , IMGFMT_YV12, NULL },
  42   { "nv12", &format, CONF_TYPE_FLAG, 0, 0 , IMGFMT_NV12, NULL },
  43   { "hm12", &format, CONF_TYPE_FLAG, 0, 0 , IMGFMT_HM12, NULL },
  44   { "yuy2", &format, CONF_TYPE_FLAG, 0, 0 , IMGFMT_YUY2, NULL },
  45   { "uyvy", &format, CONF_TYPE_FLAG, 0, 0 , IMGFMT_UYVY, NULL },
  46   { "y8", &format, CONF_TYPE_FLAG, 0, 0 , IMGFMT_Y8, NULL },
  47   // misc:
  48   { "fps", &fps, CONF_TYPE_FLOAT,CONF_RANGE,0.001,1000, NULL },
  49   { "size", &imgsize, CONF_TYPE_INT, CONF_RANGE, 1 , 8192*8192*4, NULL },
  50
  51   {NULL, NULL, 0, 0, 0, 0, NULL}
  52 };
  53
  54
  55 static demuxer_t* demux_rawvideo_open(demuxer_t* demuxer) {
  56   sh_video_t* sh_video;
  57
  58   switch(size_id){
  59   case 1: width=128; height=96; break;
  60   case 2: width=176; height=144; break;
  61   case 3: width=352; height=288; break;
  62   case 4: width=704; height=576; break;
  63   case 5: width=720; height=576; break;
  64   case 6: width=720; height=480; break;
  65   case 7: width=1408;height=1152;break;
  66   case 8: width=352; height=240; break;
  67   }
  68   if(!width || !height){
  69       mp_msg(MSGT_DEMUX,MSGL_ERR,"rawvideo: width or height not specified!\n");
  70       return 0;
  71   }
  72
  73   if(!imgsize)
  74   switch(format){
  75   case IMGFMT_I420:
  76   case IMGFMT_IYUV:
  77   case IMGFMT_NV12:
  78   case IMGFMT_HM12:
  79   case IMGFMT_YV12: imgsize=width*height+2*(width>>1)*(height>>1);break;
  80   case IMGFMT_YUY2:
  81   case IMGFMT_UYVY: imgsize=width*height*2;break;
  82   case IMGFMT_Y800:
  83   case IMGFMT_Y8: imgsize=width*height;break;
  84   default:
  85       if (IMGFMT_IS_RGB(format))
  86         imgsize = width * height * ((IMGFMT_RGB_DEPTH(format) + 7) >> 3);
  87       else if (IMGFMT_IS_BGR(format))
  88         imgsize = width * height * ((IMGFMT_BGR_DEPTH(format) + 7) >> 3);
  89       else {
  90       mp_msg(MSGT_DEMUX,MSGL_ERR,"rawvideo: img size not specified and unknown format!\n");
  91       return 0;
  92       }
  93   }
  94
  95   sh_video = new_sh_video(demuxer,0);
  96   sh_video->format=format;
  97   sh_video->fps=fps;
  98   sh_video->frametime=1.0/fps;
  99   sh_video->disp_w=width;
 100   sh_video->disp_h=height;
 101   sh_video->i_bps=fps*imgsize;
 102
 103   demuxer->movi_start = demuxer->stream->start_pos;
 104   demuxer->movi_end = demuxer->stream->end_pos;
 105
 106   demuxer->video->sh = sh_video;
 107   sh_video->ds = demuxer->video;
 108
 109   return demuxer;
 110 }
 111
 112 static int demux_rawvideo_fill_buffer(demuxer_t* demuxer, demux_stream_t *ds) {
 113   sh_video_t* sh = demuxer->video->sh;
 114   off_t pos;
 115   if(demuxer->stream->eof) return 0;
 116   if(ds!=demuxer->video) return 0;
 117   pos = stream_tell(demuxer->stream);
 118   ds_read_packet(ds,demuxer->stream,imgsize,(pos/imgsize)*sh->frametime,pos,0x10);
 119   return 1;
 120 }
 121
 122 static void demux_rawvideo_seek(demuxer_t *demuxer,float rel_seek_secs,float audio_delay,int flags){
 123   stream_t* s = demuxer->stream;
 124   sh_video_t* sh_video = demuxer->video->sh;
 125   off_t pos;
 126
 127   pos = (flags & 1) ? demuxer->movi_start : stream_tell(s);
 128   if(flags & 2)
 129     pos += ((demuxer->movi_end - demuxer->movi_start)*rel_seek_secs);
 130   else
 131     pos += (rel_seek_secs*sh_video->i_bps);
 132   if(pos < 0) pos = 0;
 133   if(demuxer->movi_end && pos > demuxer->movi_end) pos = (demuxer->movi_end-imgsize);
 134   pos/=imgsize;
 135   stream_seek(s,pos*imgsize);
 136   //sh_video->timer=pos * sh_video->frametime;
 137   demuxer->video->pts = pos * sh_video->frametime;
 138 //  printf("demux_rawvideo: streamtell=%d\n",(int)stream_tell(demuxer->stream));
 139 }
 140
 141
 142 const demuxer_desc_t demuxer_desc_rawvideo = {
 143   "Raw video demuxer",
 144   "rawvideo",
 145   "rawvideo",
 146   "?",
 147   "",
 148   DEMUXER_TYPE_RAWVIDEO,
 149   0, // no autodetect
 150   NULL,
 151   demux_rawvideo_fill_buffer,
 152   demux_rawvideo_open,
 153   NULL,
 154   demux_rawvideo_seek,
 155   NULL
 156 };