modules/codec/uleaddvaudio.c

   1 /*****************************************************************************
   2  * uleaddvaudio.c
   3  *****************************************************************************
   4  * Copyright (C) 2012 Laurent Aimar
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir _AT_ videolan _DOT_ org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify it
  10  * under the terms of the GNU Lesser General Public License as published by
  11  * the Free Software Foundation; either version 2.1 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17  * GNU Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public License
  20  * along with this program; if not, write to the Free Software Foundation,
  21  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 #ifdef HAVE_CONFIG_H
  25 # include "config.h"
  26 #endif
  27
  28 #include <vlc_common.h>
  29 #include <vlc_plugin.h>
  30 #include <vlc_codec.h>
  31
  32 #include "../demux/rawdv.h"
  33
  34 /*****************************************************************************
  35  * Module descriptor
  36  *****************************************************************************/
  37 static int  Open(vlc_object_t *);
  38 static void Close(vlc_object_t *);
  39
  40 vlc_module_begin()
  41     set_description(N_("Ulead DV audio decoder"))
  42     set_capability("decoder", 50)
  43     set_category(CAT_INPUT)
  44     set_subcategory(SUBCAT_INPUT_ACODEC)
  45     set_callbacks(Open, Close)
  46 vlc_module_end()
  47
  48 struct decoder_sys_t
  49 {
  50     date_t end_date;
  51
  52     bool     is_pal;
  53     bool     is_12bit;
  54     uint16_t shuffle[2000];
  55 };
  56
  57 static void Flush(decoder_t *dec)
  58 {
  59     decoder_sys_t *sys = dec->p_sys;
  60
  61     date_Set(&sys->end_date, 0);
  62 }
  63
  64 static block_t *Decode(decoder_t *dec, block_t **block_ptr)
  65 {
  66     decoder_sys_t *sys  = dec->p_sys;
  67
  68     if (!block_ptr || !*block_ptr)
  69         return NULL;
  70
  71     block_t *block = *block_ptr;
  72     if (block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED)) {
  73         Flush(dec);
  74         if (block->i_flags & BLOCK_FLAG_CORRUPTED) {
  75             block_Release(block);
  76             *block_ptr = NULL;
  77             return NULL;
  78         }
  79     }
  80
  81     if (block->i_pts > VLC_TS_INVALID &&
  82         block->i_pts != date_Get(&sys->end_date))
  83         date_Set(&sys->end_date, block->i_pts);
  84     block->i_pts = VLC_TS_INVALID;
  85     if (!date_Get(&sys->end_date)) {
  86         /* We've just started the stream, wait for the first PTS. */
  87         block_Release(block);
  88         return NULL;
  89     }
  90
  91     const unsigned int block_size = sys->is_pal ? 8640 : 7200;
  92     if (block->i_buffer >= block_size) {
  93         uint8_t *src = block->p_buffer;
  94
  95         block->i_buffer -= block_size;
  96         block->p_buffer += block_size;
  97
  98         int sample_count = dv_get_audio_sample_count(&src[244], sys->is_pal);
  99
 100         if( decoder_UpdateAudioFormat(dec))
 101             return NULL;
 102         block_t *output = decoder_NewAudioBuffer(dec, sample_count);
 103         if (!output)
 104             return NULL;
 105         output->i_pts    = date_Get(&sys->end_date);
 106         output->i_length = date_Increment(&sys->end_date, sample_count) - output->i_pts;
 107
 108         int16_t *dst = (int16_t*)output->p_buffer;
 109         for (int i = 0; i < sample_count; i++) {
 110           const uint8_t *v = &src[sys->shuffle[i]];
 111           if (sys->is_12bit) {
 112               *dst++ = dv_audio_12to16((v[0] << 4) | ((v[2] >> 4) & 0x0f));
 113               *dst++ = dv_audio_12to16((v[1] << 4) | ((v[2] >> 0) & 0x0f));
 114           } else {
 115               *dst++ = GetWBE(&v[0]);
 116               *dst++ = GetWBE(&v[sys->is_pal ? 4320 : 3600]);
 117           }
 118         }
 119         return output;
 120     }
 121     block_Release(block);
 122     return NULL;
 123 }
 124
 125 static int Open(vlc_object_t *object)
 126 {
 127     decoder_t *dec = (decoder_t*)object;
 128
 129     if (dec->fmt_in.i_codec != VLC_CODEC_ULEAD_DV_AUDIO_NTSC &&
 130         dec->fmt_in.i_codec != VLC_CODEC_ULEAD_DV_AUDIO_PAL)
 131         return VLC_EGENERIC;
 132     if (dec->fmt_in.audio.i_bitspersample != 12 && dec->fmt_in.audio.i_bitspersample != 16)
 133         return VLC_EGENERIC;
 134     if (dec->fmt_in.audio.i_channels != 2)
 135         return VLC_EGENERIC;
 136     if (dec->fmt_in.audio.i_rate <= 0)
 137         return VLC_EGENERIC;
 138
 139     decoder_sys_t *sys = dec->p_sys = malloc(sizeof(*sys));
 140     if (!sys)
 141         return VLC_ENOMEM;
 142
 143     sys->is_pal = dec->fmt_in.i_codec == VLC_CODEC_ULEAD_DV_AUDIO_PAL;
 144     sys->is_12bit = dec->fmt_in.audio.i_bitspersample == 12;
 145
 146     date_Init(&sys->end_date, dec->fmt_in.audio.i_rate, 1);
 147     date_Set(&sys->end_date, 0);
 148
 149     for (unsigned i = 0; i < sizeof(sys->shuffle) / sizeof(*sys->shuffle); i++) {
 150         const unsigned a = sys->is_pal ? 18 : 15;
 151         const unsigned b = 3 * a;
 152         sys->shuffle[i] = 80 * ((21 * (i % 3) + 9 * (i / 3) + ((i / a) % 3)) % b) +
 153                           (2 + sys->is_12bit) * (i / b) + 8;
 154     }
 155
 156     es_format_Init(&dec->fmt_out, AUDIO_ES, VLC_CODEC_S16N);
 157     dec->fmt_out.audio.i_rate = dec->fmt_in.audio.i_rate;
 158     dec->fmt_out.audio.i_channels = 2;
 159     dec->fmt_out.audio.i_physical_channels =
 160     dec->fmt_out.audio.i_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
 161
 162     dec->pf_decode_audio = Decode;
 163     dec->pf_flush        = Flush;
 164
 165     return VLC_SUCCESS;
 166 }
 167
 168 static void Close(vlc_object_t *object)
 169 {
 170     decoder_t *dec = (decoder_t *)object;
 171
 172     free(dec->p_sys);
 173 }
 174