ó ›ßÈ[c@`s*ddlmZmZmZddlZddlZddlmZddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZmZmZmZejejƒd e fd „ƒYƒZd efd„ƒYZdefd„ƒYZdefd„ƒYZdS(i(tprint_functiontdivisiontabsolute_importN(tPath(t Transformi(tunits(tWCS(twcs_to_celestial_frame(tsix(tSkyCoordtframe_transform_graphtSphericalRepresentationtUnitSphericalRepresentationtBaseCoordinateFrametCurvedTransformcB`sAeZdZdZdZeZd„ZeZd„Z d„Z RS(s> Abstract base class for non-affine curved transforms icC`st|j|jƒ|jƒS(s Transform a Matplotlib Path Parameters ---------- path : :class:`~matplotlib.path.Path` The path to transform Returns ------- path : :class:`~matplotlib.path.Path` The resulting path (Rt transformtverticestcodes(tselftpath((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyttransform_path%scC`stdƒ‚dS(Nt(tNotImplementedError(Rtinput((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR7scC`stdƒ‚dS(NR(R(R((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pytinverted:s( t__name__t __module__t__doc__t input_dimst output_dimstFalset is_separableRttransform_path_non_affineRR(((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRs  tWCSWorld2PixelTransformcB`sAeZdZdd„Zed„ƒZd„ZeZd„Z RS(s< WCS transformation from world to pixel coordinates cC`sÅtt|ƒjƒ||_|jjjdkr¸|dkrLtdƒ‚qÁt|ƒ|jjjkrˆtdj|jjjƒƒ‚qÁ||_ |j dƒ|_ |j dƒ|_ n d|_ dS(Nis:WCS has more than 2 dimensions, so ``slice`` should be setsHslice should have as many elements as WCS has dimensions (should be {0})txty( tsuperR!t__init__twcstnaxistNonet ValueErrortlentformattslicetindextx_indexty_index(RR&R,((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR%Cs    cC`s |jjjS(N(R&R'(R((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRSscC`s |jd|jjjkr+tdƒ‚n|jddkrPtjdƒ}n|jj|dƒd}|jdkr||S|dd…|j |j ffSdS(s  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 in the WCS. This then returns the (x, y) pixel coordinates as a Nx2 array. isGSecond dimension of input values should match number of WCS coordinatesiiN(ii( tshapeR&R'R)tnptzerost wcs_world2pixR,R(R.R/(Rtworldtpixel((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRWscC`st|jd|jƒS(s5 Return the inverse of the transform R,(tWCSPixel2WorldTransformR&R,(R((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRnsN( RRRR(R%tpropertyRRttransform_non_affineR(((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR!>s   R6cB`sAeZdZdd„Zed„ƒZd„ZeZd„Z RS(s< WCS transformation from pixel to world coordinates cC`s_tt|ƒjƒ||_||_|jdk r[|jdƒ|_|jdƒ|_ndS(NR"R#( R$R6R%R&R,R(R-R.R/(RR&R,((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR%zs   cC`s |jjjS(N(R&R'(R((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR‚scC`sf|jdkr|jƒ}ng}xv|jD]k}|dkr`|j|dd…dfƒq.|dkrŒ|j|dd…dfƒq.|j|ƒq.Wtjtj|Œƒjƒ}|d7}|jddkrêtj d ƒ}n|j j |dƒ}|j j |dƒ}tj ddƒ-tjtj||ƒd kd dƒ}WdQXtj||<|S( sÿ 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 in the WCS R"NiR#iitinvalidtignoregð?taxis(ii(R,R(tcopytappendR1tarraytbroadcast_arrayst transposeR0R2R&t wcs_pix2worldR3terrstatetanytabstnan(RR5t pixel_fullR-R4t pixel_checkR9((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR†s&     . cC`st|jd|jƒS(s5 Return the inverse of the transform R,(R!R&R,(R((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR­sN( RRRR(R%R7RRR8R(((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR6us   %tCoordinateTransformcB`sJeZd„Zed„ƒZejd„ƒZd„ZeZd„ZRS(cC`s¦tt|ƒjƒ||_||_t|jtƒrLt|jƒ|_n„t|jt j ƒr£t j |jƒ|_|jdkrÐtdj|jƒƒ‚qÐn-t|jtƒrÄ|j|_n tdƒ‚t|jtƒr÷t|jƒ|_n„t|jt j ƒrNt j |jƒ|_|jdkr{tdj|jƒƒ‚q{n-t|jtƒro|j|_n tdƒ‚|j|jkr™t|_n t|_dS(NsFrame {0} not foundsMinput_system should be a WCS instance, string, or a coordinate frame instancesNoutput_system should be a WCS instance, string, or a coordinate frame instance(R$RHR%t_input_system_namet_output_system_namet isinstanceRRt input_systemRt string_typesR t lookup_nameR(R)R+R t TypeErrort output_systemtTruet same_framesR(RRLRP((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyR%¶s0     cC`s|jS(N(t _same_frames(R((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRRÖscC`s ||_dS(N(RS(RRR((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRRÚscC`s%|jr |S|dd…df|dd…df}}t||dtjtjfd|jƒ}tjddƒ|j|jƒ}WdQXt |j t t fƒrÈ|j jj}|j jj}n|jjj}|jjj}tj|dd…tjf|dd…tjffddƒS( s= Transform one set of coordinates to another NiitunittframetallR:R;(RRR tutdegRLR1RBt transform_toRPt issubclasstrepresentationR R tdatatlontlatt sphericalt concatenatetnewaxis(Rt input_coordstx_inty_intc_intc_outR]R^((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRÞs - cC`st|j|jƒS(s5 Return the inverse of the transform (RHRJRI(R((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRûs( RRR%R7RRtsetterRR8R(((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyRH´s   ( t __future__RRRtabctnumpyR1tmatplotlib.pathRtmatplotlib.transformsRRRRWR&Rt wcs.utilsRtexternRt coordinatesR R R R R t add_metaclasstABCMetaRR!R6RH(((sGlib/python2.7/site-packages/astropy/visualization/wcsaxes/transforms.pyts  ("7?