/*----------------------------------------------------------------------------*/
/* Copyright (c) 2018-2020 FIRST. All Rights Reserved.                        */
/* Open Source Software - may be modified and shared by FRC teams. The code   */
/* must be accompanied by the FIRST BSD license file in the root directory of */
/* the project.                                                               */
/*----------------------------------------------------------------------------*/

#pragma once

#include <memory>
#include <utility>

#include <hal/simulation/PDPData.h>

#include "CallbackStore.h"

namespace frc {
namespace sim {
class PDPSim {
 public:
  explicit PDPSim(int index) { m_index = index; }

  std::unique_ptr<CallbackStore> RegisterInitializedCallback(
      NotifyCallback callback, bool initialNotify) {
    auto store = std::make_unique<CallbackStore>(
        m_index, -1, callback, &HALSIM_CancelPDPInitializedCallback);
    store->SetUid(HALSIM_RegisterPDPInitializedCallback(
        m_index, &CallbackStoreThunk, store.get(), initialNotify));
    return store;
  }

  bool GetInitialized() const { return HALSIM_GetPDPInitialized(m_index); }

  void SetInitialized(bool initialized) {
    HALSIM_SetPDPInitialized(m_index, initialized);
  }

  std::unique_ptr<CallbackStore> RegisterTemperatureCallback(
      NotifyCallback callback, bool initialNotify) {
    auto store = std::make_unique<CallbackStore>(
        m_index, -1, callback, &HALSIM_CancelPDPTemperatureCallback);
    store->SetUid(HALSIM_RegisterPDPTemperatureCallback(
        m_index, &CallbackStoreThunk, store.get(), initialNotify));
    return store;
  }

  double GetTemperature() const { return HALSIM_GetPDPTemperature(m_index); }

  void SetTemperature(double temperature) {
    HALSIM_SetPDPTemperature(m_index, temperature);
  }

  std::unique_ptr<CallbackStore> RegisterVoltageCallback(
      NotifyCallback callback, bool initialNotify) {
    auto store = std::make_unique<CallbackStore>(
        m_index, -1, callback, &HALSIM_CancelPDPVoltageCallback);
    store->SetUid(HALSIM_RegisterPDPVoltageCallback(
        m_index, &CallbackStoreThunk, store.get(), initialNotify));
    return store;
  }

  double GetVoltage() const { return HALSIM_GetPDPVoltage(m_index); }

  void SetVoltage(double voltage) { HALSIM_SetPDPVoltage(m_index, voltage); }

  std::unique_ptr<CallbackStore> RegisterCurrentCallback(
      int channel, NotifyCallback callback, bool initialNotify) {
    auto store = std::make_unique<CallbackStore>(
        m_index, channel, -1, callback, &HALSIM_CancelPDPCurrentCallback);
    store->SetUid(HALSIM_RegisterPDPCurrentCallback(
        m_index, channel, &CallbackStoreThunk, store.get(), initialNotify));
    return store;
  }

  double GetCurrent(int channel) const {
    return HALSIM_GetPDPCurrent(m_index, channel);
  }

  void SetCurrent(int channel, double current) {
    HALSIM_SetPDPCurrent(m_index, channel, current);
  }

  void GetAllCurrents(double* currents) {
    HALSIM_GetPDPAllCurrents(m_index, currents);
  }

  void SetAllCurrents(const double* currents) {
    HALSIM_SetPDPAllCurrents(m_index, currents);
  }

  void ResetData() { HALSIM_ResetPDPData(m_index); }

 private:
  int m_index;
};
}  // namespace sim
}  // namespace frc
