IR-edZddlZddlmZdddddZd Zd ZGd d ZGd deZ GddeZ GddZ Gdde Z dS)zI This module provides functions to help with testing against iraf tasks. N)log ChebyshevLegendreSpline3Spline1)g?g@g@g@ct|}|}||ddd}d|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 beginNc,g|]}t|S)Record.0rs ?lib/python3.11/site-packages/astropy/modeling/tests/irafutil.py zget_records..s'''Qvayy''')openreadclosesplit)fnamefdtbrecsrecordss r get_recordsrs[ U A &&((CGGIII 99W  abb !D''$'''G Nrctt|}|}||S)z Read an IRAF database file. Parameters ---------- fname : str name of an IRAF database file Returns ------- the database file as a string )rrr)rrrs rget_database_stringr#s. U A &&((CGGIII Jrc0eZdZdZdZdZdZdZdZdS)r a- 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 cx||_||_||_dSN)recstr get_fieldsfields get_task_nametaskname)selfr#s r__init__zRecord.__init__Ds1 oo'' **,, rcp|jddDfdDS)N c6g|]}|Sr )striprentrys rrz!Record.aslist..Ks 666U5;;==666rc`g|]*}t|dk|+S)r)lenremove)rr/reclists rrz!Record.aslist..Ls0GGG5s5zzQ  r)r#r)r(r3s @raslistz Record.aslistIsJ+##D))66g666GGGGGGGGGrc i}|}t|}t|D] }||}|r|dr|}|dz|kr||dzds@||||t |dzdz||d<dd|ddD||d<dd|ddD||d<  |S)Nrr  c3K|]}|VdSr"r rss r z$Record.get_fields..^s"3I3I!A3I3I3I3I3I3Irc3K|]}|VdSr"r r8s rr:z$Record.get_fields..`s"/E/Ea/E/E/E/E/E/Er)r4r1rangeisalpharread_array_fieldintjoin)r(r%flist numfieldsilinefields rr$zRecord.get_fieldsOsM JJ y!!  A8D Q))  q59$$ Q<?2244J+/+@+@!!a#eAh--&7!&;";<,,uQx((,/883I3IuQRRy3I3I3I+I+IuQx(('*xx/E/E59/E/E/E'E'EF58$$ rc@ |jdS#t$rYdSwxYw)Ntask)r%KeyErrorr(s rr&zRecord.get_task_namees6 ;v& &   44 s  c6d|ddD}d|D} tj|}nE#t$r8tjd|ddYnwxYw|tjS)Nc6g|]}|Sr )rr.s rrz+Record.read_array_field..ms >>>uU[[]]>>>rr c"g|] }|dd S)Nr r.s rrz+Record.read_array_field..ps 000UuRaRy000rzCould not read array field r)nparray Exceptionrdebugrastypefloat64)r( fieldlist fieldlinexyzfarrs rr>zRecord.read_array_fieldks>> !"" >>> 10i000 O8C==DD O O O IMIaL4F4F4H4H4KMM N N N N N O{{2:&&&s7?A98A9N) __name__ __module__ __qualname____doc__r)r4r$r&r>r rrr r 6si  ---  , ' ' ' ' 'rr c@eZdZdZfdZdZdZdZdZdZ xZ 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 ct||jd|_|jddddf|_||_|jddddf|_| |_ | |_ | |_||_dS)N coefficientsfeaturesrr )superr)r%flatten _flatcoeffx get_ydatayzget_model_name modelname get_ntermsnterms get_rangemrange get_coeffcoeffr(r# __class__s rr)zIdentifyRecord.__init__s    +n5==??Z(A.!!Z(A.,,..oo'' nn&& ^^%% rc2t|jdS)Nr)iraf_models_maprcrIs rrhzIdentifyRecord.get_model_namestq122rc|jdS)Nr rcrIs rrjzIdentifyRecord.get_ntermssq!!rc>|jd}|jd}||gS)NrMru)r(lowhighs rrlzIdentifyRecord.get_ranges$oa q!T{rc |jddS)NrurIs rrnzIdentifyRecord.get_coeffsqrr""rc|jd}|ddz}|d}|||}d|vr/|ddz}t||dSt|S)Nimage[r ],)r%findr?)r(r}leftrightsectionyinds rrezIdentifyRecord.get_ydatas~ G$zz#" 3U # '>>::c??Q&DuT"W~&& &w<< r) rXrYrZr[r)rhrjrlrnre __classcell__rqs@rr]r]xs0 & & & & &333""" ###        rr]c(eZdZdZfdZdZxZS)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 ct||jd|_t |jd|_|jd|_|jd|_|jd|jdg|_ |jd|jdg|_ | |_ dS) Nsurfacerr rwr{) rar)r%rb_surfacersrixorderyorderxboundsyboundsrnrorps rr)zFitcoordsRecord.__init__s     I.6688 (q)9:mA& mA&  a($-*:;  a($-*:; ^^%% rc |jddS)N)rrIs rrnzFitcoordsRecord.get_coeffs}QRR  r)rXrYrZr[r)rnrrs@rrrsQ(&&&&&!!!!!!!rrceZdZdZdZdZdS)IDBz Base class for an IRAF identify database. Attributes ---------- records: list a list of all `IdentifyRecord` in the database numrecords: int number of records cd||D|_t|j|_dS)Nc,g|]}t|Sr )r])rrstrs rrz IDB.__init__..s MMMt,,MMMr)r4rr1 numrecords)r(dtbstrs rr)z IDB.__init__s8MMV9L9LMMM dl++rc2|d} |dd}n#t$r|cYSwxYwt|dkr?|ddr$|ds |ddS|S)Nr rr+rw#r )rrPr1 startswithr-)r(rrlrl0s rr4z IDB.aslistsYYw   Q%++d##CC   III  s88q==SV..s33=CFLLNN=abb6MIs3 AAN)rXrYrZr[r)r4r rrrrs<  ,,,     rrc(eZdZdZfdZdZxZS)ReidentifyRecordzH Represents a database record for the onedspec.reidentify task. ct|tjd|jD|_||_tjd|jD|_dS)Ncg|] }|j Sr )rdrs rrz-ReidentifyRecord.__init__..555113555rcg|] }|j Sr )rgrs rrz-ReidentifyRecord.__init__..rr) rar)rNrOrrdrerfrg)r( databasestrrqs rr)zReidentifyRecord.__init__st %%%55 55566!!55 55566rctj|jj}|tjd|jDddtjfz}|S)Ncg|] }|j Sr )rfrs rrz.ReidentifyRecord.get_ydata..s444!!#444r)rNonesrdshaperOrnewaxis)r(rfs rrezReidentifyRecord.get_ydatasN GDFL ! ! 44t|44455aaamD Dr)rXrYrZr[r)rerrs@rrrsQ77777 rr) r[numpyrNastropy.loggerrrsrrr r]rrrr rrrs9 #*99UU*&?'?'?'?'?'?'?'?'D; ; ; ; ; V; ; ; | ! ! ! ! !f ! ! !F<sr