EQMap.py

#!/usr/bin/env python3
"""
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
from EQEventGatherer import EQEventGathererUSGSVolcanoAlert

# Data Sourcing
use_eu = True
use_usgs = True
eqGathererUSGSVolcano = EQEventGathererUSGSVolcanoAlert()
volcanoAlerts = []

# 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

# Wash/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
largestLOC = ''

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

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

	highestMag, trending, max_location = eventDB.getLargestEvent()
	highestMag = str(highestMag)
	eventCount = eventDB.numberOfEvents()
	displayManager.beginFrame()
	try:

		# 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)

		# Draw active volcano alerts as triangle markers.
		for alert in volcanoAlerts:
			displayManager.displayVolcanoEvent(alert['lon'], alert['lat'])

		# Draw trend graph last so map plotting does not overwrite labels
		displayManager.displayTrendingGraph(eventDB.getDayTrend())
	finally:
		displayManager.endFrame()
	return True

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

		# Display the title/ wash page
		highestMag, trending, max_location = eventDB.getLargestEvent()
		highestMag = str(highestMag)
		EQdayTrend = eventDB.getEQdailyTrend()
		displayManager.displayWashPage(highestMag, str(eventDB.getActiveRegion()), EQdayTrend, max_location)
		# Schedule next title page display
		ftForTitlePageDisplay = millis() + TITLEPAGE_DISPLAY_TIME_MS
	except Exception as e:
		print("Error displaying title page:", e)
		return False

# getUSGS Function
def getUpdatesUSGS():
	global cqIDUSGS,cqLocation,cqLon,cqLat,cqMag,cqDepth,cqTsunami,cqAlert
	# internet check test
	try:
		# Check for new earthquake event
		success = eqGathererUSGS.requestEQEvent()
		if not success:
			return False
	except Exception:
		return False

	# 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, cqAlert, cqTsunami, 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
		success = eqGathererEU.requestEQEvent()
		if not success:
			return False
	except Exception:
		return False
		
	# 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, cqAlert, cqTsunami, cqLocation)

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

			# Display the new EQ data
			repaintMap()

def getUpdatesVolcano():
	global volcanoAlerts
	# Query USGS volcano alert feed.
	try:
		has_alerts = eqGathererUSGSVolcano.requestEQEvent()
	except Exception:
		return False

	if not has_alerts:
		if volcanoAlerts:
			volcanoAlerts = []
			repaintMap()
		return False

	ids = eqGathererUSGSVolcano.getEventIDs()
	locations = eqGathererUSGSVolcano.getLocations()
	lats = eqGathererUSGSVolcano.getLats()
	lons = eqGathererUSGSVolcano.getLons()

	new_alerts = []
	for idx, volcano_id in enumerate(ids):
		lat = lats[idx] if idx < len(lats) else None
		lon = lons[idx] if idx < len(lons) else None
		if lat is None or lon is None:
			continue
		new_alerts.append({
			'id': volcano_id,
			'name': locations[idx] if idx < len(locations) else '',
			'lat': float(lat),
			'lon': float(lon)
		})

	# Repaint only when the alert set changes.
	if new_alerts != volcanoAlerts:
		volcanoAlerts = new_alerts
		repaintMap()

	return bool(volcanoAlerts)

# 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
	global volcanoAlerts

	ftForAcquisition = 0
	ftForBlink = 0
	volcanoAlerts = []
	

	last_db_clear_date = None

	# True if display is on; false if off
	displayState = False
	tForTitlePageDisplay = millis() + TITLEPAGE_DISPLAY_TIME_MS
	#exit loop handler
	running = True

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

			# Reset once per calendar day during the midnight hour.
			# This is independent of display power state, so daily save/clear is reliable.
			if now.hour == 0 and last_db_clear_date != now.date():
				print("DEBUG: Clearing DB at midnight with ", eventDB.numberOfEvents(), " events in the DB")
				eventDB.save()
				eventDB.clear()
				last_db_clear_date = now.date()

			if displayState == False:
				displayManager.backlight(True)
				displayState = True

				# Display the title page
				displayTitlePage()

				# Force a redisplay of all quake data
				repaintMap()

			# Is it time to display the title page ?
			if millis() > ftForTitlePageDisplay and displayState:
				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()

				# Always check the volcano feed during acquisition.
				getUpdatesVolcano()
				# Schedule the next acquisition
				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")
		running = False
	except Exception as e:
		print("Error in main loop:", e)
		running = False
		

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