cnt_collector_485/cnt_collector_485.py

import os, time, json
import serial
import serial.rs485
from pymodbus.client import ModbusSerialClient as ModbusClient
import paho.mqtt.client as mqtt
import struct
import argparse

if os.system("lsmod | grep max7301") == 0:
    os.system("rmmod gpio-max7301")
    os.system("rmmod gpio-max730x")
    os.system("rmmod gpio-pisosr")
    os.system("insmod /lib/modules/5.4.31/extra/max3109test.ko")

# For PION-D3W-035(Power Controller) No.1, UT35A(Temperature Controller) No.1
_serial_port_A = serial.Serial('/dev/ttyMAX1', 38400, parity=serial.PARITY_NONE, stopbits=serial.STOPBITS_ONE, bytesize=serial.EIGHTBITS, timeout = 1, rtscts=True)
_serial_port_A.rs485_mode = serial.rs485.RS485Settings()
_serial_port_A.close()
del _serial_port_A

client1 = ModbusClient(method='rtu', port='/dev/ttyMAX1', baudrate=38400,
                      parity='N', stopbits=1, bytesize=8, timeout=1)


# For PION-D3W-035(Power Controller) No.2, UT35A(Temperature Controller) No.1
_serial_port_B = serial.Serial('/dev/ttyMAX0', 38400, parity=serial.PARITY_NONE, stopbits=serial.STOPBITS_ONE, bytesize=serial.EIGHTBITS, timeout = 1, rtscts=True)
_serial_port_B.rs485_mode = serial.rs485.RS485Settings()
_serial_port_B.close()
del _serial_port_B

client2 = ModbusClient(method='rtu', port='/dev/ttyMAX0', baudrate=38400,
                      parity='N', stopbits=1, bytesize=8, timeout=1)

# For PION-D3W-035(Power Controller) No.3, UT35A(Temperature Controller) No.1
_serial_port_C = serial.Serial('/dev/ttyMAX2', 38400, parity=serial.PARITY_NONE, stopbits=serial.STOPBITS_ONE, bytesize=serial.EIGHTBITS, timeout = 1, rtscts=True)
_serial_port_C.rs485_mode = serial.rs485.RS485Settings()
_serial_port_C.close()
del _serial_port_C

client3 = ModbusClient(method='rtu', port='/dev/ttyMAX2', baudrate=38400,
                      parity='N', stopbits=1, bytesize=8, timeout=1)

# For FIX800(Gas Detector), CWT-MB307C(Flow Pump)
_serial_port_D = serial.Serial('/dev/ttyMAX3', 9600, parity=serial.PARITY_NONE, stopbits=serial.STOPBITS_ONE, bytesize=serial.EIGHTBITS, timeout = 0.3, rtscts=True)
_serial_port_D.rs485_mode = serial.rs485.RS485Settings()
# _serial_port_A.close()
# del _serial_port_A

client4 = ModbusClient(method='rtu', port='/dev/ttyMAX3', baudrate=9600,
                      parity='N', stopbits=1, bytesize=8, timeout=0.5)

client1.connect()
client2.connect()
client3.connect()
client4.connect()

parser = argparse.ArgumentParser()
parser.add_argument('-config',help='')
args = parser.parse_args()

ROOT_PATH = f'/usr/local/sdt/app/{args.config}'

with open(f'{ROOT_PATH}/connect_info.json') as f:
    info = json.load(f)

MQTT_TOPIC = info['mqtt']['topic']
# MQTT_TOPIC = "/axrcnt/device-test/S0NQ0R423090004"
MQTT_ID = info['mqtt']['id']
MQTT_PW = info['mqtt']['pw']
MQTT_HOST_IP = info['mqtt']['host_ip']
MQTT_PORT = info['mqtt']['port']

mqtt_client = mqtt.Client(info['mqtt']['assetCode'])
mqtt_client.username_pw_set(MQTT_ID, MQTT_PW)

def Read_Temperature(client, ch_num):
    if ch_num == 1:
        id = 1
        name = 'tempFn1'
    elif ch_num == 2:
        id = 2
        name = 'tempFn2'
    else: # ch_num == 3
        id = 3
        name = 'tempFn3'
        
    result = client.read_holding_registers(address=2002, count=1, slave=id)
    if not result.isError():
        # print(f'Temperature: {ch_num} {result.registers}')
        value = (result.registers[0])/10

        data = {
            "assetCode": info['mqtt']['assetCode'],
            "timestamp": int(time.time() * 1000),
            "dataType": "DATA",
            "data": {
                name: value
            }
        }

        Publish_mqtt(data)
    else:
        print(f'Temperature Err: {ch_num} \n {result}')

def Set_Temperature(client, ch_num, order, val=None):
    if ch_num == 1:
        id = 1
    elif ch_num == 2:
        id = 2
    else: # ch_num == 3
        id = 3

    if not val == None:
        val *= 10
        val = int(val)

    try:
        if order == 'start':
            # Patten 1
            client.write_register(address=8000, value=1, slave=id) # PTNO._C
            client.write_register(address=8004, value=1, slave=id) # STC
            
            # SEG 0
            client.write_register(address=8001, value=0, slave=id) # SEGNO._C
            client.write_register(address=8002, value=0, slave=id) # SSP_L1

            # SEG 1
            client.write_register(address=8001, value=1, slave=id) # SEGNO._C
            
            client.write_register(address=8102, value=val, slave=id) # TSP_L1
            client.write_register(address=8104, value=180, slave=id) # TIME
            
            # SEG 2
            client.write_register(address=8001, value=2, slave=id) # SEGNO._C
            
            client.write_register(address=8102, value=val, slave=id) # TSP_L1
            client.write_register(address=8104, value=1440, slave=id) # TIME

            # Power On
            client.write_register(address=2321, value=1, slave=id) # PTNO.
            client.write_register(address=2315, value=1, slave=id) # MODE
        elif order == 'change':
            # Power Off
            client.write_register(address=2315, value=0, slave=id) # MODE

            # Patten 1
            client.write_register(address=8000, value=1, slave=id) # PTNO._C
            client.write_register(address=8004, value=1, slave=id) # STC
            
            # SEG 0
            client.write_register(address=8001, value=0, slave=id) # SEGNO._C
            client.write_register(address=8002, value=0, slave=id) # SSP_L1
            
            # SEG 1
            client.write_register(address=8001, value=1, slave=id) # SEGNO._C
            
            client.write_register(address=8102, value=val, slave=id) # TSP_L1
            client.write_register(address=8104, value=180, slave=id) # TIME
            
            # SEG 2
            client.write_register(address=8001, value=2, slave=id) # SEGNO._C
            
            client.write_register(address=8102, value=val, slave=id) # TSP_L1
            client.write_register(address=8104, value=1440, slave=id) # TIME

            client.write_register(address=2321, value=1, slave=id) # PTNO.
            client.write_register(address=2315, value=1, slave=id) # MODE
        else: # order == 'stop'
            # Power Off
            client.write_register(address=2315, value=0, slave=id) # MODE
    except Exception as e:
        print(e)

def Read_Power(client, ch_num):
    if ch_num == 1:
        id = 11
        name = 'enerWn1'
    elif ch_num == 2:
        id = 12
        name = 'enerWn2'
    else: # ch_num == 3
        id = 13
        name = 'enerWn3'

    r_power_rate = client.read_input_registers(address=305, count=1, slave=id)
    s_power_rate = client.read_input_registers(address=306, count=1, slave=id)
    t_power_rate = client.read_input_registers(address=307, count=1, slave=id)
    
    if not r_power_rate.isError() and not s_power_rate.isError() and not t_power_rate.isError():
        max_power = 23036

        mean_power_rate = (
            (r_power_rate.registers[0] + 
            s_power_rate.registers[0] + 
            t_power_rate.registers[0]) / 30000)

        value = (max_power * mean_power_rate) / 1000
        # print(value)
        
        data = {
            "assetCode": info['mqtt']['assetCode'],
            "timestamp": int(time.time() * 1000),
            "dataType": "DATA",
            "data": {
                name: value
            }
        }
        # print(data)
        Publish_mqtt(data)

result = []
def Read_GasDetector(client, ch_num):
    global result
    if ch_num == 1:
        oxygen = '1503024e0002a6b0'
        hydrogen = '1503034e0002a74c'
        name_1 = 'oxygenOn'
        name_2 = 'hydrogenOn'
    else: # ch_num == 2
        id = 22
        oxygen = '1603024e0002a683'
        hydrogen = '1603034e0002a77f'
        name_1 = 'oxygenSb'
        name_2 = 'hydrogenSb'
    
    oxygen_value, hydrogen_value = 0, 0
    
    send_data = bytes.fromhex(oxygen)
    isRun = True
    while isRun:
        client.write(send_data)
        result += client.readline()
        if  len(result) >= 9:
            if result[0] == 21 and result[1] == 3 and result[2] == 4:
                combined_bytes = (result[5] << 24) + (result[6] << 16) + (result[3] << 8) + result[4]
                oxygen_value = struct.unpack('f', struct.pack('I', combined_bytes))[0]

                result.clear()
                client.reset_input_buffer()
                isRun = False
            else:
                while True:
                    result.pop(0)
                    if result[0] == 21:
                        break

    send_data = bytes.fromhex(hydrogen)
    isRun = True
    while isRun:
        client.write(send_data)
        result += client.readline()
        if len(result) >= 9:
            if result[0] == 21 and result[1] == 3 and result[2] == 4:
                combined_bytes = (result[5] << 24) + (result[6] << 16) + (result[3] << 8) + result[4]
                hydrogen_value = struct.unpack('f', struct.pack('I', combined_bytes))[0]

                result.clear()
                client.reset_input_buffer()
                isRun = False
            else:
                while True:
                    result.pop(0)
                    if result[0] == 21:
                        break

    data = {
        "assetCode": info['mqtt']['assetCode'],
        "timestamp": int(time.time() * 1000),
        "dataType": "DATA",
        "data": {
            name_1: oxygen_value,
            name_2: hydrogen_value
        }
    }
    # print(data)
    Publish_mqtt(data)

def Set_FlowPump(client, value):
    id = 31

    val = int(float(value) * 1000)

    register_value = val * 8 + 10000
    client.write_registers(address=3, values=register_value, slave=id)

def Publish_mqtt(data):
    send_mqtt_data = json.dumps(data, indent=4)

    while True:
        try:
            mqtt_client.connect(host=MQTT_HOST_IP, port=MQTT_PORT)
            break
        except:
            continue
        
    mqtt_client.publish(topic=MQTT_TOPIC, payload=send_mqtt_data)

    while True:
        try:
            mqtt_client.disconnect()
            break
        except:
            continue

def Command_Read():
    with open(f'{ROOT_PATH}/control.json') as f:
        cmd = json.load(f)

    # print(cmd)

    if cmd['type'] != 'null':
        if 'start' in cmd['type'] or 'change' in cmd['type']:
            if cmd['cmd']['precAc'] != 'null':
                Set_FlowPump(client=client4, value=float(cmd['cmd']['precAc']))
            
            if cmd['cmd']['tempFn1'] != 'null':
                Set_Temperature(client=client1, ch_num=1, order=cmd['type'], val=float(cmd['cmd']['tempFn1']))
            
            if cmd['cmd']['tempFn2'] != 'null':
                Set_Temperature(client=client2, ch_num=2, order=cmd['type'], val=float(cmd['cmd']['tempFn2']))
            
            if cmd['cmd']['tempFn3'] != 'null':
                Set_Temperature(client=client3, ch_num=3, order=cmd['type'], val=float(cmd['cmd']['tempFn3']))
        else: # 'stop' in cmd['type]
            Set_Temperature(client=client1, ch_num=1, order=cmd['type'])
            Set_Temperature(client=client2, ch_num=2, order=cmd['type'])
            Set_Temperature(client=client3, ch_num=3, order=cmd['type'])
            Set_FlowPump(client=client4, value=0.0)

    cmd = {
        "type": "null",
        "cmd": {
            "tempFn1": "null",
            "tempFn2": "null",
            "tempFn3": "null",
            "precAc": "null"
        }
    }
    with open(f'{ROOT_PATH}/control.json', 'w') as f:
        json.dump(cmd, f)

while True:
    start = time.time()
    # Need to add a function call for reading control.json
    Command_Read()

    # Read Temperature
    Read_Temperature(client1, 1)
    Read_Temperature(client2, 2)
    Read_Temperature(client3, 3)

    # Read Power
    Read_Power(client1, 1)
    Read_Power(client2, 2)
    Read_Power(client3, 3)
    
    # Read GasDetector
    Read_GasDetector(_serial_port_D, 1)
    # Read_GasDetector(_serial_port_D, 2)

    end = time.time()
    diff = end - start
    # print(f'diff: {diff}')

    if diff < 3:
        time.sleep(3.0 - diff)