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

41 statements  

« prev     ^ index     » next       coverage.py v7.2.1, created at 2023-03-13 02:37 +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 

25import math 

26 

27from lsst.utils import TemplateMeta 

28 

29from . import _geom 

30 

31 

32def _coordinateStr(self): 

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

34 

35 

36def _coordinateRepr(self): 

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

38 ", ".join(f"{v:0.17g}" if math.isfinite(v) else f"float('{v}')" for v in self)) 

39 

40 

41def _coordinateReduce(self): 

42 return (type(self), tuple(self)) 

43 

44 

45class CoordinateExpr(metaclass=TemplateMeta): 

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

47 """ 

48 TEMPLATE_PARAMS = ("dimensions", ) 

49 

50 __str__ = _coordinateStr 

51 __repr__ = _coordinateRepr 

52 __reduce__ = _coordinateReduce 

53 

54 

55CoordinateExpr.register(2, _geom.CoordinateExpr2) 

56CoordinateExpr.register(3, _geom.CoordinateExpr3) 

57 

58 

59class Extent(metaclass=TemplateMeta): 

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

61 """ 

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

63 TEMPLATE_DEFAULTS = (None, 2) 

64 

65 __str__ = _coordinateStr 

66 __repr__ = _coordinateRepr 

67 __reduce__ = _coordinateReduce 

68 

69 

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

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

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

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

74ExtentI = _geom.Extent2I 

75ExtentD = _geom.Extent2D 

76 

77 

78class Point(metaclass=TemplateMeta): 

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

80 """ 

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

82 TEMPLATE_DEFAULTS = (None, 2) 

83 

84 __str__ = _coordinateStr 

85 __repr__ = _coordinateRepr 

86 __reduce__ = _coordinateReduce 

87 

88 

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

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

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

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

93PointI = _geom.Point2I 

94PointD = _geom.Point2D