/***

  Olive - Non-Linear Video Editor
  Copyright (C) 2021 Olive Team

  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  This program 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program.  If not, see <http://www.gnu.org/licenses/>.

***/

#include "preferencesaudiotab.h"

#include <QGridLayout>
#include <QGroupBox>
#include <QLabel>

#include "audio/audiomanager.h"
#include "config/config.h"

namespace olive {

PreferencesAudioTab::PreferencesAudioTab()
{
  QVBoxLayout* audio_tab_layout = new QVBoxLayout(this);

  {
    // Backend Layout
    QGridLayout* main_layout = new QGridLayout();
    main_layout->setMargin(0);

    int row = 0;

    main_layout->addWidget(new QLabel(tr("Backend:")), row, 0);

    audio_backend_combobox_ = new QComboBox();
    connect(audio_backend_combobox_, static_cast<void(QComboBox::*)(int)>(&QComboBox::currentIndexChanged), this, &PreferencesAudioTab::RefreshDevices);
    main_layout->addWidget(audio_backend_combobox_, row, 1);

    audio_tab_layout->addLayout(main_layout);
  }

  {
    QGroupBox* groupbox = new QGroupBox();
    audio_tab_layout->addWidget(groupbox);

    QVBoxLayout* layout = new QVBoxLayout(groupbox);

    int row = 0;

    {
      // Output Group
      QGroupBox* output_group = new QGroupBox();
      output_group->setTitle(tr("Output"));
      layout->addWidget(output_group);

      QGridLayout* output_layout = new QGridLayout(output_group);

      output_layout->addWidget(new QLabel(tr("Device:")), row, 0);

      audio_output_devices_ = new QComboBox();
      output_layout->addWidget(audio_output_devices_, row, 1);
    }

    row = 0;

    {
      QGroupBox* input_group = new QGroupBox();
      input_group->setTitle(tr("Input"));
      layout->addWidget(input_group);

      QGridLayout* input_layout = new QGridLayout(input_group);

      input_layout->addWidget(new QLabel(tr("Device:")), row, 0);

      audio_input_devices_ = new QComboBox();
      input_layout->addWidget(audio_input_devices_, row, 1);

      row++;

      input_layout->addWidget(new QLabel(tr("Recording Mode:"), this), row, 0);

      recording_combobox_ = new QComboBox();
      recording_combobox_->addItem(tr("Mono"));
      recording_combobox_->addItem(tr("Stereo"));
      input_layout->addWidget(recording_combobox_, row, 1);
    }

    QHBoxLayout* refresh_layout = new QHBoxLayout();
    layout->addLayout(refresh_layout);
    refresh_layout->addStretch();

    refresh_devices_btn_ = new QPushButton(tr("Refresh Devices"));
    refresh_layout->addWidget(refresh_devices_btn_);

    connect(refresh_devices_btn_, &QPushButton::clicked, this, &PreferencesAudioTab::HardRefreshBackends);
  }

  audio_tab_layout->addStretch();

  // Populate lists
  RefreshBackends();
}

void PreferencesAudioTab::Accept(MultiUndoCommand *command)
{
  Q_UNUSED(command)

  // Get device indexes
  PaDeviceIndex output_device = audio_output_devices_->currentData().value<PaDeviceIndex>();
  PaDeviceIndex input_device = audio_input_devices_->currentData().value<PaDeviceIndex>();

  // Get device names, which seem to be the closest thing we have to a "unique identifier" for them
  Config::Current()[QStringLiteral("AudioOutput")] = audio_output_devices_->currentText();
  Config::Current()[QStringLiteral("AudioInput")] = audio_input_devices_->currentText();

  // Set devices to be used from now on
  AudioManager::instance()->SetOutputDevice(output_device);
  AudioManager::instance()->SetInputDevice(input_device);
}

void PreferencesAudioTab::RefreshBackends()
{
  audio_backend_combobox_->clear();
  for (PaHostApiIndex i=0, end=Pa_GetHostApiCount(); i<end; i++) {
    const PaHostApiInfo *info = Pa_GetHostApiInfo(i);

    audio_backend_combobox_->addItem(info->name);
  }

  RefreshDevices();

  AttemptToSetDevicesFromConfig();
}

void PreferencesAudioTab::RefreshDevices()
{
  if (audio_backend_combobox_->count() == 0) {
    return;
  }

  PaHostApiIndex host_index = audio_backend_combobox_->currentIndex();
  const PaHostApiInfo *host = Pa_GetHostApiInfo(host_index);

  audio_output_devices_->clear();
  audio_input_devices_->clear();

  for (int i=0; i<host->deviceCount; i++) {
    PaDeviceIndex device_index = Pa_HostApiDeviceIndexToDeviceIndex(host_index, i);
    const PaDeviceInfo *device = Pa_GetDeviceInfo(device_index);

    if (device->maxOutputChannels) {
      audio_output_devices_->addItem(device->name, device_index);
    }

    if (device->maxInputChannels) {
      audio_input_devices_->addItem(device->name, device_index);
    }
  }
}

void PreferencesAudioTab::HardRefreshBackends()
{
  AudioManager::instance()->HardReset();
  RefreshBackends();
}

void PreferencesAudioTab::AttemptToSetDevicesFromConfig()
{
  // Load with currently active devices
  PaDeviceIndex current_output_index = AudioManager::instance()->GetOutputDevice();
  PaDeviceIndex current_input_index = AudioManager::instance()->GetInputDevice();

  const PaDeviceInfo *current_output = nullptr, *current_input = nullptr;
  if (current_output_index != paNoDevice) {
    current_output = Pa_GetDeviceInfo(current_output_index);
  }
  if (current_input_index != paNoDevice) {
    current_input = Pa_GetDeviceInfo(current_input_index);
  }

  if (current_output || current_input) {
    PaHostApiIndex host = current_output ? current_output->hostApi : current_input->hostApi;

    // Set backend accordingly
    audio_backend_combobox_->setCurrentIndex(host);

    // Device comboboxes should be populated correctly now
    if (current_output) {
      audio_output_devices_->setCurrentText(current_output->name);
    }

    if (current_input) {
      audio_input_devices_->setCurrentText(current_input->name);
    }
  }
}

}