Hide keyboard shortcuts

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# 

22 

23__all__ = ["makeImageFromArray", "makeMaskFromArray", "makeMaskedImageFromArrays"] 

24 

25from .image import Image, Mask 

26from .maskedImage import MaskedImage 

27 

28 

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) 

36 

37 

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) 

45 

46 

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)