package com.ruoyi.fuzhou.utils;

import com.ruoyi.fuzhou.domain.ReceiveModuleInfo;
import com.ruoyi.fuzhou.utils.rtu.SerialPortWrapperImpl;
import com.serotonin.modbus4j.ModbusFactory;
import com.serotonin.modbus4j.ModbusMaster;
import com.serotonin.modbus4j.msg.ReadHoldingRegistersRequest;
import com.serotonin.modbus4j.msg.ReadHoldingRegistersResponse;
import com.serotonin.modbus4j.msg.WriteRegisterRequest;
import com.serotonin.modbus4j.msg.WriteRegisterResponse;
import gnu.io.SerialPort;

import java.util.Arrays;

/**
 * 模拟主站设备
 *
 * @author zhangyy
 * @date 2025/3/8
 */
public class RtuMasterUtils {

    public static void main(String[] args) throws Exception {
        createRtuMaster();
    }

    public static boolean getYdkfmStatus(ReceiveModuleInfo rmi) throws Exception {
        SerialPortWrapperImpl wrapper = new SerialPortWrapperImpl("COM4", 19200,
                SerialPort.DATABITS_8, SerialPort.STOPBITS_1, SerialPort.PARITY_NONE, 0, 0);
        ModbusFactory modbusFactory = new ModbusFactory();
        ModbusMaster master = modbusFactory.createRtuMaster(wrapper);
        master.init();

        int slaveId = rmi.getRegisterAdress(); // 从站设备ID
        readValue(master, slaveId, 7, 1);

        return true;
    }

    public static boolean ctrlYdkfm(ReceiveModuleInfo rmi, Boolean flag) throws Exception {
        SerialPortWrapperImpl wrapper = new SerialPortWrapperImpl("COM4", 19200,
                SerialPort.DATABITS_8, SerialPort.STOPBITS_1, SerialPort.PARITY_NONE, 0, 0);
        ModbusFactory modbusFactory = new ModbusFactory();
        ModbusMaster master = modbusFactory.createRtuMaster(wrapper);
        master.init();

        int slaveId = rmi.getRegisterAdress();
        writeValue(master, slaveId, 7, flag ? 2 : 1);

        return true;
    }

    private static boolean readValue(ModbusMaster master, int slaveId, int start, int len) throws Exception {
        ReadHoldingRegistersRequest request = new ReadHoldingRegistersRequest(slaveId, start, len);
        ReadHoldingRegistersResponse response = (ReadHoldingRegistersResponse) master.send(request);

        if (response.isException()) {
            System.out.println("读取保持寄存器错误，错误信息是" + response.getExceptionMessage());
        } else {
            System.out.println("读取保持寄存器=" + Arrays.toString(response.getShortData()));
        }

        return !response.isException();
    }

    private static boolean writeValue(ModbusMaster master, int slaveId, int offset, int value) throws Exception {
        WriteRegisterRequest request = new WriteRegisterRequest(slaveId, offset, value);
        WriteRegisterResponse response = (WriteRegisterResponse) master.send(request);

        if (response.isException()) {
            System.out.println("写保持寄存器错误，错误信息是" + response.getExceptionMessage());
        } else {
            System.out.println("写保持寄存器成功");
        }

        return !response.isException();
    }


    private static void createRtuMaster() throws Exception {
        // 设置串口参数，串口是COM1，波特率是9600
        SerialPortWrapperImpl wrapper = new SerialPortWrapperImpl("COM2", 9600,
                SerialPort.DATABITS_8, SerialPort.STOPBITS_1, SerialPort.PARITY_NONE, 0, 0);
        ModbusFactory modbusFactory = new ModbusFactory();
        ModbusMaster master = modbusFactory.createRtuMaster(wrapper);
        master.init();
        // 从站设备ID是1
        int slaveId = 1;
        // 读取保持寄存器
        readHoldingRegisters(master, slaveId, 0, 3);
        // 将地址为0的保持寄存器数据修改为0
        writeRegister(master, slaveId, 0, 0);
        // 再读取保持寄存器
        readHoldingRegisters(master, slaveId, 0, 3);
    }

    private static void readHoldingRegisters(ModbusMaster master, int slaveId, int start, int len) throws Exception {
        ReadHoldingRegistersRequest request = new ReadHoldingRegistersRequest(slaveId, start, len);
        ReadHoldingRegistersResponse response = (ReadHoldingRegistersResponse) master.send(request);
        if (response.isException()) {
            System.out.println("读取保持寄存器错误，错误信息是" + response.getExceptionMessage());
        } else {
            System.out.println("读取保持寄存器=" + Arrays.toString(response.getShortData()));
        }
    }

    private static void writeRegister(ModbusMaster master, int slaveId, int offset, int value) throws Exception {
        WriteRegisterRequest request = new WriteRegisterRequest(slaveId, offset, value);
        WriteRegisterResponse response = (WriteRegisterResponse) master.send(request);
        if (response.isException()) {
            System.out.println("写保持寄存器错误，错误信息是" + response.getExceptionMessage());
        } else {
            System.out.println("写保持寄存器成功");
        }
    }
}