B Qa6*@sddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZdZddZdddZd d Zdd dZdddZedkreejdddS)N)cluster timeseriesgiantTimeseries ifgramStackifgramDatasetNames)readfile writefileaexample: 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 cCsVtjdtjtd}|jddd|jdddd |jd d d d d|jdddd|S)Nz,Generates the difference of two input files.) descriptionformatter_classepilogfile1zfile to be substracted.)helpfile2+zfile used to substract)nargsr z-oz--outputoutfilez0output file name, default is file1_diff_file2.h5)destr z--force store_truezHEnforce the differencing for the shared dates only for time-series files)actionr )argparseArgumentParserRawTextHelpFormatterEXAMPLE add_argument)parserrA/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/diff.py create_parser&s rcCsNt}|j|d}t|j}|ddkrJt|jdkrJtd|d|S)N)args FILE_TYPE)rrz,ERROR: only one file2 is inputed for {} type) r parse_argsrread_attributer lenr SystemExitformat)iargsrinpsatrrrrcmd_line_parse4s   r)cCs|d|ddkrd}n|d}|dd}|dd}||ddksZ||ddkrdd}d}n|}|}|dk r|t|}|dk rt|}|||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)getint)atr1atr2ref_dateref_yref_xrrrcheck_referenceAs    r4FקAc* st}|sVtj|\}}t|dkr0tdd|tjtj|dd|}td|||t |}|d} t |d} | d} td| | | dkr$| d krt d t|dkrt d t|dt |}t |d} t | | dkr t |d d } n| d kr>t|d d} t|| \} }}fddD}tjttjd}|krtd|||rttt|}tdtd|d|tfdd|D<nt dtj||dt|dt|d}}ttt|||}tjdd||f|ddd}|r|r|||d|df}t j|d||dd| }x t|D]\}}|dkrtd|d|td|td |dt j|d||dd| }|rF|rFtd!tj|d|||t|d"ddd|j d|j d#f8}| rtd$tj|d| |!| }|t||d%d%d%d%f|j dddf8}td |t j||d&d}|d'k}|||8<d'||<~d|j d|d|d(|d|d#g}tj"||| |d)qWnlt#d*d+| | gDr t$|} | %t$|d}!|!%tt| j&t|!j&@tdkrtd,fd-dt'Dd}"td.|"|t j||"d/d}#td.|"|dt j|d|"d/d}d0|"(krtd1| j)| j*|#d%d%| j)| j*f}$|d%d%|!j)|!j*f}%xxt+|#j dD]f}|#|d%d%f|#|d%d%fd'k|$|8<||d%d%f||d%d%fd'k|%|8<qNWtj,|#|#dk<tj,||dk<|#|}~#~d'|t-|<i}&||&|"<tj.|&||d2nt |d}#t|#|#j/}xJ|D]B}'t |'d}tj|tj0dtj|tj0d}t||#j/}q.Wtd3|tj.|||d4t1t|d5\}(})td6|(|)|S)7aJcalculate/write file1 - file2 Parameters: file1 - str, path of file1 file2 - list of str, path of file2(s) out_file - str, path of output file force - bool, overwrite existing output file max_num_pixel - float, maximum number of pixels for each block r z7Output file name is needed for more than 2 files input.z {}_diff_{}{}rz{} - {} --> {}rzinput files are: {} and {}r)rrz8Input multiple dataset files are not the same file type!zFOnly 2 files substraction is supported for time-series file, {} input.g?rgMbP?csg|]}|kr|qSrr).0i) dateList2rr 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)index)r6r7) dateList1rrr9sz7To enforce the differencing anyway, use --force option.)ref_fileLENGTHWIDTHyT)box num_split dimension print_msg) datasetNamerAz5 ------- processing patch {} out of {} --------------zbox: {}zread from file: {}z(* referencing data from {} to y/x: {}/{}z&* referencing data from {} to date: {}N)rAg)datarEblockcss|]}|dkVqdS)rNr)r6r7rrr szdiff_file..z$no common dataset between two files!csg|]}|kr|qSrr)r6r7)dsNamesrrr9szreading {} from file {} ...)rEZ unwrapphasez referencing to pixel ({},{}) ...)out_filer=z writing >>> )rMmetadata<z&time used: {:02.0f} mins {:02.1f} secs)2timeospathsplitextr# ValueErrorr%basenameprintrr" Exceptionr get_date_listrr4nponesbool_listsetarrayr layout_hdf5r.ceilrsplit_box2sub_boxesread enumeratetilereshapeshaper;write_hdf5_blockallropen datasetNamesrlowerrefYrefXrangenanisnanwriter:float32divmod)*r rrMforceZ max_num_pixel start_timefbasefextr/k1r0k2Zunit_facr1r2r3ZdateListSharedZ dateSharedZ dateListExlengthwidthnum_boxbox_listref_boxref_valr7rAdata2Zref_indrImaskrJZobj1obj2dsNameZdata1Zref1Zref2ZdsDictfnamemsr)r<r8rLr diff_filebs  &             , .& 06   rcCs(t|}t|j|j|j|jd|_|jS)N)rt)r)rr rrrt)r&r'rrrmainsr__main__r )N)NFr5)N)rQsysrPrnumpyrYmintpy.objectsrrrrr mintpy.utilsrrrrr)r4rr__name__argvrrrr s  ! '