LSSTApplications  8.0.0.0+107,8.0.0.1+13,9.1+18,9.2,master-g084aeec0a4,master-g0aced2eed8+6,master-g15627eb03c,master-g28afc54ef9,master-g3391ba5ea0,master-g3d0fb8ae5f,master-g4432ae2e89+36,master-g5c3c32f3ec+17,master-g60f1e072bb+1,master-g6a3ac32d1b,master-g76a88a4307+1,master-g7bce1f4e06+57,master-g8ff4092549+31,master-g98e65bf68e,master-ga6b77976b1+53,master-gae20e2b580+3,master-gb584cd3397+53,master-gc5448b162b+1,master-gc54cf9771d,master-gc69578ece6+1,master-gcbf758c456+22,master-gcec1da163f+63,master-gcf15f11bcc,master-gd167108223,master-gf44c96c709
LSSTDataManagementBasePackage
estimateBackground.py
1 #!/usr/bin/env python
2 
3 #
4 # LSST Data Management System
5 # Copyright 2008, 2009, 2010 LSST Corporation.
6 #
7 # This product includes software developed by the
8 # LSST Project (http://www.lsst.org/).
9 #
10 # This program is free software: you can redistribute it and/or modify
11 # it under the terms of the GNU General Public License as published by
12 # the Free Software Foundation, either version 3 of the License, or
13 # (at your option) any later version.
14 #
15 # This program is distributed in the hope that it will be useful,
16 # but WITHOUT ANY WARRANTY; without even the implied warranty of
17 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 # GNU General Public License for more details.
19 #
20 # You should have received a copy of the LSST License Statement and
21 # the GNU General Public License along with this program. If not,
22 # see <http://www.lsstcorp.org/LegalNotices/>.
23 #
24 
25 import os
26 import eups
27 import lsst.afw.image as afwImage
28 import lsst.afw.math as afwMath
29 import lsst.afw.display.ds9 as ds9
30 
31 try:
32  display
33 except NameError:
34  display = not False
35 
36 ################################################
37 
38 def getImage():
39  imagePath = os.path.join(eups.productDir("afwdata"),
40  "DC3a-Sim", "sci", "v5-e0", "v5-e0-c011-a00.sci_img.fits")
41  return afwImage.MaskedImageF(imagePath)
42 
43 def simpleBackground(image):
44  binsize = 128
45  nx = int(image.getWidth()/binsize) + 1
46  ny = int(image.getHeight()/binsize) + 1
47  bctrl = afwMath.BackgroundControl(nx, ny)
48 
49  bkgd = afwMath.makeBackground(image, bctrl)
50 
51  statsImage = afwMath.cast_BackgroundMI(bkgd).getStatsImage()
52 
53  image -= bkgd.getImageF(afwMath.Interpolate.NATURAL_SPLINE)
54 
55  return bkgd
56 
57 def complexBackground(image):
58  binsize = 128
59  nx = int(image.getWidth()/binsize) + 1
60  ny = int(image.getHeight()/binsize) + 1
61 
62  sctrl = afwMath.StatisticsControl()
63  sctrl.setNumSigmaClip(3)
64  sctrl.setNumIter(4)
65  sctrl.setAndMask(afwImage.MaskU.getPlaneBitMask(["INTRP", "EDGE"]))
66  sctrl.setNoGoodPixelsMask(afwImage.MaskU.getPlaneBitMask("BAD"))
67  sctrl.setNanSafe(True)
68  if False:
69  sctrl.setWeighted(True)
70  sctrl.setCalcErrorFromInputVariance(True)
71 
72  bctrl = afwMath.BackgroundControl(nx, ny, sctrl, afwMath.MEANCLIP)
73 
74  bkgd = afwMath.makeBackground(image, bctrl)
75 
76  statsImage = afwMath.cast_BackgroundMI(bkgd).getStatsImage()
77  ds9.mtv(statsImage.getVariance())
78 
79  bkdgImages = dict(SPLINE = bkgd.getImageF(afwMath.Interpolate.NATURAL_SPLINE),
80  LINEAR = bkgd.getImageF(afwMath.Interpolate.LINEAR))
81 
82  return bkgd
83 
84 def main():
85  image = getImage()
86 
87  if display:
88  ds9.mtv(image, frame=0)
89 
90  bkgd = simpleBackground(image)
91  image = getImage()
92  bkgd = complexBackground(image)
93 
94  if display:
95  ds9.mtv(image, frame=1)
96  ds9.mtv(afwMath.cast_BackgroundMI(bkgd).getStatsImage(), frame=2)
97 
98  order = 2
99  actrl = afwMath.ApproximateControl(afwMath.ApproximateControl.CHEBYSHEV, order, order)
100  approx = bkgd.getApproximate(actrl)
101 
102  approx.getImage()
103  approx.getMaskedImage()
104  approx.getImage(order - 1)
105 
106 #################################################
107 if __name__ == '__main__':
108  main()
Generated on Mon Dec 8 2014 18:36:40 for LSSTApplications by   doxygen 1.8.5