hBddlmZddlmZmZmZmZmZddlm Z m Z m Z m Z m Z mZddlmZerddlmZgdZeddd f dd Zd ZGd d ZGddZGddeZy)) annotations) TYPE_CHECKINGSequenceIterableOptionalIterator)Vec3UVecX_AXISY_AXISZ_AXISMatrix44)RGB) BaseLayout)OCSUCS PassTroughUCScXt||D]\}}|j||d|iy)Ncolor) dxfattribs)zipadd_line)layoutstartpointscolorspointrs .lib/python3.12/site-packages/ezdxf/math/ucs.py render_axisr"s0 FF+ uu'51AB,g?ceZdZdZefddZeddZeddZeddZ ddZ ddZ ddZ dd Z d ed d d f dd Zy)rzkEstablish an :ref:`OCS` for a given extrusion vector. Args: extrusion: extrusion vector. ct|j}|jt |_|jrt |j tkr2t |jtkrtj|}ntj|}|j}|j|j}tj||||_ yyN)r normalizeiscloser transformabsx _1_OVER_64yr crossrucsmatrix)self extrusionAzAxAys r!__init__z OCS.__init__#s )_ & & (ZZ// >>BDD J&SY-C\\"%\\"%B"'')B",,r2r2DK r#cR|jr|jjStSzx-axis unit vector)r)r0uxr r1s r!r9zOCS.ux/"&t{{~~;V;r#cR|jr|jjStSzy-axis unit vector)r)r0uyr r:s r!r>zOCS.uy4r;r#cR|jr|jjStSzz-axis unit vector)r)r0uzr r:s r!rAzOCS.uz9r;r#cjt|}|jr|jj|S|S)z#Returns OCS vector for WCS `point`.)r r)r0 ocs_from_wcs)r1r p3s r!from_wcsz OCS.from_wcs>s, %[ >>;;++B/ /Ir#c#Ktj|}|jr(|jj}|D] }||y|Ed{y7w)z2Returns iterable of OCS vectors from WCS `points`.N)r generater)r0rC)r1r_pointsrEr s r!points_from_wcszOCS.points_from_wcsFsI--' >>{{//H uo%!  AAAAcf|jr|jj|St|S)z#Returns WCS vector for OCS `point`.)r)r0 ocs_to_wcsr r1r s r!to_wcsz OCS.to_wcsPs( >>;;))%0 0; r#c#Ktj|}|jr(|jj}|D] }||y|Ed{y7w)z1Returns iterable of WCS vectors for OCS `points`.N)r rGr)r0rL)r1rrHrNr s r! points_to_wcszOCS.points_to_wcsWsI--' >>[[++F Um#!  rJrrrc t|d|jt|z|jt|z|jt|zf|y)(Render axis as 3D lines into a `layout`.rrrrrrN)r"rNr r r r1rlengthrs r!r"zOCS.render_axisasL   FVO, FVO, FVO,  r#N)r2r returnr )r r rXr )rzIterable[UVec]rXIterator[Vec3])rrrVfloatrrrXNone)__name__ __module__ __qualname____doc__r r6propertyr9r>rArErIrNrPrr"r#r!rrs*0 3<<<<<<!Ql        r#rc<eZdZdZ d' d(dZed)dZed)dZed)dZed)dZ e jd*dZ d+d Z d,d Z d-d Z d.d Zd,d Zd-dZd.dZd,dZd-dZd/dZd/dZd0dZd1dZd2dZd2dZd2dZd3dZd4dZd5dZed6dZe d7dZ e d7dZ!e d7dZ"e d7d Z#e d7d!Z$e d7d"Z%d#e&d#d$d%f d8d&Z'y)9raEstablish a user coordinate system (:ref:`UCS`). The UCS is defined by the origin and two unit vectors for the x-, y- or z-axis, all axis in :ref:`WCS`. The missing axis is the cross product of the given axis. If x- and y-axis are ``None``: ux = ``(1, 0, 0)``, uy = ``(0, 1, 0)``, uz = ``(0, 0, 1)``. Unit vectors don't have to be normalized, normalization is done at initialization, this is also the reason why scaling gets lost by copying or rotating. Args: origin: defines the UCS origin in world coordinates ux: defines the UCS x-axis as vector in :ref:`WCS` uy: defines the UCS y-axis as vector in :ref:`WCS` uz: defines the UCS z-axis as vector in :ref:`WCS` NcF||t}t}t}nc|\t|j }t|j }t|j |j }n|[t|j }t|j }t|j |j }n|[t|j }t|j }t|j |j }nKt|j }t|j }t|j }t j|||t||_yr&) r r r r r'r.rr/r0)r1originr9r>rA_ux_uy_uzs r!r6z UCS.__init__s8 :"*CCC Zr($$&Cr($$&Cr(..$..0C Zr($$&Cr($$&Cr(..$..0C Zr($$&Cr($$&Cr(..$..0Cr($$&Cr($$&Cr($$&C ( S#sDL I r#c.|jjSr8)r0r9r:s r!r9zUCS.ux{{~~r#c.|jjSr=)r0r>r:s r!r>zUCS.uyrir#c.|jjSr@)r0rAr:s r!rAzUCS.uzrir#c.|jjS)zReturns the originr0rdr:s r!rdz UCS.origins{{!!!r#c&||j_y)z Set origin.Nrm)r1vs r!rdz UCS.origins r#cnt|j|j|j|jS)zReturns a copy of this UCS.)rrdr9r>rAr:s r!copyzUCS.copys#4;;$''::r#c8|jj|S)z"Returns WCS point for UCS `point`.)r0r)rMs r!rNz UCS.to_wcss{{$$U++r#c8|jj|S)z1Returns iterable of WCS vectors for UCS `points`.)r0transform_verticesr1rs r!rPzUCS.points_to_wcss{{--f55r#c8|jj|S)zAReturns WCS direction for UCS `vector` without origin adjustment.)r0transform_directionr1vectors r!direction_to_wcszUCS.direction_to_wcss{{..v66r#c8|jj|S)z"Returns UCS point for WCS `point`.)r0ucs_vertex_from_wcsrMs r!rEz UCS.from_wcss{{..u55r#c#FK|j}|D] }||yw)z2Returns iterable of UCS vectors from WCS `points`.N)rE)r1rrEr s r!rIzUCS.points_from_wcss#==E5/ !s!c8|jj|S)z>Returns UCS vector for WCS `vector` without origin adjustment.)r0ucs_direction_from_wcsrxs r!direction_from_wcszUCS.direction_from_wcss{{11&99r#cl|j|}t|jj|S)zuReturns OCS vector for UCS `point`. The :class:`OCS` is defined by the z-axis of the :class:`UCS`. rNrrArE)r1r wpoints r!to_ocsz UCS.to_ocss+ U#477|$$V,,r#c#K|j}t|j}|D]}|j||yw)zReturns iterable of OCS vectors for UCS `points`. The :class:`OCS` is defined by the z-axis of the :class:`UCS`. Args: points: iterable of UCS vertices Nr)r1rwcsocsr s r! points_to_ocszUCS.points_to_ocss;kk$''lE,,s5z* *sAAc^|jtj|jS)aTransforms `angle` from current UCS to the parent coordinate system (most likely the WCS) including the transformation to the OCS established by the extrusion vector :attr:`UCS.uz`. Args: angle: in UCS in degrees )ucs_direction_to_ocs_directionr from_deg_angle angle_degr1angles r!to_ocs_angle_degzUCS.to_ocs_angle_degs*22    & ) r#c^|jtj|jS)aTransforms `angle` from current UCS to the parent coordinate system (most likely the WCS) including the transformation to the OCS established by the extrusion vector :attr:`UCS.uz`. Args: angle: in UCS in radians )rr from_anglerrs r!to_ocs_angle_radzUCS.to_ocs_angle_rads$224??53IJPPPr#cht|jj|j|S)zTransforms UCS `direction` vector into OCS direction vector of the parent coordinate system (most likely the WCS), target OCS is defined by the UCS z-axis. )rrArErz)r1 directions r!rz"UCS.ucs_direction_to_ocs_direction s( 477|$$T%:%:9%EFFr#ctjt||}|j|j|j |j g\}}}t|j|||S)a[Returns a new rotated UCS, with the same origin as the source UCS. The rotation vector is located in the origin and has :ref:`WCS` coordinates e.g. (0, 0, 1) is the WCS z-axis as rotation vector. Args: axis: arbitrary rotation axis as vector in :ref:`WCS` angle: rotation angle in radians rdr9r>rA) r axis_rotater rtr9r>rArrd)r1axisrtr9r>rAs r!rotatez UCS.rotatesX  dU 3))477DGGTWW*EF B$++";;r#ctj|j|}|j|j|j g\}}t |j|j||S)zReturns a new rotated UCS, rotation axis is the local x-axis. Args: angle: rotation angle in radians r)rrr9rtr>rArrd)r1rrr>rAs r!rotate_local_xzUCS.rotate_local_x!sT  % 0%%tww&89B$++$''bR@@r#ctj|j|}|j|j|j g\}}t |j||j|S)zReturns a new rotated UCS, rotation axis is the local y-axis. Args: angle: rotation angle in radians r)rrr>rtr9rArrd)r1rrr9rAs r!rotate_local_yzUCS.rotate_local_y,sT  % 0%%tww&89B$++"R@@r#ctj|j|}|j|j|j g\}}t |j|||jS)zReturns a new rotated UCS, rotation axis is the local z-axis. Args: angle: rotation angle in radians r)rrrArtr9r>rrd)r1rrr9r>s r!rotate_local_zzUCS.rotate_local_z7sT  % 0%%tww&89B$++"@@r#cB|xjt|z c_|S)ztShifts current UCS by `delta` vector and returns `self`. Args: delta: shifting vector )rdr )r1deltas r!shiftz UCS.shiftBs tE{"  r#c&t||_|S)zPlace current UCS at new origin `location` and returns `self`. Args: location: new origin in WCS )r rd)r1locations r!movetoz UCS.movetoLs8n  r#c0|xj|zc_|S)zGeneral inplace transformation interface, returns `self` (floating interface). Args: m: 4x4 transformation matrix (:class:`ezdxf.math.Matrix44`) )r0)r1ms r!r)z UCS.transformVs q  r#c.|jjS)z0Returns ``True`` if cartesian coordinate system.)r0 is_cartesianr:s r!rzUCS.is_cartesianas{{'''r#cnt|}|jt||z }t|||S)aReturns a new :class:`UCS` defined by the origin, the x-axis vector and an arbitrary point in the xy-plane. Args: origin: UCS origin as (x, y, z) tuple in :ref:`WCS` axis: x-axis vector as (x, y, z) tuple in :ref:`WCS` point: arbitrary point unlike the origin in the xy-plane as (x, y, z) tuple in :ref:`WCS` rdr9rAr r.r)rdrr x_axisz_axiss r!from_x_axis_and_point_in_xyzUCS.from_x_axis_and_point_in_xyf3dd5kF23&V77r#crt|}t||z }|j|}t|||S)aReturns a new :class:`UCS` defined by the origin, the x-axis vector and an arbitrary point in the xz-plane. Args: origin: UCS origin as (x, y, z) tuple in :ref:`WCS` axis: x-axis vector as (x, y, z) tuple in :ref:`WCS` point: arbitrary point unlike the origin in the xz-plane as (x, y, z) tuple in :ref:`WCS` rdr9r>r)rdrr r xz_vectory_axiss r!from_x_axis_and_point_in_xzzUCS.from_x_axis_and_point_in_xzx8dK&( (&V77r#crt|}t||z }|j|}t|||S)aReturns a new :class:`UCS` defined by the origin, the y-axis vector and an arbitrary point in the xy-plane. Args: origin: UCS origin as (x, y, z) tuple in :ref:`WCS` axis: y-axis vector as (x, y, z) tuple in :ref:`WCS` point: arbitrary point unlike the origin in the xy-plane as (x, y, z) tuple in :ref:`WCS` rdr>rAr)rdrr r xy_vectorrs r!from_y_axis_and_point_in_xyzUCS.from_y_axis_and_point_in_xyrr#crt|}t||z }|j|}t|||S)aReturns a new :class:`UCS` defined by the origin, the y-axis vector and an arbitrary point in the yz-plane. Args: origin: UCS origin as (x, y, z) tuple in :ref:`WCS` axis: y-axis vector as (x, y, z) tuple in :ref:`WCS` point: arbitrary point unlike the origin in the yz-plane as (x, y, z) tuple in :ref:`WCS` rr)rdrr r yz_vectorrs r!from_y_axis_and_point_in_yzzUCS.from_y_axis_and_point_in_yzrr#cnt|}|jt||z }t|||S)aReturns a new :class:`UCS` defined by the origin, the z-axis vector and an arbitrary point in the xz-plane. Args: origin: UCS origin as (x, y, z) tuple in :ref:`WCS` axis: z-axis vector as (x, y, z) tuple in :ref:`WCS` point: arbitrary point unlike the origin in the xz-plane as (x, y, z) tuple in :ref:`WCS` rr)rdrr rrs r!from_z_axis_and_point_in_xzzUCS.from_z_axis_and_point_in_xzrr#crt|}t||z }|j|}t|||S)aReturns a new :class:`UCS` defined by the origin, the z-axis vector and an arbitrary point in the yz-plane. Args: origin: UCS origin as (x, y, z) tuple in :ref:`WCS` axis: z-axis vector as (x, y, z) tuple in :ref:`WCS` point: arbitrary point unlike the origin in the yz-plane as (x, y, z) tuple in :ref:`WCS` rr)rdrr rrrs r!from_z_axis_and_point_in_yzzUCS.from_z_axis_and_point_in_yzrr#rrrc t||j|jt|z|jt|z|jt |zf|y)rRrTN)r"rdrNr r r rUs r!r"zUCS.render_axissP  ++ FVO, FVO, FVO,  r#)rSNNN)rdr r9Optional[UVec]r>rrArrW)ror rXr[)rXrr r rXr rzIterable[Vec3]rXrY)ryr rXr rrZrXrZ)rr rXr )rr rrZrXr)rrZrXr)rr rXr)rr rXr)rrrXr)rXbool)rdr rr r r rXr)rrrVrZrr)(r\r]r^r_r6r`r9r>rArdsetterrqrNrPrzrErIrrrrrrrrrrrrr)r staticmethodrrrrrrrr"rar#r!rrts*!!!! JJ J  J  J<"" ]];,676" :- +  QG < A A A ((88 8)-8 88"88 8)-8 88$88 8)-8 88$88 8)-8 88$88 8)-8 88"88 8)-8 88*!Ql      r#rcbeZdZdZfdZd dZd dZd dZd dZd dZ d dZ d d Z d d Z xZ S)rz5UCS is equal to the WCS and OCS (extrusion = 0, 0, 1)c"t|yr&)superr6)r1 __class__s r!r6zPassTroughUCS.__init__s r#c|Sr&rarMs r!rNzPassTroughUCS.to_wcs r#ct|Sr&iterrus r!rPzPassTroughUCS.points_to_wcs F|r#c|Sr&rarMs r!rzPassTroughUCS.to_ocsrr#ct|Sr&rrus r!rzPassTroughUCS.points_to_ocsrr#c|Sr&rars r!rzPassTroughUCS.to_ocs_angle_degrr#c|Sr&rars r!rzPassTroughUCS.to_ocs_angle_radrr#c|Sr&rarMs r!rEzPassTroughUCS.from_wcsrr#ct|Sr&rrus r!rIzPassTroughUCS.points_from_wcsrr#rrr)r\r]r^r_r6rNrPrrrrrErI __classcell__)rs@r!rrs3?r#rN) rrrr rzSequence[UVec]rrrXr[) __future__rtypingrrrrr ezdxf.mathr r r r r r ezdxf.colorsr ezdxf.layoutsr__all__r"r,rrrrar#r!rs#HHCC( )aA, C C C C  C  C V V rr r j Cr#