pyorbital_sims.py

################################################################################
# Copyright 2015 Samuel Gongora Garcia (s.gongoragarcia@gmail.com)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
################################################################################
# Author: s.gongoragarcia[at]gmail.com
################################################################################

class Do_list:

	def __init__(self):

		import sys
		from os import getcwd
		actual_dir = getcwd()
		
		file = actual_dir + '/TLEs/' + sys.argv[1]
				
		open_tle = open(file, 'r')
		lista_nombres_satelites = open_tle.readlines()
		lista_nombres_satelites = [item.rstrip('\n') for item in lista_nombres_satelites]
		
		length_list = len(lista_nombres_satelites)
		y = length_list/3

		list_numbers = map(self.return_list, range(y))

		self.mostrar_lista_satelites = []
		self.mostrar_lista_linea1 = []
		self.mostrar_lista_linea2 = []
		i = 0
		j = 1
		k = 2
		
		for i in range(len(list_numbers)):
			self.mostrar_lista_satelites.append(lista_nombres_satelites[list_numbers[i]])
			self.mostrar_lista_linea1.append(lista_nombres_satelites[j])
			self.mostrar_lista_linea2.append(lista_nombres_satelites[k])
			j = list_numbers[i] + 4
			k = list_numbers[i] + 5
			
			self.return_values()

	def return_list(self, x):
		
		return 3*x

	def return_values(self):
		
		return self.mostrar_lista_satelites
		return self.mostrar_lista_linea1
		return self.mostrar_lista_linea2


class Solve_coordinates:

	def __init__(self, lista_elementos, lista_prueba, lista_prueba2):

		satellites_number = len(lista_elementos)
		self.get_location()
		
		# Provide data to pyephem_routine
		for i in range(len(lista_elementos)):
			j = i + 1
			try:
				self.pyorbital_routine(lista_elementos[i], lista_prueba[i], lista_prueba2[i], i, satellites_number)
			except NotImplementedError:
				print "pyorbital - Simulation [%d/%d] error!" %(j, len(lista_elementos))
				print "Deep space satellite - Propagation not available"
			i = i + 1

	def pyorbital_routine(self, satellite_name, tle_line1, tle_line2, i, satellites_number):

		import sys
		import pyorbital.orbital
		import datetime

		satellite = pyorbital.orbital.Orbital(satellite_name, tle_file = None, line1 = tle_line1, line2 = tle_line2)
		
		start_time = int(sys.argv[2])
		end_time = int(sys.argv[3])

		iterations = end_time - start_time
		iterations = iterations - 1
	
 		time1 = datetime.datetime.utcfromtimestamp(start_time)

		(lon, lat, ele) = self.get_location()

		az1, alt1 = satellite.get_observer_look(time1, lon, lat, ele)

		if alt1 > 0:
			self.output_data(satellite_name, start_time, alt1, az1)

		n2 = start_time

		j = 0

		for j in range(iterations):
			import datetime
			n2 = n2 + 1

			timeN = datetime.datetime.utcfromtimestamp(n2)

			azN, altN = satellite.get_observer_look(timeN, lon, lat, ele)

			if altN > 0:
				self.output_data(satellite_name, n2, altN, azN)

			j = j + 1


		i = i + 1
		print "PyOrbital - Simulation [%s/%d] done!" %(i, satellites_number)

	def output_data(self, name, time, alt, az):

		from os import getcwd, chdir
		script_dir = getcwd()
		chdir(script_dir + '/results/PyOrbital')

		create_file = open(name, 'a')
		create_file.writelines("%d\t" % time)
		create_file.writelines("%0.6f\t" % alt)
		create_file.writelines("%0.6f\n" % az)
		create_file.close
		chdir(script_dir)

	def get_location(self):
		
		from os import getenv

		open_file = open(getenv("HOME") + '/.predict/predict.qth')
		lines = open_file.readlines()
		lines = [item.rstrip('\n') for item in lines]

		site = lines[0]
		lat = float(lines[1])
		lon = '-' + lines[2]
		print lon
		lon = float(lines[2])
		print lon
		ele = int(lines[3])

		return lon, lat, ele


if __name__ == '__main__':
	print ""
	print "PyOrbital data"
	do_list = Do_list()

	solve_coordinates = Solve_coordinates(do_list.mostrar_lista_satelites, do_list.mostrar_lista_linea1, do_list.mostrar_lista_linea2)