File: query_usno.py

package info (click to toggle)
pysolar 0.10%2Brepack-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 2,804 kB
  • sloc: python: 2,774; makefile: 146
file content (217 lines) | stat: -rw-r--r-- 7,642 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
#!/usr/bin/python3

# Copyright Brandon Stafford
#
# This file is part of Pysolar.
#
# Pysolar 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.
#
# Pysolar 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 Pysolar. If not, see <http://www.gnu.org/licenses/>.

"""Tool for requesting data from US Naval Observatory

"""
import datetime
import random
from time import strptime

try:
  from urllib.request import Request, urlopen
  from urllib.parse import urlencode
except:
  from urllib2 import Request, urlopen
  from urllib import urlencode

from pysolar.solar import *



class Ephemeris:
    def __init__(self, timestamp, latitude, longitude, elevation, azimuth=0, altitude=0):
        self.timestamp = timestamp
        self.latitude = latitude
        self.longitude = longitude
        self.elevation = float(elevation)
        self.azimuth = float(azimuth)
        self.altitude = float(altitude)

class EphemerisComparison:
    def __init__(self, name1, eph1, name2, eph2):
        self.timestamp = eph1.timestamp
        self.latitude = eph1.latitude
        self.longitude = eph1.longitude
        self.elevation = eph1.elevation
        self.name1 = name1
        self.alt1 = eph1.altitude
        self.az1 = eph1.azimuth
        self.name2 = name2
        self.alt2 = eph2.altitude
        self.az2 = eph2.azimuth
        self.alt_error = abs(eph1.altitude - eph2.altitude)
        self.az_error = abs(eph1.azimuth - eph2.azimuth)

def RequestEphemerisData(datum):
    data = EncodeRequest(datum.latitude, datum.longitude, datum.timestamp, datum.elevation)
    url = 'http://aa.usno.navy.mil/cgi-bin/aa_topocentric2.pl'
    if type(data) == str:
      req = Request(url, data.encode())
    else:
      req = Request(url, data)
    response = urlopen(req)

    lines = response.readlines()
    response.close()
    #print(lines)
    #print(lines[21]) # should not we do some try catch here?
    result = lines[21]
    tokens = [x for x in result.split(b' ') if x not in b' ']
    print('Tokens: \n', tokens)

    usno_alt = float(tokens[4]) + float(tokens[5])/60.0 + float(tokens[6])/3600.0
    usno_az = float(tokens[7]) + float(tokens[8])/60.0 + float(tokens[9])/3600.0

#   print(usno_alt)
#   print(usno_az)

    result  = Ephemeris(datum.timestamp, datum.latitude, datum.longitude, datum.elevation, usno_az, usno_alt)

    return result

def ComparePysolarToUSNO(datum):
    alt = get_altitude(float(datum.latitude), float(datum.longitude), datum.timestamp, datum.elevation)
    pysolar_alt = (90.0 - alt)
    az = get_azimuth(float(datum.latitude), float(datum.longitude), datum.timestamp, datum.elevation)
    pysolar_az = (180.0 - az)%360.0

#   print(pysolar_alt)
#   print(pysolar_az)

    py_solar = Ephemeris(datum.timestamp, datum.latitude, datum.longitude, datum.elevation, pysolar_az, pysolar_alt)
    c = EphemerisComparison('pysolar', py_solar, 'usno', datum)
    return c

def EncodeRequest(latitude, longitude, timestamp, elevation):
    """Builds a string of arguments to be passed to the Perl script at the USNO

    Note that the degree arguments must be integers, or the USNO script chokes."""
    params = {}
    params['FFX'] = '2' # use worldwide locations script
    params['ID'] = 'pysolar'
    params['pos'] = '9'
    params['obj'] = '10' # Sun
    params['xxy'] = str(timestamp.year)
    params['xxm'] = str(timestamp.month)
    params['xxd'] = str(timestamp.day)
    params['t1'] = str(timestamp.hour)
    params['t2'] = str(timestamp.minute)
    params['t3'] = str(timestamp.second)
    params['intd'] = '1.0'
    params['unit'] = '1'
    params['rep'] = '1'
    params['place'] = 'Name omitted'

    sign = lambda x: ('1', '-1')[x < 0]
    (deg, rem) = divmod(longitude, 1)
    (min, sec) = divmod(rem, 1.0/60.0)
    params['xx0'] = sign(deg)# longitude (1 = east, -1 = west)
    params['xx1'] = str(abs(int(deg))) # degrees
    params['xx2'] = str(int(min)) # minutes
    params['xx3'] = str(sec) # seconds

    (deg, rem) = divmod(latitude, 1)
    (min, sec) = divmod(rem, 1.0/60.0)
    params['yy0'] = sign(deg) # latitude (1 = north, -1 = south)
    params['yy1'] = str(abs(int(deg))) # degrees
    params['yy2'] = str(int(min)) # minutes
    params['yy3'] = str(sec) # seconds

    params['hh1'] = str(elevation) # height above sea level in meters
    params['ZZZ'] = 'END'
    data = urlencode(params)
    return data

def GatherRandomEphemeris():
    latitude = random.randrange(-90, 90)
    longitude = random.randrange(0, 360)
    elevation = 0.0
    t = datetime.datetime(random.randrange(2013,2014), random.randrange(1, 13), random.randrange(1, 28), random.randrange(0, 24), random.randrange(0, 60), random.randrange(0,60))
    query = Ephemeris(t, latitude, longitude, elevation)
    PrintEphemerisDatum(query)
    d = RequestEphemerisData(query)
    PrintEphemerisDatum(d)
    WriteEphemerisDatumToFile(d, 'usno_data.txt')

def PrintEphemerisDatum(datum):
    print(datum.timestamp, datum.latitude, datum.longitude, datum.elevation, datum.azimuth, datum.altitude)

def ReadEphemeridesLog(logname):
    data = []
    log = open(logname, 'r')
    lines = log.readlines()
    log.close()
    for line in lines:
        args = line.split(' ')
        unaware = datetime.datetime(*(strptime(args[0] + ' ' + args[1], '%Y-%m-%d %H:%M:%S')[0:6]))
        d = unaware.replace(tzinfo=datetime.timezone.utc)
        e = Ephemeris(d, args[2], args[3], args[4], args[5], args[6])
        data.append(e)
    return data

def WriteEphemerisDatumToFile(d, filename):
    log = open(filename, 'a')
    log.write('%s %s %s %s %s %s\n' % (d.timestamp, d.latitude, d.longitude, d.elevation, d.azimuth, d.altitude))
    log.close()

def WriteComparisonsToCSV(comps, filename):
    out = open(filename, 'w')
    out.write('timestamp,latitude,longitude,elevation,alt1,alt2,alt_error,az1,az2,az_error\n')
    for c in comps:
        out.write('%s,%s,%s,%s,%s,%s,%s,%s,%s,%s\n' % (c.timestamp, c.latitude, c.longitude, c.elevation, c.alt1, c.alt2, c.alt_error, c.az1, c.az2, c.az_error))
    out.close()


if __name__ == '__main__':
    from scipy import stats
    import numpy as np
    import sys

    if len(sys.argv) >= 2:
                ephemerides = ReadEphemeridesLog(sys.argv[1])
    else:
        for i in range(100):
            GatherRandomEphemeris()
            ephemerides = ReadEphemeridesLog('usno_data.txt')

    comps = []
    for e in ephemerides:
        c = ComparePysolarToUSNO(e)
        comps.append(c)

    az_errors = np.array([c.az_error for c in comps])
    alt_errors = np.array([c.alt_error for c in comps])

    print('---------------------')
    print('Azimuth stats')
    print('Mean error: ' + str(np.mean(az_errors)))
    print('Std dev: ' + str(np.std(az_errors)))
    print('Min error: ' + str(stats.tmin(az_errors, None)))
    print('Max error: ' + str(stats.tmax(az_errors, None)))

    print('----------------------')
    print('Altitude stats')

    print('Mean error: ' + str(np.mean(alt_errors)))
    print('Std dev: '+ str(np.std(alt_errors)))
    print('Min error: ' + str(stats.tmin(alt_errors, None)))
    print('Max error: ' + str(stats.tmax(alt_errors, None)))

    WriteComparisonsToCSV(comps, 'pysolar_v_usno.csv')