B Qaq@sddlZddlZddlZddlZddlmZddlmZmZ ddgZ dZ ddZ dd d Z d d Zdd dZedkreejdddS)N) RAMP_LIST)readfileutilsz mintpy.derampzmintpy.deramp.maskFilezexample: remove_ramp.py timeseries.h5 -m maskTempCoh.h5 remove_ramp.py ifgramStack.h5 -m maskTempCoh.h5 -d unwrapPhase_bridging remove_ramp.py 090214_101120.unw -m maskTempCoh.h5 -s quadratic cCstjdtjtd}|jddd|jdddd d d |jd d dtdd|jddddd|jdddd|jddddd|jdddd d|S)!NzRemove phase ramp) descriptionformatter_classepilogfilezFile for ramp removal)helpz-mz--mask mask_filezmaskTempCoh.h5zXmask for pixels used in ramp estimation default - maskTempCoh.h5 no - use the whole area)destdefaultr z-s surface_typelinearz1type of surface/ramp to remove, linear by default)r r choicesr z-dz--dsetdsetz]dataset name to be derampped in ifgramStack file e.g.: unwrapPhase unwrapPhase_bridging)r r z-oz --outfilezOutput file name.z--save-ramp-coeffsave_ramp_coeff store_truez4Save the estimated ramp coefficients into text file.)r actionr z--update update_modezEnable update mode, and skip inversion if: 1) output file already exists, readable and newer than input file 2) all configuration parameters are the same.)argparseArgumentParserRawTextHelpFormatterEXAMPLE add_argumentr)parserrH/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/remove_ramp.py create_parsers      rcCs2t}|j|d}|jr.|js.d|_td|S)N)argsFzRupdate_mode is chosen but NOT turned on because the required --outfile is missing.)r parse_argsroutfilewarningswarn)iargsrinpsrrrcmd_line_parse:s    r%cs(tdtdd}tj|js8d}td|jnttd|j|jg}|jrb||jt dd|D}tj |j}||krd}td |ntd ||dkri|j d <|jd <t |jtfd dtDrd}tdtntdttd||S)Nz2--------------------------------------------------zupdate mode: ONskiprunz1) output file {} NOT found.z!1) output file {} already exists.css|]}tj|VqdS)N)ospathgetmtime).0irrr Sszrun_or_skip..z02) output file is NOT newer than input file: {}.z,2) output file is newer than input file: {}.z mintpy.derampzmintpy.deramp.maskFilec3s&|]}t||dkVqdS)NoneN)strget)r+key)atriDictrrr-asz73) NOT all key configuration parameters are the same:{}z33) all key configuration parameters are the same:{}zrun or skip: {}.)printr(r)isfiler formatrr appendmaxr*r rread_attributeany configKeys)r$flaginfilestitor)r2r3r run_or_skipEs6     r@cCsxt|}|jr t|dkr |jStj|j|j|j|j|j |j d}t d t i}|j|d<|j|d<t||dS)Nr&) ramp_typer out_file datasetNamerz;add/update the following configuration metadata to file: {}z mintpy.derampzmintpy.deramp.maskFile)r%rr@r ut run_deramprr r rrr4r6r; add_attribute)r#r$rBatr_newrrrmainms    rH__main__)N)N)r(sysrr!mintpy.objectsr mintpy.utilsrrrDr;rrr%r@rH__name__argvrrrr s   (