Updated FSF address in all files. Fixes ticket:51
[debian/gnuradio] / gr-radio-astronomy / src / python / local_calibrator.py
1 #!/usr/bin/env python
2 #
3 # Copyright 2003,2004,2005 Free Software Foundation, Inc.
4
5 # This file is part of GNU Radio
6
7 # GNU Radio is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 2, or (at your option)
10 # any later version.
11
12 # GNU Radio is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 # GNU General Public License for more details.
16
17 # You should have received a copy of the GNU General Public License
18 # along with GNU Radio; see the file COPYING.  If not, write to
19 # the Free Software Foundation, Inc., 51 Franklin Street,
20 # Boston, MA 02110-1301, USA.
21
22
23 import Numeric
24 import math
25 import ephem
26 import time
27
28 #
29 # Simple class for allowing local definition of a calibration function
30 #  for raw samples coming from the RA detector chain.  Each observatory
31 #  is different, and rather than hacking up the main code in usrp_ra_receiver
32 #  we define the appropriate function here.
33 #
34 # For example, one could calibrate the output in Janskys, rather than
35 #  dB.
36 #
37 #
38
39 def calib_default_total_power(data):
40     r = 10.0*math.log10(data)
41     return(r)
42
43 def calib_numogate_ridge_observatory_total_power(data):
44
45     me = ephem.Observer()
46
47     #
48     # PyEphem wants lat/long as strings, rather than floats--took me quite
49     #  a long time to figure that out.  If they don't arrive as strings,
50     #  the calculations for sidereal time are complete garbage
51     #
52     me.long = str(-76.043)
53     me.lat = str(44.967)
54
55     me.date = ephem.now()
56     sidtime = me.sidereal_time()
57
58     foo = time.localtime()
59     if not "calib_prefix" in globals():
60         pfx = "./"
61     else:
62         pfx = globals()["calib_prefix"]
63     filenamestr = "%s/%04d%02d%02d%02d" % (pfx, foo.tm_year, 
64        foo.tm_mon, foo.tm_mday, foo.tm_hour)
65
66     rainbow_file = open (filenamestr+".tpdat","a")
67   
68     r = (data / 4096.0)
69     flt = "%6.3f" % r
70     #r = calib_default_total_power(data)
71     inter = globals()["calib_decln"]
72     rainbow_file.write(str(ephem.hours(sidtime))+" "+flt+" "+str(inter)+"\n")
73     rainbow_file.close()
74     return(r)
75
76 def calib_numogate_ridge_observatory_fft(data,l):
77
78     me = ephem.Observer()
79
80     #
81     # PyEphem wants lat/long as strings, rather than floats--took me quite
82     #  a long time to figure that out.  If they don't arrive as strings,
83     #  the calculations for sidereal time are complete garbage
84     #
85     me.long = str(-76.043)
86     me.lat = str(44.967)
87
88     me.date = ephem.now()
89     sidtime = me.sidereal_time()
90
91     foo = time.localtime()
92     
93     if not "calib_prefix" in globals():
94         pfx = "./"
95     else:
96         pfx = globals()["calib_prefix"]
97     filenamestr = "%s/%04d%02d%02d%02d" % (pfx, foo.tm_year, 
98        foo.tm_mon, foo.tm_mday, foo.tm_hour)
99
100     now = time.time()
101
102     if not "calib_then" in globals():
103         globals()["calib_then"] = now
104
105     delta = (l/1024)*5
106                 
107     if (now - globals()["calib_then"]) >= delta:
108
109         globals()["calib_then"] = now
110         rainbow_file = open (filenamestr+".sdat","a")
111   
112         r = calib_default_fft(data,l)
113         inter = globals()["calib_decln"]
114         rainbow_file.write("data:"+str(ephem.hours(sidtime))+" "+str(inter)+" "+str(r)+"\n")
115         rainbow_file.close()
116         return(r)
117
118     return(data)
119
120 def calib_default_fft(db,l):
121     return(db)
122
123 #
124 # We capture various parameters from the receive chain here, because
125 #  they can affect the calibration equations.
126 #
127 #
128 def calib_set_gain(gain):
129     globals()["calib_gain_setting"] = gain
130
131 def calib_set_integ(integ):
132     globals()["calib_integ_setting"] = integ
133
134 def calib_set_bw(bw):
135     globals()["calib_bw_setting"] = bw
136
137 def calib_set_freq(freq):
138     globals()["calib_freq_setting"] = freq
139
140 def calib_set_avg_alpha(alpha):
141     globals()["calib_avg_alpha"] = alpha
142
143 def calib_set_interesting(inter):
144     globals()["calib_is_interesting"] = inter
145
146 def calib_set_decln(dec):
147     globals()["calib_decln"] = dec
148
149 def calib_set_prefix(pfx):
150     globals()["calib_prefix"] = pfx