#!/usr/bin/python

#uncomment for debbug purposes
#import logging
#logging.basicConfig(level=logging.DEBUG)

import time
import datetime
import sys
import os
import minimalmodbus

from pymlab import config
from mlabutils import ejson

parser = ejson.Parser()

#### Script Arguments ###############################################

if len(sys.argv) != 3:
    sys.stderr.write("Invalid number of arguments.\n")
    sys.stderr.write("Usage: %s PORT_ADDRESS CONFIG_FILE\n" % (sys.argv[0], ))
    sys.exit(1)


port = eval(sys.argv[1])

value = parser.parse_file(sys.argv[2])
path = value['data_path']
interval = value['raw_sample_interval']
stationName = value['origin']



if (interval<5) or (interval>3600):
    sys.stderr.write("Invalid sample interval arguments.\n")
    sys.stderr.write("The interval has to be in the range from 5 to 3600 seconds\n")
    sys.exit(2)

#### Sensor Configuration ###########################################
# I2C
cfg = config.Config(
    i2c = {
            "port": port,
    },
    bus = [
        {
            "name":          "current_sensor1",
            "type":        "vcai2c01",
            "address":        0x68,
        },
    ],
)
cfg.initialize()

# modbus
instrument = minimalmodbus.Instrument('/dev/ttyUSB0', 1) # port name, slave address (in decimal)

instrument.serial.port          # this is the serial port name
instrument.serial.baudrate = 9600   # Baud
instrument.serial.bytesize = 8
instrument.serial.stopbits = 2
instrument.serial.timeout  = 0.5   # seconds

instrument.mode = minimalmodbus.MODE_RTU   # rtu or ascii mode


sys.stdout.write("Current loop and modbus sensor example \r\n")
sys.stdout.write("Time, water-level,  temp1,  conduc, salinity, tds_kcl, temp2, pH, redox \r\n")
#sys.stdout.write("Time, channel #1,  channel #2,  channel #3 ,  channel #4,  channel #5  channel #6 ,  channel #7,  channel #8   \r\n")
sensor1 = cfg.get_device("current_sensor1")
#time.sleep(0.5)

#### Data Logging ###################################################

try:
    before = time.time()-interval
    while True:
        
        filename = path + time.strftime("%Y%m%d%H", time.gmtime()) +"0000_"+ stationName + "_data.csv"        
        now = time.time()
        
        if (now - before >= interval - 2.5):     #   0.5*5 channels= 2.5s
            with open(filename, "a") as f:
                
                ##Measuremment settings
                sensor1.setADC(channel = 1, gain = 1, sample_rate = 3.75);
                
                instrument.address = 0x1E    # this is the slave address number (1E-conductivity)
                instrument.write_register(0x01, 0x1F, 0) # Registernumber, value, number of decimals for storage
                
                instrument.address = 0x14    # this is the slave address number (14 - pH)
                instrument.write_register(0x01, 0x1F, 0) # Registernumber, value, number of decimals for storage
                                
                time.sleep(0.5)

                ##Reading
                ## Read data from analog sensors ##
                channel1 = sensor1.readCurrent();

                ## Read data from conductivity sensor ##
                instrument.address = 0x1E    # this is the slave address number (1E-conductivity)

                temperature1 = instrument.read_float(0x53, 3, 2) # Registernumber, number of decimals                
                conductivity = instrument.read_float(0x55, 3, 2) # Registernumber, number of decimals                
                salinity = instrument.read_float(0x57, 3, 2) # Registernumber, number of decimals                
                tds_kcl = instrument.read_float(0x59, 3, 2) # Registernumber, number of decimals                
                
                ## Read data from pH sensor ##
                instrument.address = 0x14    # this is the slave address number (14 - pH)
                
                temperature2 = instrument.read_float(0x53, 3, 2) # Registernumber, number of decimals                
                pH = instrument.read_float(0x55, 3, 2) # Registernumber, number of decimals                
                redox = instrument.read_float(0x57, 3, 2) # Registernumber, number of decimals                

                sys.stdout.write("%s \t %0.3f \t  %0.3f \t %0.3f \t %0.3f \t %0.3f \t %0.3f \t %0.3f \t %0.3f \t \n" % (datetime.datetime.now().isoformat(), channel1, temperature1, conductivity, salinity, tds_kcl, temperature2, pH, redox))

                f.write("%d;%0.3f;%0.3f;%0.3f;%0.3f;%0.3f;%0.3f;%0.3f;%0.3f\n" % (time.time(), channel1, temperature1, conductivity, salinity, tds_kcl, temperature2, pH, redox))
                f.flush()

                sys.stdout.flush()
                before = time.time()
        else:
            time.sleep(0.1)

except KeyboardInterrupt:
    f.close()
    sys.exit(0)