// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef ZIRCON_SYSTEM_UAPP_IHDA_INTEL_HDA_CODEC_H_
#define ZIRCON_SYSTEM_UAPP_IHDA_INTEL_HDA_CODEC_H_

#include <memory>

#include <fbl/intrusive_wavl_tree.h>
#include <fbl/string.h>
#include <intel-hda/utils/codec-commands.h>

#include "codec_state.h"
#include "intel_hda_device.h"

namespace audio {
namespace intel_hda {

struct CodecVerb;
struct CodecResponse;

class IntelHDACodec : public fbl::WAVLTreeContainable<std::unique_ptr<IntelHDACodec>> {
 public:
  using CodecTree = fbl::WAVLTree<uint32_t, std::unique_ptr<IntelHDACodec>>;

  template <typename TARGET_TYPE>
  struct CommandListEntry {
    CodecVerb verb;
    zx_status_t (*parser)(TARGET_TYPE& target, const CodecResponse& resp);
  };

  zx_status_t DumpCodec(int argc, const char** argv);

  uint32_t id() const { return codec_id_; }
  uint32_t GetKey() const { return id(); }
  fbl::String dev_name() const { return dev_name_; }

  static zx_status_t Enumerate();
  static CodecTree& codecs() { return codecs_; }

  zx_status_t Probe(IntelHDADevice* result);

  void Disconnect() { device_.Disconnect(); }

 private:
  friend class std::default_delete<IntelHDACodec>;

  zx_status_t DoCodecCmd(uint16_t nid, const CodecVerb& verb, CodecResponse* resp_out);
  zx_status_t ReadCodecState();
  zx_status_t ReadFunctionGroupState(FunctionGroupStatePtr& ptr, uint16_t nid);
  zx_status_t ReadAudioFunctionGroupState(AudioFunctionGroupState& afg);
  zx_status_t ReadAudioWidgetState(AudioWidgetState& widget);
  zx_status_t ReadConnList(AudioWidgetState& widget);
  zx_status_t ReadAmpState(uint16_t nid, bool is_input, uint8_t ndx, const AmpCaps& caps,
                           AudioWidgetState::AmpState* state_out);

  template <typename T>
  zx_status_t RunCommandList(T& target, uint16_t nid, const CommandListEntry<T>* cmds,
                             size_t cmd_count);

  IntelHDACodec(uint32_t codec_id, const char* const dev_name)
      : device_(dev_name, ZirconDevice::Type::Codec), codec_id_(codec_id), dev_name_(dev_name) {}

  ~IntelHDACodec() {}

  ZirconDevice device_;

  const uint32_t codec_id_;
  CodecState codec_state_;
  const fbl::String dev_name_;

  static CodecTree codecs_;
};

}  // namespace intel_hda
}  // namespace audio

#endif  // ZIRCON_SYSTEM_UAPP_IHDA_INTEL_HDA_CODEC_H_