jTdZddlmZddlmZddlmZddlmZGddZ d S) zm Schemdraw transformations for converting local element definition to global position within the drawing ) annotations)Sequence)Point)XYcHeZdZdZdedfdd Zd ZddZddZdS) Transforma Class defining transformation matrix Args: theta: Rotation angle in degrees globalshift: X-Y shift (applied after zoom and rotation) localshift: Local X-Y shift (applied before zoom and rotation) zoom: Zoom factor )rr)rrthetafloat globalshiftr localshiftzoom XY | floatc||_t||_t||_t |t t frt||f}||_dS)N)r rshiftr isinstanceintr r)selfr r r rs R/data/share/cad/hardware/.venv/lib/python3.11/site-packages/schemdraw/transform.py__init__zTransform.__init__sY ;''  ++ dS%L ) ) '$&&D cFd|jd|jd|jd|jS)NzTransform: xy=z; theta=z; scale=z ; lshift=)rr rr )rs r__repr__zTransform.__repr__s2m mmDJmm mm\`\kmmmrptreturnrct||jz|jz|j|jzS)z Apply the transform to the point Args: pt: Original (x, y) coordinates Returns: Transformed (x, y) coordinates )rr rrotater r)rrs r transformzTransform.transform!s7rT_, 9AA$*MMPTPZZZrpts Sequence[XY] list[Point]c fd|DS)z Apply the transform to multiple points Args: pts: List of (x,y) points to transform Returns: List of transformed (x, y) points c:g|]}|S)r).0rrs r z-Transform.transform_array..5s%111rr""111rr$)rrs` rtransform_arrayzTransform.transform_array,s2111S1111rN)r r r rr rrr)rrrr)rr rr!) __name__ __module__ __qualname____doc__rrrrr'r$rrr r s#)UU6]]nnn [ [ [ [ 2 2 2 2 2 2rr N) r+ __future__rtypingrutilrtypesrr r$rrr0s#""""")2)2)2)2)2)2)2)2)2)2r