B q\x@sdZddlmZddlZdddddZd d Zd d ZGd ddZGdddeZ GdddeZ GdddZ Gddde Z dS)zH This module provides functions to help with testing against iraf tasks )logNZ ChebyshevZLegendreZSpline3ZSpline1)g?g@g@g@cCs<t|}|}||ddd}dd|D}|S)z 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 beginNcSsg|] }t|qS)Record).0rrr>lib/python3.7/site-packages/astropy/modeling/tests/irafutil.py "szget_records..)openreadclosesplit)fnamefdtbZrecsrecordsrrr get_recordss rcCst|}|}||S)z Read an IRAF database file Parameters ---------- fname : str name of an IRAF database file Returns ------- the database file as a string )r r r )rrrrrr get_database_string&s rc@s8eZdZdZddZddZddZdd Zd d Zd S) ra, 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 cCs||_||_||_dS)N)recstr get_fieldsfields get_task_nameZtaskname)selfrrrr __init__Gs zRecord.__init__cs0|jdddDfddDS)N cSsg|] }|qSr)strip)rlrrr r Nsz!Record.aslist..cs"g|]}t|dkr|qS)r)lenremove)rr)reclistrr r Os)rr)rr)r r aslistLs z Record.aslistc Csi}|}t|}xt|D]}||}|r|dr|}|d|kr||dds||||t|dd||d<qddd|ddD||d<qddd|ddD||d<qqqW|S)Nrr css|] }|VqdS)Nr)rsrrr `sz$Record.get_fields..css|] }|VqdS)Nr)rr#rrr r$bs)r!rrangeisalpharread_array_fieldintjoin)rrZflistZ numfieldsilineZfieldrrr rRs (&&zRecord.get_fieldscCs$y |jdStk rdSXdS)NZtask)rKeyError)rrrr rgs zRecord.get_task_namec Cspdd|ddD}dd|D}yt|}Wn0tk rbtd|ddYnX|tjS)NcSsg|] }|qSr)r)rrrrr r osz+Record.read_array_field..rcSsg|]}|ddqS)Nr)rrrrr r rszCould not read array field {}r) nparray ExceptionrdebugformatrZastypeZfloat64)rZ fieldlistZ fieldlineZxyzZfarrrrr r'ms"zRecord.read_array_fieldN) __name__ __module__ __qualname____doc__rr!rrr'rrrr r9s  rcsHeZdZdZfddZddZddZdd Zd d Zd d Z Z S)IdentifyRecorda 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 cst||jd|_|jddddf|_||_|jddddf|_| |_ | |_ | |_||_dS)NZ coefficientsZfeaturesrr)superrrflatten _flatcoeffx get_ydatayzget_model_name modelname get_ntermsZnterms get_rangeZmrange get_coeffcoeff)rr) __class__rr rs     zIdentifyRecord.__init__cCst|jdS)Nr)iraf_models_mapr:)rrrr r?szIdentifyRecord.get_model_namecCs |jdS)Nr)r:)rrrr rAszIdentifyRecord.get_ntermscCs|jd}|jd}||gS)Nr-)r:)rZlowZhighrrr rBs  zIdentifyRecord.get_rangecCs|jddS)N)r:)rrrr rCszIdentifyRecord.get_coeffcCs`|jd}|dd}|d}|||}d|krT|dd}t||dSt|SdS)Nimage[r],)rfindr()rrIleftrightZsectionZyindrrr r<s   zIdentifyRecord.get_ydata) r3r4r5r6rr?rArBrCr< __classcell__rr)rEr r7zs r7cs(eZdZdZfddZddZZS)FitcoordsRecorda 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 cs~t||jd|_t|jd|_|jd|_|jd|_|jd|jdg|_ |jd|jdg|_ | |_ dS) NZsurfacerrrGrH) r8rrr9_surfacerFr@ZxorderZyorderZxboundsZyboundsrCrD)rr)rErr rs   zFitcoordsRecord.__init__cCs|jddS)N)rV)rrrr rCszFitcoordsRecord.get_coeff)r3r4r5r6rrCrQrr)rEr rRs rRc@s eZdZdZddZddZdS)IDBz Base class for an IRAF identify database Attributes ---------- records: list a list of all `IdentifyRecord` in the database numrecords: int number of records cCs&dd||D|_t|j|_dS)NcSsg|] }t|qSr)r7)rZrstrrrr r sz IDB.__init__..)r!rrZ numrecords)rZdtbstrrrr rsz IDB.__init__cCsl|d}y|dd}Wntk r0|SXt|dkrd|ddrd|dsd|ddS|SdS)NrrrrG#r)rr0r startswithr)rrZrlZrl0rrr r!s & z IDB.aslistN)r3r4r5r6rr!rrrr rXs rXcs(eZdZdZfddZddZZS)ReidentifyRecordzG Represents a database record for the onedspec.reidentify task csJt|tdd|jD|_||_tdd|jD|_dS)NcSsg|] }|jqSr)r;)rrrrr r sz-ReidentifyRecord.__init__..cSsg|] }|jqSr)r>)rrrrr r s) r8rr.r/rr;r<r=r>)rZ databasestr)rErr rs  zReidentifyRecord.__init__cCs:t|jj}|tdd|jDddtjf}|S)NcSsg|] }|jqSr)r=)rrrrr r sz.ReidentifyRecord.get_ydata..)r.Zonesr;shaper/rZnewaxis)rr=rrr r<s(zReidentifyRecord.get_ydata)r3r4r5r6rr<rQrr)rEr r[s r[) r6Zastropy.loggerrZnumpyr.rFrrrr7rRrXr[rrrr s A>#