1616# You should have received a copy of the GNU General Public License
1717# along with this program. If not, see <http://www.gnu.org/licenses/>.
1818
19+ from __future__ import division
20+ from __future__ import print_function
21+
1922from builtins import str
2023from builtins import range
2124from builtins import object
2528import ehtim .observing .obs_simulate as simobs
2629import ehtim .io .save
2730import ehtim .io .load
31+ import ehtim .const_def as ehc
2832
29- from ehtim .const_def import *
30- from ehtim .observing .obs_helpers import *
31-
32- ###########################################################################################################################################
33+ ###################################################################################################
3334# Array object
34- ###########################################################################################################################################
35+ ###################################################################################################
36+
37+
3538class Array (object ):
39+
3640 """A VLBI array of telescopes with site locations, SEFDs, and other data.
3741
3842 Attributes:
3943 tarr (numpy.recarray): The array of telescope data with datatype DTARR
4044 tkey (dict): A dictionary of rows in the tarr for each site name
41- ephem (dict): A dictionary of TLEs for each space antenna, Space antennas have x=y=z=0 in the tarr
45+ ephem (dict): A dictionary of TLEs for each space antenna,
46+ Space antennas have x=y=z=0 in the tarr
4247 """
4348
4449 def __init__ (self , tarr , ephem = {}):
@@ -47,39 +52,40 @@ def __init__(self, tarr, ephem={}):
4752
4853 # check to see if ephemeris is correct
4954 for line in self .tarr :
50- if np .any (np .isnan ([line ['x' ],line ['y' ],line ['z' ]])):
55+ if np .any (np .isnan ([line ['x' ], line ['y' ], line ['z' ]])):
5156 sitename = str (line ['site' ])
5257 try :
5358 elen = len (ephem [sitename ])
5459 except NameError :
55- raise Exception ('no ephemeris for site %s !' % sitename )
60+ raise Exception ('no ephemeris for site %s !' % sitename )
5661 if elen != 3 :
5762
58- raise Exception ('wrong ephemeris format for site %s !' % sitename )
63+ raise Exception ('wrong ephemeris format for site %s !' % sitename )
5964
6065 # Dictionary of array indices for site names
6166 self .tkey = {self .tarr [i ]['site' ]: i for i in range (len (self .tarr ))}
6267
6368 def listbls (self ):
6469 """List all baselines.
65-
70+
6671 Args:
67- Returns:
72+ Returns:
6873 numpy.array : array of baselines
6974 """
7075
7176 bls = []
7277 for i1 in sorted (self .tarr ['site' ]):
7378 for i2 in sorted (self .tarr ['site' ]):
74- if not ([i1 ,i2 ] in bls ) and not ([i2 ,i1 ] in bls ) and i1 != i2 :
75- bls .append ([i1 ,i2 ])
76-
77- return np .array (bls )
79+ if not ([i1 , i2 ] in bls ) and not ([i2 , i1 ] in bls ) and i1 != i2 :
80+ bls .append ([i1 , i2 ])
81+ bls = np .array (bls )
7882
79- def obsdata (self , ra , dec , rf , bw , tint , tadv , tstart , tstop , mjd = MJD_DEFAULT , timetype = 'UTC' ,
80- polrep = 'stokes' , elevmin = ELEV_LOW , elevmax = ELEV_HIGH ,
81- tau = TAUDEF , fix_theta_GMST = False ):
83+ return bls
8284
85+ def obsdata (self , ra , dec , rf , bw , tint , tadv , tstart , tstop ,
86+ mjd = ehc .MJD_DEFAULT , timetype = 'UTC' , polrep = 'stokes' ,
87+ elevmin = ehc .ELEV_LOW , elevmax = ehc .ELEV_HIGH ,
88+ tau = ehc .TAUDEF , fix_theta_GMST = False ):
8389 """Generate u,v points and baseline uncertainties.
8490
8591 Args:
@@ -102,23 +108,23 @@ def obsdata(self, ra, dec, rf, bw, tint, tadv, tstart, tstop, mjd=MJD_DEFAULT, t
102108 """
103109
104110 obsarr = simobs .make_uvpoints (self , ra , dec , rf , bw ,
105- tint , tadv , tstart , tstop ,
106- mjd = mjd , polrep = polrep , tau = tau ,
107- elevmin = elevmin , elevmax = elevmax ,
108- timetype = timetype , fix_theta_GMST = fix_theta_GMST )
111+ tint , tadv , tstart , tstop ,
112+ mjd = mjd , polrep = polrep , tau = tau ,
113+ elevmin = elevmin , elevmax = elevmax ,
114+ timetype = timetype , fix_theta_GMST = fix_theta_GMST )
109115
110116 uniquetimes = np .sort (np .unique (obsarr ['time' ]))
111- scans = np .array ([[time - 0.5 * tadv , time + 0.5 * tadv ] for time in uniquetimes ])
112- source = str (ra ) + ":" + str (dec )
113- obs = ehtim .obsdata .Obsdata (ra , dec , rf , bw , obsarr , self .tarr ,
117+ scans = np .array ([[time - 0.5 * tadv , time + 0.5 * tadv ] for time in uniquetimes ])
118+ source = str (ra ) + ":" + str (dec )
119+ obs = ehtim .obsdata .Obsdata (ra , dec , rf , bw , obsarr , self .tarr ,
114120 source = source , mjd = mjd , timetype = timetype , polrep = polrep ,
115- ampcal = True , phasecal = True , opacitycal = True ,
121+ ampcal = True , phasecal = True , opacitycal = True ,
116122 dcal = True , frcal = True ,
117123 scantable = scans )
118124 return obs
119125
120126 def make_subarray (self , sites ):
121- """Make a subarray from the Array object array that only includes the sites listed in sites .
127+ """Make a subarray from the Array object array that only includes the sites listed.
122128
123129 Args:
124130 sites (list) : list of sites in the subarray
@@ -127,22 +133,25 @@ def make_subarray(self, sites):
127133 """
128134 all_sites = [t [0 ] for t in self .tarr ]
129135 mask = np .array ([t in sites for t in all_sites ])
130- return Array (self .tarr [mask ],ephem = self .ephem )
136+ subarr = Array (self .tarr [mask ], ephem = self .ephem )
137+ return subarr
131138
132139 def save_txt (self , fname ):
133140 """Save the array data in a text file.
134141
135142 Args:
136143 fname (str) : path to output array file
137144 """
138- ehtim .io .save .save_array_txt (self ,fname )
145+ ehtim .io .save .save_array_txt (self , fname )
139146 return
140147
141- ###########################################################################################################################################
142- #Array creation functions
143- ###########################################################################################################################################
148+ ##########################################################################
149+ # Array creation functions
150+ ##########################################################################
151+
152+
144153def load_txt (fname , ephemdir = 'ephemeris' ):
145- """Read an array from a text file.
154+ """Read an array from a text file.
146155 Sites with x=y=z=0 are spacecraft, TLE ephemerides read from ephemdir.
147156
148157 Args:
0 commit comments