o BdAC@sddlZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z mZmZddZddd Zdd d Zd d ZddZdddZdS)N)IFGRAM_DSET_NAMESclustergiantTimeseries ifgramStack timeseries)ptimereadfile time_func writefilec Cs|dd}|dd}||krd}n|}|dd|ddg}|dd|ddg}||kr6d\}}n|\}}|durBt|nd}|durLt|nd}|||fS)zCheck reference date and point Parameters: atr1/2 - dict, metadata of file1/2 Returns: ref_date - str, None for re-referencing in time is NOT needed ref_y/x - int, None for re-referencing in space is NOT needed REF_DATENREF_YREF_X)NN)getint) atr1atr2Z ref_date1Z ref_date2ref_dateZref_yx1Zref_yx2ref_yref_xrD/home/exouser/operations/rsmas_insar/tools/MintPy/src/mintpy/diff.pycheck_references   rFקAc s>t|}t|}|d}|d}t||dkr%t|d} n |dkr1t|d} t||\} } } fddD} tjttj d}| krt d |d ||rt t t | }t d t d |d |t fdd|D<ntd| r| r| | | d| df}tj|| |dd | }nd}tj||dt|dt|d}}ttt|||}tjd d ||f|ddd\}}t|D]\}}|dkrt d|dd|dt d|t d|tj|| |dd | }|dur2t dtj|d| d | |t|d!ddd|jd|jd"f8}| r_t dtj|d#| | | }|t||ddddf|jd ddf8}t d|tj||d$d }|d%k}|||8<d%||<~d |jd |d|d&|d |d"g}tj||||d'q|S)(aCalculate the difference between two time-series files. Parameters: file1 - str, path of file1 file2 - str, path of file2 out_file - str, path of output file force_diff - bool, overwrite existing output file max_num_pixel - float, maximum number of pixels for each block Returns: out_file - str, path of output file FILE_TYPErg?rgMbP?cg|]}|vr|qSrr.0i) date_list2rr Vz#diff_timeseries..dtypez WARNING: z does not contain all dates in zBContinue and enforce the differencing for their shared dates only.z4 with following dates are ignored for differencing: rcsg|]}|qSr)indexr) date_list1rrr^z7To enforce the differencing anyway, use --force option. datasetNameboxNref_fileLENGTHWIDTHyTr)Z num_split dimension print_msg ------- processing patch  out of  --------------box: zread from file: * referencing data from to y/x: / to date: r)datar(block)rread_attributer get_date_listrrnponeslenbool_printlistsetarray Exceptionreadr layout_hdf5rceilrsplit_box2sub_boxes enumerateospathbasenametilereshapeshaper#write_hdf5_block)file1file2out_file force_diff max_num_pixelrrk1k2Zunit_facrrrZdate_list_sharedZdate_flag_sharedZ date_list_exref_boxref_vallengthwidthnum_boxbox_listrr)data2ref_indr@maskrAr)r$rrdiff_timeseries:sr       ", .&ric! Cst|}t|}t|}t|}t||\}} } | r5| r5| | | d| df} tj|d| dd} nd} t|} dd| D} d| vrOtd|d | dgkrat d t d d| Dt j ||d t |d t |d}}t t t||||}tjdd||f|ddd\}}t|D]\}}|d|d}|d|d}||}|dkrt d|dd|dt d|t d|tj|d|dd}| durt d| d| d| ddd|| 8}ddi}tj||d }t d!tj|d"|t t ||g}t ||}|d#||}d$|vrUt d%tj|d&|d$||d$}|t ||ddddf|ddf8}t d'|tj||d(d}|d)k}||8}d)||<~d||d|d|d|dg} t j||d*| d+q|S),aCalculate the difference between a time-series file and a velocity file. Parameters: file1 - str, path of file1 (time series) file2 - str, path of file2 (velocity) out_file - str, path of output file max_num_pixel - float, maximum number of pixels for each block Returns: out_file - str, path of output file r&velocityr'rNcSsg|] }|ds|qS)Std)endswithrxrrrrsz0diff_timeseries_and_velocity..z$No velocity dataset found in file2: !zPWARNING: ONLY velocity is supported, ignore the following datasets and continue:cSsg|]}|dkr|qS)rjrrmrrrrr r*r,r-r.Tr/r:r>r2r3r4r5zread velocity from file2: z* referencing velocity to y/x: r8z with value of dz.2fz cm/year polynomial)modelz"* reconstructing time-series from z with model r9r z&* referencing time-series from file2: r;zread time-series from file1: r<r=rr?) rrBrrCrFrrMget_dataset_list ValueErrorrHr rNrrDrOrrPrQr Zget_design_matrix4time_funcrRrSrTvstackzerosflattenmatmulrVkeysr#rUrX)!rYrZr[r]rr date_listnum_date_rrr`rads_namesrbrcrdrerr)box_widbox_len num_pixelZvelorrZG_fitmZts_fitrfrgr@rhrArrrdiff_timeseries_and_velocitysp      "  ( rc st|}|t|}|tt|jt|j@tdkr&tdfddtDd}td|d|dt j ||dd}td|d|dt j ||dd}d | vrtd |j d |j d |d d |j |j f}|d d |j |j f} t|jdD]2} || d d f|| d d fdk|| 8<|| d d f|| d d fdk| | 8<qtj||dk<tj||dk<||} ~~d| t| <|| i} tj| ||d|S)aCalculate the difference between two ifgramStack files. Parameters: file1 - str, path of file1 file2 - str, path of file2 out_file - str, path of output file Returns: out_file - str, path of output file rz$no common dataset between two files!crrrrr}rrr r z%diff_ifgram_stack..zreading z from file  ...r(Z unwrapphasezreferencing to pixel (,z) ...Nr=)r[r+)ropenrIrJ datasetNamesrFrtrrHrrMlowerrefYrefXrangerWrDnanisnanr write) rYrZr[Zobj1obj2ds_namedata1rfZref1Zref2rr@ds_dictrrrdiff_ifgram_stacks6   02rc Cszt|}|dd}|drdnd}td|d|tj||d\}}t|d d \}}td |d |d |d|tj||dd } | tj||dd 8} |drqtd|d| dt j t |d9} || 8}d| vrt |dt |d} } ||| | f8}td| d| d|ddkrtd|tj|ddd } | |d} n|} td|t| ||dD]q}|d|}|d|}tj|r:t||td|d |d!|d|}t| }|}Wd"n 1s wY|tj|tj|}t|d# }||Wd"n 1s5wYq|S)$asCalculate the difference between two unwrapped interferogram files. Parameters: unw_file - str, path of the interferogram file ts_file - str, path of the time-series file, e.g. ERA5.h5, SET.h5 cor_file - str, path of output corrected interferogram file Returns: cor_file - str, path of output corrected interferogram file UNITradianZradphasezread z from rDATE12-zcalc z for rzconvert z from radian to meterg WAVELENGTHr r zre-referencing to pixel (z, )r.unwzread magnitude from magnitude)rrzwrite corrected data to )xmlZvrt.zcopy z to z and update the filenameNw)rrBr startswithrHrMryyyymmddsplitrDpifloatryrr rrRrSisfileshutilcopy2rreplacerT)unw_filets_filecor_fileatrdunitZdnamer@date1date2delayrrmagrextZ unw_meta_fileZ cor_meta_fileZ meta_filefsrrrdiff_ifgram_and_timeseries+sT            rc Cst}t|d|d|t|}t|d}|d}|d} td||dkrX| dvr;tdtd | d vrJt||d|||n| d krWt||d||ntd d || fDrlt||d|n|dvr~| dkr~t ||d|dndd|g|D} t t j t t | } tdd | Dr| d} td| t| dkrtd|g|i} | D]} td| dtj|| dd}|j}t|D]\\}}t| |ddkrdn| }tj||dd}| d krt||dd\}}|r|rtdtj|dd|d|||||f8}tj|tjdtj|tjd}qtj||d| | <qtd|tj| |||dtt|d \}}td!|d"d#|d$d%|S)&a_calculate/write file1 - file2 Parameters: file1 - str, path of file1 file2 - list(str), path of file2(s) out_file - str, path of output file force_diff - bool, overwrite existing output file max_num_pixel - float, maximum number of pixels for each block z - z --> rrzthe 1st input file is: r)rrrjzZIf the first file is timeseries, the following file must be either timeseries or velocity.z8Input multiple dataset files are not the same file type!)rrrjcss|]}|dkVqdS)rNrrrrr szdiff_file..)r displacementz.off)rrrcSsg|]}t|qSr)rrsrmrrrrr%zdiff_file..css|] }t|dkVqdS)r&N)rFrmrrrrsz&List of common datasets across files: r&z&No common datasets found among files: z differencing rrNr6r7r8r!z use metadata from the 1st file: )r[metadatar+<z time used: z02.0fz mins z02.1fz secs)timerHrrBrLrirallrrrIrJ intersectionmaprFrtrMr"rQrrRrSrTrDrKfloat32r rdivmod)rYrZr[r\r] start_timerrr^r_ ds_names_listr}dsDictrr@r"rfname ds_name2readrfrrrrrrr diff_filens^      &"r)Fr)r)rRrrnumpyrDmintpy.objectsrrrrr mintpy.utilsrrr r rrirrrrrrrrs ! ^d/C