cameraFactory.py

Go to the documentation of this file.

# This file is part of afw.
#
# Developed for the LSST Data Management System.
# This product includes software developed by the LSST Project
# (https://www.lsst.org).
# See the COPYRIGHT file at the top-level directory of this distribution
# for details of code ownership.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
__all__ = ["addDetectorBuilderFromConfig",
           "makeCameraFromPath", "makeCameraFromAmpLists"]
 
import os.path
from lsst.afw.table import BaseCatalog
from .cameraGeomLib import FOCAL_PLANE, FIELD_ANGLE, PIXELS, TAN_PIXELS, ACTUAL_PIXELS, CameraSys, \
    Amplifier
from .camera import Camera
from .makePixelToTanPixel import makePixelToTanPixel
from .pupil import PupilFactory
 
cameraSysList = [FIELD_ANGLE, FOCAL_PLANE, PIXELS, TAN_PIXELS, ACTUAL_PIXELS]
cameraSysMap = dict((sys.getSysName(), sys) for sys in cameraSysList)
 
 
def addDetectorBuilderFromConfig(cameraBuilder, detectorConfig, amplifiers, focalPlaneToField):
    """Build a dictionary of Detector constructor keyword arguments.
 
    The returned dictionary can be passed as keyword arguments to the Detector
    constructor, providing all required arguments.  However, using these
    arguments directly constructs a Detector with knowledge of only the
    coordinate systems that are *directly* mapped to its own PIXELS coordinate
    system.  To construct Detectors with a shared TransformMap for the full
    Camera, use makeCameraFromCatalogs or makeCameraFromPath instead of
    calling this function or makeDetector directly.
 
    Parameters
    ----------
    cameraBuilder : `lsst.afw.cameraGeonm.Camera.Builder`
        Camera builder object to which the new Detector Builder
        should be added.
    detectorConfig : `lsst.pex.config.Config`
        Configuration for this detector.
    amplifiers : `list` [`~lsst.afw.cameraGeom.Amplifier`]
        amplifier information for this detector
    focalPlaneToField : `lsst.afw.geom.TransformPoint2ToPoint2`
        FOCAL_PLANE to FIELD_ANGLE Transform
 
    Returns
    -------
    detectorBuilder : `lsst.afw.cameraGeom.Detector.InCameraBuilder`
        A builder object for a detector corresponding to the given config,
        associated with the given camera builder object.
    """
    detectorBuilder = cameraBuilder.add(detectorConfig.name, detectorConfig.id)
    detectorBuilder.fromConfig(detectorConfig)
 
    for ampBuilder in amplifiers:
        detectorBuilder.append(ampBuilder)
 
    transforms = makeTransformDict(detectorConfig.transformDict.transforms)
 
    # It seems the C++ code has always assumed that the "nativeSys" for
    # detectors is PIXELS, despite the configs here giving the illusion of
    # choice.  We'll use PIXELS if the config value is None, and assert that
    # the value is PIXELS otherwise.  Note that we can't actually get rid of
    # the nativeSys config option without breaking lots of on-disk camera
    # configs.
    detectorNativeSysPrefix = cameraSysMap.get(detectorConfig.transformDict.nativeSys, PIXELS)
    assert detectorNativeSysPrefix == PIXELS, "Detectors with nativeSys != PIXELS are not supported."
 
    for toSys, transform in transforms.items():
        detectorBuilder.setTransformFromPixelsTo(toSys, transform)
    tanPixSys = CameraSys(TAN_PIXELS, detectorConfig.name)
    transforms[tanPixSys] = makePixelToTanPixel(
        bbox=detectorBuilder.getBBox(),
        orientation=detectorBuilder.getOrientation(),
        focalPlaneToField=focalPlaneToField,
        pixelSizeMm=detectorBuilder.getPixelSize(),
    )
 
    for toSys, transform in transforms.items():
        detectorBuilder.setTransformFromPixelsTo(toSys, transform)
 
    detectorBuilder.setCrosstalk(detectorConfig.getCrosstalk(len(amplifiers)))
 
    return detectorBuilder
 
 
def makeTransformDict(transformConfigDict):
    """Make a dictionary of CameraSys: lsst.afw.geom.Transform from a config dict.
 
    Parameters
    ----------
    transformConfigDict : value obtained from a `lsst.pex.config.ConfigDictField`
        registry; keys are camera system names.
 
    Returns
    -------
    transforms : `dict` [`CameraSys` or `CameraSysPrefix`, `lsst.afw.geom.Transform`]
        A dict of CameraSys or CameraSysPrefix: lsst.afw.geom.Transform
    """
    resMap = dict()
    if transformConfigDict is not None:
        for key in transformConfigDict:
            transform = transformConfigDict[key].transform.apply()
            resMap[CameraSys(key)] = transform
    return resMap
 
 
def makeCameraFromPath(cameraConfig, ampInfoPath, shortNameFunc,
                       pupilFactoryClass=PupilFactory):
    """Make a Camera instance from a directory of ampInfo files
 
    The directory must contain one ampInfo fits file for each detector in cameraConfig.detectorList.
    The name of each ampInfo file must be shortNameFunc(fullDetectorName) + ".fits".
 
    Parameters
    ----------
    cameraConfig : `CameraConfig`
        Config describing camera and its detectors.
    ampInfoPath : `str`
        Path to ampInfo data files.
    shortNameFunc : callable
        A function that converts a long detector name to a short one.
    pupilFactoryClass : `type`, optional
        Class to attach to camera; default is `lsst.afw.cameraGeom.PupilFactory`.
 
    Returns
    -------
    camera : `lsst.afw.cameraGeom.Camera`
        New Camera instance.
    """
    ampListDict = dict()
    for detectorConfig in cameraConfig.detectorList.values():
        shortName = shortNameFunc(detectorConfig.name)
        ampCatPath = os.path.join(ampInfoPath, shortName + ".fits")
        catalog = BaseCatalog.readFits(ampCatPath)
        ampListDict[detectorConfig.name] = [Amplifier.Builder.fromRecord(record)
                                            for record in catalog]
 
    return makeCameraFromAmpLists(cameraConfig, ampListDict, pupilFactoryClass)
 
 
def makeCameraFromAmpLists(cameraConfig, ampListDict,
                           pupilFactoryClass=PupilFactory):
    """Construct a Camera instance from a dictionary of detector name: list of
    Amplifier.Builder
 
    Parameters
    ----------
    cameraConfig : `CameraConfig`
        Config describing camera and its detectors.
    ampListDict : `dict` [`str`, `list` [`Amplifier.Builder`]]
        A dictionary of detector name: list of Amplifier.Builder
    pupilFactoryClass : `type`, optional
        Class to attach to camera; `lsst.default afw.cameraGeom.PupilFactory`.
 
    Returns
    -------
    camera : `lsst.afw.cameraGeom.Camera`
        New Camera instance.
    """
    nativeSys = cameraSysMap[cameraConfig.transformDict.nativeSys]
 
    # nativeSys=FOCAL_PLANE seems is baked into the camera class definition,
    # despite CameraConfig providing the illusion that it's configurable. Note
    # that we can't actually get rid of the nativeSys config option without
    # breaking lots of on-disk camera configs.
    assert nativeSys == FOCAL_PLANE, "Cameras with nativeSys != FOCAL_PLANE are not supported."
 
    cameraBuilder = Camera.Builder(cameraConfig.name)
    cameraBuilder.setPupilFactoryClass(pupilFactoryClass)
 
    transformDict = makeTransformDict(cameraConfig.transformDict.transforms)
    focalPlaneToField = transformDict[FIELD_ANGLE]
 
    for toSys, transform in transformDict.items():
        cameraBuilder.setTransformFromFocalPlaneTo(toSys, transform)
 
    for detectorConfig in cameraConfig.detectorList.values():
        addDetectorBuilderFromConfig(
            cameraBuilder,
            detectorConfig=detectorConfig,
            amplifiers=ampListDict[detectorConfig.name],
            focalPlaneToField=focalPlaneToField,
        )
 
    return cameraBuilder.finish()