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

   1 /*
   2  * IPU video demuxer
   3  * Copyright (c) 2020 Paul B Mahol
   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 #include "avformat.h"
  23 #include "internal.h"
  24 #include "avio_internal.h"
  25 #include "rawdec.h"
  26
  27 #include "libavutil/intreadwrite.h"
  28 #include "libavcodec/internal.h"
  29
  30 static int ipu_read_probe(const AVProbeData *p)
  31 {
  32     if (AV_RB32(p->buf) != MKBETAG('i', 'p', 'u', 'm'))
  33         return 0;
  34
  35     if (AV_RL32(p->buf + 4) == 0)
  36         return 0;
  37
  38     if (AV_RL16(p->buf + 8) == 0)
  39         return 0;
  40
  41     if (AV_RL16(p->buf + 10) == 0)
  42         return 0;
  43
  44     if (AV_RL32(p->buf + 12) == 0)
  45         return 0;
  46
  47     return AVPROBE_SCORE_MAX;
  48 }
  49
  50 static int ipu_read_header(AVFormatContext *s)
  51 {
  52     AVIOContext *pb = s->pb;
  53     AVStream *st = avformat_new_stream(s, NULL);
  54
  55     if (!st)
  56         return AVERROR(ENOMEM);
  57     avio_skip(pb, 8);
  58     st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
  59     st->codecpar->codec_id = AV_CODEC_ID_IPU;
  60     st->codecpar->width    = avio_rl16(pb);
  61     st->codecpar->height   = avio_rl16(pb);
  62     st->start_time         = 0;
  63     st->duration           =
  64     st->nb_frames          = avio_rl32(pb);
  65     st->need_parsing       = AVSTREAM_PARSE_FULL_RAW;
  66     avpriv_set_pts_info(st, 64, 1, 25);
  67
  68     return 0;
  69 }
  70
  71 static const AVClass ipu_demuxer_class = {
  72     .class_name = "ipu demuxer",
  73     .item_name  = av_default_item_name,
  74     .option     = ff_raw_options,
  75     .version    = LIBAVUTIL_VERSION_INT,
  76 };
  77
  78 const AVInputFormat ff_ipu_demuxer = {
  79     .name           = "ipu",
  80     .long_name      = NULL_IF_CONFIG_SMALL("raw IPU Video"),
  81     .read_probe     = ipu_read_probe,
  82     .read_header    = ipu_read_header,
  83     .read_packet    = ff_raw_read_partial_packet,
  84     .extensions     = "ipu",
  85     .flags          = AVFMT_GENERIC_INDEX,
  86     .raw_codec_id   = AV_CODEC_ID_IPU,
  87     .priv_data_size = sizeof(FFRawDemuxerContext),
  88     .priv_class     = &ipu_demuxer_class,
  89 };