// Copyright 2018 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 SRC_GRAPHICS_DISPLAY_DRIVERS_VIM_DISPLAY_VIM_AUDIO_UTILS_H_
#define SRC_GRAPHICS_DISPLAY_DRIVERS_VIM_DISPLAY_VIM_AUDIO_UTILS_H_

#include <lib/device-protocol/platform-device.h>
#include <lib/mmio/mmio.h>
#include <lib/zx/vmo.h>
#include <zircon/compiler.h>
#include <zircon/types.h>

#include <utility>

#include <ddk/protocol/platform/device.h>
#include <fbl/macros.h>
#include <fbl/ref_counted.h>
#include <fbl/ref_ptr.h>

namespace audio {
namespace vim2 {

class Registers : public fbl::RefCounted<Registers>, public ddk::MmioBuffer {
 public:
  static fbl::RefPtr<Registers> Create(const pdev_protocol_t* pdev, uint32_t which_mmio,
                                       zx_status_t* out_res);
  DISALLOW_COPY_ASSIGN_AND_MOVE(Registers);

 private:
  friend class fbl::RefPtr<Registers>;

  Registers(mmio_buffer_t& mmio) : ddk::MmioBuffer(mmio) {}
  ~Registers() = default;
};

class RefCountedVmo : public fbl::RefCounted<RefCountedVmo> {
 public:
  static fbl::RefPtr<RefCountedVmo> Create(zx::vmo vmo);
  DISALLOW_COPY_ASSIGN_AND_MOVE(RefCountedVmo);

  const zx::vmo& vmo() { return vmo_; }

 private:
  friend class fbl::RefPtr<RefCountedVmo>;

  explicit RefCountedVmo(zx::vmo vmo) : vmo_(std::move(vmo)) {}
  ~RefCountedVmo() = default;

  const zx::vmo vmo_;
};

}  // namespace vim2
}  // namespace audio

#endif  // SRC_GRAPHICS_DISPLAY_DRIVERS_VIM_DISPLAY_VIM_AUDIO_UTILS_H_