def phase_ft1(ft1,phase,savedir):
if np.allclose(phase.phase_fraction,1):
phased_ft1 = ft1
else:
# apply phase cut to ft1 file
phased_ft1 = join(savedir,'ft1_phased.fits')
if not os.path.exists(phased_ft1):
phasetools.phase_cut(ft1,phased_ft1,phaseranges=phase.tolist(dense=False))
return phased_ft1
def phase_ft1(ft1, phase, savedir):
if np.allclose(phase.phase_fraction, 1):
phased_ft1 = ft1
else:
# apply phase cut to ft1 file
phased_ft1 = join(savedir, 'ft1_phased.fits')
if not os.path.exists(phased_ft1):
phasetools.phase_cut(ft1,
phased_ft1,
phaseranges=phase.tolist(dense=False))
return phased_ft1
def setup_pwn(name,pwndata,phase, free_radius=5, tempdir=None, emin=1.0e2, emax=1.0e5,maxroi=10,model=None,**kwargs):
"""Name of the source
pwndata Yaml file
returns pointlike ROI.
"""
sources=yaml.load(open(pwndata))
catalog_name=sources[name]['catalog']
ltcube=sources[name]['ltcube']
pulsar_position=SkyDir(*sources[name]['dir'])
ft2=sources[name]['ft2']
ft1=sources[name]['ft1']
# in case no list was passed
if len(phase)==2 and isinstance(phase[0],numbers.Real) and \
isinstance(phase[1],numbers.Real):
# write in case phase wraps around.
if phase[0]>phase[1]:
phase=[[phase[0],1.0],[0.0,phase[1]]]
else:
phase = [phase]
phase_factor=get_phase_factor(phase)
print "phase"
print phase
print "phase_factor=%.2f"%phase_factor
catalog=FermiCatalog(e("$FERMI/catalogs/gll_psc_v02.fit"),free_radius=free_radius)
catalog_source=[i for i in catalog.get_sources(SkyDir(),180) if i.name==catalog_name][0]
center=catalog_source.skydir
if tempdir is None: tempdir=mkdtemp(prefix='/scratch/')
binfile=j(tempdir,'binned_phased.fits')
# apply phase cut to ft1 file
phased_ft1 = j(tempdir,'ft1_phased.fits')
phasetools.phase_cut(ft1,phased_ft1,phaseranges=phase)
# create a temporary ltcube scaled by the phase factor
# phased_ltcube=j(tempdir,'phased_ltcube.fits')
# phase_ltcube(ltcube,phased_ltcube, phase=phase)
phased_ltcube=ltcube
from uw.like.pointspec import DataSpecification
data_specification = DataSpecification(
ft1files = phased_ft1,
ft2files = ft2,
ltcube = phased_ltcube,
binfile = binfile)
spectral_analysis = SpectralAnalysis(data_specification,
binsperdec = 4,
emin = 100,
emax = 100000,
irf = "P6_V3_DIFFUSE",
roi_dir = center,
maxROI = maxroi,
minROI = maxroi)
if model == None :
roi=spectral_analysis.roi(
roi_dir=center,
diffuse_sources=get_default_diffuse(diffdir=e("$FERMI/diffuse"),
gfile="gll_iem_v02.fit",
ifile="isotropic_iem_v02.txt"),
catalogs = catalog,
phase_factor = 1.0,
fit_emin = [emin,emin],
fit_emax = [emax,emax],
**kwargs)
else :
roi=spectral_analysis.roi(
roi_dir=center,
xmlfile = model,
phase_factor =1.0,
fit_emin = [emin,emin],
fit_emax = [emax,emax],
**kwargs)
print "---------------------Energy range--------------------"
print "emin="+str(roi.bands[0].emin)+"\n"
print "emax="+str(roi.bands[len(roi.bands)-1].emax)+"\n"
# keep overall flux of catalog source,
# but change the starting index to 2.
roi.modify(which=catalog_name, name=name, index=2,
keep_old_flux=True)
return roi
def setup_region(name,pwndata,phase, free_radius=5, tempdir=None, maxroi=10,
xml=None, **kwargs):
"""Name of the source
pwndata Yaml file
returns pointlike ROI.
"""
phase = PhaseRange(phase)
sources=yaml.load(open(pwndata))
catalog_name=sources[name]['catalog']['2fgl']
ltcube=sources[name]['ltcube']
pulsar_position=SkyDir(*sources[name]['dir'])
ft2=sources[name]['ft2']
ft1=sources[name]['ft1']
catalog=FermiCatalog(e("$FERMI/catalogs/gll_psc_v02.fit"))
catalog=Catalog2FGL('$FERMI/catalogs/gll_psc_v05.fit',
latextdir='$FERMI/extended_archives/gll_psc_v05_templates',
free_radius=free_radius)
catalog_source=catalog.get_source(catalog_name)
center=catalog_source.skydir
if tempdir is None: tempdir=mkdtemp(prefix='/scratch/')
binfile=j(tempdir,'binned_phased.fits')
if np.allclose(phase.phase_fraction,1):
phased_ltcube = ltcube
phased_ft1 = ft1
else:
# create a temporary ltcube scaled by the phase factor
phased_ltcube=j(tempdir,'phased_ltcube.fits')
phase_ltcube(ltcube,phased_ltcube, phase=phase)
# apply phase cut to ft1 file
phased_ft1 = j(tempdir,'ft1_phased.fits')
phasetools.phase_cut(ft1,phased_ft1,phaseranges=phase.tolist(dense=False))
from uw.like.pointspec import DataSpecification
ds = DataSpecification(
ft1files = phased_ft1,
ft2files = ft2,
ltcube = phased_ltcube,
binfile = binfile)
sa = SpectralAnalysis(ds,
binsperdec = 8,
emin = 100,
emax = 100000,
irf = "P6_V11_DIFFUSE",
roi_dir = center,
maxROI = maxroi,
minROI = maxroi)
if xml is None:
roi=sa.roi(
diffuse_sources=get_default_diffuse(diffdir="/afs/slac/g/glast/groups/diffuse/mapcubes",
gfile="gll_iem_v02.fit",
ifile="isotropic_iem_v02.txt"),
catalogs = catalog,
phase_factor =1,
**kwargs)
else:
roi=sa.roi_from_xml(
roi_dir=center,
xmlfile = xml,
phase_factor =1,
**kwargs)
print 'bins ',roi.bin_edges
roi.del_source(catalog_name)
return roi
def setup_pwn(name,pwndata,phase, free_radius=5, tempdir=None, **kwargs):
"""Name of the source
pwndata Yaml file
returns pointlike ROI.
"""
sources=yaml.load(open(pwndata))
catalog_name=sources[name]['catalog']
ltcube=sources[name]['ltcube']
pulsar_position=SkyDir(*sources[name]['dir'])
ft2=sources[name]['ft2']
ft1=sources[name]['ft1']
# in case no list was passed
if len(phase)==2 and isinstance(phase[0],numbers.Real) and \
isinstance(phase[1],numbers.Real):
# write in case phase wraps around.
if phase[0]>phase[1]:
phase=[[phase[0],1.0],[0.0,phase[1]]]
else:
phase = [phase]
phase_factor=get_phase_factor(phase)
catalog=FermiCatalog(e("$FERMI/catalogs/gll_psc_v02.fit"),free_radius=free_radius)
catalog_source=[i for i in catalog.get_sources(SkyDir(),180) if i.name==catalog_name][0]
center=catalog_source.skydir
if tempdir is None: tempdir=mkdtemp(prefix='/scratch/')
binfile=j(tempdir,'binned_phased.fits')
# apply phase cut to ft1 file
phased_ft1 = j(tempdir,'ft1_phased.fits')
phasetools.phase_cut(ft1,phased_ft1,phaseranges=phase)
# create a temporary ltcube scaled by the phase factor
# phased_ltcube=j(tempdir,'phased_ltcube.fits')
# phase_ltcube(ltcube,phased_ltcube, phase=[0.0,1.0])
phased_ltcube=ltcube
from uw.like.pointspec import DataSpecification
data_specification = DataSpecification(
ft1files = phased_ft1,
ft2files = ft2,
ltcube = phased_ltcube,
binfile = binfile)
spectral_analysis = SpectralAnalysis(data_specification,
binsperdec = 4,
emin = 100,
emax = 100000,
irf = "P6_V3_DIFFUSE",
roi_dir = center,
maxROI = 10,
minROI = 10)
roi=spectral_analysis.roi(
roi_dir=center,
diffuse_sources=get_default_diffuse(diffdir=e("$FERMI/diffuse"),
gfile="gll_iem_v02.fit",
ifile="isotropic_iem_v02.txt"),
catalogs = catalog,
phase_factor = phase_factor,
**kwargs) # phaseing already done to the ltcube
print "phase_factor=%.2f"%phase_factor
# keep overall flux of catalog source,
# but change the starting index to 2.
roi.modify(which=catalog_name, name=name, index=2,
keep_old_flux=True)
roi.toXML(filename="essai")
print roi
roi.print_summary()
for names in roi.get_names():
try :
roi.modify(names,Norm=roi.get_model(names)[0]*roi.phase_factor)
except :
try :
roi.modify(names,Int_flux=roi.get_model(names)[0]*roi.phase_factor)
except :
print names
table=roi.get_names()
print roi.modify(which=table[len(table)-2],model=PowerLaw(p=[1.0*phase_factor,0.1]),free=[True,False])
print roi.modify(which=table[len(table)-1],model=PowerLaw(p=[1.0*phase_factor,0.1]),free=[True,False])
# print roi.modify(which='eg_v02',free=[False])
print roi
return roi
