/*
 * Copyright (c) 2018-2020 Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#pragma once
#include <vapours.hpp>
#include <stratosphere/i2c/i2c_types.hpp>

namespace ams::i2c {

    enum I2cBus {
        I2cBus_I2c1 = 0,
        I2cBus_I2c2 = 1,
        I2cBus_I2c3 = 2,
        I2cBus_I2c4 = 3,
        I2cBus_I2c5 = 4,
        I2cBus_I2c6 = 5,
    };

    constexpr inline const DeviceCode DeviceCode_I2c1 = 0x02000001;
    constexpr inline const DeviceCode DeviceCode_I2c2 = 0x02000002;
    constexpr inline const DeviceCode DeviceCode_I2c3 = 0x02000003;
    constexpr inline const DeviceCode DeviceCode_I2c4 = 0x02000004;
    constexpr inline const DeviceCode DeviceCode_I2c5 = 0x02000005;
    constexpr inline const DeviceCode DeviceCode_I2c6 = 0x02000006;

    constexpr inline DeviceCode ConvertToDeviceCode(I2cBus bus) {
        switch (bus) {
            case I2cBus_I2c1: return DeviceCode_I2c1;
            case I2cBus_I2c2: return DeviceCode_I2c2;
            case I2cBus_I2c3: return DeviceCode_I2c3;
            case I2cBus_I2c4: return DeviceCode_I2c4;
            case I2cBus_I2c5: return DeviceCode_I2c5;
            case I2cBus_I2c6: return DeviceCode_I2c6;
            AMS_UNREACHABLE_DEFAULT_CASE();
        }
    }

    constexpr inline DeviceCode ConvertToI2cBus(DeviceCode dc) {
        switch (dc.GetInternalValue()) {
            case DeviceCode_I2c1.GetInternalValue(): return I2cBus_I2c1;
            case DeviceCode_I2c2.GetInternalValue(): return I2cBus_I2c2;
            case DeviceCode_I2c3.GetInternalValue(): return I2cBus_I2c3;
            case DeviceCode_I2c4.GetInternalValue(): return I2cBus_I2c4;
            case DeviceCode_I2c5.GetInternalValue(): return I2cBus_I2c5;
            case DeviceCode_I2c6.GetInternalValue(): return I2cBus_I2c6;
            AMS_UNREACHABLE_DEFAULT_CASE();
        }
    }

    enum I2cDevice : u32 {
        I2cDevice_ClassicController    =  0,
        I2cDevice_Ftm3bd56             =  1,
        I2cDevice_Tmp451               =  2,
        I2cDevice_Nct72                =  3,
        I2cDevice_Alc5639              =  4,
        I2cDevice_Max77620Rtc          =  5,
        I2cDevice_Max77620Pmic         =  6,
        I2cDevice_Max77621Cpu          =  7,
        I2cDevice_Max77621Gpu          =  8,
        I2cDevice_Bq24193              =  9,
        I2cDevice_Max17050             = 10,
        I2cDevice_Bm92t30mwv           = 11,
        I2cDevice_Ina226Vdd15v0Hb      = 12,

        I2cDevice_Ina226VsysCpuDs      = 13,
        I2cDevice_Ina226VddCpuAp       = 13,

        I2cDevice_Ina226VsysGpuDs      = 14,
        I2cDevice_Ina226VddGpuAp       = 14,

        I2cDevice_Ina226VsysDdrDs      = 15,
        I2cDevice_Ina226VddDdr1V1Pmic  = 15,

        I2cDevice_Ina226VsysAp         = 16,
        I2cDevice_Ina226VsysBlDs       = 17,
        I2cDevice_Bh1730               = 18,

        I2cDevice_Ina226VsysCore       = 19,
        I2cDevice_Ina226VddCoreAp      = 19,

        I2cDevice_Ina226Soc1V8         = 20,
        I2cDevice_Ina226VddSoc1V8      = 20,

        I2cDevice_Ina226Lpddr1V8       = 21,
        I2cDevice_Ina226Vdd1V8         = 21,

        I2cDevice_Ina226Reg1V32        = 22,
        I2cDevice_Ina226Vdd3V3Sys      = 23,
        I2cDevice_HdmiDdc              = 24,
        I2cDevice_HdmiScdc             = 25,
        I2cDevice_HdmiHdcp             = 26,
        I2cDevice_Fan53528             = 27,
        I2cDevice_Max77812_3           = 28,
        I2cDevice_Max77812_2           = 29,
        I2cDevice_Ina226VddDdr0V6      = 30,
        I2cDevice_HoagNfcIc            = 31, /* TODO */
        I2cDevice_PmicUnknownAula_4_18 = 32, /* TODO */
    };

    /* TODO: Better place for this? */
    constexpr inline const DeviceCode DeviceCode_ClassicController    = 0x350000C9;
    constexpr inline const DeviceCode DeviceCode_Ftm3bd56             = 0x35000033;
    constexpr inline const DeviceCode DeviceCode_Tmp451               = 0x3E000001;
    constexpr inline const DeviceCode DeviceCode_Nct72                = 0x3E000001;
    constexpr inline const DeviceCode DeviceCode_Alc5639              = 0x33000001;
    constexpr inline const DeviceCode DeviceCode_Max77620Rtc          = 0x3B000001;
    constexpr inline const DeviceCode DeviceCode_Max77620Pmic         = 0x3A000001;
    constexpr inline const DeviceCode DeviceCode_Max77621Cpu          = 0x3A000003;
    constexpr inline const DeviceCode DeviceCode_Max77621Gpu          = 0x3A000004;
    constexpr inline const DeviceCode DeviceCode_Bq24193              = 0x39000001;
    constexpr inline const DeviceCode DeviceCode_Max17050             = 0x39000033;
    constexpr inline const DeviceCode DeviceCode_Bm92t30mwv           = 0x040000C9;
    constexpr inline const DeviceCode DeviceCode_Ina226Vdd15v0Hb      = 0x3F000401;

    constexpr inline const DeviceCode DeviceCode_Ina226VsysCpuDs      = 0x3F000001;
    constexpr inline const DeviceCode DeviceCode_Ina226VddCpuAp       = 0x3F000001;

    constexpr inline const DeviceCode DeviceCode_Ina226VsysGpuDs      = 0x3F000002;
    constexpr inline const DeviceCode DeviceCode_Ina226VddGpuAp       = 0x3F000002;

    constexpr inline const DeviceCode DeviceCode_Ina226VsysDdrDs      = 0x3F000003;
    constexpr inline const DeviceCode DeviceCode_Ina226VddDdr1V1Pmi   = 0x3F000003;

    constexpr inline const DeviceCode DeviceCode_Ina226VsysAp         = 0x3F000402;
    constexpr inline const DeviceCode DeviceCode_Ina226VsysBlDs       = 0x3F000403;
    constexpr inline const DeviceCode DeviceCode_Bh1730               = 0x35000047;

    constexpr inline const DeviceCode DeviceCode_Ina226VsysCore       = 0x3F000404;
    constexpr inline const DeviceCode DeviceCode_Ina226VddCoreAp      = 0x3F000404;

    constexpr inline const DeviceCode DeviceCode_Ina226Soc1V8         = 0x3F000405;
    constexpr inline const DeviceCode DeviceCode_Ina226VddSoc1V8      = 0x3F000405;

    constexpr inline const DeviceCode DeviceCode_Ina226Lpddr1V8       = 0x3F000406;
    constexpr inline const DeviceCode DeviceCode_Ina226Vdd1V8         = 0x3F000406;

    constexpr inline const DeviceCode DeviceCode_Ina226Reg1V32        = 0x3F000407;
    constexpr inline const DeviceCode DeviceCode_Ina226Vdd3V3Sys      = 0x3F000408;
    constexpr inline const DeviceCode DeviceCode_HdmiDdc              = 0x34000001;
    constexpr inline const DeviceCode DeviceCode_HdmiScdc             = 0x34000002;
    constexpr inline const DeviceCode DeviceCode_HdmiHdcp             = 0x34000003;
    constexpr inline const DeviceCode DeviceCode_Fan53528             = 0x3A000005;
    constexpr inline const DeviceCode DeviceCode_Max77812_3           = 0x3A000002;
    constexpr inline const DeviceCode DeviceCode_Max77812_2           = 0x3A000006;
    constexpr inline const DeviceCode DeviceCode_Ina226VddDdr0V6      = 0x3F000409;
    constexpr inline const DeviceCode DeviceCode_HoagNfcIc            = 0x36000001;
    constexpr inline const DeviceCode DeviceCode_PmicUnknownAula_4_18 = 0x3A000007;

    constexpr inline DeviceCode ConvertToDeviceCode(I2cDevice dv) {
        switch (dv) {
            case I2cDevice_ClassicController:    return DeviceCode_ClassicController;
            case I2cDevice_Ftm3bd56:             return DeviceCode_Ftm3bd56;
            case I2cDevice_Tmp451:               return DeviceCode_Tmp451;
            case I2cDevice_Nct72:                return DeviceCode_Nct72;
            case I2cDevice_Alc5639:              return DeviceCode_Alc5639;
            case I2cDevice_Max77620Rtc:          return DeviceCode_Max77620Rtc;
            case I2cDevice_Max77620Pmic:         return DeviceCode_Max77620Pmic;
            case I2cDevice_Max77621Cpu:          return DeviceCode_Max77621Cpu;
            case I2cDevice_Max77621Gpu:          return DeviceCode_Max77621Gpu;
            case I2cDevice_Bq24193:              return DeviceCode_Bq24193;
            case I2cDevice_Max17050:             return DeviceCode_Max17050;
            case I2cDevice_Bm92t30mwv:           return DeviceCode_Bm92t30mwv;
            case I2cDevice_Ina226Vdd15v0Hb:      return DeviceCode_Ina226Vdd15v0Hb;
            case I2cDevice_Ina226VsysCpuDs:      return DeviceCode_Ina226VsysCpuDs;
            case I2cDevice_Ina226VsysGpuDs:      return DeviceCode_Ina226VsysGpuDs;
            case I2cDevice_Ina226VsysDdrDs:      return DeviceCode_Ina226VsysDdrDs;
            case I2cDevice_Ina226VsysAp:         return DeviceCode_Ina226VsysAp;
            case I2cDevice_Ina226VsysBlDs:       return DeviceCode_Ina226VsysBlDs;
            case I2cDevice_Bh1730:               return DeviceCode_Bh1730;
            case I2cDevice_Ina226VsysCore:       return DeviceCode_Ina226VsysCore;
            case I2cDevice_Ina226Soc1V8:         return DeviceCode_Ina226Soc1V8;
            case I2cDevice_Ina226Lpddr1V8:       return DeviceCode_Ina226Lpddr1V8;
            case I2cDevice_Ina226Reg1V32:        return DeviceCode_Ina226Reg1V32;
            case I2cDevice_Ina226Vdd3V3Sys:      return DeviceCode_Ina226Vdd3V3Sys;
            case I2cDevice_HdmiDdc:              return DeviceCode_HdmiDdc;
            case I2cDevice_HdmiScdc:             return DeviceCode_HdmiScdc;
            case I2cDevice_HdmiHdcp:             return DeviceCode_HdmiHdcp;
            case I2cDevice_Fan53528:             return DeviceCode_Fan53528;
            case I2cDevice_Max77812_3:           return DeviceCode_Max77812_3;
            case I2cDevice_Max77812_2:           return DeviceCode_Max77812_2;
            case I2cDevice_Ina226VddDdr0V6:      return DeviceCode_Ina226VddDdr0V6;
            case I2cDevice_HoagNfcIc:            return DeviceCode_HoagNfcIc;
            case I2cDevice_PmicUnknownAula_4_18: return DeviceCode_PmicUnknownAula_4_18;
            AMS_UNREACHABLE_DEFAULT_CASE();
        }
    }

    constexpr inline I2cDevice ConvertToI2cDevice(DeviceCode dc) {
        switch (dc.GetInternalValue()) {
            case DeviceCode_ClassicController   .GetInternalValue(): return I2cDevice_ClassicController;
            case DeviceCode_Ftm3bd56            .GetInternalValue(): return I2cDevice_Ftm3bd56;
            case DeviceCode_Tmp451              .GetInternalValue(): return I2cDevice_Tmp451;
         /* case DeviceCode_Nct72               .GetInternalValue(): return I2cDevice_Nct72; */
            case DeviceCode_Alc5639             .GetInternalValue(): return I2cDevice_Alc5639;
            case DeviceCode_Max77620Rtc         .GetInternalValue(): return I2cDevice_Max77620Rtc;
            case DeviceCode_Max77620Pmic        .GetInternalValue(): return I2cDevice_Max77620Pmic;
            case DeviceCode_Max77621Cpu         .GetInternalValue(): return I2cDevice_Max77621Cpu;
            case DeviceCode_Max77621Gpu         .GetInternalValue(): return I2cDevice_Max77621Gpu;
            case DeviceCode_Bq24193             .GetInternalValue(): return I2cDevice_Bq24193;
            case DeviceCode_Max17050            .GetInternalValue(): return I2cDevice_Max17050;
            case DeviceCode_Bm92t30mwv          .GetInternalValue(): return I2cDevice_Bm92t30mwv;
            case DeviceCode_Ina226Vdd15v0Hb     .GetInternalValue(): return I2cDevice_Ina226Vdd15v0Hb;
            case DeviceCode_Ina226VsysCpuDs     .GetInternalValue(): return I2cDevice_Ina226VsysCpuDs;
            case DeviceCode_Ina226VsysGpuDs     .GetInternalValue(): return I2cDevice_Ina226VsysGpuDs;
            case DeviceCode_Ina226VsysDdrDs     .GetInternalValue(): return I2cDevice_Ina226VsysDdrDs;
            case DeviceCode_Ina226VsysAp        .GetInternalValue(): return I2cDevice_Ina226VsysAp;
            case DeviceCode_Ina226VsysBlDs      .GetInternalValue(): return I2cDevice_Ina226VsysBlDs;
            case DeviceCode_Bh1730              .GetInternalValue(): return I2cDevice_Bh1730;
            case DeviceCode_Ina226VsysCore      .GetInternalValue(): return I2cDevice_Ina226VsysCore;
            case DeviceCode_Ina226Soc1V8        .GetInternalValue(): return I2cDevice_Ina226Soc1V8;
            case DeviceCode_Ina226Lpddr1V8      .GetInternalValue(): return I2cDevice_Ina226Lpddr1V8;
            case DeviceCode_Ina226Reg1V32       .GetInternalValue(): return I2cDevice_Ina226Reg1V32;
            case DeviceCode_Ina226Vdd3V3Sys     .GetInternalValue(): return I2cDevice_Ina226Vdd3V3Sys;
            case DeviceCode_HdmiDdc             .GetInternalValue(): return I2cDevice_HdmiDdc;
            case DeviceCode_HdmiScdc            .GetInternalValue(): return I2cDevice_HdmiScdc;
            case DeviceCode_HdmiHdcp            .GetInternalValue(): return I2cDevice_HdmiHdcp;
            case DeviceCode_Fan53528            .GetInternalValue(): return I2cDevice_Fan53528;
            case DeviceCode_Max77812_3          .GetInternalValue(): return I2cDevice_Max77812_3;
            case DeviceCode_Max77812_2          .GetInternalValue(): return I2cDevice_Max77812_2;
            case DeviceCode_Ina226VddDdr0V6     .GetInternalValue(): return I2cDevice_Ina226VddDdr0V6;
            case DeviceCode_HoagNfcIc           .GetInternalValue(): return I2cDevice_HoagNfcIc;
            case DeviceCode_PmicUnknownAula_4_18.GetInternalValue(): return I2cDevice_PmicUnknownAula_4_18;
            AMS_UNREACHABLE_DEFAULT_CASE();
        }
    }

    constexpr bool IsPowerBusDeviceCode(DeviceCode device_code) {
        switch (device_code.GetInternalValue()) {
            case DeviceCode_Max77620Pmic.GetInternalValue():
            case DeviceCode_Max77812_3  .GetInternalValue():
            case DeviceCode_Max77621Cpu .GetInternalValue():
            case DeviceCode_Max77621Gpu .GetInternalValue():
            case DeviceCode_Fan53528    .GetInternalValue():
            case DeviceCode_Max77812_2  .GetInternalValue():
            case DeviceCode_Max77620Rtc .GetInternalValue():
                return true;
            default:
                return false;
        }
    }

}