ó šßÈ[c@`súdZddlmZmZmZmZddlmZddlm Z ddl Z idd6d d 6d d 6d d6Z d„Z d„Zdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdS(uH This module provides functions to help with testing against iraf tasks i(tabsolute_importtdivisiontprint_functiontunicode_literalsi(tlog(trangeNu Chebyshevgð?uLegendreg@uSpline3g@uSpline1g@cC`sXt|ƒ}|jƒ}|jƒ|jdƒd}g|D]}t|ƒ^q<}|S(uÑ Read the records of an IRAF database file into a python list Parameters ---------- fname : str name of an IRAF database file Returns ------- A list of records ubegini(topentreadtclosetsplittRecord(tfnametftdtbtrecstrtrecords((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyt get_recordss   cC`s&t|ƒ}|jƒ}|jƒ|S(uº Read an IRAF database file Parameters ---------- fname : str name of an IRAF database file Returns ------- the database file as a string (RRR(R R R ((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pytget_database_string)s   R cB`s;eZdZd„Zd„Zd„Zd„Zd„ZRS(u, A base class for all records - represents an IRAF database record Attributes ---------- recstr: string the record as a string fields: dict the fields in the record taskname: string the name of the task which created the database file cC`s+||_|jƒ|_|jƒ|_dS(N(trecstrt get_fieldstfieldst get_task_namettaskname(tselfR((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyt__init__Js cC`sg|jjdƒ}g|D]}|jƒ^q}g|D]'}t|ƒdkr8|j|ƒ^q8|S(Nu i(RR tstriptlentremove(Rtreclisttl((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pytaslistOs2cC`si}|jƒ}t|ƒ}xît|ƒD]à}||}|r+|djƒr+|jƒ}|d|krà||ddjƒs¸|j|||t|dƒd!ƒ||dlib/python2.7/site-packages/astropy/modeling/tests/irafutil.pys cscs`s|] }|VqdS(N((R R!((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pys es(RRRtisalphaR tread_array_fieldtinttjoin(RRtflistt numfieldstitlinetfield((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRUs    -((cC`s(y|jdSWntk r#dSXdS(Nutask(RtKeyErrortNone(R((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRjs cC`s›g|dD]}|jƒ^q }g|D]}|d ^q*}ytj|ƒ}Wn5tk rŠtjdj|djƒdƒƒnX|jtjƒS(NiiuCould not read array field {}i( R tnptarrayt ExceptionRtdebugtformattastypetfloat64(Rt fieldlistRt fieldlinetxyztfarr((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyR#ps# ((t__name__t __module__t__doc__RRRRR#(((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyR <s      tIdentifyRecordcB`sDeZdZd„Zd„Zd„Zd„Zd„Zd„ZRS(u Represents a database record for the onedspec.identify task Attributes ---------- x: array the X values of the identified features this represents values on axis1 (image rows) y: int the Y values of the identified features (image columns) z: array the values which X maps into modelname: string the function used to fit the data nterms: int degree of the polynomial which was fit to the data in IRAF this is the number of coefficients, not the order mrange: list the range of the data coeff: array function (modelname) coefficients cC`s»tt|ƒj|ƒ|jdjƒ|_|jddd…df|_|jƒ|_|jddd…df|_ |j ƒ|_ |j ƒ|_ |jƒ|_|jƒ|_dS(Nu coefficientsufeaturesii(tsuperR;RRtflattent _flatcoefftxt get_ydatatytztget_model_namet modelnamet get_ntermstntermst get_rangetmranget get_coefftcoeff(RR((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyR–s  cC`st|jdS(Ni(tiraf_models_mapR>(R((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRC¡scC`s |jdS(Ni(R>(R((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRE¤scC`s$|jd}|jd}||gS(Nii(R>(Rtlowthigh((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRG§s  cC`s |jdS(Ni(R>(R((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRI¬scC`sz|jd}|jdƒd}|jdƒ}|||!}d|krl|jdƒd}t||d!ƒSt|ƒSdS(Nuimageu[iu]u,iÿÿÿÿ(RtfindR$(Rtimagetlefttrighttsectiontyind((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyR@¯s   ( R8R9R:RRCRERGRIR@(((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyR;}s    tFitcoordsRecordcB`s eZdZd„Zd„ZRS(uƒ Represents a database record for the longslit.fitccords task Attributes ---------- modelname: string the function used to fit the data xorder: int number of terms in x yorder: int number of terms in y xbounds: list data range in x ybounds: list data range in y coeff: array function coefficients cC`s­tt|ƒj|ƒ|jdjƒ|_t|jd|_|jd|_|jd|_ |jd|jdg|_ |jd|jdg|_ |j ƒ|_ dS( Nusurfaceiiiiiii(R<RTRRR=t_surfaceRKRDtxordertyordertxboundstyboundsRIRJ(RR((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRÐscC`s |jdS(Ni(RU(R((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRIÚs(R8R9R:RRI(((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRT»s tIDBcB`s eZdZd„Zd„ZRS(uÒ Base class for an IRAF identify database Attributes ---------- records: list a list of all `IdentifyRecord` in the database numrecords: int number of records cC`sAg|j|ƒD]}t|ƒ^q|_t|jƒ|_dS(N(RR;RRt numrecords(Rtdtbstrtrstr((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRês+cC`s|jdƒ}y|djdƒ}Wntk r:|SXt|ƒdkry|djdƒry|djƒ ry|dS|SdS(Nubeginiu iu#i(R R/Rt startswithR(RR trltrl0((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRîs 6(R8R9R:RR(((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRZÞs  tReidentifyRecordcB`s eZdZd„Zd„ZRS(uG Represents a database record for the onedspec.reidentify task cC`stt|ƒj|ƒtjg|jD]}|j^q&ƒ|_|jƒ|_tjg|jD]}|j ^q`ƒ|_ dS(N( R<RaRR-R.RR?R@RARB(Rt databasestrR((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRs+cC`sXtj|jjƒ}|tjg|jD]}|j^q(ƒdd…tjf}|S(N(R-tonesR?tshapeR.RRAtnewaxis(RRAR((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyR@s?(R8R9R:RR@(((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyRaüs (R:t __future__RRRRtloggerRtextern.six.movesRtnumpyR-RKRRtobjectR R;RTRZRa(((s>lib/python2.7/site-packages/astropy/modeling/tests/irafutil.pyts"     A>#