def G_plus_C_four(freq):
"""plots fourier of GSM and CMB added together
User defines GSM frequency in MHz
WARNING:takes some times to run
Note:
GSM default shape is 512
CMB default shape is 2048
GSM and CMB shape must match to add arrays
GSM has been cast to a higher nside of 2048
"""
gsm = GlobalSkyModel()
#creates a model of the GSM
gsmdata = gsm.generate(freq)
#upgrades the GSM to 2048
gsmdata = hp.ud_grade(gsmdata, 2048)
cmbdata = hp.read_map(
'/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits'
)
LMAX = 1024
bothmap = hp.anafast(gsmdata + cmbdata, lmax=LMAX)
ell = np.arange(len(bothmap))
plt.figure()
freq_str = str(freq)
plt.plot(ell,
ell * (ell + 1) * bothmap,
color='b',
label='GSM at ' + freq_str + ' MHz +CMB')
plt.xlabel('ell')
plt.ylabel('ell(ell+1)')
plt.grid()
#delete this line?
#hp.write_bothmap('bothmap.fits',bothmap)
plt.legend(loc='upper right')
plt.show()
def G_plus_C_four(freq):
"""plots fourier of GSM and CMB added together
User defines GSM frequency in MHz
WARNING:takes some times to run
Note:
GSM default shape is 512
CMB default shape is 2048
GSM and CMB shape must match to add arrays
GSM has been cast to a higher nside of 2048
"""
gsm=GlobalSkyModel()
#creates a model of the GSM
gsmdata=gsm.generate(freq)
#upgrades the GSM to 2048
gsmdata=hp.ud_grade(gsmdata,2048)
cmbdata=hp.read_map('/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits')
LMAX=1024
bothmap=hp.anafast(gsmdata+cmbdata,lmax=LMAX)
ell=np.arange(len(bothmap))
plt.figure()
freq_str=str(freq)
plt.plot(ell,ell*(ell+1)*bothmap,color='b',label='GSM at '+freq_str+' MHz +CMB')
plt.xlabel('ell');plt.ylabel('ell(ell+1)');plt.grid()
#delete this line?
#hp.write_bothmap('bothmap.fits',bothmap)
plt.legend(loc='upper right')
plt.show()
def compare_to_gsm():
""" Compare output of python version to fortran version.
Compares against output of original GSM. Note that the interpolation
functions used in this and in the original differ. So, the outputs
differ too, by as much as 5%. This just goes to show that there's only
so much accuracy that one can get via interpolation.
If the frequency matches a map used to generate the GSM, then the
fortran and python should match up pretty much exactly as no interpolation
is required. In between maps is where the differences will become more apparent.
Note: The gsm.f that is supplied only ever uses the haslam map. It also
has no option to change the interpolation method.
Note: Because each output
"""
gsm = GlobalSkyModel(basemap='haslam', interpolation='cubic')
gsm_fortran = h5py.File('gsm_fortran_test_data.h5')
for freq in (10, 100, 408, 1000, 1420, 2326, 10000, 23000, 40000, 90000, 94000):
print("\nComparing at %i MHz..." % freq)
a = gsm_fortran['map_%imhz.out' % freq][:]
b = gsm.generate(freq)
print "FORTRAN: ", a
print "PYTHON: ", b
print "FRAC AVG: %2.6f" % np.average(np.abs(1 - a/b))
def GSM_four(freq, scale):
"""plots fourier of the GSM model
used defines freq in MHz such as 5000
user defines power of 10 scale down such as -4
"""
gsm = GlobalSkyModel()
#generates the GSM model
gsmdata = gsm.generate(freq)
LMAX = 1024
gsmmap = hp.anafast(gsmdata * (10**scale), lmax=LMAX)
ell = np.arange(len(gsmmap))
plt.figure()
#convert variables to text for labels
scale_str = str(scale)
freq_str = str(freq)
plt.plot(ell,
ell * (ell + 1) * gsmmap,
color='b',
label='GSM*10^ ' + scale_str + ' at ' + freq_str + ' MHz')
plt.xlabel('ell')
plt.ylabel('ell(ell+1)')
plt.grid()
#delete this line?
#hp.write_gsmmap('gsmmap.fits',gsmmap)
plt.legend(loc='upper right')
plt.show()
def G_plus_C_cross_C_four(freq,scale):
"""plot of fourier of
GSM(scale)+CMB crossed with CMB
compared to CMB only
User defines GSM freq in MHz
User defines scale of GSM such as 0.1 or 0.01
WARNING: takes some time to run
"""
gsm=GlobalSkyModel()
#create the GSM model
gsmdata=gsm.generate(freq)
#rather than upgrading GSM, downgrade the CMB to 512
cmbdata=hp.read_map('/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits')
cmbmap=hp.ud_grade(cmbdata,512)
LMAX=1024
both=(gsmdata*scale)+cmbmap
#cross cmb+gsm with the cmb
c_plus_g_cross_c=hp.sphtfunc.anafast(both,map2=cmbmap,lmax=LMAX)
ell=np.arange(len(c_plus_g_cross_c))
plt.figure()
#format variables as text for labels
freq_str=str(freq)
scale_str=str(scale)
plt.plot(ell,ell*(ell+1)*c_plus_g_cross_c,color='b',label='GSM('+scale_str+') at '+freq_str+' MHZ+CMB cross CMB')
cmbmaponly=hp.anafast(cmbdata,lmax=LMAX)
ell=np.arange(len(cmbmaponly))
plt.plot(ell,ell*(ell+1)*cmbmaponly,color='r',label='CMB')
#delete this line?
#hp.write_cmbmaponly('cmbmaponly.fits',cmbmaponly)
plt.xlabel('ell');plt.ylabel('ell(ell+1)');plt.grid()
#delete this line?
#hp.write_cp_plus_g_cross_c('c_plus_g_cross_c.fits',c_plus_g_cross_c)
plt.legend(loc='upper right')
plt.show()
def compare_gsm_to_old():
g = GlobalSkyModel2016(freq_unit='GHz', resolution='hi', unit='TCMB')
d = g.generate(0.408)
g.view()
g_old = GlobalSkyModel(freq_unit='GHz', basemap='haslam')
d_old = g_old.generate(0.408)
g_old.view()
def compare_gsm_to_old():
g = GlobalSkyModel2016(freq_unit='GHz', resolution='hi', unit='TCMB')
d = g.generate(0.408)
g.view()
g_old = GlobalSkyModel(freq_unit='GHz', basemap='haslam')
d_old = g_old.generate(0.408)
g_old.view()
def test_speed():
gsm = GlobalSkyModel(basemap='haslam')
t1 = time.time()
for ff in np.linspace(10, 10000, 100):
gsm.generate(ff)
t2 = time.time()
print("Time taken: %2.2fs" % (t2 - t1))
def Beam_G_plus_C_cross_C_four(freq,scale,itheta,iphi,nside,sig=np.pi/10):
"""plot of fourier for
GSM(scale)+CMB crossed with CMB
beam applied then accounted for
all compared to CMB alone
User defines frequency of GSM and scale to decrease it
User selects point of interest: theta and phi
User may designate sigma for beam
User selects nside
Default sigma is pi/10
"""
gsm=GlobalSkyModel()
#generates GSM model
gsmdata=gsm.generate(freq)
#rather than upgrading GSM, downgrade the CMB to 512
cmbdata=hp.read_map('/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits')
cmbmap=hp.ud_grade(cmbdata,512)
LMAX=1024
both=(gsmdata*scale)+cmbmap
#run the beam function with the given input
showbeam=beam(itheta,iphi,nside,sig)
npix=12*nside**2
#correct the data to account for the beam
correction=npix/(np.sum(showbeam))
both=both*showbeam*correction
#cross GSM+CMB with CMB and transform
c_plus_g_cross_c=hp.sphtfunc.anafast(both,map2=cmbmap,lmax=LMAX)
ell=np.arange(len(c_plus_g_cross_c))
plt.figure()
#convert variables into text for labels
freq_str=str(freq)
scale_str=str(scale)
plt.plot(ell,ell*(ell+1)*c_plus_g_cross_c,color='b',label='GSM('+scale_str+') at '+freq_str+' MHZ+CMB cross CMB')
#transform CMB
cmbmaponly=hp.anafast(cmbmap,lmax=LMAX)
ell=np.arange(len(cmbmaponly))
plt.plot(ell,ell*(ell+1)*cmbmaponly,color='r',label='CMB')
#delete this line?
#hp.write_cmbmaponly('cmbmaponly.fits',cmbmaponly)
plt.xlabel('ell');plt.ylabel('ell(ell+1)');plt.grid()
#delete this line?
#hp.write_c_plus_g_cross_c('c_plus_g_cross_c.fits',c_plus_g_cross_c)
#convert variable into text for labels
theta_str=str(itheta)
phi_str=str(iphi)
plt.legend(loc='upper right')
plt.title('Point of Interest:('+theta_str+','+phi_str+')')
#set the limits for the y axis
plt.ylim(-0.00000001,0.00000006)
plt.show()
def __init__(self, freq=350 * u.MHz, model='2008'):
assert str(model) in ['2008', '2016']
if not isinstance(freq, astropy.units.quantity.Quantity):
# assume MHz
freq = freq * u.MHz
if str(model) == '2008':
self.gsm = GlobalSkyModel()
elif str(model) == '2016':
self.gsm = GlobalSkyModel2016()
self.map = self.gsm.generate(freq.to(u.MHz).value)
self.freq = freq
self.model = model
def healpix_brightness(self, sbar):
if self.params['use_2008']:
gsm = GlobalSkyModel(unit="MJysr", theta_rot = self.params['theta0'], phi_rot = self.params['phi0'],
resolution="lo" if self.params['low_res'] else "hi")
else:
gsm = GlobalSkyModel2016(unit="MJysr", theta_rot = self.params['theta0'], phi_rot = self.params['phi0'],
resolution="lo" if self.params['low_res'] else "hi") #This only works for the steven-murray/PyGSM fork.
gsm.generate(self.params['nu0'] * self.f0)
data = np.atleast_2d(gsm.generated_map_data) * 1e6
# rot_map = np.zeros_like(data)
#
# for i in range(len(self.f0)):
# rot_map[i] = 1e6 * rotate_map(data[i], self.params['theta0'], self.params['phi0']) #1e6 because units are MJy
return data
def _generate_gsm_map(self) -> da.Array:
""" """
# Generate the GSM map at the given frequency
gsm = GlobalSkyModel(freq_unit="MHz")
gsm_map = gsm.generate(self.frequency)
# Resize the GSM HEALPix map to the required dimensions
gsm_map_nside = hp.pixelfunc.npix2nside(gsm_map.shape[-1])
if gsm_map_nside != self.nside:
gsm_map = hp.pixelfunc.ud_grade(
map_in=gsm_map,
nside_out=self.nside
)
# gsm_map = gal_to_eq.rotate_map_alms(
# gsm_map
# )
# Add frequency if size=1
if self.frequency.size == 1:
gsm_map = np.expand_dims(gsm_map, axis=0)
# Convert the map, currently in Galactic coordinates to equatorial
gal_to_eq = hp.rotator.Rotator(
deg=True,
rot=[0, 0],
coord=['G', 'C']
)
for i in range(self.frequency.size):
with HiddenPrints():
gsm_map[i, :] = gal_to_eq.rotate_map_pixel(
gsm_map[i, :]
)
# Transform into dask array
gsm_map = da.from_array(gsm_map)
# Add time/polarization dimensions
gsm_map = np.tile(gsm_map, (self.time.size, 1, 1, 1))
gsm_map = np.moveaxis(gsm_map, source=2, destination=1)
return gsm_map
def G_plus_C_cross_C_beam_chi(freq, scale, itheta, iphi, nside, sig, cmbmap,
cmbmaponly):
"""returns the ratio of chi/signal
for beam applied to GSM(scale)+CMB
crossed with CMB
versus CMB alone
uses same input as Beam_G_plus_C_cross_C_four()
but returns value of ratio only, no plot
CANNOT BE RUN ALONE, REFERENCES ANOTHER FUNCTION
"""
gsm = GlobalSkyModel()
#generates GSM model
gsmdata = gsm.generate(freq)
#rather than upgrading GSM, downgrade the CMB to 512
#following lines moved into find chi function
#cmbdata=hp.read_map('/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits')
#cmbmap=hp.ud_grade(cmbdata,512)
LMAX = 1024
both = (gsmdata * scale) + cmbmap
#run the beam function with the given input
showbeam = beam(itheta, iphi, nside, sig)
npix = 12 * nside**2
#correct the data to account for the beam
correction = npix / (np.sum(showbeam))
both = both * showbeam * correction
#cross GSM+CMB with CMB and transform
c_plus_g_cross_c = hp.sphtfunc.anafast(both, map2=cmbmap, lmax=LMAX)
#moved to find function
#cmbmaponly=hp.anafast(cmbmap,lmax=LMAX)
ell = np.arange(len(cmbmaponly))
#find chi squared, the deviation from the desired data
difference = c_plus_g_cross_c - cmbmaponly
chi_squared = (np.sum(
(ell * (ell + 1) * difference)**2)) / (len(c_plus_g_cross_c))
signal_squared = (np.sum(
(ell * (ell + 1) * cmbmaponly)**2)) / (len(c_plus_g_cross_c))
chi = np.sqrt(chi_squared)
signal = np.sqrt(signal_squared)
ratio = chi / signal
return ratio
def G_plus_C_cross_C_four(freq, scale):
"""plot of fourier of
GSM(scale)+CMB crossed with CMB
compared to CMB only
User defines GSM freq in MHz
User defines scale of GSM such as 0.1 or 0.01
WARNING: takes some time to run
"""
gsm = GlobalSkyModel()
#create the GSM model
gsmdata = gsm.generate(freq)
#rather than upgrading GSM, downgrade the CMB to 512
cmbdata = hp.read_map(
'/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits'
)
cmbmap = hp.ud_grade(cmbdata, 512)
LMAX = 1024
both = (gsmdata * scale) + cmbmap
#cross cmb+gsm with the cmb
c_plus_g_cross_c = hp.sphtfunc.anafast(both, map2=cmbmap, lmax=LMAX)
ell = np.arange(len(c_plus_g_cross_c))
plt.figure()
#format variables as text for labels
freq_str = str(freq)
scale_str = str(scale)
plt.plot(ell,
ell * (ell + 1) * c_plus_g_cross_c,
color='b',
label='GSM(' + scale_str + ') at ' + freq_str +
' MHZ+CMB cross CMB')
cmbmaponly = hp.anafast(cmbdata, lmax=LMAX)
ell = np.arange(len(cmbmaponly))
plt.plot(ell, ell * (ell + 1) * cmbmaponly, color='r', label='CMB')
#delete this line?
#hp.write_cmbmaponly('cmbmaponly.fits',cmbmaponly)
plt.xlabel('ell')
plt.ylabel('ell(ell+1)')
plt.grid()
#delete this line?
#hp.write_cp_plus_g_cross_c('c_plus_g_cross_c.fits',c_plus_g_cross_c)
plt.legend(loc='upper right')
plt.show()
def G_plus_C_cross_C_beam_chi(freq,scale,itheta,iphi,nside,sig,cmbmap,cmbmaponly):
"""returns the ratio of chi/signal
for beam applied to GSM(scale)+CMB
crossed with CMB
versus CMB alone
uses same input as Beam_G_plus_C_cross_C_four()
but returns value of ratio only, no plot
CANNOT BE RUN ALONE, REFERENCES ANOTHER FUNCTION
"""
gsm=GlobalSkyModel()
#generates GSM model
gsmdata=gsm.generate(freq)
#rather than upgrading GSM, downgrade the CMB to 512
#following lines moved into find chi function
#cmbdata=hp.read_map('/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits')
#cmbmap=hp.ud_grade(cmbdata,512)
LMAX=1024
both=(gsmdata*scale)+cmbmap
#run the beam function with the given input
showbeam=beam(itheta,iphi,nside,sig)
npix=12*nside**2
#correct the data to account for the beam
correction=npix/(np.sum(showbeam))
both=both*showbeam*correction
#cross GSM+CMB with CMB and transform
c_plus_g_cross_c=hp.sphtfunc.anafast(both,map2=cmbmap,lmax=LMAX)
#moved to find function
#cmbmaponly=hp.anafast(cmbmap,lmax=LMAX)
ell=np.arange(len(cmbmaponly))
#find chi squared, the deviation from the desired data
difference=c_plus_g_cross_c-cmbmaponly
chi_squared=(np.sum((ell*(ell+1)*difference)**2))/(len(c_plus_g_cross_c))
signal_squared=(np.sum((ell*(ell+1)*cmbmaponly)**2))/(len(c_plus_g_cross_c))
chi=np.sqrt(chi_squared)
signal=np.sqrt(signal_squared)
ratio=chi/signal
return ratio
def GSM_four(freq,scale):
"""plots fourier of the GSM model
used defines freq in MHz such as 5000
user defines power of 10 scale down such as -4
"""
gsm=GlobalSkyModel()
#generates the GSM model
gsmdata=gsm.generate(freq)
LMAX=1024
gsmmap=hp.anafast(gsmdata*(10**scale),lmax=LMAX)
ell=np.arange(len(gsmmap))
plt.figure()
#convert variables to text for labels
scale_str=str(scale)
freq_str=str(freq)
plt.plot(ell,ell*(ell+1)*gsmmap,color='b',label='GSM*10^ '+scale_str+' at '+freq_str+' MHz')
plt.xlabel('ell');plt.ylabel('ell(ell+1)');plt.grid()
#delete this line?
#hp.write_gsmmap('gsmmap.fits',gsmmap)
plt.legend(loc='upper right')
plt.show()
def test_gsm_generate():
""" Test maps generate successfully, and that output shapes are correct. """
freq = 40
gsm = GlobalSkyModel()
map = gsm.generate(freq)
assert map.shape == (3145728,)
freqs = [40, 80, 120, 160]
gsm = GlobalSkyModel(interpolation='cubic')
map = gsm.generate(freqs)
assert map.shape == (4, 3145728)
freqs = np.linspace(1, 90, 10)
freq_unit = 'GHz'
for map_id in ['5deg', 'haslam', 'wmap']:
gsm = GlobalSkyModel(basemap=map_id, interpolation='pchip', freq_unit=freq_unit)
map = gsm.generate(freqs)
assert map.shape == (10, 3145728)
def mk_map_GSM(f=150.,frange=None,nbins=None,write2fits=True):
"""
Make a Stokes I map/cube using pyGSM
Can also provide frange=[lo,hi] and nbins to get an image cube (ignores f)
Both f arguments need to be in MHz
"""
from pygsm import GlobalSkyModel
gsm = GlobalSkyModel()
if frange==None:
gsm = gsm.generate(f)
if write2fits: gsm.write_fits("gsm_%sMHz.fits"%str(int(f)))
M = gsm
else:
flo,fhi=frange[0],frange[1]
try: freqs = np.linspace(flo,fhi,nbins)
except:
print 'Did you provide "nbins"?'
return None
map_cube = gsm.generate(freqs)
#if write2fits: map_cube.write_fits("gsm_cube_%s-%sMHz"%(str(int(flo)), str(int(fhi))))
M = np.swapaxes(map_cube,0,1) #conform to other map making routines
return M
class GSModel:
"""
class interface to GSM model
so that structures/data stay in memory
gsm=GSModel(freq=freq, model=model)
to use on astropy SkyCoord:
gsm.SkyCoord_Tsky(source)
to use on pulsar parfile:
gsm.pulsar_Tsky(parfile)
"""
def __init__(self, freq=350 * u.MHz, model='2008'):
assert str(model) in ['2008', '2016']
if not isinstance(freq, astropy.units.quantity.Quantity):
# assume MHz
freq = freq * u.MHz
if str(model) == '2008':
self.gsm = GlobalSkyModel()
elif str(model) == '2016':
self.gsm = GlobalSkyModel2016()
self.map = self.gsm.generate(freq.to(u.MHz).value)
self.freq = freq
self.model = model
def SkyCoord_Tsky(self, source):
T = healpy.pixelfunc.get_interp_val(self.map,
source.galactic.l.value,
source.galactic.b.value,
lonlat=True)
return T * u.K
def pulsar_Tsky(self, parfile):
m = models.get_model(parfile)
try:
psr = SkyCoord(m.RAJ.value, m.DECJ.value, unit='deg')
except:
try:
psr = SkyCoord(m.ELONG.value * u.deg,
m.ELAT.value * u.deg,
frame='pulsarecliptic')
except:
raise KeyError, 'Cannot find RAJ,DECJ or ELONG,ELAT in:\n%s' % m.as_parfile(
)
return self.SkyCoord_Tsky(psr)
def test_write_fits():
gsm = GlobalSkyModel()
gsm.generate(1000)
gsm.write_fits("test_write_fits.fits")
d_fits = hp.read_map("test_write_fits.fits")
d_gsm = gsm.generated_map_data
assert d_fits.shape == d_gsm.shape
assert np.allclose(d_fits, d_gsm)
os.remove("test_write_fits.fits")
def mk_map_GSM(f=150., frange=None, nbins=None, write2fits=True):
"""
Make a Stokes I map/cube using pyGSM
Can also provide frange=[lo,hi] and nbins to get an image cube (ignores f)
Both f arguments need to be in MHz
"""
from pygsm import GlobalSkyModel
gsm = GlobalSkyModel()
if frange == None:
gsm = gsm.generate(f)
if write2fits: gsm.write_fits("gsm_%sMHz.fits" % str(int(f)))
M = gsm
else:
flo, fhi = frange[0], frange[1]
try:
freqs = np.linspace(flo, fhi, nbins)
except:
print 'Did you provide "nbins"?'
return None
map_cube = gsm.generate(freqs)
#if write2fits: map_cube.write_fits("gsm_cube_%s-%sMHz"%(str(int(flo)), str(int(fhi))))
M = np.swapaxes(map_cube, 0, 1) #conform to other map making routines
return M
import matplotlib.pyplot as plt
import numpy as np
import healpy as hp
from pygsm import GlobalSkyModel
gsm = GlobalSkyModel()
class CMB_GSM_analysis:
"""description"""
#init with things that will not change through out.
#user will have to specify these the first time
#allow for pass in of cmb file location, but default for ease
def __init__(
self,
threshold,
sig,
freq,
nside,
LMAX,
CMB_file_name='/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits'
):
self.threshold = threshold
self.sig = sig
self.freq = freq
#check that nside value is valid
if not np.all(hp.isnsideok(nside)):
raise ValueError(
"%s is not a valid nside parameter (must be a power of 2, less than 2**30)"
% str(nside))
self.nside = nside
# Select time steps
if max_n_timestamps is None:
max_n_timestamps = len(timestamps)
else:
max_n_timestamps = min(max_n_timestamps, len(timestamps))
timestamps = timestamps[min_timestamp:min_timestamp + max_n_timestamps]
# *** Set up sky model
lst = 299.28404 * NP.pi / 180
if use_GSM:
print 'Getting the GSM'
nside = 128
# Load in GSM
gsm = GlobalSkyModel()
sky = gsm.generate(f0 * 10**(-6))
sky = HP.ud_grade(sky,
nside) # provided with nside=512, convert to lower res
inds = NP.arange(HP.nside2npix(nside))
theta, phi = HP.pixelfunc.pix2ang(nside, inds)
gc = Galactic(l=phi,
b=NP.pi / 2 - theta,
unit=(units.radian, units.radian))
radec = gc.fk5
eq = aipy.coord.radec2eq((-lst + radec.ra.radian, radec.dec.radian))
xyz = NP.dot(aipy.coord.eq2top_m(0, lat * NP.pi / 180), eq)
# Keep just pixels above horizon
include = NP.where(xyz[2, :] > 0)
sky = sky[include]
def _load_gsm(freq):
"""
"""
gsm = GlobalSkyModel(freq_unit='MHz')
gsmap = gsm.generate(freq)
return gsmap
# Select time steps
if max_n_timestamps is None:
max_n_timestamps = len(timestamps)
else:
max_n_timestamps = min(max_n_timestamps, len(timestamps))
timestamps = timestamps[min_timestamp:min_timestamp+max_n_timestamps]
#### Set up sky model
lst = 299.28404*NP.pi/180
if use_GSM:
print 'Getting the GSM'
nside = 128
# Load in GSM
gsm = GlobalSkyModel()
sky = gsm.generate(f0*10**(-6))
sky = HP.ud_grade(sky,nside) # provided with nside=512, convert to lower res
inds = NP.arange(HP.nside2npix(nside))
theta,phi = HP.pixelfunc.pix2ang(nside,inds)
gc = Galactic(l=phi, b=NP.pi/2-theta, unit=(units.radian,units.radian))
radec = gc.fk5
eq = aipy.coord.radec2eq((-lst+radec.ra.radian,radec.dec.radian))
xyz = NP.dot(aipy.coord.eq2top_m(0,lat*NP.pi/180),eq)
# Keep just pixels above horizon
include = NP.where(xyz[2,:]>0)
sky = sky[include]
xyz = xyz[:,include].squeeze()
n_src = sky.shape[0]
plt.imshow(mycov, extent=[freqs.min(),
freqs.max(),
freqs.max(),
freqs.min()])
plt.colorbar()
plt.title('Full Unweighted Correlation, ' + tag)
plt.xlabel('Freq (MHz)')
plt.ylabel('Freq (MHz)')
plt.savefig('corr_mat_nowt_' + tag + '.png')
cc = mycov[freqs < 90, :]
cc = cc[:, freqs < 90]
#get the GSM, in this case evaluated at a single frequency (since this script was designed
#to investigate beam effects on the most boring possibly foregrounds)
gsm = GlobalSkyModel()
gsm_map_org = gsm.generate(80)
gsm_map_org = healpy.ud_grade(gsm_map_org, nside)
#if desired, cap the max flux in the map to be some number times the mean
if False:
thresh = 10
mn = np.mean(gsm_map_org)
gsm_map_org[gsm_map_org > thresh * mn] = thresh * mn
gsm_tag = '_capped'
else:
gsm_tag = ''
#rotate GSM into equatorial coordinates.
myrot = healpy.rotator.Rotator(coord=['G', 'E'])
gsm_map = myrot.rotate_map_pixel(gsm_map_org)
def Beam_G_plus_C_cross_C_four(freq,
scale,
itheta,
iphi,
nside,
sig=np.pi / 10):
"""plot of fourier for
GSM(scale)+CMB crossed with CMB
beam applied then accounted for
all compared to CMB alone
User defines frequency of GSM and scale to decrease it
User selects point of interest: theta and phi
User may designate sigma for beam
User selects nside
Default sigma is pi/10
"""
gsm = GlobalSkyModel()
#generates GSM model
gsmdata = gsm.generate(freq)
#rather than upgrading GSM, downgrade the CMB to 512
cmbdata = hp.read_map(
'/home/amber/data/cmb/COM_CMB_IQU-commander-field-Int_2048_R2.01_full.fits'
)
cmbmap = hp.ud_grade(cmbdata, 512)
LMAX = 1024
both = (gsmdata * scale) + cmbmap
#run the beam function with the given input
showbeam = beam(itheta, iphi, nside, sig)
npix = 12 * nside**2
#correct the data to account for the beam
correction = npix / (np.sum(showbeam))
both = both * showbeam * correction
#cross GSM+CMB with CMB and transform
c_plus_g_cross_c = hp.sphtfunc.anafast(both, map2=cmbmap, lmax=LMAX)
ell = np.arange(len(c_plus_g_cross_c))
plt.figure()
#convert variables into text for labels
freq_str = str(freq)
scale_str = str(scale)
plt.plot(ell,
ell * (ell + 1) * c_plus_g_cross_c,
color='b',
label='GSM(' + scale_str + ') at ' + freq_str +
' MHZ+CMB cross CMB')
#transform CMB
cmbmaponly = hp.anafast(cmbmap, lmax=LMAX)
ell = np.arange(len(cmbmaponly))
plt.plot(ell, ell * (ell + 1) * cmbmaponly, color='r', label='CMB')
#delete this line?
#hp.write_cmbmaponly('cmbmaponly.fits',cmbmaponly)
plt.xlabel('ell')
plt.ylabel('ell(ell+1)')
plt.grid()
#delete this line?
#hp.write_c_plus_g_cross_c('c_plus_g_cross_c.fits',c_plus_g_cross_c)
#convert variable into text for labels
theta_str = str(itheta)
phi_str = str(iphi)
plt.legend(loc='upper right')
plt.title('Point of Interest:(' + theta_str + ',' + phi_str + ')')
#set the limits for the y axis
plt.ylim(-0.00000001, 0.00000006)
plt.show()
# A script to plot the GSM to match LWA observation # IMPORTANT: Run this AFTER running the calibrate_LWA_data script - it relies on having certain things already calculated. import lwa_ant from datetime import datetime import scipy.ndimage as ndimage from pygsm import GlobalSkyModel #resolution = (FCNST.speed_of_light / f0) / 100.0 # rough resolution of LWA (in radians) print 'Getting the GSM' nside = 512 # Load in GSM gsm = GlobalSkyModel( ) # Note that I'm not using the 2016 GSM because it excludes certain sources (importantly CygA) sky = gsm.generate(f0 * 10**(-6)) # convert sky to Jy/(LWA beam) Aeff = 100.0**2 * NP.pi / 4.0 # basically a filled aperture sky *= 2 * FCNST.Boltzmann / Aeff * 10**26 # Jy per LWA beam #sky *= 2*FCNST.Boltzmann / (FCNST.speed_of_light / f0)**2 * 10**26 # Jy/str #sky = HP.smoothing(sky, fwhm=resolution) # Don't need this anymore because now I'm convolving with PSF sky = HP.ud_grade(sky, nside) # provided with nside=512, convert to lower res inds = NP.arange(HP.nside2npix(nside)) theta, phi = HP.pixelfunc.pix2ang(nside, inds) gc = Galactic(l=phi, b=NP.pi / 2 - theta, unit=(units.radian, units.radian)) radec = gc.fk5 eq = aipy.coord.radec2eq((-lst + radec.ra.radian, radec.dec.radian)) xyz = NP.dot(aipy.coord.eq2top_m(0, lat * NP.pi / 180), eq)
def test_set_methods():
gsm = GlobalSkyModel()
gsm.generate(40)
#gsm.view()
gsm.set_basemap('haslam')
gsm.view()
gsm.set_basemap('5deg')
gsm.view()
gsm.set_basemap('wmap')
gsm.view()
gsm.set_freq_unit('GHz')
gsm.view()