o whether simulate troposphere/ionosphere ramp for each subswathz--lls1latlons1)ZLowLatZLowLonZHighLatZHighLonzVlat and lon for the two points located at the intersect corner of swath 1 and swath 2.z--lls2latlons2zVlat and lon for the two points located at the intersect corner of swath 2 and swath 3.z--tiramptiramp ) s1xramps1yramps1offsets2xramps2yramps2offsets3xramps3yramps3offsetzaresidual troposphere / ionosphere linear ramp for swath1, swath2 and swath3 in X and Y direction.z--wrapz!whether wrap the synthetic data. z--atmozwhether add atmo noise data.z --atmofileatmofilezatmospheric noise)rr r r z--outdirz output dir)argparseArgumentParserRawTextHelpFormatterEXAMPLE add_argumentstrfloat)parserr6H/home/exouser/operations/rsmas_insar/tools/MimtPy/mimtpy/synthetic_S1.py create_parser%s:r8cCst}|j|d}|S)N)args)r8 parse_args)iargsr5inpsr6r6r7cmd_line_parseRs r=c Cst|jd|_t|j\|_|_t|jdd|_ |jd|jdd}|jd|jdd}t |j dd}t |j dd}t ||f}t j||gtd}t t||d }|d d df}|d d df} || |fS) z&transfer lat/lon to local coordinationrrr)r)rr?r?r dtypeN)rread_attributer metadataut get_lat_lonlatlonreadmasknp transposereshapevstackarrayr4mutllh2xy) r<Z origin_latZ origin_lonrHrIllhoriginXYXYr6r6r7ll2xyXs  rXc!Cst|\}}}t|jd}t|jd}t|jd}t|jd}tj||g||ggtd} tt| |d} | d} | d} | d } | d }t|jd}t|jd}t|jd}t|jd}tj||g||ggtd}tt||d}|d}|d}|d }|d }|j \}}tj ||ft dtj }| d d}|| || | || | | |}||||||||||}d||dk<|dk|dk}d||<d||dk<| ||}||} tj| |jdd |jd | S)zPgenerate mask to distinguish the 3 swaths based on the intersection cover pointsrr r@rrArCr>)rr )r r)r r r?zmask_swaths.h5out_filerE)rXr4rrLrPrMrQrRr!shapezerosintnanrNrwriteoutdirrE)!r< original_maskrVrWrTZ LowLat_s1Z LowLon_s1Z HighLat_s1Z HighLon_s1s1Zs1_XYZLowX_s1ZLowY_s1ZHighX_s1ZHighY_s1Z LowLat_s2Z LowLon_s2Z HighLat_s2Z HighLon_s2s2Zs2_XYZLowX_s2ZLowY_s2ZHighX_s2ZHighY_s2rowcolmZ mask_s123Z mask_s123_tmpZ swath1_indiZ swath2_indiidx mask_swathsr6r6r7generate_swaths_mask{sD  ((   rhcCs*|}t|d}|dd}t|}|S)z read h5 file generated by matlabrz/modelr6)h5pyFilerLrM)filesZ model_fileZ model_idxdata model_datar6r6r7 read_model_h5s    rocCs|j\|_|_tj|j|jftd}|dd}|dd}|jr-t|j}|dd}n't|j d}t|j d}t|j d} ||dk|||dk|| }||7}|j rt|j d} t|j d} t|j d} t|j d} t|j d}t|j d}t|j d }t|j d }t|j d }tj|j|jftddd}|||dk| ||dk| | 7}|||dk| ||dk||7}|||dk|||dk||7}n't|j d}t|j d}t|j d}||dk|||dk||}||7}|j r1t|jd|_|jd|jd|jf|_t|j||jrkt|j|_t|j}t|||jj\}}||jksX||jkr\td |jdd|_|j|7_|jrt|jd  d}t|jd d}|t||||}|t||||}|t||||}|jr|t|j||||_t||||dddn t||||dd|jrt||||dddSt||||dS)z+generate linear ramp for the synthetic datarAr?r rr@rr zDError! The dimension of modeled data and simulated ramp is disagree! WAVELENGTHyes)model_data_ramp wrap_flag)rw)rvN)r[rdrerLr\r4rNrrorZsubswathr"rZatmor- atmo_datawrite_atmo_datarrncopydeepcopywrite_model_data ExceptionwraprEmodmodel_data_wrap write_data)rgrVrWr< data_rampZmask_swaths_flat data_tramprrrr$r%r&r'r(r)r*r+r, data_tiramprrrmodel_data_copyZ row_modelZ colm_modelvminvmaxZdata_ramp_wrapZdata_tramp_wrapZdata_tiramp_wrapr6r6r7generate_linear_rampsv   $((*$     rc Csd|jd<|r d|jd<nd|jd<||j|j}||j|j}||j|j}|rC|r9|j|j|j|_n |j|j|j|_tj||jdk<tj||jdk<tj||jdk<|rr|ritj|j|jdk<n tj|j|jdk<|r|j dd}|j dd }|j dd }|r|j d|j d dd } n&|j dd }|j dd}|j dd}|r|j d|j d dd} t j |||jdt j |||jdt j |||jd|r|rt j |j| |jddSt j |j| |jddSdS)z write datamUNITr~ FILE_TYPEz.unwFrztectonic_ramp_wrap.h5z#troposphere_ionosphere_ramp_wrap.h5zdata_ramped_wrap.h5.z_ramped_wrap.h5ztectonic_ramp.h5ztroposphere_ionosphere_ramp.h5zdata_ramped.h5z _ramped.h5rYN)rErNrdrerrnrLr^rKr`rsplitrr_) r<rrrrvrwZ tramp_nameZ tiramp_nameZ ramp_nameZmodel_ramp_namer6r6r7rsL   rcCstj||jdk<|jd|jddd}tj|||jdt |jd d}t |jdd}|t ||||}|jd|jddd}tj|||jdd S) z,change simulated model data to mintpy formatFrr _mintpy.h5rYrtr z_mintpy_wrap.h5N) rLr^rKr`rrrr_rEr4r)rr<outfilerrZmodel_data_copy_wrapr6r6r7r|Nsr|cCsHtj||jdk<|jd|jdddd}tj|||jddS)z.change atmospheric noise data to mintpy formatFrrrrYN) rLr^rKr`r-rrr_rE)rxr<rr6r6r7ryZs"ryc Cst|}t|\}}}t|jdd}tt|g}tt|g}|jrU|j rD|j \}}tj ||ft d}tj ||jdk<nt|||||}t||||dS|jrit|j} t| } t| |dSdS)NrrAF)r=rXrrJr rLrMrPramprr[onesr4r^rKrhrrrorzr{r|) r;r<rVrWrTrardrergrnrr6r6r7main`s"   r__main__)N)NN)osr.numpyrLrzrjmintpy mintpy.utilsrrrrF mimtpy.utilsrrQr1r8r=rXrhrorrr|ryr__name__r6r6r6r7s,  -#: ^2