@@ -335,6 +335,15 @@ def __init__(self, file_name):
335335 self ._phase_centre = vis_header [self .VisHeader .PhaseCentre ][0 ]['data' ]
336336 self ._cross_correlation = vis_header [self .VisHeader .FlagCrossCorrelation ][0 ]['data' ]
337337 self ._auto_correlation = vis_header [self .VisHeader .FlagAutoCorrelation ][0 ]['data' ]
338+
339+ self .telescope_path = vis_header [self .VisHeader .TelescopePath ][0 ]['data' ].tostring ().decode ()[:- 1 ]
340+ self .telescope_lon = vis_header [self .VisHeader .TelescopeLon ][0 ]['data' ]
341+ self .telescope_lat = vis_header [self .VisHeader .TelescopeLat ][0 ]['data' ]
342+ self .telescope_alt = vis_header [self .VisHeader .TelescopeAlt ][0 ]['data' ]
343+ self .station_x = vis_header [self .VisHeader .StationX ][0 ]['data' ]
344+ self .station_y = vis_header [self .VisHeader .StationY ][0 ]['data' ]
345+ self .station_z = vis_header [self .VisHeader .StationZ ][0 ]['data' ]
346+
338347 #
339348 # block_dims = self.data[self.Group.VisBlock][self.VisBlock.Dims]
340349 # for index in block_dims:
@@ -447,13 +456,10 @@ def stations(self, flatten=False):
447456 assert self ._cross_correlation , \
448457 "Reading non-cross-correlation data not fully supported yet!"
449458
450- # Order according to documentation is 0-1, 0-2, 0-3... 1-2,
451- # 1-3... Auto-correlation isn't specified, but I guess we
452- # simply add 0-0, 1-1 etc in the appropriate place.
453- ac = 1 - int (self ._auto_correlation )
459+ # Order according to documentation is 0-1, 0-2, 0-3... 1-2, ...
454460 station1 = numpy .repeat (numpy .arange (self .num_stations ),
455- self .num_stations - ac - numpy .arange (self .num_stations ))
456- station2 = numpy .hstack ([numpy .arange (start + ac , self .num_stations )
461+ self .num_stations - 1 - numpy .arange (self .num_stations ))
462+ station2 = numpy .hstack ([numpy .arange (start + 1 , self .num_stations )
457463 for start in numpy .arange (self .num_stations )])
458464 # Tile in one or two dimensions depending on whether we want a
459465 # flat result
0 commit comments