U Qaq@sddlZddlZddlZddlZddlmZddlmZmZ ddgZ dZ ddZ dd d Z d d Zdd dZedkreejdddS)N) RAMP_LIST)readfileutils mintpy.derampmintpy.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 rchoicesr 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: {}.rrc3s&|]}t||dkVqdS)NoneN)strget)r-keyatriDictrrr/asz73) NOT all key configuration parameters are the same:{}z33) all key configuration parameters are the same:{}zrun or skip: {}.)printr*r+isfiler"formatr r appendmaxr,rrread_attributeany configKeys)r&flaginfilestitorr4r run_or_skipEs6     rCcCsxt|}|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: {}rr)r'rrCr"ut run_derampr rr rrr7r9r> add_attribute)r%r&rEatr_newrrrmainms    rK__main__)N)N)r*sysrr#mintpy.objectsr mintpy.utilsrrrGr>rrr'rCrK__name__argvrrrr s  (