libavcodec/pcm-blurayenc.c - third_party/ffmpeg - Git at Google

 /*
  * LPCM codecs for PCM formats found in Blu-ray m2ts streams
  *
  * This file is part of FFmpeg.
  *
  * FFmpeg 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.
  *
  * FFmpeg 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 FFmpeg; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */

 #include "libavutil/channel_layout.h"
 #include "avcodec.h"
 #include "bytestream.h"
 #include "codec_internal.h"
 #include "encode.h"

 typedef struct BlurayPCMEncContext {
     uint16_t header;      // Header added to every frame
 } BlurayPCMEncContext;

 static av_cold int pcm_bluray_encode_init(AVCodecContext *avctx)
 {
     BlurayPCMEncContext *s = avctx->priv_data;
     uint8_t ch_layout;
     int quant, freq, frame_size;

     switch (avctx->sample_fmt) {
     case AV_SAMPLE_FMT_S16:
         avctx->bits_per_coded_sample = 16;
         frame_size = 240;
         quant = 1;
         break;
     case AV_SAMPLE_FMT_S32:
         frame_size = 360;
         avctx->bits_per_coded_sample = 24;
         quant = 3;
         break;
     default:
         return AVERROR_BUG;
     }

     switch (avctx->sample_rate) {
     case 48000:
         freq = 1;
         break;
     case 96000:
         freq = 4;
         break;
     case 192000:
         freq = 5;
         break;
     default:
         return AVERROR_BUG;
     }

     switch (av_channel_layout_subset(&avctx->ch_layout, ~(uint64_t)0)) {
     case AV_CH_LAYOUT_MONO:
         ch_layout = 1;
         break;
     case AV_CH_LAYOUT_STEREO:
         ch_layout = 3;
         break;
     case AV_CH_LAYOUT_SURROUND:
         ch_layout = 4;
         break;
     case AV_CH_LAYOUT_2_1:
         ch_layout = 5;
         break;
     case AV_CH_LAYOUT_4POINT0:
         ch_layout = 6;
         break;
     case AV_CH_LAYOUT_2_2:
         ch_layout = 7;
         break;
     case AV_CH_LAYOUT_5POINT0:
         ch_layout = 8;
         break;
     case AV_CH_LAYOUT_5POINT1:
         ch_layout = 9;
         break;
     case AV_CH_LAYOUT_7POINT0:
         ch_layout = 10;
         break;
     case AV_CH_LAYOUT_7POINT1:
         ch_layout = 11;
         break;
     default:
         return AVERROR_BUG;
     }

     s->header = (((ch_layout << 4) | freq) << 8) | (quant << 6);
     avctx->frame_size = frame_size;

     return 0;
 }

 static int pcm_bluray_encode_frame(AVCodecContext *avctx, AVPacket *avpkt,
                                    const AVFrame *frame, int *got_packet_ptr)
 {
     BlurayPCMEncContext *s = avctx->priv_data;
     int sample_size, samples, channel, num_dest_channels;
     const int16_t *src16;
     const int32_t *src32;
     unsigned pkt_size;
     PutByteContext pb;
     int ret;

     num_dest_channels = FFALIGN(avctx->ch_layout.nb_channels, 2);
     sample_size = (num_dest_channels *
                    (avctx->sample_fmt == AV_SAMPLE_FMT_S16 ? 16 : 24)) >> 3;
     samples = frame->nb_samples;

     pkt_size = sample_size * samples + 4;

     if ((ret = ff_get_encode_buffer(avctx, avpkt, pkt_size, 0)) < 0)
         return ret;

     AV_WB16(avpkt->data, pkt_size - 4);
     AV_WB16(avpkt->data + 2, s->header);

     src16 = (const int16_t *)frame->data[0];
     src32 = (const int32_t *)frame->data[0];

     bytestream2_init_writer(&pb, avpkt->data + 4, avpkt->size - 4);

     switch (avctx->ch_layout.u.mask) {
     /* cases with same number of source and coded channels */
     case AV_CH_LAYOUT_STEREO:
     case AV_CH_LAYOUT_4POINT0:
     case AV_CH_LAYOUT_2_2:
         samples *= num_dest_channels;
         if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
 #if HAVE_BIGENDIAN
             bytestream2_put_bufferu(&pb, frame->data[0], samples * 2);
 #else
             do {
                 bytestream2_put_be16u(&pb, *src16++);
             } while (--samples);
 #endif
         } else {
             do {
                 bytestream2_put_be24u(&pb, (*src32++) >> 8);
             } while (--samples);
         }
         break;
     /* cases where number of source channels = coded channels + 1 */
     case AV_CH_LAYOUT_MONO:
     case AV_CH_LAYOUT_SURROUND:
     case AV_CH_LAYOUT_2_1:
     case AV_CH_LAYOUT_5POINT0:
         if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
             do {
 #if HAVE_BIGENDIAN
                 bytestream2_put_bufferu(&pb, (const uint8_t *)src16, avctx->ch_layout.nb_channels * 2);
                 src16 += avctx->ch_layout.nb_channels;
 #else
                 channel = avctx->ch_layout.nb_channels;
                 do {
                     bytestream2_put_be16u(&pb, *src16++);
                 } while (--channel);
 #endif
                 bytestream2_put_ne16(&pb, 0);
             } while (--samples);
         } else {
             do {
                 channel = avctx->ch_layout.nb_channels;
                 do {
                     bytestream2_put_be24u(&pb, (*src32++) >> 8);
                 } while (--channel);
                 bytestream2_put_ne24(&pb, 0);
             } while (--samples);
         }
         break;
         /* remapping: L, R, C, LBack, RBack, LF */
     case AV_CH_LAYOUT_5POINT1:
         if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
             do {
                 bytestream2_put_be16u(&pb, src16[0]);
                 bytestream2_put_be16u(&pb, src16[1]);
                 bytestream2_put_be16u(&pb, src16[2]);
                 bytestream2_put_be16u(&pb, src16[4]);
                 bytestream2_put_be16u(&pb, src16[5]);
                 bytestream2_put_be16u(&pb, src16[3]);
                 src16 += 6;
             } while (--samples);
         } else {
             do {
                 bytestream2_put_be24u(&pb, src32[0] >> 8);
                 bytestream2_put_be24u(&pb, src32[1] >> 8);
                 bytestream2_put_be24u(&pb, src32[2] >> 8);
                 bytestream2_put_be24u(&pb, src32[4] >> 8);
                 bytestream2_put_be24u(&pb, src32[5] >> 8);
                 bytestream2_put_be24u(&pb, src32[3] >> 8);
                 src32 += 6;
             } while (--samples);
         }
         break;
         /* remapping: L, R, C, LSide, LBack, RBack, RSide, <unused> */
     case AV_CH_LAYOUT_7POINT0:
         if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
             do {
                 bytestream2_put_be16u(&pb, src16[0]);
                 bytestream2_put_be16u(&pb, src16[1]);
                 bytestream2_put_be16u(&pb, src16[2]);
                 bytestream2_put_be16u(&pb, src16[5]);
                 bytestream2_put_be16u(&pb, src16[3]);
                 bytestream2_put_be16u(&pb, src16[4]);
                 bytestream2_put_be16u(&pb, src16[6]);
                 src16 += 7;
                 bytestream2_put_ne16(&pb, 0);
             } while (--samples);
         } else {
             do {
                 bytestream2_put_be24u(&pb, src32[0] >> 8);
                 bytestream2_put_be24u(&pb, src32[1] >> 8);
                 bytestream2_put_be24u(&pb, src32[2] >> 8);
                 bytestream2_put_be24u(&pb, src32[5] >> 8);
                 bytestream2_put_be24u(&pb, src32[3] >> 8);
                 bytestream2_put_be24u(&pb, src32[4] >> 8);
                 bytestream2_put_be24u(&pb, src32[6] >> 8);
                 src32 += 7;
                 bytestream2_put_ne24(&pb, 0);
             } while (--samples);
         }
         break;
         /* remapping: L, R, C, LSide, LBack, RBack, RSide, LF */
     case AV_CH_LAYOUT_7POINT1:
         if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
             do {
                 bytestream2_put_be16u(&pb, src16[0]);
                 bytestream2_put_be16u(&pb, src16[1]);
                 bytestream2_put_be16u(&pb, src16[2]);
                 bytestream2_put_be16u(&pb, src16[6]);
                 bytestream2_put_be16u(&pb, src16[4]);
                 bytestream2_put_be16u(&pb, src16[5]);
                 bytestream2_put_be16u(&pb, src16[7]);
                 bytestream2_put_be16u(&pb, src16[3]);
                 src16 += 8;
             } while (--samples);
         } else {
             do {
                 bytestream2_put_be24u(&pb, src32[0]);
                 bytestream2_put_be24u(&pb, src32[1]);
                 bytestream2_put_be24u(&pb, src32[2]);
                 bytestream2_put_be24u(&pb, src32[6]);
                 bytestream2_put_be24u(&pb, src32[4]);
                 bytestream2_put_be24u(&pb, src32[5]);
                 bytestream2_put_be24u(&pb, src32[7]);
                 bytestream2_put_be24u(&pb, src32[3]);
                 src32 += 8;
             } while (--samples);
         }
         break;
     default:
         return AVERROR_BUG;
     }

     *got_packet_ptr = 1;

     return 0;
 }

 const FFCodec ff_pcm_bluray_encoder = {
     .p.name                = "pcm_bluray",
     CODEC_LONG_NAME("PCM signed 16|20|24-bit big-endian for Blu-ray media"),
     .p.type                = AVMEDIA_TYPE_AUDIO,
     .p.id                  = AV_CODEC_ID_PCM_BLURAY,
     .priv_data_size        = sizeof(BlurayPCMEncContext),
     .init                  = pcm_bluray_encode_init,
     FF_CODEC_ENCODE_CB(pcm_bluray_encode_frame),
     .p.supported_samplerates = (const int[]) { 48000, 96000, 192000, 0 },
     .p.ch_layouts   = (const AVChannelLayout[]) {
         AV_CHANNEL_LAYOUT_MONO,
         AV_CHANNEL_LAYOUT_STEREO,
         AV_CHANNEL_LAYOUT_SURROUND,
         AV_CHANNEL_LAYOUT_2_1,
         AV_CHANNEL_LAYOUT_4POINT0,
         AV_CHANNEL_LAYOUT_2_2,
         AV_CHANNEL_LAYOUT_5POINT0,
         AV_CHANNEL_LAYOUT_5POINT1,
         AV_CHANNEL_LAYOUT_7POINT0,
         AV_CHANNEL_LAYOUT_7POINT1,
         { 0 } },
     .p.sample_fmts         = (const enum AVSampleFormat[]) {
         AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_NONE },
     .p.capabilities        = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
 };
	/*
	* LPCM codecs for PCM formats found in Blu-ray m2ts streams
	*
	* This file is part of FFmpeg.
	*
	* FFmpeg 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.
	*
	* FFmpeg 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 FFmpeg; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	*/

	#include "libavutil/channel_layout.h"
	#include "avcodec.h"
	#include "bytestream.h"
	#include "codec_internal.h"
	#include "encode.h"

	typedef struct BlurayPCMEncContext {
	uint16_t header; // Header added to every frame
	} BlurayPCMEncContext;

	static av_cold int pcm_bluray_encode_init(AVCodecContext *avctx)
	{
	BlurayPCMEncContext *s = avctx->priv_data;
	uint8_t ch_layout;
	int quant, freq, frame_size;

	switch (avctx->sample_fmt) {
	case AV_SAMPLE_FMT_S16:
	avctx->bits_per_coded_sample = 16;
	frame_size = 240;
	quant = 1;
	break;
	case AV_SAMPLE_FMT_S32:
	frame_size = 360;
	avctx->bits_per_coded_sample = 24;
	quant = 3;
	break;
	default:
	return AVERROR_BUG;
	}

	switch (avctx->sample_rate) {
	case 48000:
	freq = 1;
	break;
	case 96000:
	freq = 4;
	break;
	case 192000:
	freq = 5;
	break;
	default:
	return AVERROR_BUG;
	}

	switch (av_channel_layout_subset(&avctx->ch_layout, ~(uint64_t)0)) {
	case AV_CH_LAYOUT_MONO:
	ch_layout = 1;
	break;
	case AV_CH_LAYOUT_STEREO:
	ch_layout = 3;
	break;
	case AV_CH_LAYOUT_SURROUND:
	ch_layout = 4;
	break;
	case AV_CH_LAYOUT_2_1:
	ch_layout = 5;
	break;
	case AV_CH_LAYOUT_4POINT0:
	ch_layout = 6;
	break;
	case AV_CH_LAYOUT_2_2:
	ch_layout = 7;
	break;
	case AV_CH_LAYOUT_5POINT0:
	ch_layout = 8;
	break;
	case AV_CH_LAYOUT_5POINT1:
	ch_layout = 9;
	break;
	case AV_CH_LAYOUT_7POINT0:
	ch_layout = 10;
	break;
	case AV_CH_LAYOUT_7POINT1:
	ch_layout = 11;
	break;
	default:
	return AVERROR_BUG;
	}

	s->header = (((ch_layout << 4) \| freq) << 8) \| (quant << 6);
	avctx->frame_size = frame_size;

	return 0;
	}

	static int pcm_bluray_encode_frame(AVCodecContext avctx, AVPacket avpkt,
	const AVFrame frame, int got_packet_ptr)
	{
	BlurayPCMEncContext *s = avctx->priv_data;
	int sample_size, samples, channel, num_dest_channels;
	const int16_t *src16;
	const int32_t *src32;
	unsigned pkt_size;
	PutByteContext pb;
	int ret;

	num_dest_channels = FFALIGN(avctx->ch_layout.nb_channels, 2);
	sample_size = (num_dest_channels *
	(avctx->sample_fmt == AV_SAMPLE_FMT_S16 ? 16 : 24)) >> 3;
	samples = frame->nb_samples;

	pkt_size = sample_size * samples + 4;

	if ((ret = ff_get_encode_buffer(avctx, avpkt, pkt_size, 0)) < 0)
	return ret;

	AV_WB16(avpkt->data, pkt_size - 4);
	AV_WB16(avpkt->data + 2, s->header);

	src16 = (const int16_t *)frame->data[0];
	src32 = (const int32_t *)frame->data[0];

	bytestream2_init_writer(&pb, avpkt->data + 4, avpkt->size - 4);

	switch (avctx->ch_layout.u.mask) {
	/* cases with same number of source and coded channels */
	case AV_CH_LAYOUT_STEREO:
	case AV_CH_LAYOUT_4POINT0:
	case AV_CH_LAYOUT_2_2:
	samples *= num_dest_channels;
	if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
	#if HAVE_BIGENDIAN
	bytestream2_put_bufferu(&pb, frame->data[0], samples * 2);
	#else
	do {
	bytestream2_put_be16u(&pb, *src16++);
	} while (--samples);
	#endif
	} else {
	do {
	bytestream2_put_be24u(&pb, (*src32++) >> 8);
	} while (--samples);
	}
	break;
	/* cases where number of source channels = coded channels + 1 */
	case AV_CH_LAYOUT_MONO:
	case AV_CH_LAYOUT_SURROUND:
	case AV_CH_LAYOUT_2_1:
	case AV_CH_LAYOUT_5POINT0:
	if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
	do {
	#if HAVE_BIGENDIAN
	bytestream2_put_bufferu(&pb, (const uint8_t )src16, avctx->ch_layout.nb_channels 2);
	src16 += avctx->ch_layout.nb_channels;
	#else
	channel = avctx->ch_layout.nb_channels;
	do {
	bytestream2_put_be16u(&pb, *src16++);
	} while (--channel);
	#endif
	bytestream2_put_ne16(&pb, 0);
	} while (--samples);
	} else {
	do {
	channel = avctx->ch_layout.nb_channels;
	do {
	bytestream2_put_be24u(&pb, (*src32++) >> 8);
	} while (--channel);
	bytestream2_put_ne24(&pb, 0);
	} while (--samples);
	}
	break;
	/* remapping: L, R, C, LBack, RBack, LF */
	case AV_CH_LAYOUT_5POINT1:
	if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
	do {
	bytestream2_put_be16u(&pb, src16[0]);
	bytestream2_put_be16u(&pb, src16[1]);
	bytestream2_put_be16u(&pb, src16[2]);
	bytestream2_put_be16u(&pb, src16[4]);
	bytestream2_put_be16u(&pb, src16[5]);
	bytestream2_put_be16u(&pb, src16[3]);
	src16 += 6;
	} while (--samples);
	} else {
	do {
	bytestream2_put_be24u(&pb, src32[0] >> 8);
	bytestream2_put_be24u(&pb, src32[1] >> 8);
	bytestream2_put_be24u(&pb, src32[2] >> 8);
	bytestream2_put_be24u(&pb, src32[4] >> 8);
	bytestream2_put_be24u(&pb, src32[5] >> 8);
	bytestream2_put_be24u(&pb, src32[3] >> 8);
	src32 += 6;
	} while (--samples);
	}
	break;
	/* remapping: L, R, C, LSide, LBack, RBack, RSide, <unused> */
	case AV_CH_LAYOUT_7POINT0:
	if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
	do {
	bytestream2_put_be16u(&pb, src16[0]);
	bytestream2_put_be16u(&pb, src16[1]);
	bytestream2_put_be16u(&pb, src16[2]);
	bytestream2_put_be16u(&pb, src16[5]);
	bytestream2_put_be16u(&pb, src16[3]);
	bytestream2_put_be16u(&pb, src16[4]);
	bytestream2_put_be16u(&pb, src16[6]);
	src16 += 7;
	bytestream2_put_ne16(&pb, 0);
	} while (--samples);
	} else {
	do {
	bytestream2_put_be24u(&pb, src32[0] >> 8);
	bytestream2_put_be24u(&pb, src32[1] >> 8);
	bytestream2_put_be24u(&pb, src32[2] >> 8);
	bytestream2_put_be24u(&pb, src32[5] >> 8);
	bytestream2_put_be24u(&pb, src32[3] >> 8);
	bytestream2_put_be24u(&pb, src32[4] >> 8);
	bytestream2_put_be24u(&pb, src32[6] >> 8);
	src32 += 7;
	bytestream2_put_ne24(&pb, 0);
	} while (--samples);
	}
	break;
	/* remapping: L, R, C, LSide, LBack, RBack, RSide, LF */
	case AV_CH_LAYOUT_7POINT1:
	if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
	do {
	bytestream2_put_be16u(&pb, src16[0]);
	bytestream2_put_be16u(&pb, src16[1]);
	bytestream2_put_be16u(&pb, src16[2]);
	bytestream2_put_be16u(&pb, src16[6]);
	bytestream2_put_be16u(&pb, src16[4]);
	bytestream2_put_be16u(&pb, src16[5]);
	bytestream2_put_be16u(&pb, src16[7]);
	bytestream2_put_be16u(&pb, src16[3]);
	src16 += 8;
	} while (--samples);
	} else {
	do {
	bytestream2_put_be24u(&pb, src32[0]);
	bytestream2_put_be24u(&pb, src32[1]);
	bytestream2_put_be24u(&pb, src32[2]);
	bytestream2_put_be24u(&pb, src32[6]);
	bytestream2_put_be24u(&pb, src32[4]);
	bytestream2_put_be24u(&pb, src32[5]);
	bytestream2_put_be24u(&pb, src32[7]);
	bytestream2_put_be24u(&pb, src32[3]);
	src32 += 8;
	} while (--samples);
	}
	break;
	default:
	return AVERROR_BUG;
	}

	*got_packet_ptr = 1;

	return 0;
	}

	const FFCodec ff_pcm_bluray_encoder = {
	.p.name = "pcm_bluray",
	CODEC_LONG_NAME("PCM signed 16\|20\|24-bit big-endian for Blu-ray media"),
	.p.type = AVMEDIA_TYPE_AUDIO,
	.p.id = AV_CODEC_ID_PCM_BLURAY,
	.priv_data_size = sizeof(BlurayPCMEncContext),
	.init = pcm_bluray_encode_init,
	FF_CODEC_ENCODE_CB(pcm_bluray_encode_frame),
	.p.supported_samplerates = (const int[]) { 48000, 96000, 192000, 0 },
	.p.ch_layouts = (const AVChannelLayout[]) {
	AV_CHANNEL_LAYOUT_MONO,
	AV_CHANNEL_LAYOUT_STEREO,
	AV_CHANNEL_LAYOUT_SURROUND,
	AV_CHANNEL_LAYOUT_2_1,
	AV_CHANNEL_LAYOUT_4POINT0,
	AV_CHANNEL_LAYOUT_2_2,
	AV_CHANNEL_LAYOUT_5POINT0,
	AV_CHANNEL_LAYOUT_5POINT1,
	AV_CHANNEL_LAYOUT_7POINT0,
	AV_CHANNEL_LAYOUT_7POINT1,
	{ 0 } },
	.p.sample_fmts = (const enum AVSampleFormat[]) {
	AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_NONE },
	.p.capabilities = AV_CODEC_CAP_DR1 \| AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
	};