U bbUR@sddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZe dZdZdZdd Zd&d d Zd'd d ZddZddZddZd(ddZd)ddZd*ddZd+ddZddZd d!Zd,d"d#Ze d$kreej!d%ddS)-N) timeseries)get_template_content)readfile writefileutilsreference_pointanote: Reference value cannot be nan, thus, all selected reference point must be: a. non zero in mask, if mask is given b. non nan in data (stack) Priority: input reference_lat/lon input reference_y/x input selection_method existing REF_Y/X attributes (can be ignored by --force option) default selection methods: maxCoherence random The recommended reference pixel should meets the following criteria: 1) not in deforming areas 2) not in areas affected by strong atmospheric turbulence, such as ionospheric streaks 3) close but outside of deforming area of interest with similar elevation, to minimize the spatial correlation effect of atmosspheric delay, especially for shot-wavelength deformation (Chaussard et al., 2013; Morales-Rivera et al., 2016) 4) in high coherent area to minimize the decorrelation effect aexample: # for ifgramStack file, update metadata only # add --write-data to update data matrix value reference_point.py inputs/ifgramStack.h5 -t smallbaselineApp.cfg -c avgSpatialCoh.h5 reference_point.py inputs/ifgramStack.h5 --method manual reference_point.py inputs/ifgramStack.h5 --method random # for all the other files, update both metadata and data matrix value reference_point.py 091120_100407.unw -y 257 -x 151 -m Mask.h5 reference_point.py geo_velocity.h5 -l 34.45 -L -116.23 -m Mask.h5 cCs^tjdtjtdtdtd}|jdtdd|jddd d d |jd d ddd |jddtddd|jddddd|jdddd|jdddd|d}|jdd d!t d"d#|jd$d%d&t d'd#|jd(d)d*t d+d#|jd,d-d.t d/d#|jd0d1d2d3d |jd4d5d6d7d |jd8d9d:d;dd?t d@dAdB|jdCtdDdEdFgdGdH|S)INz%Reference to the same pixel in space. ) descriptionformatter_classepilogfilezfile to be referenced.)typehelpz-tz --template template_fileztemplate with reference info)destrz-mz--maskmaskFilez mask filez-oz --outfilezoutput file name (default: %(default)s). This option is diabled for ifgramStack file. None (default) for update data value directly without writing to a new file. )r defaultrz --write-data write_data store_truezU(option for ifgramStack file only) update data value, in addition to update metadata.)ractionrz--resetz>remove reference pixel information from attributes in the file)rrz--forcez,Enforce the re-selection of reference point.zinput coordinatesz-yz--rowref_yz&row/azimuth number of reference pixel)rr rz-xz--colref_xz&column/range number of reference pixelz-lz--latref_latzlatitude of reference pixelz-Lz--lonref_lonzlongitude of reference pixelz-rz --referencereference_filez$use reference/seed info of this filez--lookupz --lookup-file lookup_filez{Lookup table file from SAR to DEM, i.e. geomap_4rlks.trans Needed for radar coord input file with --lat/lon seeding option.z-cz --coherence coherenceFilezaverageSpatialCoherence.h5zRuse input coherence file to find the pixel with max coherence for reference pixel.)rrrz--min-coherence minCoherence333333?z>minimum coherence of reference pixel for max-coherence method.)rr rrz--method maxCoherencemanualrandomaZmethods to select reference pixel if not given in specific y/x or lat/lon: maxCoherence : select pixel with highest coherence value as reference point enabled when there is --coherence option input manual : display stack of input file and manually select reference point random : random select pixel as reference point )r choicesr) argparseArgumentParserRawTextHelpFormatterNOTETEMPLATEEXAMPLE add_argumentstradd_argument_groupintfloat)parsercoordr0L/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/reference_point.py create_parser:sh            r2cCsDt}|j|d}t|j}|ddkr2d|_n|jr@td|S)zCommand line parser.)args FILE_TYPE ifgramStackTz3--outfile is disabled for "ifgramStack" input file!)r2 parse_argsrread_attributer routfile SystemExit)iargsr.inpsatrr0r0r1cmd_line_parsems   r=cs.|stdg}t|}t|tdfddt|D}|D]:}|}|rJ|dkrp|||<qJ|dkrJt|||<qJd}|krֈ|}|r| dd d d}d d| d D\|_ |_ d }|kr*|}|r*| dd d d}d d| d D\|_ |_|S)zFRead seed/reference info from template file and update input namespacezmintpy.reference.cs g|]}|kr|qSr0)keys.0iprefixtemplater0r1 sz+read_template_file2inps..)rrrZyx[]cSsg|] }t|qSr0)r,r@r0r0r1rFs,ZlalocSsg|] }t|qSr0)r-r@r0r0r1rFs)r=varsr read_templateutcheck_template_auto_valuelistr-r?replacesplitrrrr)rr; inps_dictkey_listkeyvaluer0rCr1read_template_file2inpss6      rUcCs:t||d}t|t|kr2|t|k}ng}|S)z find nearest neighbour )npsqrtminabs)xtbasexstepdistindxr0r0r1nearests r`c Cs|stdg}t|j}|jsn|jdk rn|jt|ddkrn|jdk rn|jt|ddkrnt d|jS|ddkrd d t |jDd }nd}t j |j|d d dd }t t ||dk}t |dkrtd|jdk r|jdk r||j|jfdkrtdn~|jdkr@t|j||jd\|_|_n6|jdkr^t|\|_|_n|jdkrvt|||}|jdks|jdkrtdt||j|jd}|jst d|jt |t |j||_n|js|j|_|d}tj|jd}|dkr|j|jkrTt d|dkrt |jdd}|d} t!| j"d D]2} | | ddddf| | |j|jf8<qTt d|j#$|W5QRXnt |jd}||} t%| j"dkrt!| j"d D]2} | | ddddf| | |j|jf8<qn | dd| |j|jf8<t d|j#$|W5QRXnt d &|jt'|j\} }t%| j"dkrt!| j"d D]2} | | ddddf| | |j|jf8<qn| | |j|jf8} |$|t(j)| |j||jd!nd"d#g} t |j} i}| D]V}tj'|j|d$d } || krR| | |j|jf8} nt d%|d&| | ||<q|$|t(j)||j|d't *|j|j+g|jS)(ztSeed input file with option from input namespace Return output file name if succeed; otherwise, return None r>NREF_YiREF_XzFSAME reference pixel is already selected/saved in file, skip updating.r4r5cSsg|]}|dkr|qS)) unwrapPhase rangeOffset azimuthOffsetr0r@r0r0r1rFsz"reference_file..rTF) datasetName updateModeoutFilez6no pixel found with valid phase value in all datasets.z?reference y/x have nan value in some dataset. Please re-select.r)coh_filemaskmin_cohr!r zERROR: no reference y/x found.)yr[z&Add/update ref_x/y attribute to file: z.h5z4updating data value without re-writing to a new filezr+rczupdate metadataz)writing the referenced data into file: {})metadataref_filephase displacementrfzskip spatial referencing for z, as it's not in )out_filerp),r=rr7r forcerr,getrprintget_dataset_listrLtemporal_averagerWmultiplyisnannansum ValueErrormethodselect_max_coherence_yxrrrandom_select_reference_yxmanual_select_reference_yxreference_point_attributer add_attributer8ospathsplitexth5pyFilerangeshapeattrsupdatelenformatreadrwritetouchr)r;r<ds_namestackrkatrNewkfextfdsrBdataZ dis_namesds_namesZds_dictr0r0r1rs          02 2     rcCsdt}t||d<t||d<t|}d|kr`t|j|dd|d<t|j|dd|d<|S) NrarbX_FIRSTrm) coord_typeREF_LATr[REF_LON)dictr*rL coordinater?yx2lalo)r<rmr[rr/r0r0r1r0s    rcsddlm}tdtdtdtd|dk rBtj|dk<|}|d}|fd d }|j d |}| td j j fS) zManually select reference point in row/column number. Parameters: data : 2D np.ndarray, stack of input file inps : namespace, with key 'REF_X' and 'REF_Y', which will be updated mask : 2D np.ndarray r)pyplotz" Manual select reference point ...z9Click on a pixel that you want to choose as the refernce z& pixel in the time-series analysis;z-Then close the displayed window to continue. Nocs~|jdkrztdt|jd}t|jd}t||sbtdt||g|_|_ ntdtdtddS)Nrnclickg?zvalid input reference y/x: z WARNING:z(The selectd pixel has NaN value in data.z!Try a difference location please.) buttonrxr,xdataydatarWr|r*rr)eventr[rmrr;r0r1onclickSs z+manual_select_reference_yx..onclickbutton_press_eventy/x: {}) matplotlibrrxrWnanfigure add_subplotimshowcanvas mpl_connectshowrrr)rr;rkpltfigaxrcidr0rr1r<s   rrc Cstd|td|t|\}}|dk r min_coh in randomz'random select pixel with coherence > {}z based on coherence file: Nrirz_No pixel with average spatial coherence > {} are found for automatic reference point selection!z Try the following:zQ 1) manually specify the reference point using mintpy.reference.yx/lalo option.z< 2) change mintpy.reference.minCoherence to a lower value.F) print_msgr)rxrrrrWall RuntimeErrorr) rjrkrlcohZcoh_atrZcoh_maskmsgrmr[r0r0r1rhs"  rTcCst|\}}ttt|}ttt|}|||fdkrhttt|}ttt|}q2|r~td||f||fS)Nrzrandom select pixel y/x: {})rWrr!choicerNrrxr)Zdata_matrnrowncolrmr[r0r0r1r~srcCs|stdg}t|j}|jdks.|jdkrVd|krVt|d|_t|d|_|jdksj|j dkrd|krt |d|_t |d|_ |S)zBRead reference info from reference file and update input namespacer>Nrbrarr) r=rr7rrrr?r,rrr-)rr;ZatrRefr0r0r1read_reference_file2inpss    rcCst|j}t|d}t|d}d|_|jrLt|j|j|_d|_|Std|j rttd|j t |j |}|j rtd|j t |j |}|j rt|j dkrd |krd |j d }|d 7}t|tj||jd }|j r6|jr6|t|j t|jdd\|_|_td|j |jf|jdk r$|jdk r$td|j|jfd|jkr~|krnnd|jkr|ksnd\|_|_td|jrtj|jrtd|jtj|jddd}||j|jfdkrd\|_|_d|j}t|ntd|j s|j!sd|krdt"|dkrj|krnn\dt"|dkr|krnn8tdtd|dtd|d|j|_d|_n&|j#rtj|j#rd|_ nd|_ td t$|j td|S)!NLENGTHWIDTHTFz2--------------------------------------------------z&reading reference info from template: z'reading reference info from reference: ZUTM_ZONEzinput reference latitude (z) > 90 deg in magnitude!z4 This does not make sense, double check your inputs!)rrrVz$input reference point in lat/lon: {}z input reference point in y/x: {})NNz.input reference point is OUT of data coverage!zmask: rkrtz:input reference point is in masked OUT area defined by {}!zno input reference y/x.rbraz,REF_Y/X exists in input file, skip updating.zREF_Y: zREF_X: rr!z"reference point selection method: )%rr7r r, go_referenceresetremove_reference_pixelr8rxrrUrrrrWrZr?r~rLrrr geo2radararrayrrrrrrisfilerrrvr-rr*)r;r<lengthwidthrr/rkr0r0r1read_reference_inputs      " :         rcCs2td|i}dD] }d||<qt||}|S)z+Remove reference pixel info from input filez-remove REF_Y/X and/or REF_LAT/LON from file: )rbrarrNone)rxrLr)rZatrDroprBr0r0r1rs    rcCs<t|}t|jd|_t|}|jr0t|tddS)NrzDone.)r=rL get_file_listr rrrrx)r:r;r0r0r1mainsr__main__rn)N)N)N)Nr)T)N)N)"rsysr#rnumpyrWr!mintpy.objectsrmintpy.defaults.templater mintpy.utilsrrrrLr'r&r(r2r=rUr`rrrrrrrrr__name__argvr0r0r0r1 s6   3  %  ,  N