#!/usr/bin/env python

import re, datetime, collections

bodyss = [
 {'name':'Mercury'     ,'is_probe':False,'is_cohoweb':True },
 {'name':'Venus'       ,'is_probe':False,'is_cohoweb':True },
 {'name':'Earth'       ,'is_probe':False,'is_cohoweb':True },
 {'name':'Mars'        ,'is_probe':False,'is_cohoweb':True },
 {'name':'Jupiter'     ,'is_probe':False,'is_cohoweb':True },
 {'name':'Saturn'      ,'is_probe':False,'is_cohoweb':True },
 {'name':'Uranus'      ,'is_probe':False,'is_cohoweb':True },
 {'name':'Neptune'     ,'is_probe':False,'is_cohoweb':True },
 {'name':'Pluto'       ,'is_probe':False,'is_cohoweb':True },
 {'name':'Ultima Thule','is_probe':False,'is_cohoweb':False},
 {'name':'Pioneer 10'  ,'is_probe':True ,'is_cohoweb':True },
 {'name':'Pioneer 11'  ,'is_probe':True ,'is_cohoweb':True },
 {'name':'Voyager 2'   ,'is_probe':True ,'is_cohoweb':True },
 {'name':'Voyager 1'   ,'is_probe':True ,'is_cohoweb':True },
 {'name':'New Horizons','is_probe':True ,'is_cohoweb':True },
]
yyyymmdd_min = '19700101'
yyyymmdd_max = '20300101'

def tabbify(cellss, separator='|'):
 cellpadss = [list(rows) + [''] * (len(max(cellss, key=len)) - len(rows)) for rows in cellss]
 fmts = ['%%%ds' % (max([len(str(cell)) for cell in cols])) for cols in zip(*cellpadss)]
 return '\n'.join([separator.join(fmts) % tuple(rows) for rows in cellpadss])

## Convert (year, day of year) to (month, day)

month_day_nonleaps = [None]
month_day_leaps    = [None]
for doy in range(1, 365 + 1):
 dt = datetime.datetime.strptime('2001 %d' % (doy), '%Y %j')
 month_day_nonleaps.append((dt.month, dt.day))
for doy in range(1, 366 + 1):
 dt = datetime.datetime.strptime('2000 %d' % (doy), '%Y %j')
 month_day_leaps.append((dt.month, dt.day))
# print(month_day_nonleaps)
# print(month_day_leaps)
def is_leap(year):
 y = year // 100 if (year % 100 == 0) else year
 return y % 4 == 0
def year_doy_to_month_day(year, doy):
 return (month_day_leaps if (is_leap(year)) else month_day_nonleaps)[doy]
# print(year_doy_to_month_day(2018, 365))

## Convert month name to month number

months = collections.OrderedDict()
for month in range(1, 12 + 1):
 index = (month - 1) * 3
 months['JanFebMarAprMayJunJulAugSepOctNovDec'[index:index + 3].upper()] = month
print(months)

## Read files

datasss = {} ## [yyyymmdd][name][au,lat,lon]
for bodys in bodyss:
 name       = bodys['name']
 is_cohoweb = bodys['is_cohoweb']
 is_probe   = bodys['is_probe']
 path = '%s/%s.txt' % ('cohoweb' if (is_cohoweb) else 'horizons',
                       name.replace(' ', '').lower())
 with open(path) as f: lines = [line.strip() for line in f.readlines()]
 # lines = lines[:100]
 is_launch = is_probe # True
 for line in lines:
  if (not re.findall(r'^\d', line)): continue
  fields = line.split()
  if (is_cohoweb):
   year         = int  (fields[0])
   doy          = int  (fields[1])
   au           = float(fields[-3])
   lat          = float(fields[-2])
   lon          = float(fields[-1])
   (month, day) = year_doy_to_month_day(year, doy)
  else: ## not cohoweb
   ymd                =       fields[0]
   if (not '-' in ymd): continue
   au                 = float(fields[4])
   lon                = float(fields[2])
   lat                = float(fields[3])
   (year, month, day) = ymd.split('-')
   (year, month, day) = (int(year), months[month.upper()], int(day))
  if (not (is_launch or day == 1)): continue
  is_launch = False
  yyyymmdd = '%04d%02d%02d' % (year, month, day)
  if (yyyymmdd < yyyymmdd_min): continue
  if (yyyymmdd > yyyymmdd_max): continue
  if (not yyyymmdd in datasss): datasss[yyyymmdd] = {}
  datasss[yyyymmdd][name] = (au, lat, lon)

## Create and write table

outss = [['BODY->'  ] + [bodys['name']     for bodys in bodyss],
         ['YYYYMMDD'] + ['AU  SElat SElon' for bodys in bodyss]]
for yyyymmdd in sorted(datasss):
 outs   = [yyyymmdd]
 datass = datasss[yyyymmdd]
 for bodys in bodyss:
  name = bodys['name']
  outs.append('%.2f%+5.1f%6.1f' % datass[name] if (name in datass) else '')
 outss.append(outs)
# outss = outss[:80]

with open(re.sub(r'py$', r'txt', __file__), 'w') as f:
 f.write('%s\n' % (tabbify(outss)))