EQMap.py

"""
Earthquake Display Program
For the Raspberry Pi Model 3B and the official 7" display
Concept, Design by: Craig A. Lindley
"""

import time
from datetime import datetime
from DisplayManager import displayManager
from EQEventGatherer import eqGathererEU
from EQEventGatherer import eqGathererUSGS
from EventDB import eventDB

# Data Sourcing
use_eu = True
use_usgs = True

# Colors for display
BLACK  = (0, 0, 0)
WHITE  = (255, 255, 255)
RED    = (255, 0, 0)
YELLOW = (255, 255, 0)

# Acquire new EQ data every 30 seconds
ACQUISITION_TIME_MS = 30000

# Blink every .5 seconds
BLINK_TIME_MS = 500

# Title page display every 15 minutes
TITLEPAGE_DISPLAY_TIME_MS = 900000

# Times in the future for actions to occur
ftForAcquisition = 0
ftForBlink = 0
ftForTitlePageDisplay = 0

# Current quake data
cqIDUK  = ""
cqIDUSGS = ""
cqLocation = "loading..."
cqLon = 0.0
cqLat = 0.0
cqMag = 0.0
cqDepth = 0.0
cqAlert = None
cqTsunami = 0
dataToggle = False
blinkToggle = False

# Return system millisecond count
def millis():
	return int(round(time.time() * 1000))

# Repaint the map from the events in the DB
def repaintMap():

	highestMag, trending, max_location = eventDB.getLargestEvent()
	highestMag = str(highestMag)
	eventCount = eventDB.numberOfEvents()

	# Display fresh map
	displayManager.displayMap()

	# Display current local time upper left
	displayManager.displayCurrentTime()

	# Display EQ location in color under map
	displayManager.displayEventLong(cqLocation, cqMag, cqDepth)

	# Display map Draw data with event count and date
	displayManager.displayBottomDataFeed(max_location,eventCount)

	# Display EQ depth and last EQ timestamp upper right
	isCluster = cqLocation in (str(eventDB.getActiveRegion(preserve=True)))

	if isCluster and eventCount > 4:
		displayManager.displayDBStats(cqMag, eventCount, highestMag, cqTsunami, cqAlert, cluster=True)
	else:
		displayManager.displayDBStats(cqMag, eventCount, highestMag, cqTsunami, cqAlert)
	

	# Display all of the EQ events in the DB as circles
	count = eventDB.numberOfEvents()
	if count > 0:
		for i in range(count):
			lon, lat, mag, alert, tsunami, location = eventDB.getEvent(i)
			# Color depends upon magnitude
			color = displayManager.colorFromMag(mag)
			displayManager.mapEarthquake(lon, lat, mag, color)
	return True

# Display title page and schedule next display event
def displayTitlePage():

	global ftForTitlePageDisplay

	# Display the title/ wash page
	highestMag, trending, max_location = eventDB.getLargestEvent()
	highestMag = str(highestMag)
	dayTrend = str(eventDB.getDayTrend())
	displayManager.displayWashPage(highestMag, str(eventDB.getActiveRegion()), dayTrend, max_location)

	# Schedule next title page display
	ftForTitlePageDisplay = millis() + TITLEPAGE_DISPLAY_TIME_MS

# getUSGS Function
def getUpdatesUSGS():
	global cqIDUSGS,cqLocation,cqLon,cqLat,cqMag,cqDepth,cqTsunami,cqAlert
	# internet check test
	try:
		# Check for new earthquake event
		eqGathererUSGS.requestEQEvent()
	except:
		pass

	# Determine if we have seen this event before If so ignore it
	if cqIDUSGS != eqGathererUSGS.getEventID():

		#if data has no ID dont use it
		if eqGathererUSGS.getEventID  is None:
			return False
		else:

			# Extract the EQ data
			try:
				cqLocation = eqGathererUSGS.getLocation()
				cqLon = eqGathererUSGS.getLon()
				cqLat = eqGathererUSGS.getLat()
				cqMag = eqGathererUSGS.getMag()
				cqDepth = eqGathererUSGS.getDepth()
				cqTsunami = eqGathererUSGS.getTsunami()
				cqAlert = eqGathererUSGS.getAlert()
			except Exception:
				return False

			# Add new event to DB if it isnt also from the other source
			if not eventDB.checkDupLonLat(cqLon, cqLat):
				eventDB.addEvent(cqLon, cqLat, cqMag, cqTsunami, cqAlert, cqLocation)

				# Update the current event ID
				cqIDUSGS = eqGathererUSGS.getEventID()

				# Display the new EQ data
				repaintMap()
				return cqIDUSGS,cqLocation,cqLon,cqLat,cqMag,cqDepth,cqTsunami,cqAlert

	return False

# getEU Function
def getUpdatesEU():
	global cqIDUK,cqLocation,cqLon,cqLat,cqMag,cqDepth,cqTsunami,cqAlert
	# internet check test
	try:
		# Check for new earthquake event
		eqGathererEU.requestEQEvent()
	except:
		pass
		
	# Determine if we have seen this event before If so ignore it
	if cqIDUK != eqGathererEU.getEventID():
		# Extract the EQ data
		cqLocation = eqGathererEU.getLocation()
		cqLon = eqGathererEU.getLon()
		cqLat = eqGathererEU.getLat()
		cqMag = eqGathererEU.getMag()
		cqDepth = eqGathererEU.getDepth()
		cqTsunami = 0
		cqAlert = None

		# Add new event to DB if it isnt also from the other source
		if not eventDB.checkDupLonLat(cqLon, cqLat):
			eventDB.addEvent(cqLon, cqLat, cqMag, cqTsunami, cqAlert, cqLocation)

			# Update the current event ID
			cqIDUK = eqGathererEU.getEventID()

			# Display the new EQ data
			repaintMap()

# Code execution start
def main():
	# Setup for global variable access
	global ftForAcquisition
	global ftForBlink
	global cqIDUK
	global cqIDUSGS
	global cqLocation
	global cqLon
	global cqLat
	global cqMag
	global cqDepth
	global cqTsunami
	global cqAlert
	global blinkToggle
	global dataToggle
	global largestLOC

	ftForAcquisition = 0
	ftForBlink = 0

	dbCleared = False

	# True if display is on; false if off
	displayState = False

	#exit loop handler
	running = True

	#loop
	try:

		while running:
			
			# Get the current time
			now = datetime.now()

			# Reset the DB at 0:00 so display show EQs per day TODO settings menu
			# And we don't loose the EQ events
			if now.hour == 0 and dbCleared == False:
				eventDB.save()
				eventDB.clear()
				dbCleared = True

			# Now check to see if the display should be off or on TODO settings menu
			if now.hour > 6 and now.hour < 22: 
				# Normal viewing hours have arrived TODO is this working?
				# If display is off, turn it on
				if displayState == False:
					displayManager.backlight(True)
					displayState = True
					dbCleared = False

					# Display the title page
					displayTitlePage()

					# Force a redisplay of all quake data
					repaintMap()
			else:
				# Normal viewing hours over. Turn the display off
				if displayState == True:
					# Turn the display off
					displayManager.backlight(False)
					displayState = False

			# Is it time to display the title page ?
			if millis() > ftForTitlePageDisplay:
				displayTitlePage()

				# Force a redisplay of all quake data
				repaintMap()
				#eventDB.save() #Save Database #DEBUG

			# Is it time to acquire new earthquake data ?
			if millis() > ftForAcquisition:
				# server requests from EU and USGS balanced and dupe checked in function
				if dataToggle:
					dataToggle = False
					if use_usgs:
						getUpdatesUSGS()
					else:
						getUpdatesEU()
						
				else:
					dataToggle = True
					if use_eu:
						getUpdatesEU()
					else:
						getUpdatesUSGS()

				ftForAcquisition = millis() + ACQUISITION_TIME_MS

			# Is it time to blink EQ circle?
			if millis() > ftForBlink:

				if blinkToggle:
					color = displayManager.colorFromMag(cqMag)
					displayManager.mapEarthquake(cqLon, cqLat, cqMag, color)
					blinkToggle = False
				else:
					displayManager.mapEarthquake(cqLon, cqLat, cqMag, BLACK)
					blinkToggle = True
					# Update current display time on the off beat
					displayManager.displayCurrentTime()

				ftForBlink = millis() + BLINK_TIME_MS
	except KeyboardInterrupt:
		print("closing EQMap")
		

# Earthquake Map Program Entry Point
if __name__ == '__main__':
	main()