Coverage for python/lsst/obs/cfht/_instrument.py: 57%

42 statements  

« prev     ^ index     » next       coverage.py v7.5.1, created at 2024-05-08 03:53 -0700

1# This file is part of obs_cfht. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (http://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <http://www.gnu.org/licenses/>. 

21 

22"""Butler instrument description for the CFHT MegaCam camera. 

23""" 

24 

25__all__ = ("MegaPrime",) 

26 

27import os 

28from functools import lru_cache 

29 

30from astro_metadata_translator import MegaPrimeTranslator 

31from lsst.afw.cameraGeom import makeCameraFromPath, CameraConfig 

32from lsst.obs.base import Instrument, VisitSystem 

33from .cfhtFilters import MEGAPRIME_FILTER_DEFINITIONS 

34 

35from lsst.utils.introspection import get_full_type_name 

36from lsst.utils import getPackageDir 

37 

38 

39class MegaPrime(Instrument): 

40 filterDefinitions = MEGAPRIME_FILTER_DEFINITIONS 

41 policyName = "megacam" 

42 obsDataPackage = None 

43 translatorClass = MegaPrimeTranslator 

44 

45 def __init__(self, **kwargs): 

46 super().__init__(**kwargs) 

47 packageDir = getPackageDir("obs_cfht") 

48 self.configPaths = [os.path.join(packageDir, "config")] 

49 

50 @classmethod 

51 def getName(cls): 

52 return "MegaPrime" 

53 

54 def getCamera(self): 

55 path = os.path.join(getPackageDir("obs_cfht"), self.policyName, "camera") 

56 return self._getCameraFromPath(path) 

57 

58 @staticmethod 

59 @lru_cache() 

60 def _getCameraFromPath(path): 

61 """Return the camera geometry given solely the path to the location 

62 of that definition.""" 

63 config = CameraConfig() 

64 config.load(os.path.join(path, "camera.py")) 

65 return makeCameraFromPath( 

66 cameraConfig=config, 

67 ampInfoPath=path, 

68 shortNameFunc=lambda name: name.replace(" ", "_"), 

69 ) 

70 

71 def register(self, registry, update=False): 

72 camera = self.getCamera() 

73 obsMax = 2**31 

74 with registry.transaction(): 

75 registry.syncDimensionData( 

76 "instrument", 

77 { 

78 "name": self.getName(), "detector_max": 36, "visit_max": obsMax, "exposure_max": obsMax, 

79 "class_name": get_full_type_name(self), 

80 # Some schemas support default visit_system 

81 "visit_system": VisitSystem.ONE_TO_ONE.value, 

82 }, 

83 update=update 

84 ) 

85 

86 for detector in camera: 

87 registry.syncDimensionData( 

88 "detector", 

89 { 

90 "instrument": self.getName(), 

91 "id": detector.getId(), 

92 "full_name": detector.getName(), 

93 "name_in_raft": detector.getName(), 

94 "raft": None, # MegaPrime does not have rafts 

95 "purpose": str(detector.getType()).split(".")[-1], 

96 }, 

97 update=update 

98 ) 

99 

100 self._registerFilters(registry, update=update) 

101 

102 def getRawFormatter(self, dataId): 

103 # local import to prevent circular dependency 

104 from .rawFormatter import MegaPrimeRawFormatter 

105 return MegaPrimeRawFormatter