U c0@sddlZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z ddlmZdZdddZddd Zd d ZdddZdddZedkreejdddS)N)cluster timeseriesgiantTimeseries ifgramStackifgramDatasetNames)readfile writefile)create_argument_parseraexample: diff.py velocity.h5 velocity_demErr.h5 diff.py timeseries.h5 inputs/ERA5.h5 -o timeseries_ERA5.h5 diff.py timeseries.h5 inputs/ERA5.h5 -o timeseries_ERA5.h5 --force diff.py timeseries_ERA5_ramp_demErr.h5 ../GIANT/Stack/LS-PARAMS.h5 -o mintpy_giant.h5 diff.py reconUnwrapIfgram.h5 ./inputs/ifgramStack.h5 -o diffUnwrapIfgram.h5 # multiple files diff.py waterMask.h5 maskSantiago.h5 maskFernandina.h5 -o maskIsabela.h5 cCspd}t}tdd}t|||||d}|jddd|jdd d d |jd d ddd|jdddddd|S)Nz+Generate the difference of two input files..)synopsis descriptionepilog subparsersfile1zfile to be subtracted.)helpfile2+zfile used to subtract)nargsrz-oz--outputout_filez0output file name, default is file1_diff_file2.h5)destrz--forcez --force-diff force_diff store_truezHEnforce the differencing for the shared dates only for time-series files)ractionr)EXAMPLE__name__splitr add_argument)rr rnameparserr B/home/exouser/operations/rsmas_insar/sources/MintPy/mintpy/diff.py create_parser&s&  r"cCst}|j|d}t|jd}|dkrHt|jdkrHtd|d|jst|jdkrdt dt j |j\}}t j t j |jdd}|d |||_|S) N)args FILE_TYPE)rrz%ERROR: ONLY ONE file2 is inputed for z type!z#--output is required for >=2 files!rZ_diff_)r" parse_argsrread_attributerlenr SystemExitr ValueErrorospathsplitextbasename)iargsrinpsftypeZfbase1fextZfbase2r r r!cmd_line_parse6s r3c Cs|dd}|dd}||kr&d}n|}|dd|ddg}|dd|ddg}||krld\}}n|\}}|dk rt|nd}|dk rt|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_xr r r!check_referenceLs   r>FקAc( s td|||t|}t|d}|d}|d}td||dkr|dkrbtdt|}t|d}t||dkrt|dd} n|d krt|dd } t||\} } } fd d D} t j t t j d }| krftd|||r^t tt| }tdtd|d|t fdd |D<ntdtj||dt|dt|d}}tt t |||}tjdd||f|ddd}| r| r| | | d| df}tj|d| |dd| }t|D]\}}|dkr@td|d|td|td|dtj|d| |dd| }| r| rtdtj|d| | |t |dddd|jd|jd f8}| rtd!tj|d| | | }|t ||d"d"d"d"f|jdddf8}td|tj||d#d}|d$k}|||8<d$||<~d|jd|d|d%|d|d g}tj||||d&q n|t d'd(||fDrt!|}|"t!|d}|"t t|j#t|j#@t dkr t$d)fd*d t%Dd}td+||tj||d,d} td+||dtj|d|d,d}d-|&kr0td.|j'|j(| d"d"|j'|j(f}!|d"d"|j'|j(f}"t)| jdD]f}| |d"d"f| |d"d"fd$k|!|8<||d"d"f||d"d"fd$k|"|8<qt j*| | dk<t j*||dk<| |}~ ~d$|t +|<i}#||#|<tj,|#||d/nd0d |g|D}$t tj-t.t|$t d1d(|$Dr|$dtd2t dkrt$d3|g|i}#D]}td4|tj||d,d}|j/}%t|D]\}}&t |$|ddkrRd"n|}'tj|&|'d,d}|d5krt||dd"\} } | r| rtdtj|d| | ||| | f8}t j|t j0d t j|t j0d }q0t j||%d |#|<qtd6|tj,|#|||d7|S)8a_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{} - {} --> {}rr$zthe 1st input file is: {}r)rrz8Input multiple dataset files are not the same file type!g?rgMbP?csg|]}|kr|qSr r .0i) dateList2r r! szdiff_file..)dtypez,WARNING: {} does not contain all dates in {}zBContinue and enforce the differencing for their shared dates only.z6 with following dates are ignored for differencing: {}csg|]}|qSr )indexr@) dateList1r r!rDsz7To enforce the differencing anyway, use --force option.)ref_fileLENGTHWIDTHyT)box num_split dimension print_msgr%) datasetNamerLz5 ------- processing patch {} out of {} --------------zbox: {}zread from file: {}z(* referencing data from {} to y/x: {}/{}r z&* referencing data from {} to date: {}N)rLg)datarPblockcss|]}|dkVqdS)rNr r@r r r! szdiff_file..z$no common dataset between two files!csg|]}|kr|qSr r r@)ds_namesr r!rDszreading {} from file {} ...)rPZ unwrapphasez referencing to pixel ({},{}) ...)rrHcSsg|]}t|qSr )rget_dataset_listrAxr r r!rDscss|]}t|dkVqdS)r%N)r(rXr r r!rUsz&List of common datasets across files: z(No common datasets found among files: {}zdifferencing {} ...velocityz"use metadata from the 1st file: {})rmetadatarH)1printformatrr' Exceptionr get_date_listrr>nponesr(bool_listsetarrayr layout_hdf5r8ceilrsplit_box2sub_boxesread enumerater+r,r.tilereshapeshaperFwrite_hdf5_blockallropen datasetNamesr*rlowerrefYrefXrangenanisnanwrite intersectionmaprEfloat32)(rrrrZ max_num_pixelr9r:k1k2Zunit_facr;r<r=ZdateListSharedZ dateSharedZ dateListExlengthwidthnum_boxbox_listref_boxref_valrBrLdata2Zref_indrSmaskrTobj1obj2ds_namedata1Zref1Zref2ZdsDictZ ds_names_listrEfnameZ ds_name2readr )rGrCrVr! diff_filems           , .& 04   $rcCsRt|}t}t|j|j|j|jdtt|d\}}td ||dS)N)rrrr<z&time used: {:02.0f} mins {:02.1f} secs) r3timerrrrrdivmodr\r])r/r0 start_timemsr r r!main#sr__main__r%)N)N)Fr?)N)r+sysrnumpyr`mintpy.objectsrrrrr mintpy.utilsrrmintpy.utils.arg_utilsr rr"r3r>rrrargvr r r r! s   ! 7