// SPDX-License-Identifier: Apache-2.0
#pragma once

#include <stdbool.h>
#include <stdint.h>

#include <tactility/error.h>

struct Device;

#ifdef __cplusplus
extern "C" {
#endif

struct Axp192Config {
    /** Address on bus */
    uint8_t address;
};

/** Switchable/adjustable power rails of the AXP192. */
enum Axp192Rail {
    AXP192_RAIL_DCDC1,
    AXP192_RAIL_DCDC2,
    AXP192_RAIL_DCDC3,
    AXP192_RAIL_LDO2,
    AXP192_RAIL_LDO3,
    /** EXTEN output switch; does not support voltage control. */
    AXP192_RAIL_EXTEN,
};

/**
 * @brief Checks whether a power rail is currently enabled.
 */
error_t axp192_is_rail_enabled(struct Device* device, enum Axp192Rail rail, bool* enabled);

/**
 * @brief Enables or disables a power rail.
 */
error_t axp192_set_rail_enabled(struct Device* device, enum Axp192Rail rail, bool enabled);

/**
 * @brief Sets the output voltage of a power rail.
 * @retval ERROR_NOT_SUPPORTED for AXP192_RAIL_EXTEN, which has no voltage control
 * @retval ERROR_INVALID_ARGUMENT when millivolts is outside the rail's supported range
 * (DCDC1/DCDC3: 700-3500mV, DCDC2: 700-2275mV, LDO2/LDO3: 1800-3300mV)
 */
error_t axp192_set_rail_voltage(struct Device* device, enum Axp192Rail rail, uint16_t millivolts);

/** Battery voltage in millivolts. */
error_t axp192_get_battery_voltage(struct Device* device, uint16_t* millivolts);

/** Battery charge current in milliamps (0 when not charging). */
error_t axp192_get_battery_charge_current(struct Device* device, uint16_t* milliamps);

/** Battery discharge current in milliamps (0 when not discharging). */
error_t axp192_get_battery_discharge_current(struct Device* device, uint16_t* milliamps);

/** Battery power in microwatts. */
error_t axp192_get_battery_power(struct Device* device, uint32_t* microwatts);

/** Whether the battery is currently charging. */
error_t axp192_is_charging(struct Device* device, bool* charging);

/** Whether the charger is allowed to charge the battery. */
error_t axp192_is_charge_enabled(struct Device* device, bool* enabled);

/** Enables or disables battery charging. */
error_t axp192_set_charge_enabled(struct Device* device, bool enabled);

/** Powers off the system (does not return on success). */
error_t axp192_power_off(struct Device* device);

/** Configures GPIO1 as the PWM1 output (rather than GPIO/ADC input/output). */
error_t axp192_set_gpio1_pwm1_output(struct Device* device);

/** Sets the PWM1 duty cycle (0 = always low, 255 = always high). */
error_t axp192_set_pwm1_duty_cycle(struct Device* device, uint8_t duty);

#ifdef __cplusplus
}
#endif