1 from lsst.pipe.base import ArgumentParser, ButlerInitializedTaskRunner, ConfigDatasetType
3 from lsst.pex.config
import Config, Field, ConfigurableField, ListField
8 processCcd = ConfigurableField(
9 target=ProcessCcdTask, doc=
"CCD processing task")
10 ignoreCcdList = ListField(dtype=int, default=[],
11 doc=
"List of CCDs to ignore when processing")
12 ccdKey = Field(dtype=str, default=
"ccd",
13 doc=
"DataId key corresponding to a single sensor")
17 """Run batches, and initialize Task using a butler"""
21 class SingleFrameDriverTask(BatchParallelTask):
22 """Process CCDs in parallel
24 ConfigClass = SingleFrameDriverConfig
25 _DefaultName =
"singleFrameDriver"
26 RunnerClass = SingleFrameTaskRunner
28 def __init__(self, butler=None, psfRefObjLoader=None, astromRefObjLoader=None, photoRefObjLoader=None,
33 The psfRefObjLoader, astromRefObjLoader, photoRefObjLoader should
34 be an instance of LoadReferenceObjectsTasks that supplies an external
35 reference catalog. They may be None if the butler argument is
36 provided or the particular reference catalog is not required.
38 @param[in] butler The butler is passed to the refObjLoader constructor in case it is
39 needed. Ignored if the refObjLoader argument provides a loader directly.
40 @param[in] psfRefObjLoader Reference catalog loader for PSF determination.
41 @param[in] astromRefObjLoader Reference catalog loader for astrometric calibration.
42 @param[in] photoRefObjLoader Reference catalog loader for photometric calibration.
43 @param[in,out] kwargs other keyword arguments for lsst.ctrl.pool.BatchParallelTask
45 BatchParallelTask.__init__(self, *args, **kwargs)
46 self.
ignoreCcds = set(self.config.ignoreCcdList)
47 self.makeSubtask(
"processCcd", butler=butler, psfRefObjLoader=psfRefObjLoader,
48 astromRefObjLoader=astromRefObjLoader, photoRefObjLoader=photoRefObjLoader)
51 def _makeArgumentParser(cls, *args, **kwargs):
52 kwargs.pop(
"doBatch",
False)
53 parser = ArgumentParser(name=
"singleFrameDriver", *args, **kwargs)
54 parser.add_id_argument(
"--id",
55 datasetType=ConfigDatasetType(
56 name=
"processCcd.isr.datasetType"),
58 help=
"data ID, e.g. --id visit=12345 ccd=67")
62 """Process a single CCD, with scatter-gather-scatter using MPI.
64 if sensorRef.dataId[self.config.ccdKey]
in self.
ignoreCcds:
65 self.log.warn(
"Ignoring %s: CCD in ignoreCcdList" %
69 with self.
logOperation(
"processing %s" % (sensorRef.dataId,)):