U e{j@sddlZddlZddlmZddlmZddlmZ ddl m Z m Z m Z mZdddd gZGd ddeejd ZGd ddeZGd ddeejd ZGdd d eejd ZdS)N)Path) Transform)units)BaseCoordinateFrameSkyCoordUnitSphericalRepresentationframe_transform_graphCurvedTransformCoordinateTransformWorld2PixelTransformPixel2WorldTransformc@s8eZdZdZdZdZdZddZeZddZ dd Z d S) r z> Abstract base class for non-affine curved transforms FcCst||j|jS)a Transform a Matplotlib Path Parameters ---------- path : :class:`~matplotlib.path.Path` The path to transform Returns ------- path : :class:`~matplotlib.path.Path` The resulting path )r transformZverticescodes)selfpathrg/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/visualization/wcsaxes/transforms.pytransform_path(szCurvedTransform.transform_pathcCs tddSNNotImplementedError)rinputrrrr:szCurvedTransform.transformcCs tddSrrrrrrinverted=szCurvedTransform.invertedN) __name__ __module__ __qualname____doc__ input_dims output_dimsZ is_separablerZtransform_path_non_affinerrrrrrr s) metaclasscsNeZdZdZfddZeddZejddZddZeZ d d Z Z S) r Tcst||_||_t|jtrTt|j}|dkrJtd|jdqr||_ nt|jt rj|j|_ nt dt|jtrt|j}|dkrtd|jdq||_ nt|jt r|j|_ nt d|j |j krd|_ nd|_ dS)NzFrame z not foundzMinput_system should be a WCS instance, string, or a coordinate frame instancezNoutput_system should be a WCS instance, string, or a coordinate frame instanceTF)super__init___input_system_name_output_system_name isinstancestrr lookup_name ValueError input_systemr TypeError output_system same_frames)rr+r-Z frame_cls __class__rrr$Ds4            zCoordinateTransform.__init__cCs|jSNZ _same_framesrrrrr.kszCoordinateTransform.same_framescCs ||_dSr1r2)rr.rrrr.osc Cs|jr |S|tj}|dddf|dddf}}tt|||jd}tjdd||j }W5QRX|j j j}|j j j}tj |ddtjf|ddtjffddS)z= Transform one set of coordinates to another Nr)frameignore)all)Zaxis)r.udegrrr+npZerrstateZ transform_tor-Z sphericallonlatZ concatenateZnewaxis)rZ input_coordsZx_inZy_inZc_inZc_outr:r;rrrrss "  zCoordinateTransform.transformcCst|j|jS)5 Return the inverse of the transform )r r&r%rrrrrszCoordinateTransform.inverted) rrr has_inverser$propertyr.setterrZtransform_non_affiner __classcell__rrr/rr As '  c@sFeZdZdZdZdZeejddZ ejddZ ejdd Z dS) r z= Base transformation from world to pixel coordinates TNcCsdS)z6 The number of input world dimensions Nrrrrrr szWorld2PixelTransform.input_dimscCsdS)z Transform world to pixel coordinates. You should pass in a NxM array where N is the number of points to transform, and M is the number of dimensions. This then returns the (x, y) pixel coordinates as a Nx2 array. Nr)rZworldrrrrszWorld2PixelTransform.transformcCsdSr<NrrrrrrszWorld2PixelTransform.inverted) rrrrr=Zframe_inr>abcabstractmethodr rrrrrrr s  c@sFeZdZdZdZdZeejddZ ejddZ ejdd Z dS) r z= Base transformation from pixel to world coordinates TNcCsdS)z7 The number of output world dimensions Nrrrrrr!sz Pixel2WorldTransform.output_dimscCsdS)z Transform pixel to world coordinates. You should pass in a Nx2 array of (x, y) pixel coordinates to transform to world coordinates. This will then return an NxM array where M is the number of dimensions. Nr)rZpixelrrrrszPixel2WorldTransform.transformcCsdSrArrrrrrszPixel2WorldTransform.inverted) rrrrr=Z frame_outr>rBrCr!rrrrrrr s  )rBnumpyr9Zmatplotlib.pathrZmatplotlib.transformsrZastropyrr7Zastropy.coordinatesrrrr__all__ABCMetar r r r rrrr s   "T