// Copyright 2020 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_DEVICES_THERMAL_DRIVERS_ASTRO_THERMISTOR_THERMISTOR_CHANNEL_H_
#define SRC_DEVICES_THERMAL_DRIVERS_ASTRO_THERMISTOR_THERMISTOR_CHANNEL_H_

#include <fuchsia/hardware/adc/llcpp/fidl.h>
#include <fuchsia/hardware/temperature/llcpp/fidl.h>
#include <lib/mmio/mmio.h>
#include <lib/thermal/ntc.h>
#include <lib/zx/interrupt.h>

#include <ddk/device.h>
#include <ddktl/device.h>
#include <ddktl/protocol/empty-protocol.h>
#include <fbl/mutex.h>
#include <fbl/ref_ptr.h>
#include <soc/aml-common/aml-g12-saradc.h>

namespace thermal {

class ThermistorChannel;
using DeviceType2 = ddk::Device<ThermistorChannel, ddk::Unbindable, ddk::Messageable>;

namespace FidlTemperature = llcpp::fuchsia::hardware::temperature;

class ThermistorChannel : public DeviceType2,
                          public ddk::EmptyProtocol<ZX_PROTOCOL_TEMPERATURE>,
                          public FidlTemperature::Device::Interface {
 public:
  ThermistorChannel(zx_device_t* device, fbl::RefPtr<AmlSaradcDevice> adc, uint32_t ch,
                    NtcInfo ntc_info, uint32_t pullup_ohms)
      : DeviceType2(device), adc_(adc), adc_channel_(ch), ntc_(ntc_info, pullup_ohms) {}

  void GetTemperatureCelsius(GetTemperatureCelsiusCompleter::Sync& completer) override;
  zx_status_t DdkMessage(fidl_incoming_msg_t* msg, fidl_txn_t* txn);
  void DdkRelease() { delete this; }
  void DdkUnbind(ddk::UnbindTxn txn) { txn.Reply(); }

 private:
  const fbl::RefPtr<AmlSaradcDevice> adc_;
  const uint32_t adc_channel_;
  const Ntc ntc_;
};

class RawChannel;
using DeviceType3 = ddk::Device<RawChannel, ddk::Unbindable, ddk::Messageable>;

namespace FidlAdc = llcpp::fuchsia::hardware::adc;

class RawChannel : public DeviceType3,
                   public ddk::EmptyProtocol<ZX_PROTOCOL_ADC>,
                   public FidlAdc::Device::Interface {
 public:
  RawChannel(zx_device_t* device, fbl::RefPtr<AmlSaradcDevice> adc, uint32_t ch)
      : DeviceType3(device), adc_(adc), adc_channel_(ch) {}

  void GetSample(GetSampleCompleter::Sync& completer) override;
  void GetNormalizedSample(GetNormalizedSampleCompleter::Sync& completer) override;
  void GetResolution(GetResolutionCompleter::Sync& completer) override;
  zx_status_t DdkMessage(fidl_incoming_msg_t* msg, fidl_txn_t* txn);
  void DdkRelease() { delete this; }
  void DdkUnbind(ddk::UnbindTxn txn) { txn.Reply(); }

 private:
  const fbl::RefPtr<AmlSaradcDevice> adc_;
  const uint32_t adc_channel_;
};

}  // namespace thermal

#endif  // SRC_DEVICES_THERMAL_DRIVERS_ASTRO_THERMISTOR_THERMISTOR_CHANNEL_H_