PowerDistribution.h

#pragma once
#include <frc/PowerDistribution.h>
#include <units/current.h>
#include <units/energy.h>
#include <units/power.h>
#include <units/temperature.h>
#include <units/voltage.h>
 
namespace valor {

class PowerDistribution : public frc::PowerDistribution {
   public:
    enum class ModuleType {
        kCTRE_PDP,
        kCTRE_PDP2_0,
        kRev
    };

    PowerDistribution();

    inline PowerDistribution(int module, ModuleType moduleType)
          : frc::PowerDistribution{module, ConvertModuleType(moduleType)}, type{moduleType} {}

    int GetNumChannels() const;

    units::volt_t GetVoltage() const;

    inline units::celsius_t GetTemperature() const { return units::celsius_t{frc::PowerDistribution::GetTemperature()}; }

    inline units::ampere_t GetCurrent(int channel) const { return units::ampere_t{frc::PowerDistribution::GetCurrent(channel)}; }

    std::vector<units::ampere_t> GetAllCurrents() const;

    inline units::ampere_t GetTotalCurrent() const { return units::ampere_t{frc::PowerDistribution::GetTotalCurrent()}; }

    inline units::watt_t GetTotalPower() const { return units::watt_t{frc::PowerDistribution::GetTotalPower()}; }

    inline units::joule_t GetTotalEnergy() const { return units::joule_t{frc::PowerDistribution::GetTotalEnergy()}; }

    constexpr ModuleType GetType() const { return type; }
 
    void InitSendable(wpi::SendableBuilder& builder) override;
 
   private:
    static constexpr frc::PowerDistribution::ModuleType ConvertModuleType(ModuleType type) {
        if (type == ModuleType::kRev)
            return frc::PowerDistribution::ModuleType::kRev;
        return frc::PowerDistribution::ModuleType::kCTRE;
    }
 
    ModuleType type;
};
 
}  // namespace valor