def testM5andSeeingAssignment(self):
"""
Test assignment of m5 and seeing seeing and bandpass in ObservationMetaData
"""
obsMD = ObservationMetaData(bandpassName=['u','g'], m5=[15.0, 16.0], seeing=[0.7, 0.6])
self.assertAlmostEqual(obsMD.m5['u'], 15.0, 10)
self.assertAlmostEqual(obsMD.m5['g'], 16.0, 10)
self.assertAlmostEqual(obsMD.seeing['u'], 0.7, 10)
self.assertAlmostEqual(obsMD.seeing['g'], 0.6, 10)
obsMD.setBandpassM5andSeeing(bandpassName=['i','z'], m5=[25.0, 22.0], seeing=[0.5, 0.4])
self.assertAlmostEqual(obsMD.m5['i'], 25.0, 10)
self.assertAlmostEqual(obsMD.m5['z'], 22.0, 10)
self.assertAlmostEqual(obsMD.seeing['i'], 0.5, 10)
self.assertAlmostEqual(obsMD.seeing['z'], 0.4, 10)
with self.assertRaises(KeyError):
obsMD.m5['u']
with self.assertRaises(KeyError):
obsMD.m5['g']
obsMD.m5 = [13.0, 14.0]
obsMD.seeing = [0.2, 0.3]
self.assertAlmostEqual(obsMD.m5['i'], 13.0, 10)
self.assertAlmostEqual(obsMD.m5['z'], 14.0, 10)
self.assertAlmostEqual(obsMD.seeing['i'], 0.2, 10)
self.assertAlmostEqual(obsMD.seeing['z'], 0.3, 10)
obsMD.setBandpassM5andSeeing(bandpassName=['k', 'j'], m5=[21.0, 23.0])
self.assertAlmostEqual(obsMD.m5['k'], 21.0, 10)
self.assertAlmostEqual(obsMD.m5['j'], 23.0, 10)
self.assertIsNone(obsMD.seeing)
obsMD.setBandpassM5andSeeing(bandpassName=['w', 'x'], seeing=[0.9, 1.1])
self.assertAlmostEqual(obsMD.seeing['w'], 0.9, 10)
self.assertAlmostEqual(obsMD.seeing['x'], 1.1, 10)
phoSimMD = {'Opsim_filter':[4]}
obsMD.phoSimMetaData = phoSimMD
self.assertEqual(obsMD.bandpass, 4)
self.assertTrue(obsMD.m5 is None)
self.assertTrue(obsMD.seeing is None)
def makePhoSimTestDB(filename='PhoSimTestDatabase.db', size=1000, seedVal=32, radius=0.1,
displacedRA=None, displacedDec=None,
bandpass='******', m5=None, seeing=None, **kwargs):
"""
Make a test database to storing cartoon information for the test phoSim input
catalog to use.
The method will return an ObservationMetaData object guaranteed to encompass the
objects in this database.
@param [in] filename is a string indicating the name of the DB file to be created
@param [in] size is the number of objects int he database
@param [in] seedVal is the seed passed to the random number generator
@param [in] radius is the radius (in degrees) of the field of view to be returned
@param [in] bandpass is the bandpas(es) of the observation to be passed to
ObservationMetaData (optional)
@param [in] m5 is the m5 value(s) to be passed to ObservationMetaData
(optional)
@param [in] seeing is the seeing value(s) in arcseconds to be passed to
ObservationMetaData (optional)
@param [in] displacedRA/Dec are numpy arrays that indicate where (in relation to the center
of the field of view) objects should be placed. These coordinates are in degrees. Specifying
either of these paramters will overwrite size. If you only specify one of these parameters, the other
will be set randomly. These parameters are optional.
"""
if os.path.exists(filename):
os.unlink(filename)
#just an example of some valid SED file names
galaxy_seds = ['Const.80E07.02Z.spec','Inst.80E07.002Z.spec','Burst.19E07.0005Z.spec']
agn_sed = 'agn.spec'
star_seds = ['km20_5750.fits_g40_5790','m2.0Full.dat','bergeron_6500_85.dat_6700']
numpy.random.seed(seedVal)
if displacedRA is not None and displacedDec is not None:
if len(displacedRA) != len(displacedDec):
raise RuntimeError("WARNING in makePhoSimTestDB displacedRA and displacedDec have different lengths")
if displacedRA is not None:
size = len(displacedRA)
elif displacedDec is not None:
size = len(displacedDec)
#create the ObservationMetaData object
mjd = 52000.0
alt = numpy.pi/2.0
az = 0.0
testSite = Site(name='LSST')
obsTemp = ObservationMetaData(mjd=mjd, site=testSite)
centerRA, centerDec = _raDecFromAltAz(alt, az, obsTemp)
rotTel = _getRotTelPos(centerRA, centerDec, obsTemp, 0.0)
rotSkyPos = _getRotSkyPos(centerRA, centerDec, obsTemp, rotTel)
obs_metadata = ObservationMetaData(pointingRA=numpy.degrees(centerRA),
pointingDec=numpy.degrees(centerDec),
rotSkyPos=numpy.degrees(rotSkyPos),
bandpassName=bandpass,
mjd=mjd,
boundType = 'circle', boundLength = 2.0*radius,
site=testSite,
m5=m5, seeing=seeing)
moon_alt = -90.0
sun_alt = -90.0
moon_ra, moon_dec = raDecFromAltAz(moon_alt, 0.0, obs_metadata)
dist2moon = haversine(numpy.radians(moon_ra), numpy.radians(moon_dec),
obs_metadata._pointingRA, obs_metadata._pointingDec)
obs_metadata.phoSimMetaData = {'Opsim_moonra': moon_ra,
'Opsim_moondec': moon_dec,
'Opsim_moonalt': moon_alt,
'Opsim_sunalt': sun_alt,
'Opsim_dist2moon': dist2moon,
'Opsim_rottelpos': numpy.degrees(rotTel)}
#Now begin building the database.
#First create the tables.
conn = sqlite3.connect(filename)
c = conn.cursor()
try:
c.execute('''CREATE TABLE galaxy_bulge
(galtileid int, galid int, bra real, bdec real, ra real, dec real, magnorm_bulge real,
sedname_bulge text, a_b real, b_b real, pa_bulge real, bulge_n int,
ext_model_b text, av_b real, rv_b real, u_ab real, g_ab real, r_ab real, i_ab real,
z_ab real, y_ab real, redshift real, BulgeHalfLightRadius real)''')
conn.commit()
except:
raise RuntimeError("Error creating galaxy_bulge table.")
try:
c.execute('''CREATE TABLE galaxy
(galtileid int, galid int, ra real, dec real,
bra real, bdec real, dra real, ddec real,
agnra real, agndec real,
magnorm_bulge, magnorm_disk, magnorm_agn,
sedname_bulge text, sedname_disk text, sedname_agn text,
varParamStr text,
a_b real, b_b real, pa_bulge real, bulge_n int,
a_d real, b_d real, pa_disk real, disk_n int,
ext_model_b text, av_b real, rv_b real,
ext_model_d text, av_d real, rv_d real,
u_ab real, g_ab real, r_ab real, i_ab real,
z_ab real, y_ab real,
redshift real, BulgeHalfLightRadius real, DiskHalfLightRadius real)''')
conn.commit()
except:
raise RuntimeError("Error creating galaxy table.")
try:
c.execute('''CREATE TABLE galaxy_agn
(galtileid int, galid int, agnra real, agndec real, ra real, dec real,
magnorm_agn real, sedname_agn text, varParamStr text, u_ab real,
g_ab real, r_ab real, i_ab real, z_ab real, y_ab real, redshift real)''')
except:
raise RuntimeError("Error creating galaxy_agn table.")
try:
c.execute('''CREATE TABLE StarAllForceseek
(simobjid int, ra real, decl real, magNorm real,
mudecl real, mura real, galacticAv real, vrad real, varParamStr text, sedFilename text, parallax real)''')
except:
raise RuntimeError("Error creating StarAllForceseek table.")
#Now generate the data to be stored in the tables.
rr = numpy.random.sample(size)*numpy.radians(radius)
theta = numpy.random.sample(size)*2.0*numpy.pi
if displacedRA is None:
ra = numpy.degrees(centerRA + rr*numpy.cos(theta))
else:
ra = numpy.degrees(centerRA) + displacedRA
if displacedDec is None:
dec = numpy.degrees(centerDec + rr*numpy.sin(theta))
else:
dec = numpy.degrees(centerDec) + displacedDec
bra = numpy.radians(ra+numpy.random.sample(size)*0.01*radius)
bdec = numpy.radians(dec+numpy.random.sample(size)*0.01*radius)
dra = numpy.radians(ra + numpy.random.sample(size)*0.01*radius)
ddec = numpy.radians(dec + numpy.random.sample(size)*0.01*radius)
agnra = numpy.radians(ra + numpy.random.sample(size)*0.01*radius)
agndec = numpy.radians(dec + numpy.random.sample(size)*0.01*radius)
magnorm_bulge = numpy.random.sample(size)*4.0 + 17.0
magnorm_disk = numpy.random.sample(size)*5.0 + 17.0
magnorm_agn = numpy.random.sample(size)*5.0 + 17.0
b_b = numpy.random.sample(size)*0.2
a_b = b_b+numpy.random.sample(size)*0.05
b_d = numpy.random.sample(size)*0.5
a_d = b_d+numpy.random.sample(size)*0.1
BulgeHalfLightRadius = numpy.random.sample(size)*0.2
DiskHalfLightRadius = numpy.random.sample(size)*0.5
pa_bulge = numpy.random.sample(size)*360.0
pa_disk = numpy.random.sample(size)*360.0
av_b = numpy.random.sample(size)*0.4
av_d = numpy.random.sample(size)*0.4
rv_b = numpy.random.sample(size)*0.1 + 3.0
rv_d = numpy.random.sample(size)*0.1 + 3.0
u_ab = numpy.random.sample(size)*4.0 + 17.0
g_ab = numpy.random.sample(size)*4.0 + 17.0
r_ab = numpy.random.sample(size)*4.0 + 17.0
i_ab = numpy.random.sample(size)*4.0 + 17.0
z_ab = numpy.random.sample(size)*4.0 + 17.0
y_ab = numpy.random.sample(size)*4.0 +17.0
redshift = numpy.random.sample(size)*2.0
t0_mjd = mjd - numpy.random.sample(size)*1000.0
agn_tau = numpy.random.sample(size)*1000.0 + 1000.0
agnSeed = numpy.random.random_integers(low=2, high=4000, size=size)
agn_sfu = numpy.random.sample(size)
agn_sfg = numpy.random.sample(size)
agn_sfr = numpy.random.sample(size)
agn_sfi = numpy.random.sample(size)
agn_sfz = numpy.random.sample(size)
agn_sfy = numpy.random.sample(size)
rrStar = numpy.random.sample(size)*numpy.radians(radius)
thetaStar = numpy.random.sample(size)*2.0*numpy.pi
if displacedRA is None:
raStar = centerRA + rrStar*numpy.cos(thetaStar)
else:
raStar = centerRA + numpy.radians(displacedRA)
if displacedDec is None:
decStar = centerDec + rrStar*numpy.sin(thetaStar)
else:
decStar = centerDec + numpy.radians(displacedDec)
raStar = numpy.degrees(raStar)
decStar = numpy.degrees(decStar)
magnormStar = numpy.random.sample(size)*4.0 + 17.0
mudecl = numpy.random.sample(size)*0.0001
mura = numpy.random.sample(size)*0.0001
galacticAv = numpy.random.sample(size)*0.05*3.1
vrad = numpy.random.sample(size)*1.0
parallax = 0.00045+numpy.random.sample(size)*0.00001
period = numpy.random.sample(size)*20.0
amp = numpy.random.sample(size)*5.0
#write the data to the tables.
for i in range(size):
cmd = '''INSERT INTO galaxy_bulge VALUES (%i, %i, %f, %f, %f, %f, %f,
'%s', %f, %f, %f, %i, '%s', %f, %f, %f, %f, %f, %f, %f, %f, %f, %f)''' %\
(i, i, bra[i], bdec[i], ra[i], dec[i], magnorm_bulge[i], galaxy_seds[i%len(galaxy_seds)],
a_b[i], b_b[i], pa_bulge[i], 4, 'CCM', av_b[i], rv_b[i], u_ab[i], g_ab[i],
r_ab[i], i_ab[i], z_ab[i], y_ab[i], redshift[i], BulgeHalfLightRadius[i])
c.execute(cmd)
varParam = {'varMethodName':'applyAgn',
'pars':{'agn_tau':round(agn_tau[i],4), 't0_mjd':round(t0_mjd[i],4),
'agn_sfu':round(agn_sfu[i],4), 'agn_sfg':round(agn_sfg[i],4), 'agn_sfr':round(agn_sfr[i],4),
'agn_sfi':round(agn_sfi[i],4), 'agn_sfz':round(agn_sfz[i],4), 'agn_sfy':round(agn_sfy[i],4),
'seed':int(agnSeed[i])}}
paramStr = json.dumps(varParam)
cmd = '''INSERT INTO galaxy VALUES (%i, %i, %f, %f, %f, %f, %f, %f, %f, %f, %f, %f, %f,
'%s', '%s', '%s', '%s',
%f, %f, %f, %i,
%f, %f, %f, %i,
'%s', %f, %f,
'%s', %f, %f,
%f, %f, %f, %f, %f, %f,
%f, %f, %f)''' %\
(i, i, ra[i], dec[i], bra[i], bdec[i], dra[i], ddec[i], agnra[i], agndec[i],
magnorm_bulge[i], magnorm_disk[i], magnorm_agn[i],
galaxy_seds[i%len(galaxy_seds)], galaxy_seds[i%len(galaxy_seds)], agn_sed,
paramStr,
a_b[i], b_b[i], pa_bulge[i], 4,
a_d[i], b_d[i], pa_disk[i], 1,
'CCM', av_b[i], rv_b[i],
'CCM', av_d[i], rv_d[i],
u_ab[i], g_ab[i], r_ab[i], i_ab[i], z_ab[i], y_ab[i], redshift[i],
BulgeHalfLightRadius[i], DiskHalfLightRadius[i])
c.execute(cmd)
cmd = '''INSERT INTO galaxy_agn VALUES (%i, %i, %f, %f, %f, %f, %f, '%s', '%s',
%f, %f, %f, %f, %f, %f, %f)''' %\
(i, i, agnra[i], agndec[i], ra[i], dec[i],
magnorm_agn[i], agn_sed, paramStr,
u_ab[i], g_ab[i], r_ab[i], i_ab[i],
z_ab[i], y_ab[i], redshift[i])
c.execute(cmd)
varParam = {'varMethodName':'testVar', 'pars':{'period':period[i], 'amplitude':amp[i]}}
paramStr = json.dumps(varParam)
cmd = '''INSERT INTO StarAllForceseek VALUES (%i, %f, %f, %f, %f, %f, %f, %f, '%s', '%s', %f)''' %\
(i, raStar[i], decStar[i], magnormStar[i], mudecl[i], mura[i],
galacticAv[i], vrad[i], paramStr, star_seds[i%len(star_seds)], parallax[i])
c.execute(cmd)
conn.commit()
conn.close()
return obs_metadata
def testAssignment(self):
"""
Test that ObservationMetaData member variables get passed correctly
"""
mjd = 5120.0
RA = 1.5
Dec = -1.1
rotSkyPos = -10.0
skyBrightness = 25.0
testObsMD = ObservationMetaData()
testObsMD.pointingRA = RA
testObsMD.pointingDec = Dec
testObsMD.rotSkyPos = rotSkyPos
testObsMD.skyBrightness = skyBrightness
testObsMD.mjd = mjd
testObsMD.boundType = 'box'
testObsMD.boundLength = [1.2, 3.0]
self.assertAlmostEqual(testObsMD.pointingRA, RA, 10)
self.assertAlmostEqual(testObsMD.pointingDec, Dec, 10)
self.assertAlmostEqual(testObsMD.rotSkyPos, rotSkyPos, 10)
self.assertAlmostEqual(testObsMD.skyBrightness, skyBrightness, 10)
self.assertEqual(testObsMD.boundType, 'box')
self.assertAlmostEqual(testObsMD.boundLength[0], 1.2, 10)
self.assertAlmostEqual(testObsMD.boundLength[1], 3.0, 10)
self.assertAlmostEqual(testObsMD.mjd.TAI, mjd, 10)
#test reassignment
testObsMD.pointingRA = RA+1.0
testObsMD.pointingDec = Dec+1.0
testObsMD.rotSkyPos = rotSkyPos+1.0
testObsMD.skyBrightness = skyBrightness+1.0
testObsMD.boundLength = 2.2
testObsMD.boundType = 'circle'
testObsMD.mjd = mjd + 10.0
self.assertAlmostEqual(testObsMD.pointingRA, RA+1.0, 10)
self.assertAlmostEqual(testObsMD.pointingDec, Dec+1.0, 10)
self.assertAlmostEqual(testObsMD.rotSkyPos, rotSkyPos+1.0, 10)
self.assertAlmostEqual(testObsMD.skyBrightness, skyBrightness+1.0, 10)
self.assertEqual(testObsMD.boundType, 'circle')
self.assertAlmostEqual(testObsMD.boundLength,2.2, 10)
self.assertAlmostEqual(testObsMD.mjd.TAI, mjd+10.0, 10)
phosimMD = OrderedDict([('pointingRA', (-2.0,float)),
('pointingDec', (0.9,float)),
('Opsim_rotskypos', (1.1,float)),
('Opsim_expmjd',(4000.0,float)),
('Opsim_filter',('g',str))])
testObsMD.phoSimMetaData = phosimMD
self.assertAlmostEqual(testObsMD.pointingRA, numpy.degrees(-2.0), 10)
self.assertAlmostEqual(testObsMD.pointingDec, numpy.degrees(0.9), 10)
self.assertAlmostEqual(testObsMD.rotSkyPos, numpy.degrees(1.1))
self.assertAlmostEqual(testObsMD.mjd.TAI, 4000.0, 10)
self.assertAlmostEqual(testObsMD.bandpass, 'g')
testObsMD = ObservationMetaData(mjd=mjd, pointingRA=RA,
pointingDec=Dec, rotSkyPos=rotSkyPos, bandpassName='z',
skyBrightness=skyBrightness)
self.assertAlmostEqual(testObsMD.mjd.TAI,5120.0,10)
self.assertAlmostEqual(testObsMD.pointingRA,1.5,10)
self.assertAlmostEqual(testObsMD.pointingDec,-1.1,10)
self.assertAlmostEqual(testObsMD.rotSkyPos,-10.0,10)
self.assertEqual(testObsMD.bandpass,'z')
self.assertAlmostEqual(testObsMD.skyBrightness, skyBrightness, 10)
testObsMD = ObservationMetaData()
testObsMD.phoSimMetaData = phosimMD
self.assertAlmostEqual(testObsMD.mjd.TAI,4000.0,10)
#recall that pointingRA/Dec are stored as radians in phoSim metadata
self.assertAlmostEqual(testObsMD.pointingRA,numpy.degrees(-2.0),10)
self.assertAlmostEqual(testObsMD.pointingDec,numpy.degrees(0.9),10)
self.assertAlmostEqual(testObsMD.rotSkyPos,numpy.degrees(1.1),10)
self.assertEqual(testObsMD.bandpass,'g')
testObsMD = ObservationMetaData()
testObsMD.phoSimMetaData = phosimMD
self.assertAlmostEqual(testObsMD.mjd.TAI,4000.0,10)
#recall that pointingRA/Dec are stored as radians in phoSim metadata
self.assertAlmostEqual(testObsMD.pointingRA,numpy.degrees(-2.0),10)
self.assertAlmostEqual(testObsMD.pointingDec,numpy.degrees(0.9),10)
self.assertAlmostEqual(testObsMD.rotSkyPos,numpy.degrees(1.1),10)
self.assertEqual(testObsMD.bandpass,'g')
# test assigning ModifiedJulianDate
obs = ObservationMetaData()
mjd = ModifiedJulianDate(TAI=57388.0)
obs.mjd = mjd
self.assertEqual(obs.mjd, mjd)
mjd2 = ModifiedJulianDate(TAI=45000.0)
obs.mjd = mjd2
self.assertEqual(obs.mjd, mjd2)
self.assertNotEqual(obs.mjd, mjd)
