#!/usr/bin/python

import RPi.GPIO as GPIO
import time
import serial
import sys
import getopt
import string 
import datetime
import select
import termios
import sys
import tty
import os
import socket

def isData():
	return select.select([sys.stdin], [], [], 0) == ([sys.stdin], [], [])

def separator():
	print(" ")	
	print("==============================================================================")
	print(" ")	

def myip():
	try:
		s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
		s.connect(('8.8.8.8', 1))  # connect() for UDP doesn't send packets
		local_ip_address = s.getsockname()[0]
		return local_ip_address
	except:
		return "NO IP"	

INP_LEFT=28
INP_RIGHT=29
RELAY_LEFT=21
RELAY_RIGHT=22
WIFI_POWER=37

RED_LED=36
GREEN_LED=35
BLUE_LED=34

GROVE_1_SDA=44
GROVE_1_SCL=45

GPIO.setmode(GPIO.BCM)
GPIO.setwarnings(False)
GPIO.setup(RELAY_LEFT,GPIO.OUT)
GPIO.setup(RELAY_RIGHT,GPIO.OUT)
GPIO.setup(WIFI_POWER,GPIO.OUT)
GPIO.setup(INP_LEFT,GPIO.IN)
GPIO.setup(INP_RIGHT,GPIO.IN)

GPIO.setup(RED_LED,GPIO.OUT)
GPIO.setup(GREEN_LED,GPIO.OUT)
GPIO.setup(BLUE_LED,GPIO.OUT)

GPIO.output(RED_LED,GPIO.HIGH)
GPIO.output(GREEN_LED,GPIO.HIGH)
GPIO.output(BLUE_LED,GPIO.HIGH)

old_settings = termios.tcgetattr(sys.stdin)

try:
	tty.setcbreak(sys.stdin.fileno())

	step=0

	print("***********************")
	print(" Megatest per CM-Home ")
	print("***********************")

	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: FT4232H eeprom programming" % (step))
	print("  p=Program l='ls /dev/ttyUSB*' n=Next")

	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="p":
				os.system("sudo ftdi_eeprom --flash-eeprom cmhome.conf")

			if c=="l":
				os.system("ls /dev/ttyUSB*")

	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: RGB led" % (step))
	print("  g=Go n=Next")

	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="g":
				for i in range(2):
					GPIO.output(RED_LED,GPIO.LOW)
					GPIO.output(GREEN_LED,GPIO.HIGH)
					GPIO.output(BLUE_LED,GPIO.HIGH)
					time.sleep(0.5)

					GPIO.output(RED_LED,GPIO.HIGH)
					GPIO.output(GREEN_LED,GPIO.LOW)
					GPIO.output(BLUE_LED,GPIO.HIGH)
					time.sleep(0.5)

					GPIO.output(RED_LED,GPIO.HIGH)
					GPIO.output(GREEN_LED,GPIO.HIGH)
					GPIO.output(BLUE_LED,GPIO.LOW)
					time.sleep(0.5)

				GPIO.output(RED_LED,GPIO.HIGH)
				GPIO.output(GREEN_LED,GPIO.HIGH)
				GPIO.output(BLUE_LED,GPIO.HIGH)

	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: Inp 1/2 test" % (step))
	print("  g=Go n=Next")

	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break
				
			if c=="g":
				print("Close the input 1 and 2 lines")
				lt=0
				rt=0
				while True:
					if not GPIO.input(INP_LEFT):
						lt=1
						GPIO.output(RELAY_LEFT,GPIO.HIGH)
						while not GPIO.input(INP_LEFT):
							time.sleep(0.1)
							pass
					else:
						GPIO.output(RELAY_LEFT,GPIO.LOW)

					if not GPIO.input(INP_RIGHT):
						rt=1
						GPIO.output(RELAY_RIGHT,GPIO.HIGH)
						while not GPIO.input(INP_RIGHT):
							time.sleep(0.1)
							pass
					else:
						GPIO.output(RELAY_RIGHT,GPIO.LOW)

					if lt==1 and rt==1:
						GPIO.output(RELAY_LEFT,GPIO.LOW)
						GPIO.output(RELAY_RIGHT,GPIO.LOW)
						print("Test Inp 1/2 end. To repeat it press 'g'")
						break


	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: Port Grove 1 - GPIO mode" % (step))
	print("  g=Go n=Next")

	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="g":
			
				GPIO.setup(GROVE_1_SDA,GPIO.OUT)
				GPIO.setup(GROVE_1_SCL,GPIO.OUT)
				GPIO.output(GROVE_1_SDA,GPIO.LOW)
				GPIO.output(GROVE_1_SCL,GPIO.LOW)

				for i in range(2):
					GPIO.output(GROVE_1_SDA,GPIO.HIGH)
					GPIO.output(GROVE_1_SCL,GPIO.LOW)

					time.sleep(0.5)

					GPIO.output(GROVE_1_SDA,GPIO.LOW)
					GPIO.output(GROVE_1_SCL,GPIO.HIGH)

					time.sleep(0.5)

				GPIO.setup(GROVE_1_SDA,GPIO.IN)
				GPIO.setup(GROVE_1_SCL,GPIO.IN)

	#*************************************************************************************



	separator()
	step=step+1
	print("Step %02d: Camera" % (step))
	print("  g=Go n=Next")

	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="g":
				os.system("CAMERA=pi python /home/pi/flask-video-streaming/app.py &")
				print("http://%s:5000" % myip())

	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: Serial 3V3 test" % (step))
	print("  g=Go l='ls /dev/ttyUSBx' n=Next")


	test_running=False
	tx_counter=0
	rx_counter=0
	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				if test_running==True:
					serial_usb.close()
					serial_ttl.close()
				break

			if c=="l":
				os.system("ls /dev/ttyUSB*")

			if c=="g":
				serial_usb = serial.Serial(
					port='/dev/ttyUSB4',
					baudrate=115200,
					timeout=1,
					parity=serial.PARITY_NONE,
					stopbits=serial.STOPBITS_ONE,
					bytesize=serial.EIGHTBITS
				)
				serial_usb.flushOutput()
				serial_usb.flushInput()

				serial_ttl = serial.Serial(
					port='/dev/ttyUSB3',
					baudrate=115200,
					timeout=0.1,
					parity=serial.PARITY_NONE,
					stopbits=serial.STOPBITS_ONE,
					bytesize=serial.EIGHTBITS
				)
				serial_ttl.flushOutput()
				serial_ttl.flushInput()

				test_running=True

		if test_running==True:	
			tx_counter=tx_counter+1
			serial_usb.write("USB to Serial: %08d".encode('utf-8') % (tx_counter))
			time.sleep(0.1)
			print(serial_ttl.read(23))

			rx_counter=rx_counter+1
			serial_ttl.write("Serial to USB: %08d".encode('utf-8') % (rx_counter))
			time.sleep(0.1)
			print(serial_usb.read(23))

			if tx_counter==10:
				test_running=False

					
	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: RS485-left test" % (step))
	print("  g=Start l='ls /dev/ttyUSBx' n=Next")

	test_running=False
	tx_counter=0
	rx_counter=0
	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				if test_running==True:
					serial_usb.close()
					serial_ttl.close()
				break

			if c=="l":
				os.system("ls /dev/ttyUSB*")

			if c=="g":
				serial_usb = serial.Serial(
					port='/dev/ttyUSB5',
					baudrate=115200,
					timeout=1,
					parity=serial.PARITY_NONE,
					stopbits=serial.STOPBITS_ONE,
					bytesize=serial.EIGHTBITS
				)
				serial_usb.flushOutput()
				serial_usb.flushInput()

				serial_ttl = serial.Serial(
					port='/dev/ttyUSB0',
					baudrate=115200,
					timeout=0.1,
					parity=serial.PARITY_NONE,
					stopbits=serial.STOPBITS_ONE,
					bytesize=serial.EIGHTBITS
				)
				serial_ttl.flushOutput()
				serial_ttl.flushInput()

				test_running=True

		if test_running==True:	
			tx_counter=tx_counter+1
			serial_usb.write("USB to RS485L: %08d".encode('utf-8') % (tx_counter))
			time.sleep(0.1)
			print(serial_ttl.read(23))

			rx_counter=rx_counter+1
			serial_ttl.write("RS485L to USB: %08d".encode('utf-8') % (rx_counter))
			time.sleep(0.1)
			print(serial_usb.read(23))
			
			if tx_counter==10:
				test_running=False

					
	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: RS485-right test" % (step))
	print("  g=Start l='ls /dev/ttyUSBx' n=Next")

	test_running=False
	tx_counter=0
	rx_counter=0
	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				if test_running==True:
					serial_usb.close()
					serial_ttl.close()
				break

			if c=="l":
				os.system("ls /dev/ttyUSB*")

			if c=="g":
				serial_usb = serial.Serial(
					port='/dev/ttyUSB5',
					baudrate=115200,
					timeout=1,
					parity=serial.PARITY_NONE,
					stopbits=serial.STOPBITS_ONE,
					bytesize=serial.EIGHTBITS
				)
				serial_usb.flushOutput()
				serial_usb.flushInput()

				serial_ttl = serial.Serial(
					port='/dev/ttyUSB2',
					baudrate=115200,
					timeout=0.1,
					parity=serial.PARITY_NONE,
					stopbits=serial.STOPBITS_ONE,
					bytesize=serial.EIGHTBITS
				)
				serial_ttl.flushOutput()
				serial_ttl.flushInput()

				test_running=True

		if test_running==True:	
			tx_counter=tx_counter+1
			serial_usb.write("USB to RS485R: %08d".encode('utf-8') % (tx_counter))
			time.sleep(0.1)
			print(serial_ttl.read(23))

			rx_counter=rx_counter+1
			serial_ttl.write("RS485R to USB: %08d".encode('utf-8') % (rx_counter))
			time.sleep(0.1)
			print(serial_usb.read(23))

			if tx_counter==10:
				test_running=False

					
	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: WiFi test" % (step))
	print("  1=On 0=Off i=ifconfig n=Next")

	test_running=False
	tx_counter=0
	rx_counter=0
	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="1":
				GPIO.output(WIFI_POWER,GPIO.HIGH)
				continue

			if c=="0":
				GPIO.output(WIFI_POWER,GPIO.LOW)
				continue

			if c=="i":
				os.system("ifconfig")
				continue

	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: Audio test" % (step))
	print("  g=Go n=Next")

	test_running=False
	tx_counter=0
	rx_counter=0
	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="g":
				os.system("aplay speaker_test.wav")
				continue
					
	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: 1-wire test" % (step))
	print("  g=Go n=Next")

	test_running=False
	tx_counter=0
	rx_counter=0
	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="g":
				os.system("ls /sys/bus/w1/devices/")
				continue
					
	#*************************************************************************************

	separator()
	step=step+1
	print("Step %02d: Ethernet test" % (step))
	print("  g=Go n=Next")

	test_running=False
	tx_counter=0
	rx_counter=0
	while True:
		if isData():
			c = sys.stdin.read(1)

			if c=="n":
				break

			if c=="g":
				os.system("ping -c 4 www.acmesystems.it")
				continue

	#*************************************************************************************


finally:
	termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_settings)