'j@ddlmZddlmZmZmZmZmZddlm Z m Z m Z m Z m Z mZddlmZerddlmZgdZeddd fddZdZGddZGddZGddeZdS)) annotations) TYPE_CHECKINGSequenceIterableOptionalIterator)Vec3UVecX_AXISY_AXISZ_AXISMatrix44)RGB) BaseLayout)OCSUCS PassTroughUCSlayoutrstartr pointsSequence[UVec]colorsrreturnNonecft||D]\}}|||d|i dS)Ncolor) dxfattribs)zipadd_line)rrrrpointrs H/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/math/ucs.py render_axisr%sO FF++CC uu'51ABBBBCCg?ceZdZdZefddZeddZeddZedd Z d d Z d!dZ d dZ d!dZ dedddfd"dZdS)#rzkEstablish an :ref:`OCS` for a given extrusion vector. Args: extrusion: extrusion vector. extrusionr c t|}|t |_|jrt |jtkr2t |jtkrtj |}ntj |}|}| |}tj ||||_ dSdSN)r normalizeiscloser transformabsx _1_OVER_64yr crossrucsmatrix)selfr(AzAxAys r$__init__z OCS.__init__#s )__ & & ( (ZZ/// > 3BD J&&SYY-C-C\"%%\"%%B"''))B",r2r22DKKK 3 3r&rr c6|jr |jjntSzx-axis unit vector)r-r4uxr r5s r$r<zOCS.ux/"&;t{~~V;r&c6|jr |jjntSzy-axis unit vector)r-r4uyr r=s r$rAzOCS.uy4r>r&c6|jr |jjntSzz-axis unit vector)r-r4uzr r=s r$rDzOCS.uz9r>r&r#cft|}|jr|j|S|S)z#Returns OCS vector for WCS `point`.)r r-r4 ocs_from_wcs)r5r#p3s r$from_wcsz OCS.from_wcs>s2 %[[ > ;++B// /Ir&rIterable[UVec]Iterator[Vec3]c#Ktj|}|jr |jj}|D]}||VdS|Ed{VdS)z2Returns iterable of OCS vectors from WCS `points`.N)r generater-r4rF)r5r_pointsrHr#s r$points_from_wcszOCS.points_from_wcsFsy-'' > {/H  & &huoo%%%% & &         r&cb|jr|j|St|S)z#Returns WCS vector for OCS `point`.)r-r4 ocs_to_wcsr r5r#s r$to_wcsz OCS.to_wcsPs. > ;))%00 0;; r&c#Ktj|}|jr |jj}|D]}||VdS|Ed{VdS)z1Returns iterable of WCS vectors for OCS `points`.N)r rLr-r4rP)r5rrMrRr#s r$ points_to_wcszOCS.points_to_wcsWsy-'' > [+F  $ $fUmm#### $ $         r&rrrrrlengthfloatrrrc t|d|t|z|t|z|t|zf|dS)(Render axis as 3D lines into a `layout`.rrrrrrN)r%rRr r r r5rrUrs r$r%zOCS.render_axisasn   FVO,, FVO,, FVO,,  r&N)r(r rr )r#r rr )rrIrrJ)rrrUrVrrrr)__name__ __module__ __qualname____doc__r r9propertyr<rArDrHrNrRrTrr%r&r$rrs*0 3 3 3 3 3<<<X<<<<X<<<<X<c!Qll        r&rc eZdZdZ dEdFd ZedGd ZedGdZedGdZedGdZ e j dHdZ dIdZ dJdZ dKdZ dLdZdJdZdKdZdLdZdJd ZdKd!ZdMd$ZdMd%ZdNd'ZdOd)ZdPd*ZdPd+ZdPd,ZdQd.ZdRd0ZdSd3ZedTd5ZedUd6Z edUd7Z!edUd8Z"edUd9Z#edUd:Z$edUd;Z%d<e&dfdVdDZ'dS)WraEstablish 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` rYNoriginr r<Optional[UVec]rArDc,||t}t}t}n|xt|}t|}t||}nU|wt|}t|}t||}n|wt|}t|}t||}nct|}t|}t|}t j|||t||_dSr*) r r r r r+r2rr3r4)r5rdr<rArD_ux_uy_uzs r$r9z UCS.__init__s :"*CCCC Zr(($$&&Cr(($$&&Cr((..$$..00CC Zr(($$&&Cr(($$&&Cr((..$$..00CC Zr(($$&&Cr(($$&&Cr((..$$..00CCr(($$&&Cr(($$&&Cr(($$&&C ( S#sDLL I I r&rr c|jjSr;)r4r<r=s r$r<zUCS.ux{~r&c|jjSr@)r4rAr=s r$rAzUCS.uyrkr&c|jjSrC)r4rDr=s r$rDzUCS.uzrkr&c|jjS)zReturns the originr4rdr=s r$rdz UCS.origins{!!r&vrc||j_dS)z Set origin.Nro)r5rps r$rdz UCS.origins r&cNt|j|j|j|jS)zReturns a copy of this UCS.)rrdr<rArDr=s r$copyzUCS.copys4;$':::r&r#c6|j|S)z"Returns WCS point for UCS `point`.)r4r-rQs r$rRz UCS.to_wcss{$$U+++r&rIterable[Vec3]rJc6|j|S)z1Returns iterable of WCS vectors for UCS `points`.)r4transform_verticesr5rs r$rTzUCS.points_to_wcss{--f555r&vectorc6|j|S)zAReturns WCS direction for UCS `vector` without origin adjustment.)r4transform_directionr5rys r$direction_to_wcszUCS.direction_to_wcss{..v666r&c6|j|S)z"Returns UCS point for WCS `point`.)r4ucs_vertex_from_wcsrQs r$rHz UCS.from_wcss{..u555r&c#<K|j}|D]}||VdS)z2Returns iterable of UCS vectors from WCS `points`.N)rH)r5rrHr#s r$rNzUCS.points_from_wcss== " "E(5// ! ! ! ! " "r&c6|j|S)z>Returns UCS vector for WCS `vector` without origin adjustment.)r4ucs_direction_from_wcsr|s r$direction_from_wcszUCS.direction_from_wcss{11&999r&cz||}t|j|S)zuReturns OCS vector for UCS `point`. The :class:`OCS` is defined by the z-axis of the :class:`UCS`. rRrrDrH)r5r#wpoints r$to_ocsz UCS.to_ocss1 U##47||$$V,,,r&c#K|j}t|j}|D]"}|||V#dS)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)r5rwcsocsr#s r$ points_to_ocszUCS.points_to_ocssXk$'ll + +E,,ss5zz** * * * * + +r&anglerVcZ|tj|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_degr5rs r$to_ocs_angle_degzUCS.to_ocs_angle_degs-22   & &   r&cZ|tj|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?53I3IJJPPr& directioncvt|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. )rrDrHr})r5rs r$rz"UCS.ucs_direction_to_ocs_direction s. 47||$$T%:%:9%E%EFFFr&axisctjt||}||j|j|jg\}}}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 rdr<rArD) r axis_rotater rwr<rArDrrd)r5rrtr<rArDs r$rotatez UCS.rotates[  dU 3 3))47DGTW*EFF B$+";;;;r&ctj|j|}||j|jg\}}t |j|j||S)zReturns a new rotated UCS, rotation axis is the local x-axis. Args: angle: rotation angle in radians r)rrr<rwrArDrrd)r5rrrArDs r$rotate_local_xzUCS.rotate_local_x!sS  % 0 0%%tw&899B$+$'bR@@@@r&ctj|j|}||j|jg\}}t |j||j|S)zReturns a new rotated UCS, rotation axis is the local y-axis. Args: angle: rotation angle in radians r)rrrArwr<rDrrd)r5rrr<rDs r$rotate_local_yzUCS.rotate_local_y,sS  % 0 0%%tw&899B$+"R@@@@r&ctj|j|}||j|jg\}}t |j|||jS)zReturns a new rotated UCS, rotation axis is the local z-axis. Args: angle: rotation angle in radians r)rrrDrwr<rArrd)r5rrr<rAs r$rotate_local_zzUCS.rotate_local_z7sS  % 0 0%%tw&899B$+"@@@@r&deltac@|xjt|z c_|S)ztShifts current UCS by `delta` vector and returns `self`. Args: delta: shifting vector )rdr )r5rs r$shiftz UCS.shiftBs tE{{"  r&locationc.t||_|S)zPlace current UCS at new origin `location` and returns `self`. Args: location: new origin in WCS )r rd)r5rs r$movetoz UCS.movetoLs8nn  r&mrc&|xj|zc_|S)zGeneral inplace transformation interface, returns `self` (floating interface). Args: m: 4x4 transformation matrix (:class:`ezdxf.math.Matrix44`) )r4)r5rs r$r-z UCS.transformVs q  r&boolc|jjS)z0Returns ``True`` if cartesian coordinate system.)r4 is_cartesianr=s r$rzUCS.is_cartesianas{''r&ct|}|t||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` rdr<rDr r2r)rdrr#x_axisz_axiss r$from_x_axis_and_point_in_xyzUCS.from_x_axis_and_point_in_xyf@dd5kkF233&V7777r&ct|}t||z }||}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` rdr<rAr)rdrr#r xz_vectory_axiss r$from_x_axis_and_point_in_xzzUCS.from_x_axis_and_point_in_xzxEdKK&( ((&V7777r&ct|}t||z }||}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` rdrArDr)rdrr#r xy_vectorrs r$from_y_axis_and_point_in_xyzUCS.from_y_axis_and_point_in_xyrr&ct|}t||z }||}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&ct|}|t||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&ct|}t||z }||}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&rrrrrrUrrc t||j|t|z|t|z|t |zf|dS)rXrZN)r%rdrRr r r r[s r$r%zUCS.render_axissp  + FVO,, FVO,, FVO,,  r&)rYNNN)rdr r<rerArerDrer\)rpr rr)rrr#r rr rrurrJ)ryr rr rrVrrV)rr rr )rr rrVrr)rrVrr)rr rr)rr rr)rrrr)rr)rdr rr r#r rr)rrrUrVrr)(r]r^r_r`r9rar<rArDrdsetterrsrRrTr}rHrNrrrrrrrrrrrrr-r staticmethodrrrrrrrr%rbr&r$rrts*!!!! JJJJJ<XXX"""X" ]];;;;,,,,666677776666"""" ::::---- + + + +     Q Q Q QGGGG < < < < A A A A A A A A A A A A    (((X(888\8"888\8$888\8$888\8$888\8"888\8*c!Qll        r&rcbeZdZdZfdZddZdd Zdd Zdd ZddZ ddZ ddZ ddZ xZ S)rz5UCS is equal to the WCS and OCS (extrusion = 0, 0, 1)cHtdSr*)superr9)r5 __class__s r$r9zPassTroughUCS.__init__s r&r#r rc|Sr*rbrQs r$rRzPassTroughUCS.to_wcs r&rrurJc t|Sr*iterrxs r$rTzPassTroughUCS.points_to_wcsF||r&c|Sr*rbrQs r$rzPassTroughUCS.to_ocsrr&c t|Sr*rrxs r$rzPassTroughUCS.points_to_ocsrr&rrVc|Sr*rbrs r$rzPassTroughUCS.to_ocs_angle_degrr&c|Sr*rbrs r$rzPassTroughUCS.to_ocs_angle_radrr&c|Sr*rbrQs r$rHzPassTroughUCS.from_wcsrr&c t|Sr*rrxs r$rNzPassTroughUCS.points_from_wcsrr&rrr)r]r^r_r`r9rRrTrrrrrHrN __classcell__)rs@r$rrs??r&rN) rrrr rrrrrr) __future__rtypingrrrrr ezdxf.mathr r r r r r ezdxf.colorsr ezdxf.layoutsr__all__r%r0rrrrbr&r$rsv#"""""HHHHHHHHHHHHHHCCCCCCCCCCCCCCCC)(((((( ) ) )#aA,, CCCCC V V V V V V V V rr r r r r r r r j Cr&