U e{j@sdZddlZddlmZdddddZd d Zd d ZGd ddZGdddeZ GdddeZ GdddZ Gddde Z dS)zH This module provides functions to help with testing against iraf tasks N)logZ 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^/var/www/html/bwave/python/venv/lib/python3.8/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||_||_||_dSN)recstr get_fieldsfields get_task_nameZtasknameselfrrrr __init__Es zRecord.__init__cs0|jdddDfddDS)N cSsg|] }|qSr)striprentryrrr r Lsz!Record.aslist..cs"g|]}t|dkr|qS)r)lenremover Zreclistrr r Ms )rrrrr$r aslistJs z Record.aslistc Csi}|}t|}t|D]}||}|r|dr|}|d|kr||dds||||t|dd||d<qddd|ddD||d<qddd|ddD||d<qqq|S)Nrr css|] }|VqdSrrrsrrr _sz$Record.get_fields..css|] }|VqdSrrr(rrr r*as)r&r"rangeisalpharread_array_fieldintjoin)rrflistZ numfieldsilinefieldrrr rPs   &&zRecord.get_fieldscCs(z |jdWStk r"YdSXdS)Ntask)rKeyErrorr%rrr rfs zRecord.get_task_namecCspdd|ddD}dd|D}zt|}Wn0tk rbtd|ddYnX|tjS)NcSsg|] }|qSr)rr rrr r nsz+Record.read_array_field..rcSsg|]}|ddqS)Nrr rrr r qszCould not read array field r)nparray ExceptionrdebugrZastypeZfloat64)rZ fieldlistZ fieldlineZxyzZfarrrrr r-ls"zRecord.read_array_fieldN) __name__ __module__ __qualname____doc__rr&rrr-rrrr r6s  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 coefficientsfeaturesrr)superrrflatten _flatcoeffx get_ydatayzget_model_name modelname get_ntermsZnterms get_rangeZmrange get_coeffcoeffr __class__rr rs     zIdentifyRecord.__init__cCst|jdS)Nr)iraf_models_maprCr%rrr rHszIdentifyRecord.get_model_namecCs |jdS)NrrCr%rrr rJszIdentifyRecord.get_ntermscCs|jd}|jd}||gS)Nr6rQ)rlowhighrrr rKs  zIdentifyRecord.get_rangecCs|jddS)NrQr%rrr rLszIdentifyRecord.get_coeffcCs`|jd}|dd}|d}|||}d|krT|dd}t||dSt|SdS)Nimage[r],)rfindr.)rrVleftrightsectionZyindrrr rEs   zIdentifyRecord.get_ydata) r;r<r=r>rrHrJrKrLrE __classcell__rrrNr r?ys r?cs(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) NZsurfacerrrRrU) rArrrB_surfacerPrIZxorderZyorderZxboundsZyboundsrLrMrrNrr rs   zFitcoordsRecord.__init__cCs|jddS)N)rdr%rrr rLszFitcoordsRecord.get_coeff)r;r<r=r>rrLr_rrrNr r`s r`c@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)r?)rZrstrrrr r sz IDB.__init__..)r&rr"Z numrecords)rZdtbstrrrr rsz IDB.__init__cCsp|d}z|dd}Wntk r4|YSXt|dkrh|ddrh|dsh|ddS|SdS)NrrrrR#r)rr9r" startswithr)rrZrlZrl0rrr r&s  & z IDB.aslistN)r;r<r=r>rr&rrrr rfs rfcs(eZdZdZfddZddZZS)ReidentifyRecordzG Represents a database record for the onedspec.reidentify task csJt|tdd|jD|_||_tdd|jD|_dS)NcSsg|] }|jqSr)rDrrrr r sz-ReidentifyRecord.__init__..cSsg|] }|jqSr)rGrrrr r s) rArr7r8rrDrErFrG)rZ databasestrrNrr rs  zReidentifyRecord.__init__cCs:t|jj}|tdd|jDddtjf}|S)NcSsg|] }|jqSr)rFrrrr r sz.ReidentifyRecord.get_ydata..)r7ZonesrDshaper8rZnewaxis)rrFrrr rEs(zReidentifyRecord.get_ydata)r;r<r=r>rrEr_rrrNr ris ri) r>numpyr7Zastropy.loggerrrPrrrr?r`rfrirrrr s C?$