U c7;@sddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z dZdddZdddZd d Zd d Zdd dZedkreejdddS)N)readfile writefileutils)create_argument_parsera[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 # exclude 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 # exclude 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" c Csd}t}tdd}t|||||d}|jddd|jdd d d |jd d ddd|jddddd|jdddd|jdddtdd|jddd td!d|jd"d#d$d%td&d|d'd(}|jd)d*d+td,d-d.|jd/d0d+td1d2d.|jd3dd4d|jd5dd6d|jd7d8td+d9d:|d;d<}|jd=d>d?td+d,d@dA|jdBdCdDtd+d1dEdA|jdFdGdHtdIdJd.|jdKdLdHtdIdMd.|jdNdOtdPd|jdQdRdStdTd|jdUdVtdWdXd:|jdYddZd|jd[d\td]d|jd^d_dd`d|jdadbddcd|S)dNz"Generate mask file from input file.)synopsis descriptionepilog subparsersfilez 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.zAOI for thresholdz+Define the AOI for thresholding operations.z--vx v_subset_x)XMINXMAXzEAOI range in X for threshold operation (and keep the rest untouched.))rrrmetavarr z--vy v_subset_y)YMINYMAXzEAOI range in Y for threshold operation (and keep the rest untouched.)z --vroipolyzAAOI via interactive polygonal region of interest (ROI) selection.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-xz--sub-xsubset_xz0selection range in x/cross-track/range direction)rrrrr z-yz--sub-ysubset_yz2selection range in y/along-track/azimuth directionz --ex-circle ex_circle)XYZRADIUSz@exclude area defined by an circle (x, y, radius) in pixel numberz --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.) EXAMPLE__name__splitr add_argumentfloatintadd_argument_groupstr)r rr nameparservmaskaoir=K/home/exouser/operations/rsmas_insar/sources/MintPy/mintpy/generate_mask.py create_parser8s             r?cCstt}|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) r? parse_argsr+r,rget_dataset_listformatwarningswarnprint)iargsr:inpsZbase_dataset_listmsgr=r=r>cmd_line_parses     rJc Cstdtdd}tj|js8d}td|jntd|jt|jd(}t ||j j dtj |j}W5QRXtj |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: {}.)rFospathisfilerrCh5pyFiler r5rattrsgetgetmtime)rHflagftitor=r=r> run_or_skips ,r[cCs|jrtd|j|jfntd|jtj|j|jd\}}t|jdkrVtdt|dt|d}}t |}t j ||ft j d}|j rt|j \}}d |ddd|f<d |dd|df<|jr t|j\} } d |d| ddf<d || dddf<|jrJd d lm} | j|j|j|jd } | dk rJd || d k<td t j ||ft j d} |js~| |9} td |jrd | |dk<td|jdk r| ||||jk9<tdt|j|jdk r| ||||jk9<tdt|j|jdk r\d dlm}t | }|| |jdd} td|j|t | f|jr|ddkrztdtj|jddd }| |t |||j!||k9<td"|j!|j#dk rt|j#\}}d | d |ddf<d | ||ddf<td||f|j$dk rjt|j$\}}d | ddd |f<d | dd||f<td||f|j%r|j%\}}}t&'|||||f}d | |dk<td"||||j(r|j(\}}}t&'|||||f}d | |d k<td"||||j)r2d d lm} | j|j|j|jd } | dk r2| | 9} |j*rtj|j*|j+dd }t|jdkrnt j|d d }d | |t,|j-k<d!"t.j/0|j*}|j+r|d""|j+7}|d#"|j-7}t||j1rt j2| t j d}d$| |d%k<d%| |d$k<~d&|d<t3j4| |j5|d'|j5S)(Nz read %s %szread %s) datasetNamerz>Only 2D dataset is supported for threshold method, input is 3DLENGTHWIDTH)dtyper)plot_ext)r\r.zDcreate initial mask with the same size as the input file and all = 1zall pixels with nan value = 0gzexclude pixels with zero valuezexclude pixels with value < %szexclude pixels with value > %s)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={})r')axiszexclude pixels in base file {} z dataset {} zwith value == {}FTmask)out_filemetadata)6rrFr rreadlenshape Exceptionr6npisnanonesbool_rsortedrZvroipoly mintpy.utilsr`Z get_poly_maskr.rr/rr8rrZskimage.morphologyrasumZvstd ValueErrorabsr"rCr%r$r&utget_circular_maskr*roipolyr+r,r5r-rOrPbasenamerevertarrayrwriter)rHdataatrlengthwidthnanmaskr;Zvx0Zvx1Zvy0Zvy1r`Z poly_maskrgra num_pixelZdata_stdy0y1x0x1xyrMcmaskZ base_datarItempr=r=r>create_threshold_masks          (          rc sht}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|_W5QRX|jr|jrt|dkr|jStj|j|j|jd|_|jSt|tt|d\}}td||dS)Nrdzinput {} file: {}z maskPoly.h5temporalCoherencerbZ maskTempCohzmask.h5Zgeo_ztemporalCoherence.h5gffffff? ifgramStackrMcsg|]}|kr|qSr=)keys).0irXr=r> Ns zmain..)connectComponent unwrapPhaserrK)rhr\<z'time used: {:02.0f} mins {:02.1f} secs.)timerJrread_attributer rFrCrryr3 startswithrendswithr/rrRrSr0r[rw nonzero_maskrdivmod)rG start_timerHrksuffixmsr=rr>main2s8     "r__main__rb)N)N)N)rOsysrrDrRnumpyrnrsrrrrwmintpy.utils.arg_utilsrr1r?rJr[rrr2argvr=r=r=r> s  $ L  0