37 pixelScale = lsst.pex.config.Field(
38 dtype=float, default=0.2, optional=
False,
39 doc=
"Pixel scale for mock WCSs in arcseconds/pixel" 41 doRotate = lsst.pex.config.Field(
42 dtype=bool, default=
True, optional=
False,
43 doc=
"Whether to randomly rotate observations relative to the tract Wcs" 45 fluxMag0 = lsst.pex.config.Field(
46 dtype=float, default=1E11, optional=
False,
47 doc=
"Flux at zero magnitude used to define Calibs." 49 fluxMag0Err = lsst.pex.config.Field(
50 dtype=float, default=100.0, optional=
False,
51 doc=
"Error on flux at zero magnitude used to define Calibs; used to add scatter as well." 53 expTime = lsst.pex.config.Field(
54 dtype=float, default=60.0, optional=
False,
55 doc=
"Exposure time set in generated Calibs (does not affect flux or noise level)" 57 psfImageSize = lsst.pex.config.Field(
58 dtype=int, default=21, optional=
False,
59 doc=
"Image width and height of generated Psfs." 61 psfMinSigma = lsst.pex.config.Field(
62 dtype=float, default=1.5, optional=
False,
63 doc=
"Minimum radius for generated Psfs." 65 psfMaxSigma = lsst.pex.config.Field(
66 dtype=float, default=3.0, optional=
False,
67 doc=
"Maximum radius for generated Psfs." 69 apCorrOrder = lsst.pex.config.Field(
70 dtype=int, default=1, optional=
False,
71 doc=
"Polynomial order for aperture correction fields" 73 seed = lsst.pex.config.Field(dtype=int, default=1, doc=
"Seed for numpy random number generator")
77 """Task to generate mock Exposure parameters (Wcs, Psf, Calib), intended for use as a subtask 81 - document "pa" in detail; angle of what to what? 82 - document the catalog parameter of the run method 85 ConfigClass = MockObservationConfig
88 lsst.pipe.base.Task.__init__(self, **kwds)
90 self.
ccdKey = self.
schema.addField(
"ccd", type=np.int32, doc=
"CCD number")
91 self.
visitKey = self.
schema.addField(
"visit", type=np.int32, doc=
"visit number")
93 self.
filterKey = self.
schema.addField(
"filter", type=str, doc=
"Bandpass filter name", size=16)
94 self.
rng = np.random.RandomState(self.config.seed)
96 def run(self, butler, n, tractInfo, camera, catalog=None):
97 """Driver that generates an ExposureCatalog of mock observations. 99 @param[in] butler: a data butler 100 @param[in] n: number of pointings 101 @param[in] camera: camera geometry (an lsst.afw.cameraGeom.Camera) 102 @param[in] catalog: catalog to which to add observations (an ExposureCatalog); 103 if None then a new catalog is created. 105 @todo figure out what `pa` is and use that knowledge to set `boresightRotAng` and `rotType` 110 if not catalog.getSchema().contains(self.
schema):
111 raise ValueError(
"Catalog schema does not match Task schema")
116 exposureTime=self.config.expTime,
118 boresightRaDec=position,
120 for detector
in camera:
122 record = catalog.addNew()
123 record.setI(self.
ccdKey, detector.getId())
127 record.setWcs(self.buildWcs(position, pa, detector)) 128 record.setCalib(calib) 129 record.setVisitInfo(visitInfo) 130 record.setPsf(self.buildPsf(detector)) 133 record.setBBox(detector.getBBox()) 134 detectorId = detector.getId() 135 obj = butler.get("ccdExposureId", visit=visit, ccd=detectorId, immediate=
True)
141 """Generate (celestial) positions and rotation angles that define field locations. 143 Default implementation draws random pointings that are uniform in the tract's image 146 @param[in] n: number of pointings 147 @param[in] tractInfo: skymap tract (a lsst.skymap.TractInfo) 148 @return a Python iterable over (coord, angle) pairs: 149 - coord is an ICRS object position (an lsst.afw.geom.SpherePoint) 150 - angle is a position angle (???) (an lsst.afw.geom.Angle) 152 The default implementation returns an iterator (i.e. the function is a "generator"), 153 but derived-class overrides may return any iterable. 155 wcs = tractInfo.getWcs()
156 bbox = lsst.afw.geom.Box2D(tractInfo.getBBox())
157 bbox.grow(lsst.afw.geom.Extent2D(-0.1 * bbox.getWidth(), -0.1 * bbox.getHeight()))
159 x = self.
rng.rand() * bbox.getWidth() + bbox.getMinX()
160 y = self.
rng.rand() * bbox.getHeight() + bbox.getMinY()
161 pa = 0.0 * lsst.afw.geom.radians
162 if self.config.doRotate:
163 pa = self.
rng.rand() * 2.0 * np.pi * lsst.afw.geom.radians
164 yield wcs.pixelToSky(x, y), pa
167 """Build a simple TAN Wcs with no distortion and exactly-aligned CCDs. 169 @param[in] position: ICRS object position on sky (on lsst.afw.geom.SpherePoint) 170 @param[in] pa: position angle (an lsst.afw.geom.Angle) 171 @param[in] detector: detector information (an lsst.afw.cameraGeom.Detector) 174 pixelScale = (self.config.pixelScale * lsst.afw.geom.arcseconds).asDegrees()
175 cd = (lsst.afw.geom.LinearTransform.makeScaling(pixelScale) *
176 lsst.afw.geom.LinearTransform.makeRotation(pa))
177 crpix = detector.transform(lsst.afw.geom.Point2D(0, 0), FOCAL_PLANE, PIXELS)
182 """Build a simple Calib object with exposure time fixed by config, fluxMag0 drawn from 183 a Gaussian defined by config, and mid-time set to DateTime.now(). 187 self.
rng.randn() * self.config.fluxMag0Err + self.config.fluxMag0,
188 self.config.fluxMag0Err
193 """Build a simple Gaussian Psf with linearly-varying ellipticity and size. 195 The Psf pattern increases sigma_x linearly along the x direction, and sigma_y 196 linearly along the y direction. 198 @param[in] detector: detector information (an lsst.afw.cameraGeom.Detector) 199 @return a psf (an instance of lsst.meas.algorithms.KernelPsf) 201 bbox = detector.getBBox()
202 dx = (self.config.psfMaxSigma - self.config.psfMinSigma) / bbox.getWidth()
203 dy = (self.config.psfMaxSigma - self.config.psfMinSigma) / bbox.getHeight()
204 sigmaXFunc = lsst.afw.math.PolynomialFunction2D(1)
205 sigmaXFunc.setParameter(0, self.config.psfMinSigma - dx * bbox.getMinX() - dy * bbox.getMinY())
206 sigmaXFunc.setParameter(1, dx)
207 sigmaXFunc.setParameter(2, 0.0)
208 sigmaYFunc = lsst.afw.math.PolynomialFunction2D(1)
209 sigmaYFunc.setParameter(0, self.config.psfMinSigma)
210 sigmaYFunc.setParameter(1, 0.0)
211 sigmaYFunc.setParameter(2, dy)
212 angleFunc = lsst.afw.math.PolynomialFunction2D(0)
214 spatialFuncList.append(sigmaXFunc)
215 spatialFuncList.append(sigmaYFunc)
216 spatialFuncList.append(angleFunc)
218 self.config.psfImageSize, self.config.psfImageSize,
219 lsst.afw.math.GaussianFunction2D(self.config.psfMinSigma, self.config.psfMinSigma),
225 """Build an ApCorrMap with random linearly-varying fields for all 226 flux fields registered for aperture correction. 228 These flux field names are used only as strings; there is no 229 connection to any actual algorithms with those names or the PSF model. 231 order = self.config.apCorrOrder
233 def makeRandomBoundedField():
234 """Make an upper-left triangular coefficient array appropriate 235 for a 2-d polynomial.""" 236 array = np.zeros((order + 1, order + 1), dtype=float)
237 for n
in range(order + 1):
238 array[n, 0:order + 1 - n] = self.
rng.randn(order + 1 - n)
241 bbox = detector.getBBox()
243 for name
in getApCorrNameSet():
244 apCorrMap.set(name +
"_flux", makeRandomBoundedField())
245 apCorrMap.set(name +
"_fluxErr", makeRandomBoundedField())
249 """Build a random spacially-varying TransmissionCurve.""" 250 bbox = detector.getBBox()
251 return makeRandomTransmissionCurve(rng=self.
rng, maxRadius=max(bbox.getWidth(), bbox.getHeight()))
def buildTransmissionCurve(self, detector)
static Schema makeMinimalSchema()
def buildApCorrMap(self, detector)
def buildPsf(self, detector)
static DateTime now(void)
def run(self, butler, n, tractInfo, camera, catalog=None)
def makePointings(self, n, tractInfo)
def buildWcs(self, position, pa, detector)
std::shared_ptr< SkyWcs > makeSkyWcs(TransformPoint2ToPoint2 const &pixelsToFieldAngle, lsst::geom::Angle const &orientation, bool flipX, lsst::geom::SpherePoint const &boresight, std::string const &projection="TAN")
static CoordKey addFields(afw::table::Schema &schema, std::string const &name, std::string const &doc)