git.sesse.net Git - ffmpeg/blob - libavcodec/imx_dump_header_bsf.c

   1 /*
   2  * imx dump header bitstream filter
   3  * Copyright (c) 2007 Baptiste Coudurier
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 /**
  23  * @file
  24  * imx dump header bitstream filter
  25  * modifies bitstream to fit in mov and be decoded by final cut pro decoder
  26  */
  27
  28 #include "avcodec.h"
  29 #include "bsf.h"
  30 #include "bytestream.h"
  31
  32
  33 static int imx_dump_header(AVBSFContext *ctx, AVPacket *out)
  34 {
  35     /* MXF essence element key */
  36     static const uint8_t imx_header[16] = { 0x06,0x0e,0x2b,0x34,0x01,0x02,0x01,0x01,0x0d,0x01,0x03,0x01,0x05,0x01,0x01,0x00 };
  37
  38     AVPacket *in;
  39     int ret = 0;
  40     uint8_t *out_buf;
  41
  42     ret = ff_bsf_get_packet(ctx, &in);
  43     if (ret < 0)
  44         return ret;
  45
  46     ret = av_new_packet(out, in->size + 20);
  47     if (ret < 0)
  48         goto fail;
  49
  50     out_buf = out->data;
  51
  52     bytestream_put_buffer(&out_buf, imx_header, 16);
  53     bytestream_put_byte(&out_buf, 0x83); /* KLV BER long form */
  54     bytestream_put_be24(&out_buf, in->size);
  55     bytestream_put_buffer(&out_buf, in->data, in->size);
  56
  57     ret = av_packet_copy_props(out, in);
  58     if (ret < 0)
  59         goto fail;
  60
  61 fail:
  62     if (ret < 0)
  63         av_packet_unref(out);
  64     av_packet_free(&in);
  65     return ret;
  66 }
  67
  68 static const enum AVCodecID codec_ids[] = {
  69     AV_CODEC_ID_MPEG2VIDEO, AV_CODEC_ID_NONE,
  70 };
  71
  72 const AVBitStreamFilter ff_imx_dump_header_bsf = {
  73     .name      = "imxdump",
  74     .filter    = imx_dump_header,
  75     .codec_ids = codec_ids,
  76 };