zgrid[goodind] = NP.sqrt(1.0 - (xgrid[goodind]**2 + ygrid[goodind]**2))
xvect = xgrid.ravel()
yvect = ygrid.ravel()
zvect = zgrid.ravel()
xyzvect = NP.hstack((xvect.reshape(-1,1), yvect.reshape(-1,1), zvect.reshape(-1,1)))
if use_DSM or use_GSM:
backdrop = HP.cartview(fluxes_DSM.ravel(), coord=['G','E'], rot=[180,0,0], xsize=backdrop_xsize, return_projected_map=True)
elif use_GLEAM or use_SUMSS:
if backdrop_coords == 'radec':
backdrop = griddata(NP.hstack((ra_deg.reshape(-1,1), dec_deg.reshape(-1,1))), fpeak, NP.hstack((xvect.reshape(-1,1), yvect.reshape(-1,1))), method='cubic')
backdrop = backdrop.reshape(backdrop_xsize/2, backdrop_xsize)
elif backdrop_coords == 'dircos':
if (telescope == 'mwa_dipole') or (obs_mode == 'drift'):
backdrop = PB.primary_beam_generator(xyzvect, freq, telescope=telescope, freq_scale='Hz', skyunits='dircos', phase_center=[0.0,0.0,1.0])
backdrop = backdrop.reshape(backdrop_xsize, backdrop_xsize)
else:
if backdrop_coords == 'radec':
backdrop = griddata(NP.hstack((ra_deg.reshape(-1,1), dec_deg.reshape(-1,1))), fpeak, NP.hstack((xvect.reshape(-1,1), yvect.reshape(-1,1))), method='nearest')
backdrop = backdrop.reshape(backdrop_xsize/2, backdrop_xsize)
elif backdrop_coords == 'dircos':
if (telescope == 'mwa_dipole') or (obs_mode == 'drift'):
backdrop = PB.primary_beam_generator(xyzvect, freq, telescope=telescope, freq_scale='Hz', skyunits='dircos', phase_center=[0.0,0.0,1.0])
backdrop = backdrop.reshape(backdrop_xsize, backdrop_xsize)
## Create data for overlay
overlays = []
roi_obj_inds = []
for i in xrange(n_snaps):
catlabel = NP.concatenate((catlabel, NP.repeat('NVSS',count_valid)))
ra_deg = NP.concatenate((ra_deg, ra_deg_NVSS[NP.logical_and(NP.logical_and(not_in_SUMSS_ind, bright_source_ind), PS_ind)]))
dec_deg = NP.concatenate((dec_deg, dec_deg_NVSS[NP.logical_and(NP.logical_and(not_in_SUMSS_ind, bright_source_ind), PS_ind)]))
spindex = NP.concatenate((spindex, spindex_NVSS[NP.logical_and(NP.logical_and(not_in_SUMSS_ind, bright_source_ind), PS_ind)]))
majax = NP.concatenate((majax, nvss_majax[NP.logical_and(NP.logical_and(not_in_SUMSS_ind, bright_source_ind), PS_ind)]))
minax = NP.concatenate((minax, nvss_minax[NP.logical_and(NP.logical_and(not_in_SUMSS_ind, bright_source_ind), PS_ind)]))
fluxes = NP.concatenate((fluxes, nvss_fpeak))
ctlgobj = CTLG.Catalog(catlabel, freq_catalog, NP.hstack((ra_deg.reshape(-1,1), dec_deg.reshape(-1,1))), fluxes, spectral_index=spindex, src_shape=NP.hstack((majax.reshape(-1,1),minax.reshape(-1,1),NP.zeros(fluxes.size).reshape(-1,1))), src_shape_units=['degree','degree','degree'])
if backdrop_coords == 'radec':
backdrop = griddata(NP.hstack((ra_deg.reshape(-1,1), dec_deg.reshape(-1,1))), fluxes, NP.hstack((xvect.reshape(-1,1), yvect.reshape(-1,1))), method='cubic')
backdrop = backdrop.reshape(backdrop_xsize/2, backdrop_xsize)
elif backdrop_coords == 'dircos':
if (telescope == 'mwa_dipole') or (obs_mode == 'drift'):
backdrop = PB.primary_beam_generator(xyzvect, freq, telescope=telescope, freq_scale='Hz', skyunits='dircos', pointing_center=[0.0,0.0,1.0])
backdrop = backdrop.reshape(backdrop_xsize, backdrop_xsize)
else:
if use_PS:
catalog_file = '/data3/t_nithyanandan/project_MWA/foregrounds/PS_catalog.txt'
catdata = ascii.read(catalog_file, comment='#', header_start=0, data_start=1)
ra_deg = catdata['RA'].data
dec_deg = catdata['DEC'].data
fluxes = catdata['F_INT'].data
if backdrop_coords == 'radec':
ra_deg_wrapped = ra_deg.ravel() + 0.0
ra_deg_wrapped[ra_deg > 180.0] -= 360.0
dxvect = xgrid[0,1]-xgrid[0,0]
dyvect = ygrid[1,0]-ygrid[0,0]
def observe(
self, timestamp, Tsys, bandpass, pointing_center, skymodel, tobs, pb_min=0.1, fov_radius=None, lst=None
):
if bandpass.size != self.bp.shape[1]:
raise ValueError("bandpass length does not match.")
self.Tsys = self.Tsys + [Tsys]
self.vis_rms_freq = self.vis_rms_freq + [
2.0 * FCNST.k * Tsys / self.A_eff / self.eff_Q / NP.sqrt(2) / tobs / self.freq_resolution / CNST.Jy
]
self.tobs = self.tobs + [tobs]
self.lst = self.lst + [lst]
if self.timestamp == []:
self.bp = NP.asarray(bandpass).reshape(1, -1)
self.pointing_center = NP.asarray(pointing_center).reshape(1, -1)
else:
self.bp = NP.vstack((self.bp, NP.asarray(bandpass).reshape(1, -1)))
self.pointing_center = NP.vstack((self.pointing_center, NP.asarray(pointing_center).reshape(1, -1)))
pointing_lon = self.pointing_center[-1, 0]
pointing_lat = self.pointing_center[-1, 1]
if self.skycoords == "radec":
if self.pointing_coords == "hadec":
if lst is not None:
pointing_lon = lst - self.pointing_center[-1, 0]
pointing_lat = self.pointing_center[-1, 1]
else:
raise ValueError(
"LST must be provided. Sky coordinates are in RA-Dec format while pointing center is in HA-Dec format."
)
elif self.pointing_coords == "altaz":
pointing_lonlat = lst - GEOM.altaz2hadec(self.pointing_center[-1, :], self.latitude, units="degrees")
pointing_lon = pointing_lonlat[0]
pointing_lat = pointing_lonlat[1]
elif self.skycoords == "hadec":
if self.pointing_coords == "radec":
if lst is not None:
pointing_lon = lst - self.pointing_center[-1, 0]
pointing_lat = self.pointing_center[-1, 1]
else:
raise ValueError(
"LST must be provided. Sky coordinates are in RA-Dec format while pointing center is in HA-Dec format."
)
elif self.pointing_coords == "altaz":
pointing_lonlat = lst - GEOM.altaz2hadec(self.pointing_center[-1, :], self.latitude, units="degrees")
pointing_lon = pointing_lonlat[0]
pointing_lat = pointing_lonlat[1]
else:
if self.pointing_coords == "radec":
if lst is not None:
pointing_lonlat = GEOM.hadec2altaz(
NP.asarray([lst - self.pointing_center[-1, 0], self.pointing_center[-1, 1]]),
self.latitude,
units="degrees",
)
pointing_lon = pointing_lonlat[0]
pointing_lat = pointing_lonlat[1]
else:
raise ValueError(
"LST must be provided. Sky coordinates are in Alt-Az format while pointing center is in RA-Dec format."
)
elif self.pointing_coords == "hadec":
pointing_lonlat = GEOM.hadec2altaz(self.pointing_center, self.latitude, units="degrees")
pointing_lon = pointing_lonlat[0]
pointing_lat = pointing_lonlat[1]
pointing_phase = 0.0
baseline_in_local_frame = self.baseline
if self.baseline_coords == "equatorial":
baseline_in_local_frame = GEOM.xyz2enu(self.baseline, self.latitude, "degrees")
ptmp = self.pointing_center[-1, :] # Convert pointing center to Alt-Az coordinates
if self.pointing_coords == "hadec":
ptmp = GEOM.hadec2altaz(self.pointing_center[-1, :], self.latitude, units="degrees")
elif self.pointing_coords == "radec":
if lst is not None:
ptmp = GEOM.hadec2altaz(
NP.asarray([lst - self.pointing_center[-1, 0], self.pointing_center[-1, 1]]),
self.latitude,
units="degrees",
)
else:
raise ValueError(
"LST must be provided. Sky coordinates are in Alt-Az format while pointing center is in RA-Dec format."
)
ptmp = GEOM.altaz2dircos(ptmp, "degrees") # Convert pointing center to direction cosine coordinates
pointing_phase = (
2.0
* NP.pi
* NP.dot(baseline_in_local_frame.reshape(1, -1), ptmp.reshape(-1, 1))
* self.channels.reshape(1, -1)
/ FCNST.c
)
if fov_radius is None:
fov_radius = 90.0
# PDB.set_trace()
m1, m2, d12 = GEOM.spherematch(
pointing_lon,
pointing_lat,
skymodel.catalog.location[:, 0],
skymodel.catalog.location[:, 1],
fov_radius,
maxmatches=0,
)
# if fov_radius is not None:
# m1, m2, d12 = GEOM.spherematch(pointing_lon, pointing_lat, skymodel.catalog.location[:,0], skymodel.catalog.location[:,1], fov_radius, maxmatches=0)
# else:
# m1 = [0] * skymodel.catalog.location.shape[0]
# m2 = xrange(skymodel.catalog.location.shape[0])
# d12 = GEOM.sphdist(NP.empty(skymodel.catalog.shape[0]).fill(pointing_lon), NP.empty(skymodel.catalog.shape[0]).fill(pointing_lat), skymodel.catalog.location[:,0], skymodel.catalog.location[:,1])
if len(d12) != 0:
pb = NP.empty((len(d12), len(self.channels)))
fluxes = NP.empty((len(d12), len(self.channels)))
coords_str = self.skycoords
if self.skycoords == "radec":
coords_str = "altaz"
source_positions = GEOM.hadec2altaz(
NP.hstack(
(
NP.asarray(lst - skymodel.catalog.location[m2, 0]).reshape(-1, 1),
skymodel.catalog.location[m2, 1].reshape(-1, 1),
)
),
self.latitude,
"degrees",
)
for i in xrange(len(self.channels)):
# pb[:,i] = PB.primary_beam_generator(d12, self.channels[i]/1.0e9, 'degrees', self.telescope)
pb[:, i] = PB.primary_beam_generator(
source_positions, self.channels[i] / 1.0e9, "altaz", self.telescope
)
fluxes[:, i] = (
skymodel.catalog.flux_density[m2]
* (self.channels[i] / skymodel.catalog.frequency) ** skymodel.catalog.spectral_index[m2]
)
geometric_delays = DLY.geometric_delay(
baseline_in_local_frame,
source_positions,
altaz=(coords_str == "altaz"),
hadec=(coords_str == "hadec"),
latitude=self.latitude,
)
self.geometric_delays = self.geometric_delays + [geometric_delays.reshape(len(source_positions))]
phase_matrix = 2.0 * NP.pi * NP.repeat(
geometric_delays.reshape(-1, 1), len(self.channels), axis=1
) * NP.repeat(self.channels.reshape(1, -1), len(d12), axis=0) - NP.repeat(pointing_phase, len(d12), axis=0)
skyvis = NP.sum(
pb
* fluxes
* NP.repeat(NP.asarray(bandpass).reshape(1, -1), len(d12), axis=0)
* NP.exp(-1j * phase_matrix),
axis=0,
)
if fov_radius is not None:
self.obs_catalog_indices = self.obs_catalog_indices + [m2]
# self.obs_catalog = self.obs_catalog + [skymodel.catalog.subset(m2)]
else:
print "No sources found in the catalog within matching radius. Simply populating the observed visibilities with noise."
skyvis = NP.zeros((1, len(self.channels)))
if self.timestamp == []:
self.skyvis_freq = skyvis.reshape(1, -1)
self.vis_noise_freq = self.vis_rms_freq[-1] * (
NP.random.randn(len(self.channels)).reshape(1, -1)
+ 1j * NP.random.randn(len(self.channels)).reshape(1, -1)
)
self.vis_freq = self.skyvis_freq + self.vis_noise_freq
else:
self.skyvis_freq = NP.vstack((self.skyvis_freq, skyvis.reshape(1, -1)))
self.vis_noise_freq = NP.vstack(
(
self.vis_noise_freq,
self.vis_rms_freq[-1]
* (
NP.random.randn(len(self.channels)).reshape(1, -1)
+ 1j * NP.random.randn(len(self.channels)).reshape(1, -1)
),
)
)
self.vis_freq = NP.vstack(
(self.vis_freq, (self.skyvis_freq[-1, :] + self.vis_noise_freq[-1, :]).reshape(1, -1))
)
self.timestamp = self.timestamp + [timestamp]
freq = NP.float(freq)
freq_resolution = NP.float(freq_resolution)
chans = (freq + (NP.arange(nchan) - 0.5 * nchan) * freq_resolution)/ 1e9 # in GHz
bandpass_str = '{0:0d}x{1:.1f}_kHz'.format(nchan, freq_resolution/1e3)
theta, phi = HP.pix2ang(nside, NP.arange(HP.nside2npix(nside)))
alt = 90.0 - NP.degrees(theta)
az = NP.degrees(phi)
altaz = NP.hstack((alt.reshape(-1,1), az.reshape(-1,1)))
pinfo = {}
pinfo['pointing_center'] = pointing_altaz
pinfo['pointing_coords'] = 'altaz'
pb = PB.primary_beam_generator(altaz, chans, telescope, freq_scale='GHz', skyunits='altaz', pointing_info=pinfo, short_dipole_approx=short_dipole_approx, half_wave_dipole_approx=half_wave_dipole_approx)
colnum = 0
npix = HP.nside2npix(nside)
frequencies = chans * 1e3
if outfile is not None:
if not isinstance(outfile, str):
raise TypeError('outfile parameter must be a string')
else:
outfile = telescope_str+beam_id+'_'+ground_plane_str+'nside_{0:0d}_'.format(nside)+delaygain_err_str+'{0}_{1:.1f}_MHz'.format(bandpass_str, freq/1e6)+'.fits'
hdulist = []
hdulist += [fits.PrimaryHDU()]
hdulist[0].header['EXTNAME'] = 'PRIMARY'
hdulist[0].header['NPOL'] = (1, 'Number of polarizations')
else:
pointings_radec = NP.hstack((NP.asarray(lst-pointing_init[0]).reshape(-1,1), pointing_init[1]+NP.zeros(n_snaps).reshape(-1,1)))
pointings_hadec = NP.hstack(((lst-pointings_radec[:,0]).reshape(-1,1), pointings_radec[:,1].reshape(-1,1)))
pointings_altaz = GEOM.hadec2altaz(pointings_hadec, MWA_latitude, units='degrees')
pointings_dircos = GEOM.altaz2dircos(pointings_altaz, units='degrees')
delay_matrix = DLY.delay_envelope(intrfrmtr.baseline, pointings_dircos, units='mks')
pbeams = []
m2s = []
for i in xrange(n_snaps):
havect = lst[i] - ravect
altaz = GEOM.hadec2altaz(NP.hstack((havect.reshape(-1,1),decvect.reshape(-1,1))), MWA_latitude, units='degrees')
# m1, m2, d12 = GEOM.spherematch(pointings[i,0], pointings[i,1], ravect, decvect, 90.0, maxmatches=0)
roi_altaz = NP.asarray(NP.where(altaz[:,0] >= 0.0)).ravel()
pb = PB.primary_beam_generator(altaz[roi_altaz,:], freq, telescope='mwa', skyunits='altaz', freq_scale='Hz', phase_center=pointings_altaz[i,:])
pbeams += [pb]
m2s += [roi_altaz]
## Plotting animation
fig = PLT.figure(figsize=(14,14))
ax1 = fig.add_subplot(211)
# fig, (ax1, ax2) = PLT.subplots(2,1,figsize=(14,12))
ax1.set_xlabel(r'$\eta$ [$\mu$s]', fontsize=18)
ax1.set_ylabel('Amplitude [K Hz]', fontsize=18)
ax1.set_title('Delay Spectrum', fontsize=18, weight='semibold')
ax1.set_yscale('log')
ax1.set_xlim(1e6*NP.amin(lags)-1.0, 1e6*NP.amax(lags)+1.0)
ax1.set_ylim(0.5*NP.amin(NP.abs(vis_lag)),2.0*NP.amax(NP.abs(vis_lag)))
l1 = ax1.plot([], [], 'k+', [], [], 'k-', [], [], 'k-', [], [], 'k:', [], [], 'k:', markersize=10)
src_shape_units=['degree', 'degree', 'degree'])
if backdrop_coords == 'radec':
backdrop = griddata(NP.hstack(
(ra_deg.reshape(-1, 1), dec_deg.reshape(-1, 1))),
fluxes,
NP.hstack(
(xvect.reshape(-1, 1), yvect.reshape(-1, 1))),
method='cubic')
backdrop = backdrop.reshape(backdrop_xsize / 2, backdrop_xsize)
elif backdrop_coords == 'dircos':
if (telescope == 'mwa_dipole') or (obs_mode == 'drift'):
backdrop = PB.primary_beam_generator(
xyzvect,
freq,
telescope=telescope,
freq_scale='Hz',
skyunits='dircos',
pointing_center=[0.0, 0.0, 1.0])
backdrop = backdrop.reshape(backdrop_xsize, backdrop_xsize)
else:
if use_PS:
catalog_file = '/data3/t_nithyanandan/project_MWA/foregrounds/PS_catalog.txt'
catdata = ascii.read(catalog_file,
comment='#',
header_start=0,
data_start=1)
ra_deg = catdata['RA'].data
dec_deg = catdata['DEC'].data
fluxes = catdata['F_INT'].data
