Coverage for klayout_pex/rcx25/c/geometry_restorer.py: 92%

1#! /usr/bin/env python3 

2# 

3# -------------------------------------------------------------------------------- 

4# SPDX-FileCopyrightText: 2024 Martin Jan Köhler and Harald Pretl 

5# Johannes Kepler University, Institute for Integrated Circuits. 

6# 

7# This file is part of KPEX 

8# (see https://github.com/martinjankoehler/klayout-pex). 

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

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

22# SPDX-License-Identifier: GPL-3.0-or-later 

23# -------------------------------------------------------------------------------- 

24# 

25 

26import klayout.db as kdb 

27 

28from klayout_pex.rcx25.types import EdgeInterval 

29 

30 

31class GeometryRestorer: 

32 def __init__(self, transformation: kdb.IMatrix3d): 

33 self.transformation = transformation 

34 

35 def restore_edge_interval(self, edge_interval: EdgeInterval) -> kdb.Edge: 

36 return self.transformation * kdb.Edge(kdb.Point(edge_interval[0], 0), 

37 kdb.Point(edge_interval[1], 0)) 

38 

39 def restore_edge(self, edge: kdb.Edge) -> kdb.Edge: 

40 return self.transformation * edge 

41 

42 def restore_polygon(self, polygon: kdb.Polygon) -> kdb.Polygon: 

43 return self.transformation * polygon 

44 

45 def restore_region(self, region: kdb.Region) -> kdb.Region: 

46 return region.transformed(self.transformation)