Coverage for python/lsst/afw/image/basicUtils.py : 35%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1#
2# LSST Data Management System
3# Copyright 2008-2017 LSST/AURA.
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#
23__all__ = ["makeImageFromArray", "makeMaskFromArray", "makeMaskedImageFromArrays"]
25from .image import Image, Mask
26from .maskedImage import MaskedImage
29def makeImageFromArray(array):
30 """Construct an Image from a NumPy array, inferring the Image type from
31 the NumPy type. Return None if input is None.
32 """
33 if array is None:
34 return None
35 return Image(array, dtype=array.dtype.type)
38def makeMaskFromArray(array):
39 """Construct an Mask from a NumPy array, inferring the Mask type from the
40 NumPy type. Return None if input is None.
41 """
42 if array is None:
43 return None
44 return Mask(array, dtype=array.dtype.type)
47def makeMaskedImageFromArrays(image, mask=None, variance=None):
48 """Construct a MaskedImage from three NumPy arrays, inferring the
49 MaskedImage types from the NumPy types.
50 """
51 return MaskedImage(makeImageFromArray(image), makeMaskFromArray(mask),
52 makeImageFromArray(variance), dtype=image.dtype.type)