123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- # Created: 2019-01-04
- # Copyright (c) 2019 Manfred Moitzi
- # License: MIT License
- from typing import Union, Iterable, List, TYPE_CHECKING
- import math
- from .vector import Vec2
- from .construct2d import ConstructionTool
- from .bbox import BoundingBox2d
- if TYPE_CHECKING:
- from ezdxf.eztypes import Vertex
- class Shape2d(ConstructionTool):
- """
- 2d geometry object as vertices list which can be moved, rotated and scaled.
- """
- def __init__(self, vertices: Iterable['Vertex'] = None):
- self.vertices = [] if vertices is None else Vec2.list(vertices) # type: List[Vec2]
- def move(self, dx: float, dy: float) -> None:
- self.translate(Vec2((dx, dy)))
- def bounding_box(self)->BoundingBox2d:
- return BoundingBox2d(self.vertices)
- def translate(self, vector: 'Vertex') -> None:
- delta = Vec2(vector)
- self.vertices = [v + delta for v in self.vertices]
- def scale(self, sx: float = 1., sy: float = 1.) -> None:
- self.vertices = [Vec2((v.x * sx, v.y * sy)) for v in self.vertices]
- def scale_uniform(self, scale: float) -> None:
- self.vertices = [v * scale for v in self.vertices]
- def rotate(self, angle: float, center: 'Vertex' = None) -> None:
- self.rotate_rad(math.radians(angle), center)
- def rotate_rad(self, angle: float, center: 'Vertex' = None) -> None:
- if center is not None:
- center = Vec2(center)
- self.translate(-center) # faster than a Matrix44 multiplication
- self.vertices = [v.rotate(angle) for v in self.vertices]
- if center is not None:
- self.translate(center) # faster than a Matrix44 multiplication
- # Sequence interface
- def __len__(self) -> int:
- return len(self.vertices)
- def __getitem__(self, item: Union[int, slice]) -> Vec2:
- return self.vertices[item]
- # limited List interface
- def append(self, vertex: 'Vertex') -> None:
- self.vertices.append(Vec2(vertex))
- def extend(self, vertices: Iterable) -> None:
- self.vertices.extend(Vec2.generate(vertices))
|