Coverage for python/lsst/geom/_coordinates.py: 86%

40 statements  

« prev     ^ index     » next       coverage.py v6.4.1, created at 2022-07-11 05:56 +0000

1# 

2# Developed for the LSST Data Management System. 

3# This product includes software developed by the LSST Project 

4# (https://www.lsst.org). 

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

6# for details of code ownership. 

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 GNU General Public License 

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

20# 

21 

22__all__ = ["CoordinateExpr", "Extent", "ExtentI", "ExtentD", 

23 "Point", "PointI", "PointD"] 

24 

25from lsst.utils import TemplateMeta 

26 

27from . import _geom 

28 

29 

30def _coordinateStr(self): 

31 return "({})".format(", ".join("%0.5g" % v for v in self)) 

32 

33 

34def _coordinateRepr(self): 

35 return "{}({})".format(type(self).__name__, 

36 ", ".join("%0.10g" % v for v in self)) 

37 

38 

39def _coordinateReduce(self): 

40 return (type(self), tuple(self)) 

41 

42 

43class CoordinateExpr(metaclass=TemplateMeta): 

44 """Abstract base class and factory for CoordinateExpr objects. 

45 """ 

46 TEMPLATE_PARAMS = ("dimensions", ) 

47 

48 __str__ = _coordinateStr 

49 __repr__ = _coordinateRepr 

50 __reduce__ = _coordinateReduce 

51 

52 

53CoordinateExpr.register(2, _geom.CoordinateExpr2) 

54CoordinateExpr.register(3, _geom.CoordinateExpr3) 

55 

56 

57class Extent(metaclass=TemplateMeta): 

58 """Abstract base class and factory for Extent objects. 

59 """ 

60 TEMPLATE_PARAMS = ("dtype", "dimensions") 

61 TEMPLATE_DEFAULTS = (None, 2) 

62 

63 __str__ = _coordinateStr 

64 __repr__ = _coordinateRepr 

65 __reduce__ = _coordinateReduce 

66 

67 

68Extent.register((int, 2), _geom.Extent2I) 

69Extent.register((float, 2), _geom.Extent2D) 

70Extent.register((int, 3), _geom.Extent3I) 

71Extent.register((float, 3), _geom.Extent3D) 

72ExtentI = _geom.Extent2I 

73ExtentD = _geom.Extent2D 

74 

75 

76class Point(metaclass=TemplateMeta): 

77 """Abstract base class and factory for Point objects. 

78 """ 

79 TEMPLATE_PARAMS = ("dtype", "dimensions") 

80 TEMPLATE_DEFAULTS = (None, 2) 

81 

82 __str__ = _coordinateStr 

83 __repr__ = _coordinateRepr 

84 __reduce__ = _coordinateReduce 

85 

86 

87Point.register((int, 2), _geom.Point2I) 

88Point.register((float, 2), _geom.Point2D) 

89Point.register((int, 3), _geom.Point3I) 

90Point.register((float, 3), _geom.Point3D) 

91PointI = _geom.Point2I 

92PointD = _geom.Point2D