Coverage for python/lsst/pipe/tasks/fakes.py: 82%

18 statements  

« prev     ^ index     » next       coverage.py v6.4.4, created at 2022-08-20 02:29 -0700

1# 

2# LSST Data Management System 

3# Copyright 2008-2014 LSST Corporation. 

4# 

5# This product includes software developed by the 

6# LSST Project (http://www.lsst.org/). 

7# 

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

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

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

11# (at your option) any later version. 

12# 

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

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

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

16# GNU General Public License for more details. 

17# 

18# You should have received a copy of the LSST License Statement and 

19# the GNU General Public License along with this program. If not, 

20# see <http://www.lsstcorp.org/LegalNotices/>. 

21# 

22import abc 

23 

24from deprecated.sphinx import deprecated 

25 

26import lsst.pex.config 

27import lsst.pipe.base 

28import lsst.afw.image 

29 

30 

31class BaseFakeSourcesConfig(lsst.pex.config.Config): 

32 maskPlaneName = lsst.pex.config.Field( 

33 dtype=str, default="FAKE", 

34 doc="Mask plane to set on pixels affected by fakes. Will be added if not already present." 

35 ) 

36 

37 

38@deprecated(reason=("The fakes code has been deprecated in favor of ProcessCcdWithFakes. " 

39 "This will be removed after v24."), 

40 category=FutureWarning, 

41 version="24.0") 

42class BaseFakeSourcesTask(lsst.pipe.base.Task, metaclass=abc.ABCMeta): 

43 """An abstract base class for subtasks that inject fake sources into images to test completeness and 

44 other aspects of the processing. 

45 

46 This class simply adds a mask plane that subclasses should use to mark pixels that have been touched. 

47 

48 This is an abstract base class (abc) and is not intended to be directly used. To create a fake sources 

49 injector, create a child class and re-implement the required methods. 

50 """ 

51 

52 ConfigClass = BaseFakeSourcesConfig 

53 _DefaultName = "baseFakeSources" 

54 

55 def __init__(self, **kwargs): 

56 """Initialize the Task. 

57 

58 Subclasses that define their own __init__ should simply forward all arguments to the base 

59 class constructor. They can then assume self.config is an instance of their ConfigClass. 

60 

61 If an external catalog is used to add sources consistently to multiple overlapping images, 

62 that catalog should generally be loaded and attached to self here, so it can be used 

63 multiple times by the run() method. 

64 """ 

65 lsst.pipe.base.Task.__init__(self, **kwargs) 

66 lsst.afw.image.Mask[lsst.afw.image.MaskPixel].addMaskPlane(self.config.maskPlaneName) 

67 self.bitmask = lsst.afw.image.Mask[lsst.afw.image.MaskPixel]\ 

68 .getPlaneBitMask(self.config.maskPlaneName) 

69 

70 @abc.abstractmethod 

71 def run(self, exposure, background): 

72 """Add fake sources to the given Exposure, making use of the given BackgroundList if desired. 

73 

74 If pixels in the Exposure are replaced, not added to, extra care should be taken with the background, 

75 mask, and variance planes. The Exposure as given is background-subtracted (using the supplied 

76 background model) and should be returned in the same state. 

77 """ 

78 raise NotImplementedError("FakeSourcesTask is abstract, create a child class to use this method")