modbusproject.cpp

#include "modbusproject.h"
#include <QJsonDocument>
#include <QJsonArray>
#include <QFile>

ModbusProject::ModbusProject()
    : m_connectionType(ConnectionType::TCP)
    , m_tcpHost("127.0.0.1")
    , m_tcpPort(502)
    , m_rtuBaudRate(9600)
    , m_rtuDataBits(8)
    , m_rtuStopBits(1)
    , m_rtuParity("None")
    , m_slaveId(1)
    , m_timeout(1000)
    , m_retries(3)
{
}

void ModbusProject::addRegister(const ModbusRegister &reg)
{
    m_registers.append(reg);
}

void ModbusProject::removeRegister(int index)
{
    if (index >= 0 && index < m_registers.size()) {
        m_registers.removeAt(index);
    }
}

void ModbusProject::updateRegister(int index, const ModbusRegister &reg)
{
    if (index >= 0 && index < m_registers.size()) {
        m_registers[index] = reg;
    }
}

void ModbusProject::addTrigger(const TriggerCondition &trigger)
{
    m_triggers.append(trigger);
}

void ModbusProject::removeTrigger(int index)
{
    if (index >= 0 && index < m_triggers.size()) {
        m_triggers.removeAt(index);
    }
}

QJsonObject ModbusProject::toJson() const
{
    QJsonObject json;
    json["name"] = m_name;
    json["connectionType"] = (m_connectionType == ConnectionType::TCP) ? "TCP" : "RTU";
    
    // Connection settings
    json["tcpHost"] = m_tcpHost;
    json["tcpPort"] = m_tcpPort;
    json["rtuPort"] = m_rtuPort;
    json["rtuBaudRate"] = m_rtuBaudRate;
    json["rtuDataBits"] = m_rtuDataBits;
    json["rtuStopBits"] = m_rtuStopBits;
    json["rtuParity"] = m_rtuParity;
    json["slaveId"] = m_slaveId;
    json["timeout"] = m_timeout;
    json["retries"] = m_retries;
    
    // Registers
    QJsonArray registersArray;
    for (const auto &reg : m_registers) {
        QJsonObject regObj;
        regObj["name"] = reg.name();
        regObj["registerType"] = static_cast<int>(reg.registerType());
        regObj["address"] = reg.address();
        regObj["dataType"] = static_cast<int>(reg.dataType());
        regObj["description"] = reg.description();
        regObj["readOnly"] = reg.isReadOnly();
        regObj["pollInterval"] = reg.pollInterval();
        registersArray.append(regObj);
    }
    json["registers"] = registersArray;
    
    // Triggers
    QJsonArray triggersArray;
    for (const auto &trigger : m_triggers) {
        QJsonObject trigObj;
        trigObj["registerName"] = trigger.registerName;
        trigObj["condition"] = trigger.condition;
        trigObj["value"] = trigger.value.toString();
        trigObj["action"] = trigger.action;
        triggersArray.append(trigObj);
    }
    json["triggers"] = triggersArray;
    
    return json;
}

bool ModbusProject::fromJson(const QJsonObject &json)
{
    m_name = json["name"].toString();
    m_connectionType = (json["connectionType"].toString() == "TCP") ? ConnectionType::TCP : ConnectionType::RTU;
    
    m_tcpHost = json["tcpHost"].toString();
    m_tcpPort = json["tcpPort"].toInt();
    m_rtuPort = json["rtuPort"].toString();
    m_rtuBaudRate = json["rtuBaudRate"].toInt();
    m_rtuDataBits = json["rtuDataBits"].toInt();
    m_rtuStopBits = json["rtuStopBits"].toInt();
    m_rtuParity = json["rtuParity"].toString();
    m_slaveId = json["slaveId"].toInt();
    m_timeout = json["timeout"].toInt();
    m_retries = json["retries"].toInt();
    
    // Registers
    m_registers.clear();
    QJsonArray registersArray = json["registers"].toArray();
    for (const QJsonValue &val : registersArray) {
        QJsonObject regObj = val.toObject();
        ModbusRegister reg;
        reg.setName(regObj["name"].toString());
        reg.setRegisterType(static_cast<RegisterType>(regObj["registerType"].toInt()));
        reg.setAddress(regObj["address"].toInt());
        reg.setDataType(static_cast<DataType>(regObj["dataType"].toInt()));
        reg.setDescription(regObj["description"].toString());
        reg.setReadOnly(regObj["readOnly"].toBool());
        reg.setPollInterval(regObj["pollInterval"].toInt());
        m_registers.append(reg);
    }
    
    // Triggers
    m_triggers.clear();
    QJsonArray triggersArray = json["triggers"].toArray();
    for (const QJsonValue &val : triggersArray) {
        QJsonObject trigObj = val.toObject();
        TriggerCondition trigger;
        trigger.registerName = trigObj["registerName"].toString();
        trigger.condition = trigObj["condition"].toString();
        trigger.value = trigObj["value"].toString();
        trigger.action = trigObj["action"].toString();
        m_triggers.append(trigger);
    }
    
    return true;
}

bool ModbusProject::saveToFile(const QString &filename) const
{
    QFile file(filename);
    if (!file.open(QIODevice::WriteOnly)) {
        return false;
    }
    
    QJsonDocument doc(toJson());
    file.write(doc.toJson(QJsonDocument::Indented));
    file.close();
    return true;
}

bool ModbusProject::loadFromFile(const QString &filename)
{
    QFile file(filename);
    if (!file.open(QIODevice::ReadOnly)) {
        return false;
    }
    
    QByteArray data = file.readAll();
    file.close();
    
    QJsonDocument doc = QJsonDocument::fromJson(data);
    if (doc.isNull() || !doc.isObject()) {
        return false;
    }
    
    return fromJson(doc.object());
}