# script for creating asteroid-searching scheduler files (.esl) for Ekos/kstars
# version 1.0
# Ricardo Serpell 2021
#
# The script creates a file with instructions for ekos to run a sequence of
# captures (predefined in an Ekos sequence file .esq) at adjacent fields

import numpy as np
import math

################################## Options here ################################
width = 90			# field width (RA direction) in minutes 
height = 70			# field height (DEC direction) in minutes
overlap = 10 		# overlaps between fields

cols = 3    		# number of RA columns
rows = 3  			# number of fields along increasing DEC

RA1 = (19, 54)		# RA of first column (hours, minutes)
DE1 = (1, 0)		# DEC of first field in first column (degrees, minutes)

filename = "scan"

# IMPORTANT : profile and sequence names should match those existing in EKOS:
profile = "No GPS - TS 115 f5.6 on HEQ5 (Tak as guider)"
sequence = "/home/astroberry/Asteroid.esq"

# constraints
minimum_altitude = 30	
EnforceTwilight = True 

################################################################################
# in the future I want to allow columns to be located at different RAs #########
# RA2 = (18, 35)
# DE2 = (2, 0)
# RA3 = (18, 40)
# DE3 = (3, 0)
################################################################################

start = (RA1[0]+RA1[1]/60, DE1[0]+DE1[1]/60)

ra_delta = round((((width-overlap)/15)*math.cos(math.radians(DE1[0]+DE1[1]/60)))/60, 4) 
de_delta = round((height-overlap)/60, 4)

for c in range(cols):
	print("Column "+str(c+1)+":")
	for r in range(rows):
		print(start[0]+ra_delta*c, ",", start[1]+de_delta*r)

file = open(filename+"_"+str(RA1[0])+"_"+str(RA1[1])+".esl","w")
file.write('<?xml version="1.0" encoding="UTF-8"?>\n')
file.write("<SchedulerList version='1.4'>\n")
file.write("<Profile>"+profile+"</Profile>\n")

for c in range(cols):
	for i in ['', 'b', 'c']:
		for r in range(rows):
			file.write("<Job>\n")
			file.write("<Name>Field"+str(c+1)+str(r)+i+"</Name>\n")
			file.write("<Priority>10</Priority>\n")
			file.write("<Coordinates>\n")
			file.write("<J2000RA>"+str(start[0]+ra_delta*c)+"</J2000RA>\n")
			file.write("<J2000DE>"+str(start[1]+de_delta*r)+"</J2000DE>\n")
			file.write("</Coordinates>\n")
			file.write("<Sequence>"+sequence+"</Sequence>\n")
			file.write("<StartupCondition>\n")
			file.write("<Condition>ASAP</Condition>\n")
			file.write("</StartupCondition>\n")
			file.write("<Constraints>\n")
			if minimum_altitude>0:
				file.write("<Constraint value='"+str(minimum_altitude)+"'>MinimumAltitude</Constraint>\n")
			if EnforceTwilight:
				file.write("<Constraint>EnforceTwilight</Constraint>\n")
			file.write("</Constraints>\n")
			file.write("<CompletionCondition>\n")
			file.write("<Condition>Sequence</Condition>\n")
			file.write("</CompletionCondition>\n")
			file.write("<Steps>\n")
			file.write("<Step>Track</Step>\n")
			if (i=='')&(r==0):
				file.write("<Step>Focus</Step>\n")
			file.write("<Step>Align</Step>\n")
			file.write("<Step>Guide</Step>\n")
			file.write("</Steps>\n")
			file.write("</Job>\n")

file.write("<ErrorHandlingStrategy value='2'>\n")
file.write("<delay>0</delay>\n")
file.write("</ErrorHandlingStrategy>\n")
file.write("<StartupProcedure>\n")
file.write("</StartupProcedure>\n")
file.write("<ShutdownProcedure>\n")
file.write("<Procedure>WarmCCD</Procedure>\n")
file.write("<Procedure>ParkMount</Procedure>\n")
file.write("</ShutdownProcedure>\n")
file.write("</SchedulerList>\n")

file.close()