Libav
libdcadec.c
Go to the documentation of this file.
1 /*
2  * libdcadec decoder wrapper
3  * Copyright (C) 2015 Hendrik Leppkes
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include <libdcadec/dca_context.h>
23 
25 #include "libavutil/common.h"
26 #include "libavutil/opt.h"
27 
28 #include "avcodec.h"
29 #include "dca.h"
30 #include "dca_syncwords.h"
31 #include "internal.h"
32 
33 typedef struct DCADecContext {
34  struct dcadec_context *ctx;
38 
39 static int dcadec_decode_frame(AVCodecContext *avctx, void *data,
40  int *got_frame_ptr, AVPacket *avpkt)
41 {
42  DCADecContext *s = avctx->priv_data;
43  AVFrame *frame = data;
44  int ret, i, k;
45  int **samples, nsamples, channel_mask, sample_rate, bits_per_sample, profile;
46  uint32_t mrk;
47  uint8_t *input = avpkt->data;
48  int input_size = avpkt->size;
49 
50  /* convert bytestream syntax to RAW BE format if required */
51  if (input_size < 8) {
52  av_log(avctx, AV_LOG_ERROR, "Input size too small\n");
53  return AVERROR_INVALIDDATA;
54  }
55  mrk = AV_RB32(input);
56  if (mrk != DCA_SYNCWORD_CORE_BE && mrk != DCA_SYNCWORD_SUBSTREAM) {
58  if (!s->buffer)
59  return AVERROR(ENOMEM);
60 
61  if ((ret = ff_dca_convert_bitstream(avpkt->data, avpkt->size, s->buffer, s->buffer_size)) < 0)
62  return ret;
63 
64  input = s->buffer;
65  input_size = ret;
66  }
67 
68  if ((ret = dcadec_context_parse(s->ctx, input, input_size)) < 0) {
69  av_log(avctx, AV_LOG_ERROR, "dcadec_context_parse() failed: %d (%s)\n", -ret, dcadec_strerror(ret));
70  return AVERROR_UNKNOWN;
71  }
72  if ((ret = dcadec_context_filter(s->ctx, &samples, &nsamples, &channel_mask,
73  &sample_rate, &bits_per_sample, &profile)) < 0) {
74  av_log(avctx, AV_LOG_ERROR, "dcadec_context_filter() failed: %d (%s)\n", -ret, dcadec_strerror(ret));
75  return AVERROR_UNKNOWN;
76  }
77 
78  avctx->channels = av_get_channel_layout_nb_channels(channel_mask);
79  avctx->channel_layout = channel_mask;
80  avctx->sample_rate = sample_rate;
81 
82  if (bits_per_sample == 16)
84  else if (bits_per_sample <= 24)
86  else {
87  av_log(avctx, AV_LOG_ERROR, "Unsupported number of bits per sample: %d\n",
88  bits_per_sample);
89  return AVERROR(ENOSYS);
90  }
91 
92  avctx->bits_per_raw_sample = bits_per_sample;
93 
94  switch (profile) {
95  case DCADEC_PROFILE_DS:
96  avctx->profile = FF_PROFILE_DTS;
97  break;
98  case DCADEC_PROFILE_DS_96_24:
100  break;
101  case DCADEC_PROFILE_DS_ES:
102  avctx->profile = FF_PROFILE_DTS_ES;
103  break;
104  case DCADEC_PROFILE_HD_HRA:
106  break;
107  case DCADEC_PROFILE_HD_MA:
108  avctx->profile = FF_PROFILE_DTS_HD_MA;
109  break;
110  case DCADEC_PROFILE_EXPRESS:
112  break;
113  case DCADEC_PROFILE_UNKNOWN:
114  default:
115  avctx->profile = FF_PROFILE_UNKNOWN;
116  break;
117  }
118 
119  /* bitrate is only meaningful if there are no HD extensions, as they distort the bitrate */
120  if (profile == DCADEC_PROFILE_DS || profile == DCADEC_PROFILE_DS_96_24 || profile == DCADEC_PROFILE_DS_ES) {
121  struct dcadec_core_info *info = dcadec_context_get_core_info(s->ctx);
122  avctx->bit_rate = info->bit_rate;
123  dcadec_context_free_core_info(info);
124  } else
125  avctx->bit_rate = 0;
126 
127  frame->nb_samples = nsamples;
128  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
129  return ret;
130 
131  for (i = 0; i < avctx->channels; i++) {
132  if (frame->format == AV_SAMPLE_FMT_S16P) {
133  int16_t *plane = (int16_t *)frame->extended_data[i];
134  for (k = 0; k < nsamples; k++)
135  plane[k] = samples[i][k];
136  } else {
137  int32_t *plane = (int32_t *)frame->extended_data[i];
138  int shift = 32 - bits_per_sample;
139  for (k = 0; k < nsamples; k++)
140  plane[k] = samples[i][k] << shift;
141  }
142  }
143 
144  *got_frame_ptr = 1;
145 
146  return avpkt->size;
147 }
148 
150 {
151  DCADecContext *s = avctx->priv_data;
152  dcadec_context_clear(s->ctx);
153 }
154 
156 {
157  DCADecContext *s = avctx->priv_data;
158 
159  dcadec_context_destroy(s->ctx);
160  s->ctx = NULL;
161 
162  av_freep(&s->buffer);
163 
164  return 0;
165 }
166 
168 {
169  DCADecContext *s = avctx->priv_data;
170 
171  s->ctx = dcadec_context_create(0);
172  if (!s->ctx)
173  return AVERROR(ENOMEM);
174 
176  avctx->bits_per_raw_sample = 24;
177 
178  return 0;
179 }
180 
181 static const AVProfile profiles[] = {
182  { FF_PROFILE_DTS, "DTS" },
183  { FF_PROFILE_DTS_ES, "DTS-ES" },
184  { FF_PROFILE_DTS_96_24, "DTS 96/24" },
185  { FF_PROFILE_DTS_HD_HRA, "DTS-HD HRA" },
186  { FF_PROFILE_DTS_HD_MA, "DTS-HD MA" },
187  { FF_PROFILE_DTS_EXPRESS, "DTS Express" },
188  { FF_PROFILE_UNKNOWN },
189 };
190 
192  .name = "libdcadec",
193  .long_name = NULL_IF_CONFIG_SMALL("dcadec DCA decoder"),
194  .type = AVMEDIA_TYPE_AUDIO,
195  .id = AV_CODEC_ID_DTS,
196  .priv_data_size = sizeof(DCADecContext),
197  .init = dcadec_init,
199  .close = dcadec_close,
200  .flush = dcadec_flush,
204  .profiles = NULL_IF_CONFIG_SMALL(profiles),
205 };
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:54
struct dcadec_context * ctx
Definition: libdcadec.c:34
This structure describes decoded (raw) audio or video data.
Definition: frame.h:140
#define FF_PROFILE_DTS_HD_HRA
Definition: avcodec.h:2898
static av_cold void dcadec_flush(AVCodecContext *avctx)
Definition: libdcadec.c:149
int size
Definition: avcodec.h:1347
av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (%s)\, len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt), use_generic ? ac->func_descr_generic :ac->func_descr)
int ff_dca_convert_bitstream(const uint8_t *src, int src_size, uint8_t *dst, int max_size)
Convert bitstream to one representation based on sync marker.
Definition: dca.c:35
#define FF_PROFILE_DTS_EXPRESS
Definition: avcodec.h:2900
#define AV_CODEC_CAP_CHANNEL_CONF
Codec should fill in channel configuration and samplerate instead of container.
Definition: avcodec.h:889
int bits_per_raw_sample
Bits per sample/pixel of internal libavcodec pixel/sample format.
Definition: avcodec.h:2776
int profile
profile
Definition: avcodec.h:2880
AVCodec.
Definition: avcodec.h:3120
void av_freep(void *arg)
Free a memory block which has been allocated with av_malloc(z)() or av_realloc() and set the pointer ...
Definition: mem.c:202
enum AVSampleFormat sample_fmt
audio sample format
Definition: avcodec.h:2160
uint8_t
#define av_cold
Definition: attributes.h:66
AVOptions.
#define AV_RB32
Definition: intreadwrite.h:130
int buffer_size
Definition: libdcadec.c:36
const char data[16]
Definition: mxf.c:70
uint8_t * data
Definition: avcodec.h:1346
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:124
void * av_fast_realloc(void *ptr, unsigned int *size, size_t min_size)
Reallocate the given block if it is not large enough, otherwise do nothing.
Definition: mem.c:375
#define AVERROR(e)
Definition: error.h:43
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:148
int av_get_channel_layout_nb_channels(uint64_t channel_layout)
Return the number of channels in the channel layout.
#define FF_PROFILE_DTS_ES
Definition: avcodec.h:2896
const char * name
Name of the codec implementation.
Definition: avcodec.h:3127
uint64_t channel_layout
Audio channel layout.
Definition: avcodec.h:2203
int bit_rate
the average bitrate
Definition: avcodec.h:1473
audio channel layout utility functions
signed 32 bits, planar
Definition: samplefmt.h:70
static int decode(AVCodecContext *avctx, AVFrame *frame, int *got_frame, AVPacket *pkt)
Definition: avconv.c:1288
#define FF_PROFILE_UNKNOWN
Definition: avcodec.h:2881
int32_t
#define FF_PROFILE_DTS
Definition: avcodec.h:2895
static int dcadec_decode_frame(AVCodecContext *avctx, void *data, int *got_frame_ptr, AVPacket *avpkt)
Definition: libdcadec.c:39
int format
format of the frame, -1 if unknown or unset Values correspond to enum AVPixelFormat for video frames...
Definition: frame.h:191
#define FF_PROFILE_DTS_96_24
Definition: avcodec.h:2897
NULL
Definition: eval.c:55
static av_cold int dcadec_init(AVCodecContext *avctx)
Definition: libdcadec.c:167
Libavcodec external API header.
AVSampleFormat
Audio Sample Formats.
Definition: samplefmt.h:60
int sample_rate
samples per second
Definition: avcodec.h:2152
main external API structure.
Definition: avcodec.h:1409
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: utils.c:589
static const AVProfile profiles[]
Definition: libdcadec.c:181
static av_cold int dcadec_close(AVCodecContext *avctx)
Definition: libdcadec.c:155
mfxU16 profile
Definition: qsvenc.c:43
common internal api header.
common internal and external API header
static av_cold void flush(AVCodecContext *avctx)
Flush (reset) the frame ID after seeking.
Definition: alsdec.c:1797
AVProfile.
Definition: avcodec.h:3108
uint8_t * buffer
Definition: libdcadec.c:35
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:61
static av_cold int init(AVCodecParserContext *s)
Definition: h264_parser.c:582
#define AV_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding...
Definition: avcodec.h:638
void * priv_data
Definition: avcodec.h:1451
int channels
number of audio channels
Definition: avcodec.h:2153
static enum AVSampleFormat sample_fmts[]
Definition: adpcmenc.c:700
signed 16 bits, planar
Definition: samplefmt.h:69
#define FF_PROFILE_DTS_HD_MA
Definition: avcodec.h:2899
uint8_t ** extended_data
pointers to the data planes/channels.
Definition: frame.h:174
This structure stores compressed data.
Definition: avcodec.h:1323
AVCodec ff_libdcadec_decoder
Definition: libdcadec.c:191
int nb_samples
number of audio samples (per channel) described by this frame
Definition: frame.h:184
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:838
for(j=16;j >0;--j)