B bCa.<2@s ddlZddlZddlZddlZddlZddlZddl m Z ddlm Z ddlm Z m Z ddlmZddlmZdZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$ddgddgddgddgdZ%d d!d"d#dZ&d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdF"Z'd$d%d%d&dGd'dHdId)dBd*dJd+d,d-dKdLd8d1d2d3dMdNd5d6d7d8dOdPdQddTdUdVdWdXdYdZd[d\d]d^dTd_d`dadb1Z'dcddZ(GdedfdfZ)Gdgdhdhe)Z*didjZ+dkdlZ,dmdnZ-dodpZ.dqdrZ/dsdtZ0dudvZ1dwdxZ2dydzZ3d{d|Z4d}d~Z5ddZ6dS)N)Time)signal)makedirspath)pyplot)EEXISTFgALAgd;Ox@g`"ەx@glx@gfffffx@g?gCl?gףp= @?Bg333333?dg?bZCTIOgZSAAOrZMcDonaldkZWise)ZlscZcptelpZtlvgffffff!@g"@g$@g333333%@iiiniiiiri~ii{iiuiriiZipiRiiigiei+ipiiijiVii8iiiii4)"18351223520630269133739443587753327836682443828858873798230B100180114710115383115404126053136202149661152391154417165341A1760511821011876911940122068602204917051304954993376151120136190406iiiiii&iiiii@i,iiiiii}iipiiii!ii#iL)1rrr.rr-rr/rrr0rr1rrrrrrrr r!r2r"r#r$r%r&r'r(r)r*r3r+r,Z3427720Z5184732Z6116048Z6278762Z7871531Z7970740Z8006161Z8379927Z9139151Z10124866Z10454113Z10644253Z10963065Z12009504Z12258514cCs$x|D]}t|||q WdS)N)keysprint)headerhr8&/home/travis/obs/LCO/nreshk/helpers.py print_headersr:c@seZdZUejjjjed<e ed<e ed<e ed<e ed<e ed<e ed<e ed<e ed <e ed <e ed <e ed <e ed <e ed<e ed<e ed<dddZdS)rawDatar6specfitsFileformatwaveGridmjdsitedatestarnxnOrdalphafhfkfrfbNcCsd|_|dkr|||_|d|_|d|_t|j|_|d|_|d|_|d|_|d|_||_ ||_||_ ||_ ||_ n|j|_|j|_|j|_|j|_|j|_|j |_ |j |_ |j |_ |j |_ |j|_|j|_|j|_|j |_ |j|_|j|_|j|_dS)NzMJD-OBSZSITEIDzDATE-OBSZOBJ1ZNORDNX)r@r6rA siteAlpharFrBrCrErDr<r=r>r?rGrHrIrJ)selfr?r<r6fileNamer>copyr8r8r9__init__s>       zrawData.__init__)NNNNNNN)__name__ __module__ __qualname__astropyiofitsr6Header__annotations__liststrboolfloatintrPr8r8r8r9r;s" r;c seZdZUged<eed<ged<ged<dZeed<eed<eed<eed<eed <e ed <e ed <e ed <e ed <dfdd Z ddZ ddZ ddZddZddZddZddZZS) analyzedDataflatshkwindowoffsetdayhour decimalYrlabelbadlamGrid targOlapfaverageNc s|dk rtj|dt|drp|jdk rp|j} tt| |_tt|  dd|_ t | dd|_ d|j _ nd|_| |_||_||_||_||_||_||_||_| |_dS)N)rOr@.r )r> decimalyear)superrPhasattrr@r]npfloorrdrZsplitrerrfr> outputPathrhr_rirjrbr`rkra) rMrawrir_rjr`rbrkrhrarsr@) __class__r8r9rPs$ zanalyzedData.__init__cCs|S)Nr8)rMr8r8r9rt.szanalyzedData.rawcCs:|jdkr |t|jdS|t|jdSdS)NF_dataZ_combined_data)rk outputDirrZrerd)rMr8r8r9 data_path4s zanalyzedData.data_pathcCsV|jdkr.|t|jdt|jdS|t|jdt|jdSdS)NF_z _report.pdfz_combined_report.pdf)rkrwrZrerErd)rMr8r8r9 report_path;s $zanalyzedData.report_pathcCsFdt|jdt|jdt|jdt|jdt|jS)NzMJD: z and decYr z with z orders. SHK: z and offsets:)rZr@rfrEr`rb)rMr8r8r9rgAszanalyzedData.labelcCs,|jdkrd|jdS|j|jdSdS)Nzoutput//)rsrC)rMr8r8r9starDirDs zanalyzedData.starDircCsH|jdkr&d|jdt|jdS|j|jdt|jdSdS)Nzoutput/r{)rsrCrZrd)rMr8r8r9rwJs zanalyzedData.outputDircCsZd}|jrd}|d|jd|jdd|jjdt|jdd|j7}|S) Nrczbad zNRES spectra, z, z (z{:.6}z), S=z{:.4}) rhrArBr>rfvaluerZr@r`)rMtr8r8r9pdfTitlePs HzanalyzedData.pdfTitle) NNNNNNNNNN)rQrRrSrXr\Z outputStrr]rrZr[rYrPrtrxrzrgr|rwr __classcell__r8r8)rur9r^s, r^c CsLy t|Wn:tk rF}z|jtkr4t|r4nWdd}~XYnXdS)zECreates a directory. equivalent to using mkdir -p on the command lineN)rOSErrorerrnorrisdir)Zmypathexcr8r8r9mkdir_pZs  rcCsNt||dddd}t|dd}t|dd}||}|SdSdS)NfulliܛiTF)r fftconvolverpmeanmax maxNoiseRatio)ritarg correlationrmaxiratior8r8r9bad_spec_detection_v2is rcCsNd}t|d}t|d}d}d}|d|d}tjj|d|}|||} |||} tt| } | |} | |} | dkrd} | t | dkrt | } tj | | | <tt| }tt t }t| || | }t|||krd }t d tt||n0t d tt||d t| d t||S)NgQ?iiTir rg?Fzspec with diff: zbad spec with diff: z first: z second: )rpabsargminscndimagefiltersgaussian_filterr] nanargminlennancahLamcakLamr5rZ)rirZ allowedErrorZlowIZhighIrhwidthdLam gausdTargZnewLookZadjLamfirstlowhighsecondZtrueDistZrealDistr8r8r9bad_spec_detections2   0 rcCstr t|dS)N)debugr5)inputr8r8r9dprintsrcCs|djdS)NrzMJD-OBS)r6)hdur8r8r9 mjd_from_hdusrcsHtfdd|D}tfdd|D}||kr@|n|}|S)Ncsg|]}|kr|qSr8r8).0d)r@r8r9 sz$find_nearest_mjd..csg|]}|kr|qSr8r8)rr)r@r8r9rs)rmin)ddr@rrZnearkeyr8)r@r9find_nearest_mjdsrcs fdd|D}t||jdS)Ncsi|]}t||qSr8)r)rr)keyr8r9 szclosestKey..)r)rget)Zdicrdiffr8)rr9 closestKeysrcCs*yt|ddStk r$dSXdS)NrTF)r] ValueError)folderr8r8r9is_folder_stars  rcsfddtDS)Ncs&g|]}tjtj|r|qSr8)osrrjoin)rname)a_dirr8r9rsz0get_immediate_subdirectories..)rlistdir)rr8)rr9get_immediate_subdirectoriessrcCsL||}t|dt|d}||d}||}|||||k8}|S)Nr ) searchsortedrpclipr)Atargetidxleftrightr8r8r9 find_closests   rcCs(d}x|D]}||j7}q W|t|S)Nr)r`r)datatotrr8r8r9 find_means rcCs0t|}||}x|D]}|j|9_qWdS)N)rr`)rrZpreMeanrFrr8r8r9 set_data_means r)7rrOnumpyrpscipyrastropy.io.fitsrTZastropy.io.fits.header astropy.timerrrr matplotlibrpltrrrcrrlamBlamRconWidlineWidZ sigToFWHMlowGOrdZhighGOrdoldScalepdfModerZsingleIconSizeZaverageIconSizeZ singleOpacityZaverageOpacityZ siteColorsrL tEffLookupr:r;r^rrrrrrrrrrrrr8r8r8r9s      D^%B