B Qa/6@sddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z dZ ddZdddZdd Zd d Zdd d ZedkreejdddS)N)readfile writefileutilsa[example: generate_mask.py temporalCoherence.h5 -m 0.7 -o maskTempCoh.h5 generate_mask.py temporalCoherence.h5 -m 0.7 -o maskTempCoh.h5 --base inputs/geometryRadar.h5 --base-dset shadow --base-value 1 generate_mask.py avgSpatialCoh.h5 -m 0.7 --base waterMask.h5 -o maskSpatialCoh.h5 # exlcude area by min/max value and/or subset in row/col direction generate_mask.py 081018_090118.unw -m 3 -M 8 -y 100 700 -x 200 800 -o mask_1.h5 # exlcude pixel cluster based on minimum number of pixels generate_mask.py maskTempCoh.h5 -p 10 mask_1.h5 # exclude pixels with large velocity STD: |velocity| > cutoff (2 by default) * velocityStd generate_mask.py velocity.h5 --vstd generate_mask.py velocity.h5 --vstd --vstd-num 3 # exclude / include an circular area generate_mask.py maskTempCoh.h5 -m 0.5 --ex-circle 230 283 100 -o maskTempCoh_nonDef.h5 generate_mask.py maskTempCoh.h5 -m 0.5 --in-circle 230 283 100 -o maskTempCoh_Def.h5 # maskout an area within a circle AND with height smaller than a threshold generate_mask.py inputs/geometryGeo.h5 height --in-circle 339 370 21 -M 1400 --revert -o maskCrater.h5 # use an specific dataset from multiple dataset file generate_mask.py geometryRadar.dem height -m 0.5 -o waterMask.h5 generate_mask.py ifgramStack.h5 unwrapPhase-20101120_20110220 -m 4 # common mask file of pixels in all connected components / with non-zero unwrapped phase generate_mask.py ifgramStack.h5 --nonzero -o maskConnComp.h5 --update # interative polygon selection of region of interest # useful for custom mask generation in unwrap error correction with bridging generate_mask.py waterMask.h5 -m 0.5 --roipoly generate_mask.py azOff.h5 --roipoly --view-cmd "-v -0.1 0.1" generate_mask.py velocity.h5 --roipoly --view-cmd "--dem ./inputs/geometryGeo.h5 --contour-step 100 --contour-smooth 0.0" cCstjdtjtd}|jddd|jdddd |jd d d d d|jddddd|jdddd|jdddtdd|jdddtdd|jd d!d"d#td$d|jd%dd&d|jd'd(td)d*d+|d,d-}|jd.d/td)d0d1d2|jd3d4td)d5d6d2|jd7d8d9td:d;d<|jd=d>d9td:d?d<|jd@dAtdBd|jdCdDdEtdFd|jdGdHtdIdJd+|jdKddLd|jdMdNtdOd|jdPdQddRd|jdSdTddUd|S)VNz"Generate mask file from input file) descriptionformatter_classepilogfilez input file)helpdset?z?date of timeseries, or date12 of interferograms to be converted)nargsr z-oz--outputoutfilezoutput file name.)destr z --keep-nankeep_nan store_truez$Do not exclude pixels with NaN value)ractionr z--revertz(revert 0 and 1 value of output mask file)rr z-mz--minvminz!minimum value for selected pixels)rtyper z-Mz--maxvmaxz!maximum value for selected pixelsz-pz--mpz --minpixels minpixelsz?minimum cluster size in pixels, to remove small pixel clusters.z--vstdz;mask according to the formula: |velocity| > a * velocityStdz --vstd-numvstd_numz-multiple of velocityStd (a) to use for cutoff)rrdefaultr ZAOIz!define secondary area of interestz-xsubset_x)XMINXMAXz0selection range in x/cross-track/range direction)rrr metavarr z-ysubset_y)YMINYMAXz2selection range in y/along-track/azimuth directionz --ex-circle ex_circle)XYZRADIUSz@exclude area defined by an circle (x, y, radius) in pixel number)rr rrr z --in-circle in_circlez@include area defined by an circle (x, y, radius) in pixel numberz--base base_filezEexclude pixels == base_value output_mask[base_data == base_value] = 0z --base-dsetz--base-dataset base_datasetzdataset in base_file to be used, for file with multiple datasets. i.e.: --base inputs/geometryRadar.h5 --base-dset shadow --base-value 1z --base-value base_valuerz7value of pixels in base_file to be excluded. Default: 0z --roipolyz9Interactive polygonal region of interest (ROI) selection.z --view-cmdview_cmdzBview.py command to facilitate the AOI selection.E.g. "-v -0.1 0.1"z --nonzerononzerozDSelect all non-zero pixels. i.e. maskConnComp.h5 from ifgramStack.h5z--update update_modez,Enable update checking for --nonzero option.) argparseArgumentParserRawTextHelpFormatterEXAMPLE add_argumentfloatintadd_argument_groupstr)parserZaoir5J/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/generate_mask.py create_parser8sZ            r7cCstt}|j|d}|jrp|jrpt|j}|j|krpd|j|j}|d|7}t|t dd|_d|_|S)N)argsz*dataset {} NOT found in input base file {}z available datasets: {}z-ignore --base --base-dset option and continue) r7 parse_argsr%r&rget_dataset_listformatwarningswarnprint)iargsr4inpsZbase_dataset_listmsgr5r5r6cmd_line_parseys     rBc Cstdtdd}tj|js8d}td|jntd|jt|jd(}t ||j j dtj |j}WdQRXtj |j}||krd}td |j ntd |j td ||S) Nz2--------------------------------------------------zupdate mode: ONskiprunz1) output file {} NOT exist.z!1) output file {} already exists.rMODIFICATION_TIMEz32) output file is NOT newer than input dataset: {}.z/2) output file is newer than input dataset: {}.zrun or skip: {}.)r>ospathisfiler r;h5pyFilerr0r attrsgetgetmtime)r@flagftitor5r5r6 run_or_skips ,rScCsR|jrtd|j|jfntd|jtj|j|jd\}}t|jdkrVtdt|dt|d}}t |}tdt j ||ft j d }|j s||9}td |jr||||d k9<td |jdk r|||||jk9<td t|j|jdk rJ|||||jk9<tdt|j|jdk rddlm}t |}|||jdd}td|j|t |f|jr|ddkrtdtj|jddd} ||t |||j| |k9<td|j|jdk rTt|j\} } d|d| ddf<d|| |ddf<td| | f|jdk rt|j\} } d|ddd| f<d|dd| |f<td| | f|jr|j\}}}t |||||f}d||dk<td||||j!r2|j!\}}}t |||||f}d||dk<td||||j"rnddl#m$}|j%|j|j|j&d}|dk rn||9}|j'rtj|j'|j(dd}t|jdkrt j|dd }d||t)|j*k<d!t+j,-|j'}|j(r|d"|j(7}|d#|j*7}t||j.r2t j/|t j d }d$||d%k<d%||d$k<~d&|d<t0j1||j2|d'|j2S)(Nz read %s %szread %s) datasetNamerz>Only 2D dataset is supported for threshold method, input is 3DLENGTHWIDTHzDcreate initial mask with the same size as the input file and all = 1)dtypezall pixels with nan value = 0gzexclude pixels with zero valuezexclude pixels with value < %szexclude pixels with value > %sr)remove_small_objects) connectivityz>exclude pixel clusters with size < %d pixels: remove %d pixels FILE_TYPEvelocityz@Input file MUST be a velocity file when using the --vstd option!Z velocityStdzFexclude pixels according to the formula: |velocity| > {} * velocityStdz%exclude pixels with y OUT of [%d, %d]z%exclude pixels with x OUT of [%d, %d]z=exclude pixels inside of circle defined as (x={}, y={}, r={})z>exclude pixels outside of circle defined as (x={}, y={}, r={}))plot_ext)rTr(r!)axiszexclude pixels in base file {} z dataset {} zwith value == {}FTmask)out_filemetadata)3r r>rrreadlenshape Exceptionr1npisnanonesbool_rr)rr3rrZskimage.morphologyrXsumZvstd ValueErrorabsrr;rsortedrr utget_circular_maskr$roipoly mintpy.utilsr]Z get_poly_maskr(r%r&r0r'rGrHbasenameZrevertarrayrwriter )r@dataatrlengthwidthnanmaskr_rX num_pixelZdata_stdy0y1x0x1xyrEcmaskr]Z poly_maskZ base_datarAtempr5r5r6create_threshold_masks      (          rc slt}t|}t|j}|d}td||j|js|jrJd|_n,d|jkrp|j dd}d||_nd|_|j drd|j|_|j dkr|j d rd |_ |j r6|d kr6|jst|jd fd ddDd|_WdQRX|jr|jrt|dkr|jStj|j|j|jd|_|jSt||_tt|d\}}td|||jS)Nr[zinput {} file: {}z maskPoly.h5temporalCoherencerYZ maskTempCohzmask.h5Zgeo_ztemporalCoherence.h5gffffff? ifgramStackrEcsg|]}|kr|qSr5)keys).0i)rPr5r6 3szmain..)connectComponent unwrapPhaserrC)r`rT<z'time used: {:02.0f} mins {:02.1f} secs.)timerBrread_attributerr>r;r rpsplit startswithrendswithr)r rJrKr*rSrn nonzero_maskrdivmod)r? start_timer@rvksuffixmsr5)rPr6mains8     " r__main__rY)N)N)rGsysrr<r+rJnumpyrfrqrrrrnr.r7rBrSrr__name__argvr5r5r5r6 s%A r /