picorder.py

from os import system
import curses
import time
import datetime
import os
import random
from sense_hat import SenseHat
from time import sleep
import numpy as np
import RPi.GPIO as GPIO

GPIO.setmode(GPIO.BCM)
GPIO.setup(11,GPIO.OUT) #led Red
GPIO.setup(9,GPIO.OUT) #led Green
GPIO.setup(10,GPIO.OUT) #led Yellow

GPIO.output(11, GPIO.HIGH)
GPIO.output(9, GPIO.HIGH)
GPIO.output(10, GPIO.HIGH)

sense = SenseHat()
os.environ["NCURSES_NO_UTF8_ACS"] = "1"
curses.initscr()
curses.halfdelay(3)
curses.start_color()
curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_BLUE)
curses.init_pair(3, curses.COLOR_BLACK, curses.COLOR_YELLOW)
curses.init_pair(4, curses.COLOR_BLUE, curses.COLOR_WHITE)

start_time = time.time()
start_hour = time.time()

# get CPU temperature
def get_cpu_temp():
    res = os.popen("vcgencmd measure_temp").readline()
    t = float(res.replace("temp=","").replace("'C\n",""))
    return(t)

# use moving average to smooth readings
def get_smooth(x):
    if not hasattr(get_smooth, "t"):
        get_smooth.t = [x,x,x]
    get_smooth.t[2] = get_smooth.t[1]
    get_smooth.t[1] = get_smooth.t[0]
    get_smooth.t[0] = x
    xs = (get_smooth.t[0]+get_smooth.t[1]+get_smooth.t[2])/3
    return(xs)

# calculates the real temperature compesating CPU heating
def get_temp():
    t1 = sense.get_temperature_from_humidity()
    t2 = sense.get_temperature_from_pressure()
    t_cpu = get_cpu_temp()
    h = sense.get_humidity()
    p = sense.get_pressure()
    t = (t1+t2)/2
    t_corr = t - ((t_cpu-t)/1.25)
    t_corr = get_smooth(t_corr)
    f_corr = round(((t_corr * 9)/5)+32,1)
    return(f_corr)

def sendlog():
        os.system("mpack -s picorder -a /home/pi/picordlog.txt d10d3.email@gmail.com")

def logfile(logstring):
    global start_time
    global start_hour
    current_time=time.time()
    if current_time>start_time+180:
        x = datetime.datetime.now()
        tstamp = datetime.datetime(x.year,x.month,x.day,x.hour,x.minute)
        #local_time=time.ctime(current_time)
        f= open("/home/pi/picordlog.txt","a+")
        logstring=str(tstamp)+","+logstring
        f.write(logstring+"\r\n")
        f.close()
        start_time = time.time()
    if current_time>start_hour+3600:
        start_hour = time.time()
        sendlog()

def BlinkinLights():
    grid = []
    for i in range(0,64):
        n = random.randint(0, 4)
        if n == 0:
            r = random.randint(0,100)
            b = random.randint(0,100)
            g = random.randint(0,100)
            grid.append ([r,g,b])
        else:
            grid.append ([0,0,0])
    sense.set_pixels(grid)


def GetTemp_old():
    t = sense.get_temperature()
    t = round(t, 1)
    fconv = ((t * 9)/5)+32
    f = round(fconv,1)
    sleep (0.5)
    return f

def display():
    k = 0
    #background Window
    bgw = curses.newwin(12,37,0,0)
    bgw.border(0)
    bgw.attron(curses.color_pair(4))
    for x in range(0,37):
        for y in range(0,12):
            try:
                bgw.addch(y,x," ")
            except curses.error:
                pass
    bgw.refresh()
    #Disply loop
    while (k != ord('q')):
        BlinkinLights()
        #Temperature
        t = get_temp()
        win_t = curses.newwin(3, 32, 0, 2)
        win_t.attron(curses.color_pair(1))
        win_t.border(0)
        rangex = (30 / float(120)) * t
        pos = int(rangex)
        if pos<1:
            pos = 0
        display = ' '
        if pos != 0:
            win_t.addstr(1, 1, "{}".format(display*pos),curses.color_pair(3))
        win_t.addstr(0,2,"Temp(f): " + str(t))
        win_t.refresh()
        #Pressure
        p = sense.get_pressure()
        p = round(p, 1)
        win_p = curses.newwin(3, 32, 3, 2)
        win_p.attron(curses.color_pair(1))
        win_p.border(0)
        rangex = (30 / float(1260)) * p
        pos = int(rangex)
        display = ' '
        if pos != 0:
            win_p.addstr(1, 1, "{}".format(display*pos),curses.color_pair(3))
        win_p.addstr(0,2,"Pressure(mb): " + str(p))
        win_p.refresh()
        #Humidity
        h = sense.get_humidity()
        h = round(h, 1)
        win_h = curses.newwin(3, 32, 6, 2)
        win_h.attron(curses.color_pair(1))
        win_h.border(0)
        rangex = (30 / float(100)) * h
        pos = int(rangex)
        display = ' '
        if pos != 0:
            win_h.addstr(1, 1, "{}".format(display*pos),curses.color_pair(3))
        win_h.addstr(0,2,"Humidity(%): " + str(h))
        win_h.refresh()
        #Magnetics
        raw = sense.get_compass_raw()
        mx = round(raw["x"],5)
        my = round(raw["y"],5)
        mz = round(raw["z"],5)
        m_av = round((mx+my+mz)/3,2)
        win_m = curses.newwin(3, 32, 9, 2)
        win_m.attron(curses.color_pair(1))
        win_m.border(0)
        rangex = (30 / float(800)) * (h+600)
        pos = int(rangex)
        display = ' '
        if pos != 0:
            win_m.addstr(1, 1, "{}".format(display*pos),curses.color_pair(3))
        win_m.addstr(0,2,"Ave. Magnetic Flux(uT): " + str(m_av))
        win_m.refresh()
        #logging
        logstring =str(t)+","+str(p)+","+str(h)+","+str(m_av)
        logfile(logstring)
        k = win_t.getch()
        sleep(5)

def menu():
    k = 0
    #background Window
    win = curses.newwin(12,37,0,0)
    win.border(0)
    win.attron(curses.color_pair(4))
    while (k != ord('q')):
        for x in range(0,37):
                for y in range(0,12):
                        try:
                                win.addch(y,x," ")
                        except curses.error:
                                pass
        win.addstr(3,3,"T: Display Temperature",curses.color_pair(2))
        win.addstr(4,3,"Q: Quit",curses.color_pair(2))
        win.refresh()
        k = win.getch()
        if k == ord('t'):
            temp_graph()
try:
    display()
except curses.error:
    pass

#Close down curses properly, inc turn echo back on!
curses.nocbreak(); curses.echo()
curses.endwin()
GPIO.cleanup() # cleanup all GPIO