LSSTApplications  16.0-10-g0ee56ad+5,16.0-11-ga33d1f2+5,16.0-12-g3ef5c14+3,16.0-12-g71e5ef5+18,16.0-12-gbdf3636+3,16.0-13-g118c103+3,16.0-13-g8f68b0a+3,16.0-15-gbf5c1cb+4,16.0-16-gfd17674+3,16.0-17-g7c01f5c+3,16.0-18-g0a50484+1,16.0-20-ga20f992+8,16.0-21-g0e05fd4+6,16.0-21-g15e2d33+4,16.0-22-g62d8060+4,16.0-22-g847a80f+4,16.0-25-gf00d9b8+1,16.0-28-g3990c221+4,16.0-3-gf928089+3,16.0-32-g88a4f23+5,16.0-34-gd7987ad+3,16.0-37-gc7333cb+2,16.0-4-g10fc685+2,16.0-4-g18f3627+26,16.0-4-g5f3a788+26,16.0-5-gaf5c3d7+4,16.0-5-gcc1f4bb+1,16.0-6-g3b92700+4,16.0-6-g4412fcd+3,16.0-6-g7235603+4,16.0-69-g2562ce1b+2,16.0-8-g14ebd58+4,16.0-8-g2df868b+1,16.0-8-g4cec79c+6,16.0-8-gadf6c7a+1,16.0-8-gfc7ad86,16.0-82-g59ec2a54a+1,16.0-9-g5400cdc+2,16.0-9-ge6233d7+5,master-g2880f2d8cf+3,v17.0.rc1
LSSTDataManagementBasePackage
compareModel.py
Go to the documentation of this file.
1 """
2 Compare the input fake galaxy model to the output one
3 """
4 import argparse
5 
6 import numpy as np
7 import astropy.table
8 
9 import lsst.pipe.base
10 import lsst.afw.table
12 import lsst.daf.persistence as dafPersist
13 
14 from .matchFakes import getFakeSources
15 
16 
17 def getParams(record, galType='sersic'):
18  """
19  return the semi-major axis, axis ratio and position angle (in degrees)
20  """
21  fluxType = {'dev': 'cmodel.dev',
22  'exp': 'cmodel.exp',
23  'sersic': 'cmodel',
24  'cmodel': 'cmodel'}[galType]
25 
26  e = lsst.afw.geom.ellipses.Axes(record.get(fluxType + '.ellipse'))
27  q = (e.getB() / e.getA())
28  reff = e.getA()
29  pa = (e.getTheta() * 180.0 / np.pi)
30  return reff, q, pa
31 
32 
33 def getMag(record, fluxType='cmodel'):
34  """
35  return the magnitude and error
36  """
37  flux, fluxerr = record.get(fluxType), record.get(fluxType+'.err')
38  mag, magerr = -2.5 * np.log10(flux), 2.5/np.log(10.0)*fluxerr/flux
39  mag += record.get('zeropoint')
40  return mag, magerr
41 
42 
43 def writeNumpyTable(fakeTable):
44  """Writes output to numpy
45  """
46  npTable = np.recarray(len(fakeTable),
47  dtype={'names': ['id', 'fakeid', 'visit', 'ccd',
48  'cmodelMag', 'expMag',
49  'devMag',
50  'expReff', 'devReff',
51  'expBA', 'devBA',
52  'expPosAng', 'devPosAng',
53  'cmodelMagErr', 'expMagErr',
54  'devMagErr', 'nchild', 'parent'],
55  'formats': [int, int, int, int,
56  float, float,
57  float, float, float, float,
58  float, float, float, float,
59  float, float, int, int]})
60  for indFake, fake in enumerate(fakeTable):
61  npTable[indFake]['id'] = fake.get('id')
62  npTable[indFake]['fakeid'] = fake.get('fakeId')
63  npTable[indFake]['ccd'] = fake.get('ccd')
64  npTable[indFake]['visit'] = fake.get('visit')
65  nameMatch = {'sersic': 'cmodel',
66  'exp': 'cmodel.exp',
67  'cmodel': 'cmodel',
68  'dev': 'cmodel.dev'}
69  for name in ('cmodel', 'exp', 'dev'):
70  m1, m2 = getMag(fake, nameMatch[name]+'.flux')
71  npTable[indFake][name+'Mag'] = m1
72  npTable[indFake][name+'MagErr'] = m2
73  for name in ('exp', 'dev'):
74  params = getParams(fake, name)
75  npTable[indFake][name+'Reff'] = params[0]
76  npTable[indFake][name+'BA'] = params[1]
77  npTable[indFake][name+'PosAng'] = params[2]
78  npTable[indFake]['nchild'] = fake.get('deblend.nchild')
79  npTable[indFake]['parent'] = fake.get('parent')
80  return npTable
81 
82 
83 def main(root, visit, ccds, galType='sersic', output='outputs/'):
84  """
85  main function for controlling fake source comparison
86  """
87  butler = dafPersist.Butler(root)
88  fakeTable = None
89  for ccd in ccds:
90  dataId = {'visit': visit, 'ccd': ccd}
91  try:
92  temp = getFakeSources(butler, dataId, tol=1.0,
93  visit=True, ccd=True)
94  except Exception:
95  continue
96  if fakeTable is None:
97  fakeTable = temp.copy(True)
98  else:
99  fakeTable.extend(temp, True)
100 
101  npTable = writeNumpyTable(fakeTable)
102  rerunName = root.split('/')[-2]
103  tableFileName = output + '/' + rerunName + '_galMags.txt'
104  astropy.table.Table(npTable).write(tableFileName, format='ascii')
105 
106 
107 if __name__ == '__main__':
108  parser = argparse.ArgumentParser()
109  parser.add_argument('root', help="Root directory of data repository")
110  parser.add_argument("visit", type=int, help="Visit")
111  parser.add_argument("--ccd", nargs='+', type=int, help="CCD(s)")
112  parser.add_argument('-g', '--galtype', type=str, dest='galType',
113  choices=['exp', 'dev', 'sersic'])
114  parser.add_argument('-o', '--outputpath', dest='outpath',
115  help='path for output')
116  args = parser.parse_args()
117 
118  main(args.root, args.visit, args.ccd, galType=args.galType,
119  output=args.outpath)
def getMag(record, fluxType='cmodel')
Definition: compareModel.py:33
def getParams(record, galType='sersic')
Definition: compareModel.py:17
def getFakeSources(butler, dataId, tol=1.0, extraCols=('zeropoint', 'visit', 'ccd'), includeMissing=False, footprints=False, radecMatch=None, multiband=False, reffMatch=False, pix=0.168, minRad=None, raCol='RA', decCol='Dec')
Definition: matchFakes.py:219
def writeNumpyTable(fakeTable)
Definition: compareModel.py:43
def main(root, visit, ccds, galType='sersic', output='outputs/')
Definition: compareModel.py:83
An ellipse core for the semimajor/semiminor axis and position angle parametrization (a...
Definition: Axes.h:47