B Qa@sddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl Z e jdeddddlmZddlmZmZmZmZmZmZddlmZddlmZmZmZm Z!m"Z#dd l$m%Z%dd l&m'Z'm(Z(d Z)d Z*d dZ+d5ddZ,ddZ-ddZ.ddZ/ddZ0d6ddZ1ddZ2ggdfddZ3d d!Z4d"d#Z5d$d%Z6d&d'Z7d(d)Z8d*d+Z9d7d-d.Z:Gd/d0d0Z;d8d1d2Zd4ddS)9N)make_axes_locatableignore matplotlib)categorymodule)giantIfgramStackgeometryDatasetNamesifgramDatasetNames ifgramStacktimeseriesKeyNames timeseries)GPS) arg_groupptimereadfileutilsplot)multilook_data)subsetversionaexample: view.py velocity.h5 view.py velocity.h5 velocity --wrap --wrap-range -2 2 -c cmy --lalo-label view.py velocity.h5 --ref-yx 210 566 #change reference pixel for display view.py velocity.h5 --sub-lat 31.05 31.10 --sub-lon 130.05 130.10 #subset in lalo / yx view.py timeseries.h5 view.py timeseries.h5 -m no #do not use auto mask view.py timeseries.h5 --ref-date 20101120 #change reference date view.py timeseries.h5 --ex drop_date.txt #exclude dates to plot view.py timeseries.h5 '*2017*' '*2018*' #all acquisitions in 2017 and 2018 view.py timeseries.h5 20200616_20200908 #reconstruct interferogram on the fly view.py ifgramStack.h5 coherence view.py ifgramStack.h5 unwrapPhase- #unwrapPhase only in the presence of unwrapPhase_bridging view.py ifgramStack.h5 -n 6 #the 6th slice view.py ifgramStack.h5 20171010_20171115 #all data related with 20171010_20171115 view.py ifgramStack.h5 'coherence*20171010*' #all coherence related with 20171010 view.py ifgramStack.h5 unwrapPhase-20070927_20100217 --zero-mask --wrap #wrapped phase view.py ifgramStack.h5 unwrapPhase-20070927_20100217 --mask ifgramStack.h5 #mask using connected components # GPS (for one subplot in geo-coordinates only) view.py geo_velocity_msk.h5 velocity --show-gps --gps-label #show locations of available GPS view.py geo_velocity_msk.h5 velocity --show-gps --gps-comp enu2los --ref-gps GV01 view.py geo_timeseries_ERA5_ramp_demErr.h5 20180619 --ref-date 20141213 --show-gps --gps-comp enu2los --ref-gps GV01 # Save and Output view.py velocity.h5 --save view.py velocity.h5 --nodisplay view.py geo_velocity.h5 velocity --nowhitespace a\Plot Setting: plot.name = 'Yunjun et al., 2016, AGU, Fig 4f' plot.type = LOS_VELOCITY plot.startDate = plot.endDate = plot.displayUnit = cm/yr plot.displayMin = -2 plot.displayMax = 2 plot.colormap = jet plot.subset.lalo = 33.05:33.15, 131.15:131.27 plot.seed.lalo = 33.0651, 131.2076 c CsLtjdtjtd}|dd}|jdtdd|jdtd gd d |jd d ddtd gdd|jddddd|jddddd gdd|jddddd d|jd!d"d#td$|jd%d&dd'd|jd(d)d*d+d,d-hd.d/t |}t |}t |}t |}t |}t |}t |}t |}t |}t |}t |}|S)0NzPlot InSAR Product in 2D) descriptionformatter_classepilogz Input FilezFile/Dataset to displayfilezfile for display)typehelpdset*z8optional - dataset(s) to display (default: %(default)s).)rnargsdefaultrz-nz --dset-num dsetNumListNUMzMoptional - order number of date/dataset(s) to display (default: %(default)s).)destmetavarrrrrz --nosearch search_dset store_falsez#Disable glob search for input dset.)r"actionrz--exz --exclude exDsetListZDsetz3dates will not be displayed (default: %(default)s).)r"r#rrrz --show-keptz--show-kept-ifgramplot_drop_ifgramz@display kept interferograms only, without dropped interferogramsz--plot-settingdisp_setting_filez!Template file with plot setting. )r"rz --noverbose print_msgzz cmd_line_parse..zview.py Agg)(rI parse_argsvarsospathisfileFileNotFoundErrorformatr'rread_date_listoutfiledisp_figsave_fig lalo_step lalo_label zero_maskrLdisp_whitespace disp_axis disp_title disp_cbar disp_tickflip_lrflip_ud auto_fliplistsetrread_attributerkeysprintr*vprintjoinr)%update_inps_with_display_setting_filepltswitch_backend)iargsrEinpskeyfnameZ geo_opt_namesopt_namerGrGrHcmd_line_parsesH        rzc Cstdd}g}x`|jD]V}|tjtj|dtj|g}dd|D}t|dkrj||dqd}qW|dkrtj |j }t dd|D}||krd}ntd ||j |S) Nzupdate mode: ONskippiccSsg|]}tj|r|qSrG)rWrXrY).0irGrGrH szrun_or_skip..rruncSsg|]}tj|qSrG)rWrXgetmtime)r}r~rGrGrHrsz4{} exist and are newer than input file: {} --> skip.) rpr]rWrXrqdirnamebasenamelenappendrrminr[)rvflagZoutfilesrxfnamestitorGrGrH run_or_skips  $ rcCs0t|}|js&d|kr&|d|_|jsFd|krFt|d|_|jsfd|krft|d|_|jsd|kr|d|_|jsd|krdd|d dd  d d D|_|j sd|krd d|d dd  d d D|_ |j s,d|kr,dd|d dd  D|_ |S)z2Update inps using values from display setting filezplot.displayUnitzplot.displayMinzplot.displayMaxz plot.colormapzplot.subset.lalocSsg|] }t|qSrG)float)r}nrGrGrHrsz9update_inps_with_display_setting_file..,rSrcSsg|] }t|qSrG)r)r}rrGrGrHrszplot.seed.lalocSsg|] }t|qSrG)r)r}rrGrGrHrszplot.referenceLalo) r read_template disp_unitrndisp_minrdisp_maxcolormap subset_latreplacesplit subset_lonref_lalo)rvZ disp_set_fileZ disp_set_dictrGrGrHrrs    (( rrcs t|}tt|\__|j_|j_ddj j f_ t dt j t dt jt dt |j t dt jt dtjdjdjd jdd krd _jrt d jdkrd _tj|jjdjd_jrt djjsV|jsVtdtdg_n*|jdjd_t djjrjr|jjddd|jjdddg_n0d|krt |dt |dg_nd_t!|tj"|j#j$j%jd\_#_$d_&|'dd(_)jrj)*drj+dkrj,sfj-rd}j)*drt./_&t |dnrj)*d rd!|kr|d!}t.0|_&t |d"|j,rt1d#ntd$j)td%d&_+j2st3fd'd(d)DsXj4d*krHjd5d+dd,ksXjdd-krdd.d/g_2nj4d0ksxj$rj%_2j6sjrj7rd1_6nd/_6j8rtj9|jd2\_:_;j<stj=|d3jtd4_sd6j<j?g_>t@S)7Nrzdata coverage in y/x: zsubset coverage in y/x: zdata coverage in lat/lon: zsubset coverage in lat/lon: zH------------------------------------------------------------------------rg@@Fz=area exceed 2000 pixels, turn off default DEM contour displayT) datasetNamer*z$input reference point in lat/lon: {}zbWARNING: --ref-lalo is NOT supported when 1) file is radar-coded AND 2) no lookup table file foundz1 --> ignore the --ref-lalo input and continue.z$input reference point in y /x : {}y) coord_typexREF_LATREF_LON)rwrap wrap_ranger*Y_UNITdegrees)degmetergeoz!initiate cartopy map projection: r PlateCarreerUTM_ZONEz UTM zone z1--lalo-label is NOT supported for projection: UTMz*WARNING: Un-recognized coordinate unit: {}z1 Switch to the native Y/X and continue to plotradarc3s|]}|jkVqdS)N)rwlower)r}r~)rvrGrH ^sz1update_inps_with_file_metadata..) coherencez.corr -)rcmaskgg?)z.intg?)r* FILE_PATH) datasetNames inps_dictzfigure title: z{}{})Aut coordinatersubset_input_dict2boxrVpix_boxgeo_boxcheck_box_within_data_coverage box_pixel2geowidthlengthdata_boxrpr7maxdisp_dem_contourrK multilook_num multilookppauto_colormap_namerrr*rr[ lookup_filero geo2radarref_yxyx2lalornr read_pts2inpscheck_disp_unit_and_wraprrr map_proj_objgetr coord_unit startswith fig_coordra coastlineccrsrUTM ValueErrorvlimanyrwr transparencydisp_dem_shaderjauto_flip_directionrhri fig_titleauto_figure_titler]fig_extupdate_figure_setting)rvmetadatacoordmsgutm_zonerG)rvrHupdate_inps_with_file_metadatas           &  rcCsd|jrJy,|jd|jd}|jd|jd}Wn YnX|jdkrt|t|jd|j}t|t|jd|j}t|jdkr|||f}tj|st |s||8}t d |jn"d ||}|d7}t |d|_qPt|jdkrPd|kr"|jdkrbnn skip re-referencing.rr F)rboxr*z1input reference point {} is out of data coverage!)rrrrr*zApply math operation: {}r,r-r.r/g?)r*zdata range: {} {}zdisplay range: {} {})'rrrr8rshapenpma is_maskedisnanrpr[rosqueezerwrreadrrralltilereshaperscale_data4disp_unit_and_rewraprrrr* disp_scalerr+r,r-nanminnanmaxdlim'auto_adjust_colormap_lut_and_disp_limitcmap_lut)datarrvref_yref_xref_valrref_boxrGrGrHupdate_data_with_plot_inpsst    D ",        rc s\jr tnddas*tdgt|tjtrPtj jj j dj_j rvtj j jjjd\}tdtjj\ jrjdrjdkrtd jrtd j|jjd j rtd tj|jjd t|tdjrjrjrtjdjdd} |d|ddd\}}|jd8}|jd8}||f8tdj|d_!jdjdjdjdf}|j"jj#dj#d|ddjj$dd } jdr,jdjddkr,d_%j%rdtdj&tj'|jjj&j(j)dj*rtj+j|j,j-j.j)j/jdn|j0dd j)d!d!d!d!d"j!rj1r|j2j1dj1dj3j4d#td$j5d%k r4|j2j5d%d%dfj5d%d%dfj6j7d&d'd(td)jrvjdjdjdjdf} tj8|| |jd}j rt|j\ f d*d+} | |_9nd,_td-j rtd tj|d%jd td.jdd/jdd/jdd/jdd/f}|j"jj#dj#d|djdd0} |j0j)d1j!rd2\} } j:rj:dj:d} } n(d3|;krt<|d3t<|d4} } | r| r|j2| | j3j4d#td$j=d%k r.|j2j=d%d%dfj=d%d%dfj6j4d5d'd(td)d}|rt>?d6d7gd8d9gd8d:gd;dtBjCDtBjCE|d?d@tBjCFrFy4tGjHdAjddBd tGjHdCjddBd Wn2dDtBjCI}|dE7}t|d%YnXnd% fdFd+} | |_9d%}jJrtK|}|jLjMjNjNtOjPdG}tQ| |\}jRr|jSjTj)jUdHjVrtdI|WjXrtdJ|YjZs|[dKtdLj\r4|j]^d!|j_^d!n|`ag|bag|| |fS)Ma?Plot one slice of matrix Parameters: ax : matplot.pyplot axes object data : 2D np.array, metadata : dictionary, attributes of data inps : Namespace, optional, input options for display Returns: ax : matplot.pyplot axes object inps : Namespace for input options im : matplotlib.image.AxesImage object cbar : matplotlib.colorbar.Colorbar object Example: import matplotlib.pyplot as plt import mintpy.utils.readfile as readfile import mintpy.view as pv data, atr = readfile.read('velocity.h5') fig = plt.figure() ax = fig.add_axes([0.1,0.1,0.8,0.8]) ax = pv.plot_slice(ax, data, atr)[0] plt.show() c_sdS)NrG)rJrPrGrGrHrQrRzplot_slice..)rvlist)rrr*zdisplay data in transparency: )rrrzplot in geo-coordinatez#draw coast line with resolution: {}) resolutionzplotting DEM background ...)axrdemrvr*zplotting image ...)siteF)r*rrrzDreferencing InSAR data to the pixel nearest to GPS station: {} at {}ruppernearest) cmapvminvmaxextentorigin interpolationalphaanimatedzorderrzplot scale bar: {})runitloclabelpad font_size)r`lalo_loc lalo_max_numr projectionr*bothinT)which direction labelsizeleftrighttopbottom)mszplot reference pointNkg?)rmecmewzplot points of interestc sdd||}j|ddjd}j|ddjd}d|krRkr`nnd|krn kr`nn||f}t|stj|r|d7}n|d|7}jr<j|ddd}j|ddd}d|krkr.format_coordrzplotting in Y/X coordinate ...zplotting Data ...g?)rrrrr r r )r)NNREF_YREF_Xblackiiiiiiirrrzinputs/geometryRadar.h5latitude)rrr* longitudez4WARNING: no latitude / longitude found in file: {}, z'skip showing lat/lon in the status bar.csd||}tt|jd}tt|jd}d|krPkrnnd|krhkrnn||f}t|stj|r|d7}n|d|7}jrЈ||f}t|s|d|7}r|d||f||f7}|S)Nzx={:.1f}, y={:.1f}rrz, v=[]z , v={:.3f}z , h={:.0f} mz, lat={:.4f}, lon={:.4f}) r[r8rrintrrrrrK)rrrr%r&r'r()rr geom_filervlatslonsr-r.rGrHr/s 0     )pad axes_class)fontsizecolorzflip figure left and rightzflip figure up and downoffzturn off axis display)cr*rorprzr isinstancerr7r ColormapExtr cmap_vlistrKread_demrrrrrrrrr[ coastlinesplot_dem_backgroundrrdisp_gps gps_component ref_gps_siter Zget_stat_lat_lonrdisp_ref_pixelimshowr animation disp_scalebarZscalebar draw_scalebar scalebar_padrradraw_lalo_labelr`rrr tick_paramsrr ref_markerref_marker_sizepts_lalo pts_markerpts_marker_sizeplot_gpsr/rrnr8pts_yxrarrayset_xlimset_ylimrWrXrqrrYrrrrfr append_axescbar_loc cbar_sizersAxes plot_colorbarre set_titler font_colorrh invert_xaxisri invert_yaxisrdaxisrgxaxis set_visibleyaxis get_xaxis set_ticks get_yaxis)rrrrv dem_metadataZ ref_site_lalorrrimSNWEr/rrZ temp_testrUrcbardividercaxrG) rr)rrr*r+r,r6rvr7r8r-r.rH plot_slices4          (     $        $  0     rocCst|j}d}|jds.|d|d7}|d|dtj|j7}d|krj|d|d7}t d tj t t j t |t|d |_t|d |_|d|_tjtj |jd |_tj|jd |_t d|j|jftj|jdd|_t|\}}||fS)Nzinput file is )z.h5z.he5z{} PROCESSORz {} file: {} FILE_TYPE DATA_TYPEz in {} formatz run {} in {}LENGTHWIDTHrrzfile size in y/x: {}T) no_complex)rrmrendswithr[rWrXabspathrnrpr__file__rrelease_descriptionr8rrrwsplitextZfileBaseZfileExtget_slice_list sliceListread_dataset_input)rvatrrrGrGrHread_input_file_infos$     rTcsdd|D}|rt|tr"|g}g}|rnxX|D]8dkrFddd|fddD7}q0Wn|fdd|D7}ttt|}|fdd|D7}ttt|}fd d|D}||fS) z0Get dataset(es) from input dataset / dataset_numcSsg|]}|qSrGrG)r}rrGrGrHrsz(search_dataset_input..rz*{}*z.*cs g|]}t|dk r|qS)N)rematch)r}e)dsrGrHr)scsg|]}|kr|qSrGrG)r}r~)allListrGrHr-scsg|]}|qSrG)index)r}r)rrGrHr/scsg|] }|qSrGrG)r}r~)rrGrHr5s)r>r7r[rsortedrkrl)rZinListZ inNumListr$tempListZ outNumListZoutListrG)rrrHsearch_dataset_inputs"    rc stjdkstjdkrtjdkr:tdjjdkr\d_tdjnHjdkrtjdkrdjdkrjdd\}}|g_|_t j jjjd_njd krd g_njd krd g_ntjd krd_nbjdkr8t j }|j dd|j dddg_n&j _jdkr^ddjD_t j jjjd_t j jgj\__g}tj }js|dd krtdtj fddj D}fdd|D}ntd|dd_tttjtjt|_fddjD_tj_jrjtkrtd_t j jggjdd}|stdjd_n|_jdkr tdtjj tj td tjtd!tjnPtdtjj tj td"tjjtd#tjjjrjtkrtd$jjdkrd%}|d&j 7}t|tjj jdddd'}|fS)(zFCheck input / exclude / reference dataset input with file dataset listrzinput dataset: "{}"velocityFz#turning glob search OFF for {} filer r_r unwrapPhaseHDFEOS displacementgiantTimeseriesreconsr)r*rgeometrycSsg|]}|ds|qS)bperp)r)r}rrGrGrHrasz&read_dataset_input..rqz&do not show the dropped interferogramscs"g|]}|ddkr|qS)rr)r)r}r)date12_drop_listrGrHruscsg|]}j|qSrG)r|r)r}r)rvrGrHrvszH--show-kept option does not apply to file type: {}, ignore and continue.Tcsg|]}j|qSrG)r|)r}r~)rvrGrHr}sNzVWARNING: input reference date {} is not included in input file! Ignore it and continue)r znum of datasets in file {}: {}znum of datasets to exclude: {}znum of datasets to display: {}zdatasets to exclude ({}): {}zdatasets to display ({}): {}zinput reference date: {}zNo input dataset found!z available datasets: {})r)rrr rpr[rwr$rref_daterr|rropenr'Z exDsetNumListrrmr(r get_drop_date12_listrrkrldsetNumr rorWrXr Exception) rvdate1date2objZ drop_num_listr~Zdrop_slice_listrrrG)rrvrHr}:s &          &       r}cstjdjd}tjdjd}jdkrjsFd_jsjrjdkrtjdjd}tjdjd}tj ||fj j d_njstj _t djdjdjs d_x(jtjd kr jd7_qWjdkr|jdkr|tfd d d Dr||d |g}jddjdg}tj||j\__ttjtjjt_t dtjt dtjt dtjt dtjjsd_jjdkrd_jr^tjjd_tjtjjd\_ _!j!"_!ntjj#_tjtjj#d_ jdjdjdjd||krʈj d7_ j$r"j d7_ j%dj%ddtj&kr"j tj%dj%d7_ j'r>j dj'7_ j(rTj d7_ jdkrvdj j!g_n fddt)djdD_fddjD_S)zUpdate figure setting based on number of subplots/datasets 1) fig_size and font_size 2) for multi: figure/row/column number 3) for multi: output file name rrrrr)ds_shaperfr*zfigure size : [{:.2f}, {:.2f}]gd@c3s|]}|jkVqdS)N)argv)r}r~)rvrGrHrsz(update_figure_setting..)z--nrowsz--ncolsg?gffffff?zdataset number: zrow number: zcolumn number: zfigure number: 2_sub_wrapZ_refZ_exz{}{}cs"g|]}djt|jqS)z{}_{}{})r[ outfile_baser7r)r}j)rvrGrHrsz)update_figure_setting..csg|]}tjj|qSrG)rWrXrqoutdir)r}r])rvrGrHrs)*rrrrfig_sizerrabsrauto_figure_sizerfr*default_figsize_multirpr[fig_num fig_row_num fig_col_numrauto_row_col_numrceilastyper8r7r]rWrXrrrzrrrrrrrpirr'range)rvrr data_shapeZ fig_size4plotrG)rvrHrsv   "2 "  rc stj||tjdjdjtjdjdjftjd}tjdkr`jdkr`t dfdd t ||D}t j j |jjjjd d|d d <jd krjdd rjrj\}}|||d|df}t j j ||ddd} xt |jdD]8} || d d d d fdk} || | f| | 8<q"Wnt dtj||jd} x|t ||D]n} t j j j| jjjdd d} | || |d d d d f<| j| |dj| dddqW| jrDt djt j j jjjjdd d} || 8}tjdksjdkstfdddDsjdddgksjdd t tjDkrd}nd}|rt||\}n&tfd dd!Drtd"d _jrt d#tj |dk|}j!d k rTt d$t"j!|jdddf}tj |dk|}t#|t$|g_%|rtfd%dd&Drʈjdd'rjrʈjdd(krt&j'|d)dd*\_(_)|S)+zIRead multiple datasets for one figure into 3D matrix based on i_start/endrrrr)dtype)r rr rrzreading data as a 3D matrix ...csg|]}j|qSrG)r)r}r~)rvrGrHrsz$read_data4figure..)rrxstepystepr*Nr rF)rrr*gz)reading data as a list of 2D matrices ...)maxValuer*/r)suffixzconsider input reference date: )r Z inversionc3s|]}|jkVqdS)N)dsetFamilyList)r}d)rvrGrHr=sz#read_data4figure..) horizontalverticalrmodelresidualcSsg|]}d|dqS)zband{}r)r[)r}r~rGrGrHr?sTc3s|]}|jkVqdS)N)ru)r}r)rvrGrHrHs)z-uz--unitzeWARNING: -u/--unit option is disabled for multi-subplots with different units! Ignore it and continuezmasking pixels with zero valuez masking datac3s|]}|jkVqdS)N)r)r}arg)rvrGrHrXs)z-vz--vlimz--wrapunwrap)r ) num_multilookr*)*rzerosr8rrfloat32rrrwrprrrrr*r file_ref_yxrr progressBarrupdatercloserrrrrorrbr masked_wheremskrrrrrrrr)i_starti_endrvrr dset_listrrrref_datar~maskprog_barrZsame_unit4all_subplotsrrG)rvrHread_data4figures        ,    rc Cs|jr&tj|d|j|j|j||jd|j}|dkrHt |t |g}|j dd|j dd|j dd|j ddf}|j ||j |d|dd|j|dd }|jrd \}} |jr|jd|jd}} n&d |krt|d t|d }} |r| r|j| ||j|jd ||dd||dd|jrV|j|jdkrr|g|gdd} | |_|jrd} |j t!ks|j"d#dr|j"|$dd%dd} yt&j'(| d)dd} Wn| } YnXn|j"|} |j*+| }t,|j-dkrRd| krR| %dd} |j.ddkrR|j/+| }|j|j}|dkrn| } nfd|krdkrnnd0|| } ns|?d,|S)-zPlot one subplot for one 3D array 1) Plot DEM, data and reference pixel 2) axes setting: tick, ticklabel, title, axis etc. N)rr dem_shade dem_contourdem_contour_seqrvr*rg?rrrr)rrrr r rr )NNr0r1)rrrcSs d||S)Nzx={:.1f}, y={:.1f}, v =)r[)rrrGrGrHr/sz)plot_subplot4figure..format_coordrrrz%Y%m%drqr z{} {}rr z{})size)rprop)r;crimsonr<bold fontweightPLATFORMZSenZSENTINEL1_SENSORAZ SENTINEL1_IPFz : {}r=)@rKrrCrrrr*rrrrrrHrrrGrrnr8rrOrPrWrXrgrrrfrgrhr/rerwr rrrrdtdatetimestrptime isoformatr|rrrr~ date12Listr[ fig_title_indictradd_inner_titledropDatasetListrdisp_title4sentinel1 mplColorsr^rhr`rirardrb)r~rvrrrrrrjrrr/Z subplot_titledate_strZ title_strZ title_indZ num_subplotrkwargZ s1_sensorZs1_IPFrGrGrHplot_subplot4figurebs                rc CsXdt||j|d}tdt|tj||j|j|jddd\}}|j j || }|d|j|j}t |j||j|jg}t||||}t|jtrtj|j|j|jdj|_tdtj|||jd} xt||D]} | |} t| ||| || d d d d f|d } |jrv|jsv|j| || d d d dd} tj|j |j!| d"dd }|rv| #|| j$| d|j!| %dddqW| &~x*t||t'|D]} |(|| qWt)*|j+d|j,dgt)-|j+d|j,dgg|_+td|j,|j.|jr8td|j|j.dd}|jr|jstdtd|/|jdd|jd|||nf|||d}|jddkr|d}td |j0d!d"|1d#d$|dd%|g}t2|| |\}} n ||||j3rTtd&t4j56|j|d|j7|j8|j|dd'd|j7d(|j9sT|:d S))zPlot one figure with multiple subplots 1) create figure 2) read all data into 3D array 3) loop to plot each subplot using plot_subplot4figure() 4) common colorbar and save zFigure {} - {}rz(----------------------------------------T)numfigsizenrowsncolssharexsharey)rrz plotting ...)rr*N)rrrgQ?g?r r)rr9shrinkaspect orientation)rUNITrr)rrzdata range: {} {}zdisplay range: {} {}cSsH|jddddddd|js"|jr6|j|j|jdn|jrD|dS)Ng{Gz?g\(\?g?)rrrrwspacehspace)rr)subplots_adjust fig_wid_space fig_hei_spacefig_tight_layout tight_layout)figrvrGrGrHadjust_subplots_layouts   z+plot_figure..adjust_subplots_layoutz-Note: different color scale for EACH subplot!z0Adjust figsize for the colorbar of each subplot.g?g?g @rz show colorbarg(\?)rgGz?g?g{Gzt?zsave figure to {} with dpi={}tight) bbox_inches transparentdpi);r[r7r]rprssubplotsrrrcanvasmanagerset_window_titleflattenrrrr>rrr?rr@rrr*rrrfrcolorbarrrmrrr set_labelrrrrdelaxesrrdlim_allrrrset_size_inchesradd_axesr]r_rWrXrwfig_dpisavefigr^clf)rrvrrraxsrrrrr~idxrjrl data_unitrZ cbar_lengthrnrGrGrH plot_figures~    &      $rc stttdd|jD|_tttdd|jD|_t|jdkrv|jddkrvtttdd|jD|_|j dkr|j rt d |j n2|j r|j dkrtj|j|j|j|j|j d |_ |jd k r|j dkrt|j|j |j d d |_d |_|jd krdkrtdtd}}tdtd}}d|krb|krnn6d|kr~|krnn||g|_td |j|jdkr|jd_n|jdkr|jd_g|_|jdkr@|jr@t|j }|j!dd|"}x,|jD]"|jfdd|D7_qWtd|j#rt$%|j#t&fdddDrtd t'j()|j#t$j*|j#d |j|j |j dd!d}tj+|||j d"\|_,|_-|_.n(d |_#d#|_/d$} | d%7} | d&7} t | |S)'zPrepare for multiple subplots: 1) check multilook to save memory 2) read existed reference pixel info for unwrapPhase 3) read dropIfgram info 4) read and prepare DEM for background css|]}|ddVqdS)rrN)r)r}rrGrGrHrOsz)prepare4multi_subplots..css|]}|ddVqdS)StdrN)r)r}rrGrGrHrPsrrqr css|]}|ddVqdS)rrN)r)r}rrGrGrHrRsz/multilook {0} by {0} with nearest interpolation) max_memoryr*Nr)method)r r0r1rsrtrz#consider reference pixel in y/x: {}rg$@dg4@F)r*csg|]}d|qS)z{}-{})r[)r}r)r~rGrHr}sz*prepare4multi_subplots..z@mark interferograms with 'dropIfgram=False' in red colored titlec3s|]}||kVqdS)NrG)r}r~)rirrGrHrs)rsrtzreading DEM: {} ... height)rrrrr*)rrvr*g?z;WARNING: DEM file has a different size from the data file. zNThis feature is only supported for single subplot, and not for multi-subplots.z --> Ignore it and continue.)0rrkrlrrrr~r|rrr*ror[rrauto_multilook_numrrr maxMemoryrrrrwrnr8rprrPrrer rrrrKrrmrrWrXrrprepare_dem_backgroundrrrr) rvrrrrrrZdropDate12ListrrrG)rir~rrHprepare4multi_subplotsHsl  8      "  rFc Csdt|dd}|dd|_t|t|\}}t||}tj|j|j |j d|j |j |j |jd\|_|_ tj|j|j d|j |jd\}}|jr|tj|j|j|j ddd8}|jdkrP|j dddd rPd |krPt|d t|d }}tj|j|j d|||d|dfddd}||d k|8<|jrjtj|d k|}|jdk rtj|jd k|}ntj|jtjd |_tj|r|j|j9_|jt |j!9_n|jt |9_t"|||\}}|dkrVdd|j#D}|j$dk r2t%|j$dni} t&j'|d| d\} } |||| fS|||fSdS)aPrepare data from command line as input, for easy call plot_slice() externally Parameters: cmd - string, command to be run in terminal Returns: data - 2D np.ndarray, data to be plotted atr - dict, metadata inps - namespace, input argument for plot setup Example: subplot_kw = dict(projection=ccrs.PlateCarree()) fig, ax = plt.subplots(figsize=[4, 3], subplot_kw=subplot_kw) W, N, E, S = (-91.670, -0.255, -91.370, -0.515) # geo_box cmd = 'view.py geo_velocity.h5 velocity --mask geo_maskTempCoh.h5 --dem srtm1.dem --dem-nocontour ' cmd += f'--sub-lon {W} {E} --sub-lat {S} {N} -c jet -v -3 10 ' cmd += '--cbar-loc bottom --cbar-nbins 3 --cbar-ext both --cbar-size 5% ' cmd += '--lalo-step 0.2 --lalo-loc 1 0 1 0 --scalebar 0.3 0.80 0.05 --notitle' data, atr ,inps = view.prep_slice(cmd) ax, inps, im, cbar = view.plot_slice(ax, data, atr, inps) plt.show() rNr)rLrrrrr*)rrr*F)r rrr0r1g)rTcSsg|] }|dqS)g@rG)r}r~rGrGrHrszprep_slice..)rFigure)rr subplot_kw)(rzrrrprrr read_maskrrLrr mask_vmin mask_vmaxr*rrrrrwrrnr8rbrrronesrbool_rrrrrrrrrsr) cmdZauto_figrvr~rrrrrr rrrGrGrH prep_slices\        r'c@s*eZdZdZd ddZddZddZdS) viewerzClass for view.py Example: import matplotlib.pyplot as plt from mintpy.view import viewer cmd = 'view.py timeseries.h5' obj = viewer(cmd) obj.configure() obj.plot() NcCs$|r|dd}||_||_dS)Nr)rr&ru)selfr&rurGrGrH__init__s zviewer.__init__c Cst|j}t|j|_t|\}|_t||j}d|_|jrT|j sTt |dkrTd|_x"|j D]\}}t |||q`Wtj|j|j|jd|j|j|j|jd\|_|_|jS)Nrr{r)rLrrrrr*)rzrurkrrr~rr update_moder^r__dict__itemssetattrrr!rrLrrr"r#r*r)r)rvrwvaluerGrGrH configures"   zviewer.configurec Cst|jdkrtdtj|j|jd|jdd\}|_|jr\|tj|j|j|jddd8}|j dkr|jd dd drd |j krt |jd t |jd }}tj|j|jd|||d|dfddd}||d k|8<|jrtd tj|d k|}|jdk r:td tj|jd k|}ntj|jtjd|_tj|r|j|j9_|jt|j9_n|jt|9_t||j|\}}|jdk rt|jdni}tj|jd|d\}}|j s|j!dddddt"|||j|d}|j#rXtd$t%j&'|j(d|j)|j s\|j*|j(dd|j)d dn|j*|j(dd|j)dd|j+sX|,nddddddddg}t-t.|t.|j/@}x|D]} t0d $| qWt1||jd!}d d g|_2x*t3d|j4dD]} t5| ||jd!qW|j4dkrXtd"td#$|j2|j6|j7rXtd$$|j7|j6|j+rptd%t8dS)&Nrzreading data ...rF)rrr*)r rrr0r1gzmasking pixels with zero valuez masking data)r)rr)rrr )rrrrzsave figure to {} with dpi={}T)rr pad_inchesr)rrrz --show-gpsz --coastlinez --lalo-labelz --lalo-stepz --scalebarz--pts-yxz --pts-laloz --pts-filezHWARNING: {} is NOT supported for multi-subplots, ignore it and continue.)rz(----------------------------------------zall data range: {} {}zdisplay range: {} {}z showing ...)9rrprrrrrr~rrwrrrnr8rbrrrrr$rr%rrrrrrrrsrrrcrror_r[rWrXrwr]rrr^rrkrlrurorr rrrrrshow) r)rrrrr rrZ opt_namesryrrGrGrHrs|            z viewer.plot)NN)__name__ __module__ __qualname____doc__r*r0rrGrGrGrHr(s  r(cCs(t|d}||jdkr$|dS)N)rur)r(r0rr)rurrGrGrHmain}s   r7__main__r)N)N)F)N)?rWsysrr1rrnumpyrmatplotlib.pyplotpyplotrsmpl_toolkits.axes_grid1rwarningsfilterwarnings UserWarning cartopy.crscrsrmintpy.objectsrrr r r r Zmintpy.objects.gpsr mintpy.utilsrrrrrrrmintpy.multilookrmintpyrrr4r9rIrzrrrrrrorrr}rrrrrr'r(r7r3rrGrGrGrH sR      " / ;V #$iWhxnT R