LCOV - code coverage report
Current view: top level - libavcodec - tak.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 61 66 92.4 %
Date: 2017-12-16 01:21:47 Functions: 5 5 100.0 %

          Line data    Source code
       1             : /*
       2             :  * TAK common code
       3             :  * Copyright (c) 2012 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 "libavutil/crc.h"
      23             : #include "libavutil/intreadwrite.h"
      24             : 
      25             : #define BITSTREAM_READER_LE
      26             : #include "tak.h"
      27             : 
      28             : static const int64_t tak_channel_layouts[] = {
      29             :     0,
      30             :     AV_CH_FRONT_LEFT,
      31             :     AV_CH_FRONT_RIGHT,
      32             :     AV_CH_FRONT_CENTER,
      33             :     AV_CH_LOW_FREQUENCY,
      34             :     AV_CH_BACK_LEFT,
      35             :     AV_CH_BACK_RIGHT,
      36             :     AV_CH_FRONT_LEFT_OF_CENTER,
      37             :     AV_CH_FRONT_RIGHT_OF_CENTER,
      38             :     AV_CH_BACK_CENTER,
      39             :     AV_CH_SIDE_LEFT,
      40             :     AV_CH_SIDE_RIGHT,
      41             :     AV_CH_TOP_CENTER,
      42             :     AV_CH_TOP_FRONT_LEFT,
      43             :     AV_CH_TOP_FRONT_CENTER,
      44             :     AV_CH_TOP_FRONT_RIGHT,
      45             :     AV_CH_TOP_BACK_LEFT,
      46             :     AV_CH_TOP_BACK_CENTER,
      47             :     AV_CH_TOP_BACK_RIGHT,
      48             : };
      49             : 
      50             : static const uint16_t frame_duration_type_quants[] = {
      51             :     3, 4, 6, 8, 4096, 8192, 16384, 512, 1024, 2048,
      52             : };
      53             : 
      54          22 : static int tak_get_nb_samples(int sample_rate, enum TAKFrameSizeType type)
      55             : {
      56             :     int nb_samples, max_nb_samples;
      57             : 
      58          22 :     if (type <= TAK_FST_250ms) {
      59          20 :         nb_samples     = sample_rate * frame_duration_type_quants[type] >>
      60             :                          TAK_FRAME_DURATION_QUANT_SHIFT;
      61          20 :         max_nb_samples = 16384;
      62           2 :     } else if (type < FF_ARRAY_ELEMS(frame_duration_type_quants)) {
      63           1 :         nb_samples     = frame_duration_type_quants[type];
      64           1 :         max_nb_samples = sample_rate *
      65           1 :                          frame_duration_type_quants[TAK_FST_250ms] >>
      66             :                          TAK_FRAME_DURATION_QUANT_SHIFT;
      67             :     } else {
      68           1 :         return AVERROR_INVALIDDATA;
      69             :     }
      70             : 
      71          21 :     if (nb_samples <= 0 || nb_samples > max_nb_samples)
      72           0 :         return AVERROR_INVALIDDATA;
      73             : 
      74          21 :     return nb_samples;
      75             : }
      76             : 
      77          83 : int ff_tak_check_crc(const uint8_t *buf, unsigned int buf_size)
      78             : {
      79             :     uint32_t crc, CRC;
      80             : 
      81          83 :     if (buf_size < 4)
      82           0 :         return AVERROR_INVALIDDATA;
      83          83 :     buf_size -= 3;
      84             : 
      85          83 :     CRC = AV_RB24(buf + buf_size);
      86          83 :     crc = av_crc(av_crc_get_table(AV_CRC_24_IEEE), 0xCE04B7U, buf, buf_size);
      87          83 :     if (CRC != crc)
      88           8 :         return AVERROR_INVALIDDATA;
      89             : 
      90          75 :     return 0;
      91             : }
      92             : 
      93          22 : void ff_tak_parse_streaminfo(TAKStreamInfo *s, GetBitContext *gb)
      94             : {
      95          22 :     uint64_t channel_mask = 0;
      96             :     int frame_type, i;
      97             : 
      98          22 :     s->codec = get_bits(gb, TAK_ENCODER_CODEC_BITS);
      99          22 :     skip_bits(gb, TAK_ENCODER_PROFILE_BITS);
     100             : 
     101          22 :     frame_type = get_bits(gb, TAK_SIZE_FRAME_DURATION_BITS);
     102          22 :     s->samples = get_bits64(gb, TAK_SIZE_SAMPLES_NUM_BITS);
     103             : 
     104          22 :     s->data_type   = get_bits(gb, TAK_FORMAT_DATA_TYPE_BITS);
     105          22 :     s->sample_rate = get_bits(gb, TAK_FORMAT_SAMPLE_RATE_BITS) +
     106             :                      TAK_SAMPLE_RATE_MIN;
     107          22 :     s->bps         = get_bits(gb, TAK_FORMAT_BPS_BITS) +
     108             :                      TAK_BPS_MIN;
     109          22 :     s->channels    = get_bits(gb, TAK_FORMAT_CHANNEL_BITS) +
     110             :                      TAK_CHANNELS_MIN;
     111             : 
     112          22 :     if (get_bits1(gb)) {
     113           2 :         skip_bits(gb, TAK_FORMAT_VALID_BITS);
     114           2 :         if (get_bits1(gb)) {
     115          23 :             for (i = 0; i < s->channels; i++) {
     116          21 :                 int value = get_bits(gb, TAK_FORMAT_CH_LAYOUT_BITS);
     117             : 
     118          21 :                 if (value < FF_ARRAY_ELEMS(tak_channel_layouts))
     119           7 :                     channel_mask |= tak_channel_layouts[value];
     120             :             }
     121             :         }
     122             :     }
     123             : 
     124          22 :     s->ch_layout     = channel_mask;
     125          22 :     s->frame_samples = tak_get_nb_samples(s->sample_rate, frame_type);
     126          22 : }
     127             : 
     128           1 : int avpriv_tak_parse_streaminfo(TAKStreamInfo *s, const uint8_t *buf, int size)
     129             : {
     130             :     GetBitContext gb;
     131           1 :     int ret = init_get_bits8(&gb, buf, size);
     132             : 
     133           1 :     if (ret < 0)
     134           0 :         return AVERROR_INVALIDDATA;
     135             : 
     136           1 :     ff_tak_parse_streaminfo(s, &gb);
     137             : 
     138           1 :     return 0;
     139             : }
     140             : 
     141         124 : int ff_tak_decode_frame_header(AVCodecContext *avctx, GetBitContext *gb,
     142             :                                TAKStreamInfo *ti, int log_level_offset)
     143             : {
     144         124 :     if (get_bits(gb, TAK_FRAME_HEADER_SYNC_ID_BITS) != TAK_FRAME_HEADER_SYNC_ID) {
     145           0 :         av_log(avctx, AV_LOG_ERROR + log_level_offset, "missing sync id\n");
     146           0 :         return AVERROR_INVALIDDATA;
     147             :     }
     148             : 
     149         124 :     ti->flags     = get_bits(gb, TAK_FRAME_HEADER_FLAGS_BITS);
     150         124 :     ti->frame_num = get_bits(gb, TAK_FRAME_HEADER_NO_BITS);
     151             : 
     152         124 :     if (ti->flags & TAK_FRAME_FLAG_IS_LAST) {
     153          12 :         ti->last_frame_samples = get_bits(gb, TAK_FRAME_HEADER_SAMPLE_COUNT_BITS) + 1;
     154          12 :         skip_bits(gb, 2);
     155             :     } else {
     156         112 :         ti->last_frame_samples = 0;
     157             :     }
     158             : 
     159         124 :     if (ti->flags & TAK_FRAME_FLAG_HAS_INFO) {
     160          21 :         ff_tak_parse_streaminfo(ti, gb);
     161             : 
     162          21 :         if (get_bits(gb, 6))
     163           3 :             skip_bits(gb, 25);
     164          21 :         align_get_bits(gb);
     165             :     }
     166             : 
     167         124 :     if (ti->flags & TAK_FRAME_FLAG_HAS_METADATA)
     168           3 :         return AVERROR_INVALIDDATA;
     169             : 
     170         121 :     skip_bits(gb, 24);
     171             : 
     172         121 :     return 0;
     173             : }

Generated by: LCOV version 1.13