/*
 *  Copyright (C) 2010-2018 Team Kodi
 *  This file is part of Kodi - https://kodi.tv
 *
 *  SPDX-License-Identifier: GPL-2.0-or-later
 *  See LICENSES/README.md for more information.
 */

#pragma once

#include "cores/AudioEngine/Utils/AEAudioFormat.h"

#include <cassert>
#include <list>
#include <map>
#include <memory>
#include <utility>
#include <vector>

extern "C" {
#include <libavutil/samplefmt.h>
}

typedef std::pair<std::string, std::string> AEDevice;
typedef std::vector<AEDevice> AEDeviceList;

/* forward declarations */
class IAEStream;
class IAEStreamDeleter;
class IAESound;
class IAESoundDeleter;
class IAEPacketizer;
class IAudioCallback;
class IAEClockCallback;
class CAEStreamInfo;

namespace ADDON
{
struct Interface_AudioEngine;
}

/* sound options */
#define AE_SOUND_OFF 0 /*! disable sounds */
#define AE_SOUND_IDLE 1 /*! only play sounds while no streams are running */
#define AE_SOUND_ALWAYS 2 /*! always play sounds */

/* config options */
#define AE_CONFIG_FIXED 1
#define AE_CONFIG_AUTO  2
#define AE_CONFIG_MATCH 3

enum AEQuality
{
  AE_QUALITY_UNKNOWN = -1, /*! Unset, unknown or incorrect quality level */
  AE_QUALITY_DEFAULT = 0, /*! Engine's default quality level */

  /* Basic quality levels */
  AE_QUALITY_LOW = 20, /*! Low quality level */
  AE_QUALITY_MID = 30, /*! Standard quality level */
  AE_QUALITY_HIGH = 50, /*! Best sound processing quality */

  /* Optional quality levels */
  AE_QUALITY_REALLYHIGH = 100, /*! Uncompromised optional quality level,
                               usually with unmeasurable and unnoticeable improvement */
  AE_QUALITY_GPU = 101, /*! GPU acceleration */
};

struct SampleConfig
{
  AVSampleFormat fmt;
  uint64_t channel_layout;
  int channels;
  int sample_rate;
  int bits_per_sample;
  int dither_bits;
};

/*!
 * \brief IAE Interface
 */
class IAE
{
protected:

  IAE() = default;
  virtual ~IAE() = default;

  /*!
   * \brief Initializes the AudioEngine, called by CFactory when it is time to initialize the audio engine.
   *
   * Do not call this directly, CApplication will call this when it is ready
   */
  virtual void Start() = 0;
public:
  using StreamPtr = std::unique_ptr<IAEStream, IAEStreamDeleter>;
  using SoundPtr = std::unique_ptr<IAESound, IAESoundDeleter>;

  /*!
   * \brief Called when the application needs to terminate the engine
   */
  virtual void Shutdown() { }

  /*!
   * \brief Suspends output and de-initializes sink
   *
   * Used to avoid conflicts with external players or to reduce power consumption
   *
   * \return True if successful
   */
  virtual bool Suspend() = 0;

  /*!
   * \brief Resumes output and re-initializes sink
   *
   * Used to resume output from Suspend() state above
   *
   * \return True if successful
   */
  virtual bool Resume() = 0;

  /*!
   * \brief Get the current Suspend() state
   *
   * Used by players to determine if audio is being processed
   * Default is true so players drop audio or pause if engine unloaded
   *
   * \return True if processing suspended
   */
  virtual bool IsSuspended() {return true;}

  /*!
   * \brief Returns the current master volume level of the AudioEngine
   *
   * \return The volume level between 0.0 and 1.0
   */
  virtual float GetVolume() = 0;

  /*!
   * \brief Sets the master volume level of the AudioEngine
   *
   * \param volume The new volume level between 0.0 and 1.0
   */
  virtual void SetVolume(const float volume) = 0;

  /*!
   * \brief Set the mute state (does not affect volume level value)
   *
   * \param enabled The mute state
   */
  virtual void SetMute(const bool enabled) = 0;

  /*!
   * \brief Get the current mute state
   *
   * \return The current mute state
   */
  virtual bool IsMuted() = 0;

  /*!
   * \brief Creates and returns a new IAEStream in the format specified, this function should never fail
   *
   * The cleanup behaviour can be modified with the IAEStreamDeleter::setFinish method.
   * Per default the behaviour is the same as calling FreeStream with true.
   *
   * \param audioFormat
   * \param options A bit field of stream options (see: enum AEStreamOptions)
   * \return a new IAEStream that will accept data in the requested format
   */
  virtual StreamPtr MakeStream(AEAudioFormat& audioFormat,
                               unsigned int options = 0,
                               IAEClockCallback* clock = NULL) = 0;

  /*!
   * \brief Creates a new IAESound that is ready to play the specified file
   *
   * \param file The WAV file to load, this supports XBMC's VFS
   * \return A new IAESound if the file could be loaded, otherwise NULL
   */
  virtual SoundPtr MakeSound(const std::string& file) = 0;

  /*!
   * \brief Enumerate the supported audio output devices
   *
   * \param devices The device list to append supported devices to
   * \param passthrough True if only passthrough devices are wanted
   */
  virtual void EnumerateOutputDevices(AEDeviceList &devices, bool passthrough) = 0;

  /*!
   * \brief Returns true if the AudioEngine supports AE_FMT_RAW streams for use with formats such as IEC61937
   *
   * \see CAEPackIEC61937::CAEPackIEC61937()
   *
   * \returns true if the AudioEngine is capable of RAW output
   */
  virtual bool SupportsRaw(AEAudioFormat &format) { return false; }

  /*!
   * \brief Returns true if the AudioEngine supports drain mode which is not streaming silence when idle
   *
   * \returns true if the AudioEngine is capable of drain mode
   */
  virtual bool SupportsSilenceTimeout() { return false; }

  /*!
   * \brief Returns true if the AudioEngine is currently configured to extract the DTS Core from DTS-HD streams
   *
   * \returns true if the AudioEngine is currently configured to extract the DTS Core from DTS-HD streams
   */
  virtual bool UsesDtsCoreFallback() { return false; }

  /*!
   * \brief Returns true if the AudioEngine is currently configured for stereo audio
   *
   * \returns true if the AudioEngine is currently configured for stereo audio
   */
  virtual bool HasStereoAudioChannelCount() { return false; }

  /*!
   * \brief Returns true if the AudioEngine is currently configured for HD audio (more than 5.1)
   *
   * \returns true if the AudioEngine is currently configured for HD audio (more than 5.1)
   */
  virtual bool HasHDAudioChannelCount() { return true; }

  virtual void RegisterAudioCallback(IAudioCallback* pCallback) {}

  virtual void UnregisterAudioCallback(IAudioCallback* pCallback) {}

  /*!
   * \brief Returns true if AudioEngine supports specified quality level
   *
   * \return true if specified quality level is supported, otherwise false
   */
  virtual bool SupportsQualityLevel(enum AEQuality level) { return false; }

  /*!
   * \brief AE decides whether this settings should be displayed
   *
   * \return true if AudioEngine wants to display this setting
   */
  virtual bool IsSettingVisible(const std::string &settingId) {return false; }

  /*!
   * \brief Instruct AE to keep configuration for a specified time
   *
   * \param millis time for which old configuration should be kept
   */
  virtual void KeepConfiguration(unsigned int millis) {}

  /*!
   * \brief Instruct AE to re-initialize, e.g. after ELD change event
   */
  virtual void DeviceChange() {}

  /*!
   * \brief Instruct AE to re-initialize, e.g. after ELD change event
   */
  virtual void DeviceCountChange(const std::string& driver) {}

  /*!
   * \brief Get the current sink data format
   *
   * \param Current sink data format. For more details see AEAudioFormat.
   * \return Returns true on success, else false.
   */
  virtual bool GetCurrentSinkFormat(AEAudioFormat &SinkFormat) { return false; }

private:
  friend class IAEStreamDeleter;
  friend class IAESoundDeleter;
  friend struct ADDON::Interface_AudioEngine;

  /*!
   * \brief This method will remove the specified stream from the engine.
   *
   * For OSX/IOS this is essential to reconfigure the audio output.
   *
   * \param stream The stream to be altered
   * \param finish if true AE will switch back to gui sound mode (if this is last stream)
   * \return true on success, else false.
   */
  virtual bool FreeStream(IAEStream* stream, bool finish) = 0;

  /*!
   * \brief Free the supplied IAESound object
   *
   * \param sound The IAESound object to free
   */
  virtual void FreeSound(IAESound* sound) = 0;
};

class IAEStreamDeleter
{
private:
  IAE* m_iae;
  bool m_finish;

public:
  IAEStreamDeleter() : m_iae(nullptr), m_finish(true) {}
  explicit IAEStreamDeleter(IAE& iae) : m_iae(&iae), m_finish{true} {}
  void setFinish(bool finish) { m_finish = finish; }
  void operator()(IAEStream* stream)
  {
    assert(m_iae);
    m_iae->FreeStream(stream, m_finish);
  }
};

class IAESoundDeleter
{
private:
  IAE* m_iae;

public:
  IAESoundDeleter() : m_iae(nullptr) {}
  explicit IAESoundDeleter(IAE& iae) : m_iae(&iae) {}
  void operator()(IAESound* sound)
  {
    assert(m_iae);
    m_iae->FreeSound(sound);
  }
};