o …<ÔcÇã@sžddlZddlZddlZddlZddlmZddlmZm Z ddl Z ddlZ ddlm ZdZdd„Zddd „Zd d„Zdd „Zddd„ZedkrMeƒdSdS)éN)Ú RAMP_LIST)ÚreadfileÚ writefile)Úmultitrack_utilitiesa$example: gridsearch_ramps_relax.py KokoxiliModelData_AT143_ramped.h5 KokoxiliModel_AT143.h5 mask.h5 --ramp_estimate -s linear -o ramp_noise.h5 --ramp_file ramps.h5 --outdir ./ gridsearch_ramps_relax.py KokoxiliModelData_AT143_ramped.h5 KokoxiliModel_AT143.h5 mask.h5 --outdir ./ cCs¬tjdtjtd}|jddtdd|jddtdd|jd dtd d|jddd dd|jdddtdd|jddtddd|jddddd|jddtddd|S) NzPrepare data for Kite software)ÚdescriptionÚformatter_classÚepilogÚObs_fileéz Observed data)ÚnargsÚtypeÚhelpÚ Model_filezModeled dataÚmaskz Mask filez--ramp_estimateÚe_rampÚ store_truezestimate ramp or not)ÚdestÚactionr ú-sÚsurface_typeÚlinearz5type of surface/ramp to calculated, linear by default)rÚdefaultÚchoicesr ú-oÚoutputzoutput name for ramp+noise data)rrrr ú--ramp_fileÚ ramp_filezestimated ramp data)rrr ú--outdirÚoutdirzoutput file dir)ÚargparseÚArgumentParserÚRawTextHelpFormatterÚEXAMPLEÚadd_argumentÚstrr)Úparser©r&úR/home/exouser/operations/rsmas_insar/tools/MimtPy/mimtpy/gridsearch_ramps_relax.pyÚ create_parsers*þÿÿ ÿÿr(cCstƒ}|j|d}|S)N)Úargs)r(Ú parse_args)Úiargsr%Úinpsr&r&r'Úcmd_line_parse8sr-cCs |jd}|jd}|j}|jd}|jd}|jd}||d|dd|d|d|d|jdg }t |¡}t d|ƒt d |ƒt d |ƒt j | ¡¡||fS)ú.subtract forward model data from observed datarrz--ramprz-mrrúsubtract_h5.pyzramp_noise file is: %szestimated ramp file is: %s)r rrrrrrÚmuÚseperate_str_byspaceÚprintÚmimtpyÚsubtract_h5ÚmainÚsplit)r,Ú observed_fileÚ mask_fileÚ ramp_typerÚramp_noise_fileÚ model_fileÚscp_argsr&r&r'Úcalculate_ramp_noise>s $ r=cCsx|jddd…d}t |jd|¡d}t |jd|¡\}}||}td|ƒtj||jd||ddS)zLcalculate residual data which is observed data - model data - estimated ramprNéýÿÿÿz_residual.h5zHthe residual file after subtracting model data and estimated ramp is: %s)Úout_fileÚmetadata)rrÚreadrr2rÚwrite)r,r:rÚ residual_fileZramp_noise_dataZ ramp_dataÚatrÚresidualr&r&r'Úcalculate_residualQsrFc CsÊt|ƒ}tdƒ|jr$tdƒt|ƒ\}}tdƒtdƒt|||ƒn;tdƒ|jd}|jd}|jddd…d}|jd}||d|d |jdg}t |¡}td |ƒtj | ¡¡tdƒdS)Nz3-------------------------------------------------- r.z-calculate residual data with estimating rampsz/calculate residual data without estimating ramprr>z_residual_noramp.h5rrr/zfinish!)r-r2rr=rFr rrrr0r1r3r4r5r6) r+r,rr:r7r8rCr;r<r&r&r'r5`s$ r5Ú__main__)N)ÚosrÚnumpyÚnpÚmintpyÚmintpy.objectsrÚmintpy.utilsrrr3Úmimtpy.workflowÚmimtpy.utilsrr0r"r(r-r=rFr5Ú__name__r&r&r&r'Ús$ ÿ