libvo/vo_mpegpes.c

   1 /*
   2  * Based on:
   3  *
   4  * test_av.c - Test program for new API
   5  *
   6  * Copyright (C) 2000 Ralph  Metzler <ralph@convergence.de>
   7  *                  & Marcus Metzler <marcus@convergence.de>
   8  *                    for convergence integrated media GmbH
   9  *
  10  * libav - MPEG-PS multiplexer, part of ffmpeg
  11  * Copyright Gerard Lantau  (see http://ffmpeg.sf.net)
  12  *
  13  */
  14
  15 #include "config.h"
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18 #include <string.h>
  19 #include <errno.h>
  20 #include <sys/types.h>
  21 #include <sys/stat.h>
  22 #include <fcntl.h>
  23 #include <unistd.h>
  24
  25 #include "mp_msg.h"
  26
  27 #ifdef CONFIG_DVB
  28 #ifndef CONFIG_DVB_HEAD
  29 #include <poll.h>
  30
  31 #include <sys/ioctl.h>
  32 #include <stdio.h>
  33 #include <time.h>
  34
  35 #include <ost/dmx.h>
  36 #include <ost/frontend.h>
  37 #include <ost/sec.h>
  38 #include <ost/video.h>
  39 #include <ost/audio.h>
  40
  41 #else
  42 #define true 1
  43 #define false 0
  44 #include <poll.h>
  45
  46 #include <sys/ioctl.h>
  47 #include <stdio.h>
  48 #include <time.h>
  49
  50 #include <linux/dvb/dmx.h>
  51 #include <linux/dvb/frontend.h>
  52 #include <linux/dvb/video.h>
  53 #include <linux/dvb/audio.h>
  54 #endif
  55 #endif
  56
  57 #include "config.h"
  58 #include "video_out.h"
  59 #include "video_out_internal.h"
  60 #include "libmpdemux/mpeg_packetizer.h"
  61
  62 int vo_mpegpes_fd=-1;
  63 extern int vo_mpegpes_fd2;
  64
  65 static const vo_info_t info =
  66 {
  67 #ifdef CONFIG_DVB
  68         "MPEG-PES to DVB card",
  69 #else
  70         "MPEG-PES file",
  71 #endif
  72         "mpegpes",
  73         "A'rpi",
  74         ""
  75 };
  76
  77 const LIBVO_EXTERN (mpegpes)
  78
  79 static int
  80 config(uint32_t s_width, uint32_t s_height, uint32_t width, uint32_t height, uint32_t flags, char *title, uint32_t format)
  81 {
  82 #ifdef CONFIG_DVB
  83     switch(s_height){
  84     case 288:
  85     case 576:
  86     case 240:
  87     case 480:
  88         break;
  89     default:
  90         mp_msg(MSGT_VO,MSGL_ERR,"DVB: height=%d not supported (try 240/480 (ntsc) or 288/576 (pal)\n",s_height);
  91         return -1;
  92     }
  93 #endif
  94     return 0;
  95 }
  96
  97 static int preinit(const char *arg){
  98 #ifdef CONFIG_DVB
  99     int card = -1;
 100     char vo_file[30], ao_file[30], *tmp;
 101
 102     if(arg != NULL){
 103         if((tmp = strstr(arg, "card=")) != NULL) {
 104             card = atoi(&tmp[5]);
 105             if((card < 1) || (card > 4)) {
 106                 mp_msg(MSGT_VO, MSGL_ERR, "DVB card number must be between 1 and 4\n");
 107                 return -1;
 108             }
 109             card--;
 110             arg = NULL;
 111         }
 112     }
 113
 114     if(!arg){
 115     //|O_NONBLOCK
 116         //search the first usable card
 117         if(card==-1) {
 118           int n;
 119           for(n=0; n<4; n++) {
 120             sprintf(vo_file, "/dev/dvb/adapter%d/video0", n);
 121             if(access(vo_file, F_OK | W_OK)==0) {
 122               card = n;
 123               break;
 124             }
 125           }
 126         }
 127         if(card==-1) {
 128           mp_msg(MSGT_VO,MSGL_INFO, "Couldn't find a usable dvb video device, exiting\n");
 129           return -1;
 130         }
 131 #ifndef CONFIG_DVB_HEAD
 132         mp_msg(MSGT_VO,MSGL_INFO, "Opening /dev/ost/video+audio\n");
 133         sprintf(vo_file, "/dev/ost/video");
 134         sprintf(ao_file, "/dev/ost/audio");
 135 #else
 136         mp_msg(MSGT_VO,MSGL_INFO, "Opening /dev/dvb/adapter%d/video0+audio0\n", card);
 137         sprintf(vo_file, "/dev/dvb/adapter%d/video0", card);
 138         sprintf(ao_file, "/dev/dvb/adapter%d/audio0", card);
 139 #endif
 140         if((vo_mpegpes_fd = open(vo_file,O_RDWR)) < 0){
 141                 perror("DVB VIDEO DEVICE: ");
 142                 return -1;
 143         }
 144         if ( (ioctl(vo_mpegpes_fd,VIDEO_SET_BLANK, false) < 0)){
 145                 perror("DVB VIDEO SET BLANK: ");
 146                 return -1;
 147         }
 148         if ( (ioctl(vo_mpegpes_fd,VIDEO_SELECT_SOURCE, VIDEO_SOURCE_MEMORY) < 0)){
 149                 perror("DVB VIDEO SELECT SOURCE: ");
 150                 return -1;
 151         }
 152         if ( (ioctl(vo_mpegpes_fd,VIDEO_PLAY) < 0)){
 153                 perror("DVB VIDEO PLAY: ");
 154                 return -1;
 155         }
 156         return 0;
 157     }
 158 #endif
 159     arg = (arg ? arg : "grab.mpg");
 160     mp_msg(MSGT_VO,MSGL_INFO, "Saving PES stream to %s\n", arg);
 161     vo_mpegpes_fd=open(arg,O_WRONLY|O_CREAT,0666);
 162     if(vo_mpegpes_fd<0){
 163         perror("vo_mpegpes");
 164         return -1;
 165     }
 166     return 0;
 167 }
 168
 169
 170 static void draw_osd(void)
 171 {
 172 }
 173
 174
 175 static int my_write(unsigned char* data,int len){
 176     int orig_len = len;
 177 #ifdef CONFIG_DVB
 178 #define NFD   2
 179     struct pollfd pfd[NFD];
 180
 181 //    printf("write %d bytes  \n",len);
 182
 183         pfd[0].fd = vo_mpegpes_fd;
 184         pfd[0].events = POLLOUT;
 185
 186         pfd[1].fd = vo_mpegpes_fd2;
 187         pfd[1].events = POLLOUT;
 188
 189     while(len>0){
 190         if (poll(pfd,NFD,1)){
 191             if (pfd[0].revents & POLLOUT){
 192                 int ret=write(vo_mpegpes_fd,data,len);
 193 //              printf("ret=%d  \n",ret);
 194                 if(ret<=0){
 195                     perror("write");
 196                     usleep(0);
 197                 } else {
 198                     len-=ret; data+=ret;
 199                 }
 200             } else usleep(1000);
 201         }
 202     }
 203
 204 #else
 205     write(vo_mpegpes_fd,data,len); // write to file
 206 #endif
 207     return orig_len;
 208 }
 209
 210 void send_pes_packet(unsigned char* data,int len,int id,int timestamp){
 211     send_mpeg_pes_packet (data, len, id, timestamp, 1, my_write);
 212 }
 213
 214 void send_lpcm_packet(unsigned char* data,int len,int id,unsigned int timestamp,int freq_id){
 215     send_mpeg_lpcm_packet(data, len, id, timestamp, freq_id, my_write);
 216 }
 217
 218
 219 static int draw_frame(uint8_t * src[])
 220 {
 221     vo_mpegpes_t *p=(vo_mpegpes_t *)src[0];
 222     send_pes_packet(p->data,p->size,p->id,(p->timestamp>0)?p->timestamp:vo_pts);  // video data
 223     return 0;
 224 }
 225
 226 static void flip_page (void)
 227 {
 228 }
 229
 230 static int draw_slice(uint8_t *srcimg[], int stride[], int w,int h,int x0,int y0)
 231 {
 232     return 0;
 233 }
 234
 235
 236 static int
 237 query_format(uint32_t format)
 238 {
 239     if(format==IMGFMT_MPEGPES) return VFCAP_CSP_SUPPORTED|VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_TIMER;
 240     return 0;
 241 }
 242
 243 static void
 244 uninit(void)
 245 {
 246     if(vo_mpegpes_fd2>=0 && vo_mpegpes_fd2!=vo_mpegpes_fd) close(vo_mpegpes_fd2);
 247     vo_mpegpes_fd2=-1;
 248     if(vo_mpegpes_fd>=0){ close(vo_mpegpes_fd);vo_mpegpes_fd=-1;}
 249 }
 250
 251
 252 static void check_events(void)
 253 {
 254 }
 255
 256 static int control(uint32_t request, void *data, ...)
 257 {
 258   switch (request) {
 259   case VOCTRL_QUERY_FORMAT:
 260     return query_format(*((uint32_t*)data));
 261   }
 262   return VO_NOTIMPL;
 263 }