a ߙfbm,@sddlZddlmZmZddlZddlmZddgZddZ d d Z d d Z Gd ddej dZ ddZddZGddde ZdS)N) defaultdict OrderedDict)deserialize_classBaseHighLevelWCSHighLevelWCSMixincCs|dD]}t||}q |S)N.)splitgetattr)objZattar @lib/python3.9/site-packages/astropy/wcs/wcsapi/high_level_api.py rec_getattr s rcCs*g}|D]\}}}||vr||q|SN)append) componentsorderkey_r r r default_orders  rcCs"tjtt|dtd}|S)a Convert value to an int or an int array. Input coordinates converted to integers corresponding to the center of the pixel. The convention is that the center of the pixel is (0, 0), while the lower left corner is (-0.5, -0.5). The outputs are used to index the mask. Examples -------- >>> _toindex(np.array([-0.5, 0.49999])) array([0, 0]) >>> _toindex(np.array([0.5, 1.49999])) array([1, 1]) >>> _toindex(np.array([1.5, 2.49999])) array([2, 2]) g?)Zdtype)npZasarrayZfloorint)valueZindxr r r_toindexsrc@sNeZdZdZeejddZejddZddZ ejdd Z d d Z d S) rz Abstract base class for the high-level WCS interface. This is described in `APE 14: A shared Python interface for World Coordinate Systems `_. cCsdS)zM Returns a reference to the underlying low-level WCS object. Nr selfr r r low_level_wcs6szBaseHighLevelWCS.low_level_wcscGsdS)a Convert pixel coordinates to world coordinates (represented by high-level objects). If a single high-level object is used to represent the world coordinates (i.e., if ``len(wcs.world_axis_object_classes) == 1``), it is returned as-is (not in a tuple/list), otherwise a tuple of high-level objects is returned. See `~astropy.wcs.wcsapi.BaseLowLevelWCS.pixel_to_world_values` for pixel indexing and ordering conventions. Nr )r pixel_arraysr r rpixel_to_world=szBaseHighLevelWCS.pixel_to_worldcGs|j|dddS)a Convert array indices to world coordinates (represented by Astropy objects). If a single high-level object is used to represent the world coordinates (i.e., if ``len(wcs.world_axis_object_classes) == 1``), it is returned as-is (not in a tuple/list), otherwise a tuple of high-level objects is returned. See `~astropy.wcs.wcsapi.BaseLowLevelWCS.array_index_to_world_values` for pixel indexing and ordering conventions. N)r)rZ index_arraysr r rarray_index_to_worldKs z%BaseHighLevelWCS.array_index_to_worldcGsdS)a Convert world coordinates (represented by Astropy objects) to pixel coordinates. If `~astropy.wcs.wcsapi.BaseLowLevelWCS.pixel_n_dim` is ``1``, this method returns a single scalar or array, otherwise a tuple of scalars or arrays is returned. See `~astropy.wcs.wcsapi.BaseLowLevelWCS.world_to_pixel_values` for pixel indexing and ordering conventions. Nr r world_objectsr r rworld_to_pixelYszBaseHighLevelWCS.world_to_pixelcGs<|jdkrt|j|Stt|j|dddSdS)a Convert world coordinates (represented by Astropy objects) to array indices. If `~astropy.wcs.wcsapi.BaseLowLevelWCS.pixel_n_dim` is ``1``, this method returns a single scalar or array, otherwise a tuple of scalars or arrays is returned. See `~astropy.wcs.wcsapi.BaseLowLevelWCS.world_to_array_index_values` for pixel indexing and ordering conventions. The indices should be returned as rounded integers. rNr )Z pixel_n_dimrr$tupletolistr"r r rworld_to_array_indexfs z%BaseHighLevelWCS.world_to_array_indexN) __name__ __module__ __qualname____doc__propertyabcabstractmethodrrr!r$r'r r r rr.s   ) metaclassc Gs|j}|j}t}t|D],}|jr:t||dd||<q||||<qt|t|krptdt|t|i}d}|D]T}g} | D] \}^} } t || r| |qt| dkr||| d<q|d}qq|i} |r| D]\}^} } }}t|dkr| }n t|dkr |d}ntdddl m }t |||rtd |vrf|||d | |<n ||| |<q|||g| Ri|| |<qnt|D]\}}||^} } }}t|dkr| }n t|dkr|d}ntd||}t || s td d d d |Dddl m }t ||r`d |vrV||d | |<n|| |<n||g| Ri|| |<qg}|D]>\}} }t|r| || |n| t| ||q|S)aq Convert the input high level object to low level values. This function uses the information in ``wcs.world_axis_object_classes`` and ``wcs.world_axis_object_components`` to convert the high level objects (such as `~.SkyCoord`) to low level "values" `~.Quantity` objects. This is used in `.HighLevelWCSMixin.world_to_pixel`, but provided as a separate function for use in other places where needed. Parameters ---------- *world_objects: object High level coordinate objects. low_level_wcs: `.BaseLowLevelWCS` The WCS object to use to interpret the coordinates. FZ constructz8Number of world inputs ({}) does not match expected ({})Trr=Tuples in world_axis_object_classes should have length 3 or 4)SkyCoordframez3Expected the following order of world arguments: {}z, cSsg|]\}}}|jqSr )r().0krr r r z0high_level_objects_to_values..)world_axis_object_classesworld_axis_object_componentsrrserialized_classesrlen ValueErrorformatitems isinstancerZastropy.coordinatesr2Z transform_to enumeratejoinvaluescallabler)rr#r:rclassesrZ world_by_keyZ unique_matchwZmatchesklassrZobjectsargskwargsrest klass_genr2ZikeyZworldattrr r rhigh_level_objects_to_valuesxs|         "       rLcGs4|j}|j}|jrrrlistdictr@r?strr;rrr<)r world_valuesrrDZ classes_newrrrGrHirKrresultrFZarkwrIrJr r rvalues_to_high_level_objectss2     ,rTc@s,eZdZdZeddZddZddZdS) rz Mix-in class that automatically provides the high-level WCS API for the low-level WCS object given by the `~HighLevelWCSMixin.low_level_wcs` property. cCs|Srr rr r rr/szHighLevelWCSMixin.low_level_wcscGs t|d|ji}|jj|}|S)Nr)rLrZworld_to_pixel_values)rr#rP pixel_valuesr r rr$3s z HighLevelWCSMixin.world_to_pixelcGsH|jj|}|jdkr|f}t|d|ji}t|dkr@|dS|SdS)Nrrr)rZpixel_to_world_valuesZ world_n_dimrTr;)rrrPrUr r rr<s   z HighLevelWCSMixin.pixel_to_worldN)r(r)r*r+r,rr$rr r r rr(s   )r- collectionsrrZnumpyrZutilsr__all__rrrABCMetarrLrTrr r r rs Jx8