libmpdemux/demux_lmlm4.c

   1 /*
   2  * LMLM4 MPEG4 Compression Card stream & file parser
   3  * Copyright (C) 2003 Maxim Yevtyushkin <max@linuxmedialabs.com>
   4  * based on SMJPEG file parser by Alex Beregszaszi
   5  *
   6  * This file is part of MPlayer.
   7  *
   8  * MPlayer is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * MPlayer is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License along
  19  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  20  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <unistd.h>
  26 #include <string.h> /* strtok */
  27
  28 #include "config.h"
  29 #include "mp_msg.h"
  30
  31 #include "stream/stream.h"
  32 #include "demuxer.h"
  33 #include "stheader.h"
  34
  35 typedef struct FrameInfo
  36 {
  37     ssize_t frameSize;
  38     ssize_t paddingSize;
  39     int     frameType;
  40     int     channelNo;
  41 } FrameInfo;
  42
  43 #define FRAMETYPE_I             0
  44 #define FRAMETYPE_P             1
  45 #define FRAMETYPE_B             2
  46 #define FRAMETYPE_AUDIO_MPEG1L2 4
  47 #define FRAMETYPE_AUDIO_ULAW    5
  48 #define FRAMETYPE_AUDIO_ADPCM   6
  49
  50 #define PACKET_BLOCK_SIZE   0x00000200
  51 #define PACKET_BLOCK_LAST   0x000001FF
  52 #define PACKET_BLOCK_MASK   0xFFFFFE00
  53
  54 #define MAX_PACKET_SIZE   1048576 // 1 Mb
  55
  56 #define STREAM_START_CODE_SIZE   4
  57
  58 /*
  59 // codes in MSB first
  60 static unsigned int start_code [] =
  61 {
  62     0xB0010000,         // VISUAL_OBJECT_SEQUENCE_START_CODE
  63     0xB6010000,         // VOP_START_CODE
  64     0x04C4FDFF,          // MPEG1LAYERII_START_CODE
  65     0x00000000          // end of start codes list
  66 };
  67 */
  68
  69 static int imeHeaderValid(FrameInfo *frame)
  70 {
  71     if ( frame->channelNo > 7 ||
  72          frame->frameSize > MAX_PACKET_SIZE || frame->frameSize <= 0)
  73     {
  74         mp_msg(MSGT_DEMUX, MSGL_V, "Invalid packet in LMLM4 stream: ch=%d size=%d\n", frame->channelNo, frame->frameSize);
  75         return 0;
  76     }
  77     switch (frame->frameType) {
  78     case FRAMETYPE_I:
  79     case FRAMETYPE_P:
  80     case FRAMETYPE_B:
  81     case FRAMETYPE_AUDIO_MPEG1L2:
  82     case FRAMETYPE_AUDIO_ULAW:
  83     case FRAMETYPE_AUDIO_ADPCM:
  84         break;
  85     default:
  86         mp_msg(MSGT_DEMUX, MSGL_V, "Invalid packet in LMLM4 stream (wrong packet type %d)\n", frame->frameType);
  87         return 0;
  88     }
  89     return 1;
  90 }
  91
  92 /*
  93 int searchMPEG4Stream(demuxer_t* demuxer, IME6400Header *imeHeader)
  94 {
  95     void *data;
  96     ssize_t imeHeaderSize = sizeof(IME6400Header);
  97     ssize_t dataSize = sizeof(IME6400Header) * 3;
  98     ssize_t ptr = imeHeaderSize * 2;
  99     int errNo, startCodeNo;
 100     off_t pos;
 101
 102     data = malloc(dataSize);
 103
 104     imeHeaderSwap(imeHeader);
 105     memcpy(data + imeHeaderSize, imeHeader, imeHeaderSize);
 106
 107 //    printHex(data + imeHeaderSize, imeHeaderSize);
 108
 109     while ((errNo = stream_read(demuxer->stream, data + imeHeaderSize * 2 , imeHeaderSize)) == imeHeaderSize)
 110     {
 111 //        printHex(data + imeHeaderSize * 2, imeHeaderSize);
 112
 113         pos = stream_tell(demuxer->stream);
 114         while (dataSize - ptr >= STREAM_START_CODE_SIZE) {
 115             startCodeNo = 0;
 116             while (start_code[startCodeNo])
 117             {
 118                 if (memcmp(&start_code[startCodeNo], data + ptr, STREAM_START_CODE_SIZE) == 0) // start code match
 119                 {
 120                     memcpy(imeHeader, data + ptr - imeHeaderSize, imeHeaderSize);
 121                     imeHeaderSwap(imeHeader);
 122                     if (imeHeaderValid(imeHeader))
 123                     {
 124                         stream_seek(demuxer->stream, pos - (dataSize - ptr));
 125                         free(data);
 126                         return 0;
 127                     }
 128                 }
 129                 startCodeNo++;
 130             }
 131             ptr++;
 132         }
 133         memcpy(data,data + imeHeaderSize, imeHeaderSize * 2);
 134         ptr -= imeHeaderSize;
 135     }
 136
 137     free(data);
 138     return errNo;
 139 }
 140 */
 141
 142 static int getFrame(demuxer_t *demuxer, FrameInfo *frameInfo)
 143 {
 144     unsigned int packetSize;
 145
 146     frameInfo->channelNo = stream_read_word(demuxer->stream);
 147     frameInfo->frameType = stream_read_word(demuxer->stream);
 148     packetSize=stream_read_dword(demuxer->stream);
 149
 150     if(stream_eof(demuxer->stream)){
 151         frameInfo->frameSize = 0;
 152         return 0;
 153     }
 154
 155     frameInfo->frameSize = packetSize - 8; //sizeof(IME6400Header);
 156     frameInfo->paddingSize = (packetSize & PACKET_BLOCK_LAST) ? PACKET_BLOCK_SIZE - (packetSize & PACKET_BLOCK_LAST) : 0;
 157
 158     mp_msg(MSGT_DEMUX, MSGL_DBG2, "typ: %d chan: %d size: %d pad: %d\n",
 159             frameInfo->frameType,
 160             frameInfo->channelNo,
 161             frameInfo->frameSize,
 162             frameInfo->paddingSize);
 163
 164     if(!imeHeaderValid(frameInfo)){
 165         // skip this packet
 166         stream_skip(demuxer->stream,PACKET_BLOCK_SIZE-8);
 167         frameInfo->frameSize = 0;
 168         return -1;
 169     }
 170
 171     return 1;
 172 }
 173
 174 static int lmlm4_check_file(demuxer_t* demuxer)
 175 {
 176     FrameInfo frameInfo;
 177     unsigned int first;
 178
 179     mp_msg(MSGT_DEMUX, MSGL_V, "Checking for LMLM4 Stream Format\n");
 180
 181     if(getFrame(demuxer, &frameInfo)!=1){
 182         stream_skip(demuxer->stream,-8);
 183         mp_msg(MSGT_DEMUX, MSGL_V, "LMLM4 Stream Format not found\n");
 184         return 0;
 185     }
 186     first=stream_read_dword(demuxer->stream);
 187     stream_skip(demuxer->stream,-12);
 188
 189     mp_msg(MSGT_DEMUXER,MSGL_V,"LMLM4: first=0x%08X\n",first);
 190
 191     switch(frameInfo.frameType){
 192     case FRAMETYPE_AUDIO_MPEG1L2:
 193         if( (first & 0xffe00000) != 0xffe00000 ){
 194             mp_msg(MSGT_DEMUXER,MSGL_V,"LMLM4: not mpeg audio\n");
 195             return 0;
 196         }
 197         if((4-((first>>17)&3))!=2){
 198             mp_msg(MSGT_DEMUXER,MSGL_V,"LMLM4: not layer-2\n");
 199             return 0;
 200         }
 201         if(((first>>10)&0x3)==3){
 202             mp_msg(MSGT_DEMUXER,MSGL_V,"LMLM4: invalid audio sampelrate\n");
 203             return 0;
 204         }
 205         mp_msg(MSGT_DEMUXER,MSGL_V,"LMLM4: first packet is audio, header checks OK!\n");
 206         break;
 207     // TODO: add checks for video header too, for case of disabled audio
 208     }
 209
 210
 211 //    stream_reset(demuxer->stream);
 212     mp_msg(MSGT_DEMUX, MSGL_V, "LMLM4 Stream Format found\n");
 213
 214     return DEMUXER_TYPE_LMLM4;
 215 }
 216
 217 static int video = 0;
 218 static int frames= 0;
 219
 220 // return value:
 221 //     0 = EOF or no stream found
 222 //     1 = successfully read a packet
 223 static int demux_lmlm4_fill_buffer(demuxer_t *demux, demux_stream_t *ds)
 224 {
 225     FrameInfo frameInfo;
 226     double pts;
 227     int id=1;
 228     int ret;
 229
 230 //hdr:
 231     demux->filepos = stream_tell(demux->stream);
 232     mp_msg(MSGT_DEMUX, MSGL_DBG2, "fpos = %"PRId64"\n", (int64_t)demux->filepos);
 233
 234     ret=getFrame(demux, &frameInfo);
 235     if(ret<=0) return ret; // EOF/error
 236
 237     pts=demux->video->sh ? frames*((sh_video_t*)(demux->video->sh))->frametime : 0;
 238
 239     switch(frameInfo.frameType){
 240     case FRAMETYPE_AUDIO_MPEG1L2:
 241         mp_dbg(MSGT_DEMUX, MSGL_DBG2, "Audio Packet\n");
 242         if (!video)
 243         {
 244             stream_skip(demux->stream, frameInfo.frameSize + frameInfo.paddingSize);
 245             mp_msg(MSGT_DEMUX, MSGL_V, "Skip Audio Packet\n");
 246             return -1; //goto hdr;
 247         }
 248         if(demux->audio->id==-1){
 249             if(!demux->a_streams[id]) new_sh_audio(demux,id);
 250             demux->audio->id=id;
 251             demux->audio->sh=demux->a_streams[id];
 252             ((sh_audio_t*)(demux->audio->sh))->format=0x50; // mpeg audio layer 1/2
 253         }
 254         if(demux->audio->id==id)
 255             ds_read_packet(demux->audio, demux->stream, frameInfo.frameSize,
 256                 pts, demux->filepos, 0);
 257         else
 258             stream_skip(demux->stream,frameInfo.frameSize);
 259         break;
 260     case FRAMETYPE_I:
 261         if (!video) {
 262             video = 1;
 263             mp_dbg(MSGT_DEMUX, MSGL_DBG2, "First Video Packet\n");
 264         }
 265     case FRAMETYPE_P:
 266         frames=(frames+1)&(1024*1024-1); // wrap around at 4 hrs to avoid inaccurate float calculations
 267         if (!video)
 268         {
 269             stream_skip(demux->stream, frameInfo.frameSize + frameInfo.paddingSize);
 270             mp_msg(MSGT_DEMUX, MSGL_V, "Skip Video P Packet\n");
 271             return -1; //goto hdr;
 272         }
 273         mp_dbg(MSGT_DEMUX, MSGL_DBG2, "Video Packet\n");
 274         if(demux->video->id==-1){
 275             if(!demux->v_streams[id]) new_sh_video(demux,id);
 276             demux->video->id=id;
 277             demux->video->sh=demux->v_streams[id];
 278             ((sh_video_t*)(demux->video->sh))->format=0x10000004; // mpeg4-ES
 279         }
 280         if(demux->video->id==id)
 281             ds_read_packet(demux->video, demux->stream, frameInfo.frameSize,
 282                 pts, demux->filepos, 0);
 283         break;
 284     default:
 285         stream_skip(demux->stream,frameInfo.frameSize);
 286     }
 287
 288     stream_skip(demux->stream, frameInfo.paddingSize);
 289
 290     return 1;
 291 }
 292
 293 static demuxer_t* demux_open_lmlm4(demuxer_t* demuxer){
 294     sh_audio_t *sh_audio=NULL;
 295     sh_video_t *sh_video=NULL;
 296
 297 #if 0
 298     sh_video_t* sh_video;
 299     sh_audio_t* sh_audio;
 300     unsigned int htype = 0, hleng;
 301     int i = 0;
 302
 303     sh_video = new_sh_video(demuxer, 0);
 304     demuxer->video->sh = sh_video;
 305     sh_video->ds = demuxer->video;
 306     sh_video->disp_w = 640;
 307     sh_video->disp_h = 480;
 308     sh_video->format = mmioFOURCC('D','I','V','X');
 309
 310     sh_video->bih = malloc(sizeof(BITMAPINFOHEADER));
 311     memset(sh_video->bih, 0, sizeof(BITMAPINFOHEADER));
 312
 313     /* these are false values */
 314     sh_video->bih->biSize = 40;
 315     sh_video->bih->biWidth = sh_video->disp_w;
 316     sh_video->bih->biHeight = sh_video->disp_h;
 317     sh_video->bih->biPlanes = 3;
 318     sh_video->bih->biBitCount = 16;
 319     sh_video->bih->biCompression = sh_video->format;
 320     sh_video->bih->biSizeImage = sh_video->disp_w*sh_video->disp_h;
 321
 322     sh_audio = new_sh_audio(demuxer, 0);
 323     demuxer->audio->sh = sh_audio;
 324     sh_audio->ds = demuxer->audio;
 325
 326     sh_audio->wf = malloc(sizeof(WAVEFORMATEX));
 327     memset(sh_audio->wf, 0, sizeof(WAVEFORMATEX));
 328
 329     sh_audio->samplerate = 48000;
 330     sh_audio->wf->wBitsPerSample = 16;
 331     sh_audio->channels = 2;
 332     sh_audio->format = 0x50;
 333     sh_audio->wf->wFormatTag = sh_audio->format;
 334     sh_audio->wf->nChannels = sh_audio->channels;
 335     sh_audio->wf->nSamplesPerSec = sh_audio->samplerate;
 336     sh_audio->wf->nAvgBytesPerSec = sh_audio->wf->nChannels*
 337     sh_audio->wf->wBitsPerSample*sh_audio->wf->nSamplesPerSec/8;
 338     sh_audio->wf->nBlockAlign = sh_audio->channels *2;
 339     sh_audio->wf->cbSize = 0;
 340
 341 #endif
 342
 343     demuxer->seekable = 0;
 344
 345     if(!ds_fill_buffer(demuxer->video)){
 346         mp_msg(MSGT_DEMUXER, MSGL_INFO, "LMLM4: %s",
 347                mp_gtext("No video stream found.\n"));
 348         demuxer->video->sh=NULL;
 349     } else {
 350         sh_video=demuxer->video->sh;sh_video->ds=demuxer->video;
 351     }
 352     if(demuxer->audio->id!=-2) {
 353         if(!ds_fill_buffer(demuxer->audio)){
 354             mp_msg(MSGT_DEMUXER, MSGL_INFO, "LMLM4: %s",
 355                    mp_gtext("No audio stream found -> no sound.\n"));
 356             demuxer->audio->sh=NULL;
 357         } else {
 358             sh_audio=demuxer->audio->sh;sh_audio->ds=demuxer->audio;
 359         }
 360     }
 361
 362     return demuxer;
 363 }
 364
 365 static void demux_close_lmlm4(demuxer_t *demuxer)
 366 {
 367 //    printf("Close LMLM4 Stream\n");
 368     return;
 369 }
 370
 371
 372 const demuxer_desc_t demuxer_desc_lmlm4 = {
 373   "LMLM4 MPEG4 Compression Card stream demuxer",
 374   "lmlm4",
 375   "RAW LMLM4",
 376   "Maxim Yevtyushkin",
 377   "",
 378   DEMUXER_TYPE_LMLM4,
 379   0, // unsafe autodetect
 380   lmlm4_check_file,
 381   demux_lmlm4_fill_buffer,
 382   demux_open_lmlm4,
 383   demux_close_lmlm4,
 384   NULL,
 385   NULL
 386 };