#ifndef SAKO_VFD_H
#define SAKO_VFD_H

#include "config.h"

#ifdef ENABLE_RS485 // Assuming this VFD also uses RS485

#include <ArduinoLog.h>
#include <Component.h>
#include <modbus/ModbusRTU.h>
#include <modbus/ModbusTypes.h>
#include <xstatistics.h>
#include <ValueWrapper.h>
#include "VFD_Base.h"

#define SAKO_VFD_DEFAULT_READ_INTERVAL 100
#define SAKO_VFD_TCP_REG_RANGE 16


enum class E_SakoTcpOffset : ushort
{
    RUNNING_FREQUENCY = 1,     // Corresponds to U0-00
    SET_FREQUENCY = 2,         // Corresponds to U0-01
    OUTPUT_CURRENT = 3,        // Corresponds to U0-04
    OUTPUT_POWER_KW = 4,       // Corresponds to U0-05
    OUTPUT_TORQUE_PERCENT = 5, // Corresponds to U0-06
    FAULT_CODE = 6,            // Corresponds to U0-62
    IS_RUNNING = 7,            // Derived from U0-61
    HAS_FAULT = 8,             // Derived from U0-62
    STATE = 9,                 // E_VFD_STATE
    CMD_FREQ = 10,              // Write Frequency (Uses SAKO_REG_SET_FREQ)
    CMD_DIRECTION = 11,        // Write Direction/Control (Uses E_SAKO_REGISTERS_SET_DIR)
    CMD_COMMAND = 12,          // Internal command
    TARGET_REGISTER = 13,      // Write target register address
    TARGET_VALUE = 14,         // Write value to target register
};

class SAKO_VFD : public VFD_Base
{
public:
    static constexpr int SAKO_TCP_BLOCK_COUNT = 14;
    // Constructor
    SAKO_VFD(Component* owner, uint8_t slaveId, millis_t readInterval = SAKO_VFD_DEFAULT_READ_INTERVAL);
    virtual ~SAKO_VFD() = default;
    // --- Component Interface ---
    virtual short setup() override;
    virtual short loop() override;
    virtual short info() override;
    short reset() override;
    short getTorque();

    // --- Modbus Register Update Notification ---
    virtual bool onRegisterUpdate(uint16_t address, uint16_t newValue) override;
    virtual void onError(ushort errorCode, const char* errorMessage) override;

    // --- Getters for Specific VFD Values ---
    // Add getters relevant to a VFD, e.g., Frequency, Speed, Status, Fault codes
    bool getFrequency(uint16_t& value) const; // Returns frequency in 0.01 Hz units
    bool getSpeed(uint16_t& value) const; // Example: speed might be integer RPM
    bool isRunning() const;
    bool hasFault() const;
    uint16_t getFaultCode() const;
    E_VFD_STATE getVfdState() const;
    bool getOutputPowerKW(uint16_t& value) const;
    bool getOutputTorquePercent(uint16_t& value) const;
    bool getOutputCurrent(uint16_t& value) const;

    // --- Setters for VFD Control ---
    bool setFrequency(uint16_t value); // Expects frequency in 0.01 Hz units
    bool run();
    bool reverse();
    short stop() override;
    bool resetFault();
    bool retract();

    // --- Modbus Block Definitions ---
    virtual ModbusBlockView* mb_tcp_blocks() const override;
    virtual short mb_tcp_read(MB_Registers * reg) override;
    virtual short mb_tcp_write(MB_Registers * reg, short value) override;
    virtual ushort mb_tcp_error(MB_Registers *reg) override;

    // --- Modbus TCP Mapping Overrides ---
    virtual uint16_t mb_tcp_base_address() const override;
    virtual uint16_t mb_tcp_offset_for_rtu_address(uint16_t rtuAddress) const override;


private:
    // --- SAKO-specific State Storage ---
    uint16_t _currentSpeed = 0;
    uint16_t _outputPowerKW = 0;
    uint16_t _outputTorquePercent = 0;
    bool _speedValid = false;
    bool _outputPowerKWValid = false;
    bool _outputTorquePercentValid = false;

    uint16_t _tcpTargetRegister = 0; // Stores the value written to TARGET_REGISTER offset

    // Statistics
    Statistic _ampStats;

    MB_Registers _modbusBlocks[SAKO_TCP_BLOCK_COUNT];
    mutable ModbusBlockView _modbusBlockView;
    bool _modbusBlocksInitialized; // Added for on-demand initialization

    // Add internal helper methods if needed
    void _updateStatusFromRegister(uint16_t statusReg) override;
    void _updateVfdState() override;
    short serial_register(Bridge *bridge) override;
    short setupVFD();
};

#endif // ENABLE_RS485
#endif // SAKO_VFD_H