コード例 #1
0
 def get_TEC_pp(self,radec,pol=0,new_stat_select=None):
    TEC_out=[]      
    N_stations = len(self.stat_select[:])
    N_sources = len(self.sources)
    N_piercepoints = N_stations * N_sources
    h=self.piercepoints.attrs.height
    r_0 = self.TECfit_white.attrs.r_0
    beta = self.TECfit_white.attrs.beta
    pp_out=[]
    am_out=[]
    TEC_out=[]
    stations=self.station_positions[:]
    if not (new_stat_select is None):
       stations=stations[new_stat_select]
    for i in range(len(self.n_list)) :
       if i%10==0:
          sys.stdout.write(str(i)+'...')
          sys.stdout.flush()                
       time=self.times[i]
       piercepoints=PosTools.getPiercePoints(time,radec.reshape((1,-1)),stations,height=h)
       pp=piercepoints.positions_xyz
       am=piercepoints.zenith_angles
       pp_out.append(pp)
       am_out.append(am)
       Xp_table=reshape(self.piercepoints[i]['positions_xyz'], (N_piercepoints, 3))
       #v=self.TECfit_white[ i, :, : ,pol][self.stat_select[:]].reshape((N_piercepoints,1))
       v=self.TECfit_white[ i, :, : ,pol][self.stat_select[:]].reshape((N_piercepoints,1))
       tecs=[]
       for ist in range(stations[:].shape[0]):
          tecs.append(get_interpolated_TEC_white(Xp_table,v,beta,r_0,pp[0,ist]))
       TEC_out.append(tecs)
    return array(pp_out),array(am_out),array(TEC_out)
コード例 #2
0
   def calculate_Sage_piercepoints(self, sage_group=0,time_steps = [], station_select=[],height = 200.e3):
      if ( len( time_steps ) == 0 ):
         n_list = range( self.times[:].shape[0] )
      else:
         n_list = time_steps
      self.sage_n_list = n_list
      if 'sage%d_n_list'%sage_group in self.hdf5.root: self.hdf5.removeNode('\sage%d_n_list'%sage_group)
      self.hdf5.createArray(self.hdf5.root, 'sage%d_n_list'%sage_group, self.sage_n_list)
      if ( len( station_select ) == 0 ):
         stat_select = range( self.stations[:].shape[0] )
      else:
         stat_select = station_select
      self.sage_stat_select = stat_select
      if 'sage%d_stat_select'%sage_group in self.hdf5.root: self.hdf5.removeNode('\sage%d_stat_select'%sage_group)
      self.hdf5.createArray(self.hdf5.root,'sage%d_stat_select'%sage_group, self.sage_stat_select)


      self.sage_height = height

      N_stations=len(stat_select)
      if 'sage%d_piercepoints'%sage_group in self.hdf5.root: self.hdf5.removeNode('\sage%d_piercepoints'%sage_group)
      description = {'positions':tables.Float64Col((self.N_sources, N_stations,2)), \
                     'positions_xyz':tables.Float64Col((self.N_sources, N_stations,3)), \
                     'zenith_angles':tables.Float64Col((self.N_sources, N_stations))}
      self.sage_piercepoints = self.hdf5.createTable(self.hdf5.root,'sage%d_piercepoints'%sage_group, description)
      self.sage_piercepoints.attrs.height = self.sage_height
      piercepoints_row = self.sage_piercepoints.row
      source_positions=self.hdf5.getNode('sage_radec%d'%sage_group)[:]
      p = ProgressBar(len(n_list), "Calculating piercepoints: ")
      for (n, counter) in zip(n_list, range(len(n_list))):
         p.update(counter)
         
         piercepoints=PosTools.getPiercePoints(self.times[n],source_positions,self.station_positions[:][self.stat_select[:]],height=self.sage_height)
         #piercepoints =  PiercePoints( self.times[ n ], self.pointing, self.array_center, self.source_positions, self.station_positions[:][self.stat_select[:]], height = self.height )
         piercepoints_row['positions'] = piercepoints.positions
         piercepoints_row['positions_xyz'] = piercepoints.positions_xyz
         piercepoints_row['zenith_angles'] = piercepoints.zenith_angles
         piercepoints_row.append()
      self.sage_piercepoints.flush()
      p.finished()