git.sesse.net Git - ffmpeg/blob - libavformat/dv.c

   1 /*
   2  * Raw DV format
   3  * Copyright (c) 2002 Fabrice Bellard.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  */
  19 #include "avformat.h"
  20
  21 #define NTSC_FRAME_SIZE 120000
  22 #define PAL_FRAME_SIZE  144000
  23
  24 typedef struct DVDemuxContext {
  25     int       is_audio;
  26     uint8_t   buf[PAL_FRAME_SIZE];
  27     int       size;
  28 } DVDemuxContext;
  29
  30 /* raw input */
  31 static int dv_read_header(AVFormatContext *s,
  32                           AVFormatParameters *ap)
  33 {
  34     AVStream *vst, *ast;
  35     DVDemuxContext *c = s->priv_data;
  36
  37     vst = av_new_stream(s, 0);
  38     if (!vst)
  39         return AVERROR_NOMEM;
  40     vst->codec.codec_type = CODEC_TYPE_VIDEO;
  41     vst->codec.codec_id = CODEC_ID_DVVIDEO;
  42
  43
  44     ast = av_new_stream(s, 1);
  45     if (!ast)
  46         return AVERROR_NOMEM;
  47
  48     ast->codec.codec_type = CODEC_TYPE_AUDIO;
  49     ast->codec.codec_id = CODEC_ID_DVAUDIO;
  50     ast->codec.channels = 2;
  51     c->is_audio = 0;
  52
  53     return 0;
  54 }
  55
  56 /* XXX: build fake audio stream when DV audio decoder will be finished */
  57 static int dv_read_packet(AVFormatContext *s, AVPacket *pkt)
  58 {
  59     int ret, dsf;
  60     DVDemuxContext *c = s->priv_data;
  61
  62     if (!c->is_audio) {
  63         ret = get_buffer(&s->pb, c->buf, 4);
  64         if (ret <= 0)
  65             return -EIO;
  66         dsf = c->buf[3] & 0x80;
  67         if (!dsf)
  68             c->size = NTSC_FRAME_SIZE;
  69         else
  70             c->size = PAL_FRAME_SIZE;
  71
  72         ret = get_buffer(&s->pb, c->buf + 4, c->size - 4);
  73         if (ret <= 0)
  74             return -EIO;
  75     }
  76
  77     if (av_new_packet(pkt, c->size) < 0)
  78         return -EIO;
  79
  80     pkt->stream_index = c->is_audio;
  81     c->is_audio = !c->is_audio;
  82     memcpy(pkt->data, c->buf, c->size);
  83     return ret;
  84 }
  85
  86 static int dv_read_close(AVFormatContext *s)
  87 {
  88     return 0;
  89 }
  90
  91 static AVInputFormat dv_iformat = {
  92     "dv",
  93     "DV video format",
  94     sizeof(DVDemuxContext),
  95     NULL,
  96     dv_read_header,
  97     dv_read_packet,
  98     dv_read_close,
  99     .extensions = "dv",
 100 };
 101
 102 #if 0
 103 int dv_write_header(struct AVFormatContext *s)
 104 {
 105     return 0;
 106 }
 107
 108 int dv_write_packet(struct AVFormatContext *s,
 109                      int stream_index,
 110                      unsigned char *buf, int size, int force_pts)
 111 {
 112     put_buffer(&s->pb, buf, size);
 113     put_flush_packet(&s->pb);
 114     return 0;
 115 }
 116
 117 int dv_write_trailer(struct AVFormatContext *s)
 118 {
 119     return 0;
 120 }
 121
 122 AVOutputFormat dv_oformat = {
 123     "dv",
 124     "DV video format",
 125     NULL,
 126     "dv",
 127     0,
 128     CODEC_ID_DVVIDEO,
 129     CODEC_ID_DVAUDIO,
 130     dv_write_header,
 131     dv_write_packet,
 132     dv_write_trailer,
 133 };
 134 #endif
 135
 136 int dv_init(void)
 137 {
 138     av_register_input_format(&dv_iformat);
 139     //    av_register_output_format(&dv_oformat);
 140     return 0;
 141 }