Add support for setting OPUS header gain.

media/filters/opus_audio_decoder.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/filters/opus_audio_decoder.h"
 
 #include <cmath>
 
 #include "base/bind.h"
 #include "base/callback_helpers.h"
@@ skipping 139 matching lines @@
 
 // Size of the Opus extra data excluding optional mapping information.
 static const int kOpusExtraDataSize = 19;
 
 // Offset to the channel count byte in the Opus extra data.
 static const int kOpusExtraDataChannelsOffset = 9;
 
 // Offset to the pre-skip value in the Opus extra data.
 static const int kOpusExtraDataSkipSamplesOffset = 10;
 
+// Offset to the gain value in the Opus extra data.
+static const int kOpusExtraDataGainOffset = 16;
+
 // Offset to the channel mapping byte in the Opus extra data.
 static const int kOpusExtraDataChannelMappingOffset = 18;
 
 // Extra Data contains a stream map. The mapping values are in extra data beyond
 // the always present |kOpusExtraDataSize| bytes of data. The mapping data
 // contains stream count, coupling information, and per channel mapping values:
 //   - Byte 0: Number of streams.
 //   - Byte 1: Number coupled.
 //   - Byte 2: Starting at byte 2 are |extra_data->channels| uint8 mapping
 //             values.
 static const int kOpusExtraDataNumStreamsOffset = kOpusExtraDataSize;
 static const int kOpusExtraDataNumCoupledOffset =
     kOpusExtraDataNumStreamsOffset + 1;
 static const int kOpusExtraDataStreamMapOffset =
     kOpusExtraDataNumStreamsOffset + 2;
 
 struct OpusExtraData {
   OpusExtraData()
       : channels(0),
         skip_samples(0),
         channel_mapping(0),
         num_streams(0),
-        num_coupled(0) {
+        num_coupled(0),
+        gain_db(0) {
     memcpy(stream_map,
            kDefaultOpusChannelLayout,
            kMaxChannelsWithDefaultLayout);
   }
   int channels;
-  int skip_samples;
+  uint16 skip_samples;
   int channel_mapping;
   int num_streams;
   int num_coupled;
+  int16 gain_db;
   uint8 stream_map[kMaxVorbisChannels];
 };
 
 // Returns true when able to successfully parse and store Opus extra data in
 // |extra_data|. Based on opus header parsing code in libopusdec from FFmpeg,
 // and opus_header from Xiph's opus-tools project.
 static bool ParseOpusExtraData(const uint8* data, int data_size,
                                const AudioDecoderConfig& config,
                                OpusExtraData* extra_data) {
   if (data_size < kOpusExtraDataSize) {
     DLOG(ERROR) << "Extra data size is too small:" << data_size;
     return false;
   }
 
   extra_data->channels = *(data + kOpusExtraDataChannelsOffset);
 
   if (extra_data->channels <= 0 || extra_data->channels > kMaxVorbisChannels) {
     DLOG(ERROR) << "invalid channel count in extra data: "
                 << extra_data->channels;
     return false;
   }
 
   extra_data->skip_samples =
       ReadLE16(data, data_size, kOpusExtraDataSkipSamplesOffset);
+  extra_data->gain_db = static_cast<int16>(
+      ReadLE16(data, data_size, kOpusExtraDataGainOffset));
 
   extra_data->channel_mapping = *(data + kOpusExtraDataChannelMappingOffset);
 
   if (!extra_data->channel_mapping) {
     if (extra_data->channels > kMaxChannelsWithDefaultLayout) {
       DLOG(ERROR) << "Invalid extra data, missing stream map.";
       return false;
     }
 
     extra_data->num_streams = 1;
@@ skipping 269 matching lines @@
                                                   opus_extra_data.num_streams,
                                                   opus_extra_data.num_coupled,
                                                   channel_mapping,
                                                   &status);
   if (!opus_decoder_ || status != OPUS_OK) {
     DLOG(ERROR) << "opus_multistream_decoder_create failed status="
                 << opus_strerror(status);
     return false;
   }
 
+  status = opus_multistream_decoder_ctl(
+      opus_decoder_, OPUS_SET_GAIN(opus_extra_data.gain_db));
+  if (status != OPUS_OK) {
+    DLOG(ERROR) << "Failed to set OPUS header gain; status="
+                << opus_strerror(status);
+    return false;
+  }
+
   channel_layout_ = config.channel_layout();
   samples_per_second_ = config.samples_per_second();
   output_timestamp_helper_.reset(
       new AudioTimestampHelper(config.samples_per_second()));
   start_input_timestamp_ = kNoTimestamp();
   return true;
 }
 
 void OpusAudioDecoder::CloseDecoder() {
   if (opus_decoder_) {
@@ skipping 89 matching lines @@
   output_timestamp_helper_->AddFrames(frames_decoded);
 
   // Discard the buffer to indicate we need more data.
   if (!frames_to_output)
     *output_buffer = NULL;
 
   return true;
 }
 
 }  // namespace media