From 5846646296e377e093441dfe9eadde38ff1f7c99 Mon Sep 17 00:00:00 2001 From: Dirk Farin Date: Thu, 12 Sep 2013 09:14:45 +0200 Subject: Add raw HEVC demuxer Signed-off-by: Anton Khirnov --- libavformat/hevcdec.c | 64 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100644 libavformat/hevcdec.c (limited to 'libavformat/hevcdec.c') diff --git a/libavformat/hevcdec.c b/libavformat/hevcdec.c new file mode 100644 index 0000000000..65a3cdfb3c --- /dev/null +++ b/libavformat/hevcdec.c @@ -0,0 +1,64 @@ +/* + * RAW HEVC video demuxer + * Copyright (c) 2013 Dirk Farin + * + * This file is part of Libav. + * + * Libav is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * Libav is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with Libav; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "libavcodec/hevc.h" + +#include "avformat.h" +#include "rawdec.h" + +static int hevc_probe(AVProbeData *p) +{ + uint32_t code = -1; + int vps = 0, sps = 0, pps = 0, irap = 0; + int i; + + for (i = 0; i < p->buf_size - 1; i++) { + code = (code << 8) + p->buf[i]; + if ((code & 0xffffff00) == 0x100) { + uint8_t nal2 = p->buf[i + 1]; + int type = (code & 0x7E) >> 1; + + if (code & 0x81) // forbidden and reserved zero bits + return 0; + + if (nal2 & 0xf8) // reserved zero + return 0; + + switch (type) { + case NAL_VPS: vps++; break; + case NAL_SPS: sps++; break; + case NAL_PPS: pps++; break; + case NAL_BLA_N_LP: + case NAL_BLA_W_LP: + case NAL_BLA_W_RADL: + case NAL_CRA_NUT: + case NAL_IDR_N_LP: + case NAL_IDR_W_RADL: irap++; break; + } + } + } + + if (vps && sps && pps && irap) + return AVPROBE_SCORE_EXTENSION + 1; // 1 more than .mpg + return 0; +} + +FF_DEF_RAWVIDEO_DEMUXER(hevc, "raw HEVC video", hevc_probe, "hevc,h265,265", AV_CODEC_ID_HEVC) -- cgit v1.2.3