#!/usr/bin/python3
# -*- coding: UTF-8 -*-
# this code indented with actual 0x09 tabs

# This is a hacked up version of code from: https://gist.github.com/geoffwatts/b0b488b5a5257223ed53

import sys, serial, time, datetime, struct, fileinput, os
import numpy as np
import matplotlib.dates as mdates

sys.path.append('/home/ghz/repos/wxlib')
import wxlib as wx

wx_dir = "/home/ghz/dust"
plot_d = wx_dir+'/plots/'

# yet another wonky converter
def yawc(data):
	return mdates.strpdate2num('%Y%m%d%H%M%S')(data.decode('utf8'))

def plot(ts, n_plate):
	npoints = 3000 # ~48h

	dat_f = ["1000", "0100", "0010", "0001"]

	td = datetime.datetime.strptime(ts, "%Y%m%d%H%M%S")

	for i in range(0, 4):
		d_date = (td - datetime.timedelta(i)).strftime("%Y%m%d")
		d_year = (td - datetime.timedelta(i)).strftime("%Y")
		dat_f[3 - i] = wx_dir+'/data/'+d_year+'/'+n_plate+'.'+d_date
		wx.proof_dat_f(dat_f[3 - i])

	dust_dat  = fileinput.input(dat_f)
	# date, pm25, pm10 = np.loadtxt(dust_dat, usecols=(0, 3, 7), unpack=True, converters={ 0: mdates.strpdate2num('%Y%m%d%H%M%S')})
	date, pm25, pm10 = np.loadtxt(dust_dat, usecols=(0, 3, 7), unpack=True, converters={ 0: yawc})

	if date.size < 4:
		return 0; # not enough points yet. wait for more

	if date.size < npoints:
		npoints = date.size - 1

	f_pts  = date.size - npoints
	t_pts  = date.size

	wx.graph(date[f_pts : t_pts], pm25[f_pts : t_pts], "b-", "Particulate Matter", r'$PM_{2.5}\ (\mu g/m^3)$', plot_d+'dust_pm25.png')
	wx.graph(date[f_pts : t_pts], pm10[f_pts : t_pts], "g-", "Particulate Matter", r'$PM_{10}\ (\mu g/m^3)$', plot_d+'dust_pm10.png')

def gen_index(pm25, pm10):
        plate = wx_dir+"/dust_wx_index.html.template"
        plate_fd = open(plate, 'r')
        plate_dat = plate_fd.read()
        plate_fd.close()

        ts = datetime.datetime.fromtimestamp(time.time()).strftime("%FT%TZ")

        plate_dat = plate_dat.replace("TTTPM25", str("%.2f" % pm25))
        plate_dat = plate_dat.replace("TTTPM10", str("%.2f" % pm10))
        plate_dat = plate_dat.replace("DATE", ts)

        wx.write_out(wx_dir+'/plots/dust_wx.html', plate_dat, 'w')

if __name__ == "__main__":

	dat_fname = 'dust.dat'
	wx.proof_dir(plot_d)

	ser = serial.Serial()
	# ser.port = "/dev/ttyU0"
	ser.port = "/dev/ttyUSB0"
	ser.baudrate = 9600

	ser.open()
	ser.flushInput()

	byte, lastbyte = b"\x00", b"\x00"

	time0 = time1 = time.time()
	pm_25_val = pm_10_val = count = 0

	while True:
		lastbyte = byte
		byte = ser.read(size=1)
    
		# We got a valid packet header
		if lastbyte == b"\xAA" and byte == b"\xC0":
			sentence = ser.read(size=8) # Read 8 more bytes

			check = b"\x00\x00\x00\x00\x00\x00\x00\x00"
			check = struct.unpack('<cccccccc', sentence)

			# check tail byte, should also help eliminate short reads
			if check[7] != b"\xAB":
				print("tail byte not 0xAB")
				continue

			# if it worked move on to checksum
			b = bytearray(sentence[0:7])
			if b[6] != (sum(b[0:6]) % 256):
				print("checksum failed")
				continue

			# Decode the packet - little endian, 2 shorts for pm2.5 and pm10, 2 ID bytes, checksum, message tail
			readings = struct.unpack('<hhxxcc',sentence)
        
			pm_25_val += readings[0]/10.0
			pm_10_val += readings[1]/10.0

			count += 1
			time1 = time.time()
			if((time1 - time0) > 60):
				ts =  datetime.datetime.fromtimestamp(time1).strftime("%Y%m%d%H%M%S")
				pm_25 = pm_25_val / count
				pm_10 = pm_10_val / count
				dat_string = "%s\tPM 2.5: %.2f μg/m³\tPM 10: %.2f μg/m³\n" % (ts, pm_25, pm_10)
				wx.write_out_dat_stamp(ts, dat_fname, dat_string, wx_dir)
				plot(ts, dat_fname)
				gen_index(pm_25, pm_10)
				os.system("/usr/bin/rsync -ur --timeout=55 /home/ghz/dust/* wx3@wx3.slackology.net:/wx3/")
				pm_25_val = pm_10_val = count = 0
				time0 = time1 = time.time()