// Copyright 2015 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 "content/renderer/pepper/pepper_audio_encoder_host.h"

#include <stddef.h>
#include <utility>

#include "base/bind.h"
#include "base/macros.h"
#include "base/memory/shared_memory.h"
#include "content/public/renderer/renderer_ppapi_host.h"
#include "content/renderer/pepper/host_globals.h"
#include "content/renderer/render_thread_impl.h"
#include "media/base/bind_to_current_loop.h"
#include "ppapi/c/pp_codecs.h"
#include "ppapi/c/pp_errors.h"
#include "ppapi/host/dispatch_host_message.h"
#include "ppapi/host/ppapi_host.h"
#include "ppapi/proxy/ppapi_messages.h"
#include "ppapi/shared_impl/media_stream_buffer.h"
#include "third_party/opus/src/include/opus.h"

using ppapi::proxy::SerializedHandle;

namespace content {

namespace {

// Buffer up to 150ms (15 x 10ms per frame).
const uint32_t kDefaultNumberOfAudioBuffers = 15;

bool PP_HardwareAccelerationCompatibleAudio(bool accelerated,
                                            PP_HardwareAcceleration requested) {
  switch (requested) {
    case PP_HARDWAREACCELERATION_ONLY:
      return accelerated;
    case PP_HARDWAREACCELERATION_NONE:
      return !accelerated;
    case PP_HARDWAREACCELERATION_WITHFALLBACK:
      return true;
      // No default case, to catch unhandled PP_HardwareAcceleration values.
  }
  return false;
}

}  // namespace

// This class should be constructed and initialized on the main renderer
// thread, used and destructed on the media thread.
class PepperAudioEncoderHost::AudioEncoderImpl {
 public:
  // Callback used to signal encoded data. If |size| is negative, an error
  // occurred.
  using BitstreamBufferReadyCB = base::OnceCallback<void(int32_t size)>;

  AudioEncoderImpl();
  ~AudioEncoderImpl();

  // Used on the renderer thread.
  static std::vector<PP_AudioProfileDescription> GetSupportedProfiles();
  bool Initialize(const ppapi::proxy::PPB_AudioEncodeParameters& parameters);
  int32_t GetNumberOfSamplesPerFrame();

  // Used on the media thread.
  void Encode(uint8_t* input_data,
              size_t input_size,
              uint8_t* output_data,
              size_t output_size,
              BitstreamBufferReadyCB callback);
  void RequestBitrateChange(uint32_t bitrate);

 private:
  std::unique_ptr<uint8_t[]> encoder_memory_;
  OpusEncoder* opus_encoder_;

  // Initialization parameters, only valid if |encoder_memory_| is not
  // nullptr.
  ppapi::proxy::PPB_AudioEncodeParameters parameters_;

  DISALLOW_COPY_AND_ASSIGN(AudioEncoderImpl);
};

PepperAudioEncoderHost::AudioEncoderImpl::AudioEncoderImpl()
    : opus_encoder_(nullptr) {}

PepperAudioEncoderHost::AudioEncoderImpl::~AudioEncoderImpl() {}

// static
std::vector<PP_AudioProfileDescription>
PepperAudioEncoderHost::AudioEncoderImpl::GetSupportedProfiles() {
  std::vector<PP_AudioProfileDescription> profiles;
  static const uint32_t sampling_rates[] = {8000, 12000, 16000, 24000, 48000};

  for (uint32_t i = 0; i < arraysize(sampling_rates); ++i) {
    PP_AudioProfileDescription profile;
    profile.profile = PP_AUDIOPROFILE_OPUS;
    profile.max_channels = 2;
    profile.sample_size = PP_AUDIOBUFFER_SAMPLESIZE_16_BITS;
    profile.sample_rate = sampling_rates[i];
    profile.hardware_accelerated = PP_FALSE;
    profiles.push_back(profile);
  }
  return profiles;
}

bool PepperAudioEncoderHost::AudioEncoderImpl::Initialize(
    const ppapi::proxy::PPB_AudioEncodeParameters& parameters) {
  if (parameters.output_profile != PP_AUDIOPROFILE_OPUS)
    return false;

  DCHECK(!encoder_memory_);

  int32_t encoder_size = opus_encoder_get_size(parameters.channels);
  if (encoder_size < 1)
    return false;

  std::unique_ptr<uint8_t[]> encoder_memory(new uint8_t[encoder_size]);
  opus_encoder_ = reinterpret_cast<OpusEncoder*>(encoder_memory.get());

  if (opus_encoder_init(opus_encoder_, parameters.input_sample_rate,
                        parameters.channels, OPUS_APPLICATION_AUDIO) != OPUS_OK)
    return false;

  if (opus_encoder_ctl(opus_encoder_,
                       OPUS_SET_BITRATE(parameters.initial_bitrate <= 0
                                            ? OPUS_AUTO
                                            : parameters.initial_bitrate)) !=
      OPUS_OK)
    return false;

  encoder_memory_.swap(encoder_memory);
  parameters_ = parameters;

  return true;
}

int32_t PepperAudioEncoderHost::AudioEncoderImpl::GetNumberOfSamplesPerFrame() {
  DCHECK(encoder_memory_);
  // Opus supports 2.5, 5, 10, 20, 40 or 60ms audio frames. We take
  // 10ms by default.
  return parameters_.input_sample_rate / 100;
}

void PepperAudioEncoderHost::AudioEncoderImpl::Encode(
    uint8_t* input_data,
    size_t input_size,
    uint8_t* output_data,
    size_t output_size,
    BitstreamBufferReadyCB callback) {
  DCHECK(encoder_memory_);
  int32_t result = opus_encode(
      opus_encoder_, reinterpret_cast<opus_int16*>(input_data),
      (input_size / parameters_.channels) / parameters_.input_sample_size,
      output_data, output_size);
  std::move(callback).Run(result);
}

void PepperAudioEncoderHost::AudioEncoderImpl::RequestBitrateChange(
    uint32_t bitrate) {
  DCHECK(encoder_memory_);
  opus_encoder_ctl(opus_encoder_, OPUS_SET_BITRATE(bitrate));
}

PepperAudioEncoderHost::PepperAudioEncoderHost(RendererPpapiHost* host,
                                               PP_Instance instance,
                                               PP_Resource resource)
    : ResourceHost(host->GetPpapiHost(), instance, resource),
      renderer_ppapi_host_(host),
      initialized_(false),
      encoder_last_error_(PP_ERROR_FAILED),
      media_task_runner_(RenderThreadImpl::current()
                             ->GetMediaThreadTaskRunner()),
      weak_ptr_factory_(this) {}

PepperAudioEncoderHost::~PepperAudioEncoderHost() {
  Close();
}

int32_t PepperAudioEncoderHost::OnResourceMessageReceived(
    const IPC::Message& msg,
    ppapi::host::HostMessageContext* context) {
  PPAPI_BEGIN_MESSAGE_MAP(PepperAudioEncoderHost, msg)
    PPAPI_DISPATCH_HOST_RESOURCE_CALL_0(
        PpapiHostMsg_AudioEncoder_GetSupportedProfiles,
        OnHostMsgGetSupportedProfiles)
    PPAPI_DISPATCH_HOST_RESOURCE_CALL(PpapiHostMsg_AudioEncoder_Initialize,
                                      OnHostMsgInitialize)
    PPAPI_DISPATCH_HOST_RESOURCE_CALL(PpapiHostMsg_AudioEncoder_Encode,
                                      OnHostMsgEncode)
    PPAPI_DISPATCH_HOST_RESOURCE_CALL(
        PpapiHostMsg_AudioEncoder_RecycleBitstreamBuffer,
        OnHostMsgRecycleBitstreamBuffer)
    PPAPI_DISPATCH_HOST_RESOURCE_CALL(
        PpapiHostMsg_AudioEncoder_RequestBitrateChange,
        OnHostMsgRequestBitrateChange)
    PPAPI_DISPATCH_HOST_RESOURCE_CALL_0(PpapiHostMsg_AudioEncoder_Close,
                                        OnHostMsgClose)
  PPAPI_END_MESSAGE_MAP()
  return PP_ERROR_FAILED;
}

int32_t PepperAudioEncoderHost::OnHostMsgGetSupportedProfiles(
    ppapi::host::HostMessageContext* context) {
  std::vector<PP_AudioProfileDescription> profiles;
  GetSupportedProfiles(&profiles);

  host()->SendReply(
      context->MakeReplyMessageContext(),
      PpapiPluginMsg_AudioEncoder_GetSupportedProfilesReply(profiles));

  return PP_OK_COMPLETIONPENDING;
}

int32_t PepperAudioEncoderHost::OnHostMsgInitialize(
    ppapi::host::HostMessageContext* context,
    const ppapi::proxy::PPB_AudioEncodeParameters& parameters) {
  if (initialized_)
    return PP_ERROR_FAILED;

  if (!IsInitializationValid(parameters))
    return PP_ERROR_NOTSUPPORTED;

  std::unique_ptr<AudioEncoderImpl> encoder(new AudioEncoderImpl);
  if (!encoder->Initialize(parameters))
    return PP_ERROR_FAILED;
  if (!AllocateBuffers(parameters, encoder->GetNumberOfSamplesPerFrame()))
    return PP_ERROR_NOMEMORY;

  initialized_ = true;
  encoder_last_error_ = PP_OK;
  encoder_.swap(encoder);

  ppapi::host::ReplyMessageContext reply_context =
      context->MakeReplyMessageContext();
  reply_context.params.AppendHandle(
      SerializedHandle(renderer_ppapi_host_->ShareSharedMemoryHandleWithRemote(
                           audio_buffer_manager_->shm()->handle()),
                       audio_buffer_manager_->shm()->mapped_size()));
  reply_context.params.AppendHandle(
      SerializedHandle(renderer_ppapi_host_->ShareSharedMemoryHandleWithRemote(
                           bitstream_buffer_manager_->shm()->handle()),
                       bitstream_buffer_manager_->shm()->mapped_size()));
  host()->SendReply(reply_context,
                    PpapiPluginMsg_AudioEncoder_InitializeReply(
                        encoder_->GetNumberOfSamplesPerFrame(),
                        audio_buffer_manager_->number_of_buffers(),
                        audio_buffer_manager_->buffer_size(),
                        bitstream_buffer_manager_->number_of_buffers(),
                        bitstream_buffer_manager_->buffer_size()));

  return PP_OK_COMPLETIONPENDING;
}

int32_t PepperAudioEncoderHost::OnHostMsgEncode(
    ppapi::host::HostMessageContext* context,
    int32_t buffer_id) {
  if (encoder_last_error_)
    return encoder_last_error_;

  if (buffer_id < 0 || buffer_id >= audio_buffer_manager_->number_of_buffers())
    return PP_ERROR_FAILED;

  audio_buffer_manager_->EnqueueBuffer(buffer_id);

  DoEncode();

  return PP_OK_COMPLETIONPENDING;
}

int32_t PepperAudioEncoderHost::OnHostMsgRecycleBitstreamBuffer(
    ppapi::host::HostMessageContext* context,
    int32_t buffer_id) {
  if (encoder_last_error_)
    return encoder_last_error_;

  if (buffer_id < 0 ||
      buffer_id >= bitstream_buffer_manager_->number_of_buffers())
    return PP_ERROR_FAILED;

  bitstream_buffer_manager_->EnqueueBuffer(buffer_id);

  DoEncode();

  return PP_OK;
}

int32_t PepperAudioEncoderHost::OnHostMsgRequestBitrateChange(
    ppapi::host::HostMessageContext* context,
    uint32_t bitrate) {
  if (encoder_last_error_)
    return encoder_last_error_;

  media_task_runner_->PostTask(
      FROM_HERE, base::BindOnce(&AudioEncoderImpl::RequestBitrateChange,
                                base::Unretained(encoder_.get()), bitrate));

  return PP_OK;
}

int32_t PepperAudioEncoderHost::OnHostMsgClose(
    ppapi::host::HostMessageContext* context) {
  encoder_last_error_ = PP_ERROR_FAILED;
  Close();

  return PP_OK;
}

void PepperAudioEncoderHost::GetSupportedProfiles(
    std::vector<PP_AudioProfileDescription>* profiles) {
  DCHECK(RenderThreadImpl::current());

  *profiles = AudioEncoderImpl::GetSupportedProfiles();
}

bool PepperAudioEncoderHost::IsInitializationValid(
    const ppapi::proxy::PPB_AudioEncodeParameters& parameters) {
  DCHECK(RenderThreadImpl::current());

  std::vector<PP_AudioProfileDescription> profiles;
  GetSupportedProfiles(&profiles);

  for (const PP_AudioProfileDescription& profile : profiles) {
    if (parameters.output_profile == profile.profile &&
        parameters.input_sample_size == profile.sample_size &&
        parameters.input_sample_rate == profile.sample_rate &&
        parameters.channels <= profile.max_channels &&
        PP_HardwareAccelerationCompatibleAudio(
            profile.hardware_accelerated == PP_TRUE, parameters.acceleration))
      return true;
  }

  return false;
}

bool PepperAudioEncoderHost::AllocateBuffers(
    const ppapi::proxy::PPB_AudioEncodeParameters& parameters,
    int32_t samples_per_frame) {
  DCHECK(RenderThreadImpl::current());

  // Audio buffers size computation.
  base::CheckedNumeric<uint32_t> audio_buffer_size = samples_per_frame;
  audio_buffer_size *= parameters.channels;
  audio_buffer_size *= parameters.input_sample_size;

  base::CheckedNumeric<uint32_t> total_audio_buffer_size = audio_buffer_size;
  total_audio_buffer_size += sizeof(ppapi::MediaStreamBuffer::Audio);
  base::CheckedNumeric<size_t> total_audio_memory_size =
      total_audio_buffer_size;
  total_audio_memory_size *= kDefaultNumberOfAudioBuffers;

  // Bitstream buffers size computation (individual bitstream buffers are
  // twice the size of the raw data, to handle the worst case where
  // compression doesn't work).
  base::CheckedNumeric<uint32_t> bitstream_buffer_size = audio_buffer_size;
  bitstream_buffer_size *= 2;
  bitstream_buffer_size += sizeof(ppapi::MediaStreamBuffer::Bitstream);
  base::CheckedNumeric<size_t> total_bitstream_memory_size =
      bitstream_buffer_size;
  total_bitstream_memory_size *= kDefaultNumberOfAudioBuffers;

  if (!total_audio_memory_size.IsValid() ||
      !total_bitstream_memory_size.IsValid())
    return false;

  std::unique_ptr<base::SharedMemory> audio_memory(
      RenderThreadImpl::current()->HostAllocateSharedMemoryBuffer(
          total_audio_memory_size.ValueOrDie()));
  if (!audio_memory)
    return false;
  std::unique_ptr<ppapi::MediaStreamBufferManager> audio_buffer_manager(
      new ppapi::MediaStreamBufferManager(this));
  if (!audio_buffer_manager->SetBuffers(
          kDefaultNumberOfAudioBuffers,
          base::ValueOrDieForType<int32_t>(total_audio_buffer_size),
          std::move(audio_memory), false))
    return false;

  for (int32_t i = 0; i < audio_buffer_manager->number_of_buffers(); ++i) {
    ppapi::MediaStreamBuffer::Audio* buffer =
        &(audio_buffer_manager->GetBufferPointer(i)->audio);
    buffer->header.size = total_audio_buffer_size.ValueOrDie();
    buffer->header.type = ppapi::MediaStreamBuffer::TYPE_AUDIO;
    buffer->sample_rate =
        static_cast<PP_AudioBuffer_SampleRate>(parameters.input_sample_rate);
    buffer->number_of_channels = parameters.channels;
    buffer->number_of_samples = samples_per_frame;
    buffer->data_size = audio_buffer_size.ValueOrDie();
  }

  std::unique_ptr<base::SharedMemory> bitstream_memory(
      RenderThreadImpl::current()->HostAllocateSharedMemoryBuffer(
          total_bitstream_memory_size.ValueOrDie()));
  if (!bitstream_memory)
    return false;
  std::unique_ptr<ppapi::MediaStreamBufferManager> bitstream_buffer_manager(
      new ppapi::MediaStreamBufferManager(this));
  if (!bitstream_buffer_manager->SetBuffers(
          kDefaultNumberOfAudioBuffers,
          base::ValueOrDieForType<int32_t>(bitstream_buffer_size),
          std::move(bitstream_memory), true))
    return false;

  for (int32_t i = 0; i < bitstream_buffer_manager->number_of_buffers(); ++i) {
    ppapi::MediaStreamBuffer::Bitstream* buffer =
        &(bitstream_buffer_manager->GetBufferPointer(i)->bitstream);
    buffer->header.size = bitstream_buffer_size.ValueOrDie();
    buffer->header.type = ppapi::MediaStreamBuffer::TYPE_BITSTREAM;
  }

  audio_buffer_manager_.swap(audio_buffer_manager);
  bitstream_buffer_manager_.swap(bitstream_buffer_manager);

  return true;
}

void PepperAudioEncoderHost::DoEncode() {
  DCHECK(RenderThreadImpl::current());
  DCHECK(!encoder_last_error_);

  if (!audio_buffer_manager_->HasAvailableBuffer() ||
      !bitstream_buffer_manager_->HasAvailableBuffer())
    return;

  int32_t audio_buffer_id = audio_buffer_manager_->DequeueBuffer();
  int32_t bitstream_buffer_id = bitstream_buffer_manager_->DequeueBuffer();

  ppapi::MediaStreamBuffer* audio_buffer =
      audio_buffer_manager_->GetBufferPointer(audio_buffer_id);
  ppapi::MediaStreamBuffer* bitstream_buffer =
      bitstream_buffer_manager_->GetBufferPointer(bitstream_buffer_id);

  media_task_runner_->PostTask(
      FROM_HERE,
      base::BindOnce(&AudioEncoderImpl::Encode,
                     base::Unretained(encoder_.get()),
                     static_cast<uint8_t*>(audio_buffer->audio.data),
                     audio_buffer_manager_->buffer_size() -
                         sizeof(ppapi::MediaStreamBuffer::Audio),
                     static_cast<uint8_t*>(bitstream_buffer->bitstream.data),
                     bitstream_buffer_manager_->buffer_size() -
                         sizeof(ppapi::MediaStreamBuffer::Bitstream),
                     media::BindToCurrentLoop(base::BindOnce(
                         &PepperAudioEncoderHost::BitstreamBufferReady,
                         weak_ptr_factory_.GetWeakPtr(), audio_buffer_id,
                         bitstream_buffer_id))));
}

void PepperAudioEncoderHost::BitstreamBufferReady(int32_t audio_buffer_id,
                                                  int32_t bitstream_buffer_id,
                                                  int32_t result) {
  DCHECK(RenderThreadImpl::current());

  if (encoder_last_error_)
    return;

  if (result < 0) {
    NotifyPepperError(PP_ERROR_FAILED);
    return;
  }

  host()->SendUnsolicitedReply(
      pp_resource(), PpapiPluginMsg_AudioEncoder_EncodeReply(audio_buffer_id));

  ppapi::MediaStreamBuffer::Bitstream* buffer =
      &(bitstream_buffer_manager_->GetBufferPointer(bitstream_buffer_id)
            ->bitstream);
  buffer->data_size = static_cast<uint32_t>(result);

  host()->SendUnsolicitedReply(
      pp_resource(),
      PpapiPluginMsg_AudioEncoder_BitstreamBufferReady(bitstream_buffer_id));
}

void PepperAudioEncoderHost::NotifyPepperError(int32_t error) {
  DCHECK(RenderThreadImpl::current());

  encoder_last_error_ = error;
  Close();
  host()->SendUnsolicitedReply(
      pp_resource(),
      PpapiPluginMsg_AudioEncoder_NotifyError(encoder_last_error_));
}

void PepperAudioEncoderHost::Close() {
  DCHECK(RenderThreadImpl::current());

  // Destroy the encoder and the audio/bitstream buffers on the media thread
  // to avoid freeing memory the encoder might still be working on.
  media_task_runner_->PostTask(
      FROM_HERE,
      base::BindOnce(&StopAudioEncoder, base::Passed(std::move(encoder_)),
                     base::Passed(std::move(audio_buffer_manager_)),
                     base::Passed(std::move(bitstream_buffer_manager_))));
}

// static
void PepperAudioEncoderHost::StopAudioEncoder(
    std::unique_ptr<AudioEncoderImpl> encoder,
    std::unique_ptr<ppapi::MediaStreamBufferManager> audio_buffer_manager,
    std::unique_ptr<ppapi::MediaStreamBufferManager> bitstream_buffer_manager) {
}

}  // namespace content