U a K@sZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl Z ddlZddlmZddlmZddlmmZddlmZddlmZddlmmZddlmZdZd Zd Z d Z!d d ddddZ"dddZ#ddZ$d.ddZ%d/ddZ&ddZ'dd Z(d!d"Z)d#d$Z*d%d&Z+d'd(Z,d0d)d*Z-d1d+d,Z.e/d-krVe.dS)2N)make_axes_locatable) timeseries)datetime)GPS)utils)CGPSg@g$@g @a example: # only plot faults and reference points plot_geotiff.py geotiff_file --shpdir /data/lxrtest/Balochistan/shp/ --fault multifault.shp nearbyfault.shp --fcolor b r --fstyle s d --refpoi refpoi_DT.shp --outfile velocity --outdir /data/lxrtest/BalochistanSenAT/shp/ plot_geotiff.py geotiff_file --shpdir /data/lxrtest/Balochistan/shp/ --fault multifault.shp nearbyfault.shp --fcolor o m --fstyle s s --refpoi refpoi_DT.shp --vlim -0.02 0.02 --outfile velocity --outdir /data/lxrtest/BalochistanSenAT/shp/ # plot data using nonlinear colorbar plot_geotiff.py geotiff_file --shpdir /data/lxrtest/Balochistan/shp/ --fault multifault.shp nearbyfault.shp --fcolor b r --fstyle s d --refpoi refpoi_DT.shp --nonlinear --outfile velocity --outdir /data/lxrtest/BalochistanSenAT/shp/ # plot shape file and gps velocity vector field # plot single gps site with extracting incidence and heading angle from geometreRadar plot_geotiff.py geotiff_file --shpdir /data/lxrtest/Balochistan/shp/ --fault multifault.shp nearbyfault.shp --fcolor o m --fstyle s s --refpoi refpoi_DT.shp --gps_putton --type NGPS --gps_name GV05 --geometry $SCRATCHDIR/BalochistanSenAT115/mintpy/inputs/geometryRadar.h5 --gpsdir /data/lxr/insarlab/GPS/ --scale 1 --scale_key 0.01 --vlim -0.02 0.02 --outfile velocity --outdir /data/lxrtest/BalochistanSenAT/shp/ # plot single gps site using typical incidence and heading angle for Seitinel1 plot_geotiff.py geotiff_file --shpdir /data/lxrtest/Balochistan/shp/ --fault multifault.shp nearbyfault.shp --fcolor o m --fstyle s s --refpoi refpoi_DT.shp --gps_putton --type CGPS --gps_name AHAQ --scale 1 --scale_key 0.01 --vlim -0.02 0.02 --outfile velocity --outdir /data/lxrtest/BalochistanSenAT/shp/ # plot multiple gps sites with extracting incidence and heading angle from geometreRadar plot_geotiff.py geotiff_file --shpdir /data/lxrtest/Balochistan/shp/ --fault multifault.shp nearbyfault.shp --fcolor o m --fstyle s s --refpoi refpoi_DT.shp --gps_putton --type NGPS --search --bbox 36 40 36 39 --geometry $SCRATCHDIR/BalochistanSenAT115/mintpy/inputs/geometryRadar.h5 --gpsdir /data/lxr/insarlab/GPS/ --scale 1 --scale_key 0.01 --vlim -0.02 0.02 --outfile velocity --outdir /data/lxrtest/BalochistanSenAT/shp/ # plot multiple gps sites with extracting incidence and heading angle from geometreRadar plot_geotiff.py geotiff_file --shpdir /data/lxrtest/Balochistan/shp/ --fault multifault.shp nearbyfault.shp --fcolor o m --fstyle s s --refpoi refpoi_DT.shp --gps_putton --type CGPS --search --bbox 30 35 101 106 --scale 1 --scale_key 0.01 --vlim -0.02 0.02 --outfile velocity --outdir /data/lxrtest/BalochistanSenAT/shp/ # plot GPS data alone plot_geotiff.py velocity_201504_202010.tiff --gps_putton --type NGPS --search --bbox 37 43 -121 -115 --gpsdir /data/lxr/insarlab/GPS/ --scale 1 --scale_key 0.01 --vlim -0.02 0.02 --outfile velocity3 --outdir ./ blackredyelloworangemagenta)bryomsoliddashed)sdcCstjdtjtd}|jddtdd|jdd}|jd d dtd d |jd ddtdd |jdddtdd |jdddtdd |jdddtdd |jdd}|jddddd |jd!d"tdd#d$|jd%ddd&d |jd'd(td)d*d+d,|jd-d.dtd/d |jd0d1dtd2d |jd3d4dtd5d |jd6d7dtd8d |jd9d:dtd;d |jdtd?d |jd@d}|jdAdddBd |jdCdDdtdEd |jdFdGdtdHd |S)INz plot *.tiff) descriptionformatter_classepilog input_geotiffzgeotiff file. )nargstypehelpz4option for shape files (faults and reference point).)titlez--shpdirshpdirzdirectory of shp files. )destrrrz--faultfault+zshp files of fault. z--fcolorfcolorzTdefine color for faults:b -> black; r -> red; y -> yellow;o -> orange; m-> magenta z--fstylefstylez6define line stype for faults:s -> solid; d -> dashed z--refpoirefpoi?zshp file of reference point. z*options for plot gps velocity vector fieldz --gps_putton store_trueFz)whether plot gps velocity vector field. )actiondefaultrz--typeGtypez3CGPS: China GPS database; NGPS: Nevada GPS database)r rrrz--searchz3whether search gps sites based on the given SNWE. z--bboxSNWE)SNWEzBounding box of area to be geocoded. Include the uppler left corner of the first pixeland the lower right corner of the last pixel)r rrmetavarrz --gps_namegps_namezgps site name. z --geometrygeometryzgeometry data. z--gpsdirgpsdirz2directory to store Nevada Geodetic Lab gps data. z--scalescalezparameter for quiver.scalez --scale_key scale_keyzparameter for quiverkey.U z--vlimvlimz?max and min value for displacement to display. The unit is m. z?options for the colorbar: linear colorbar or nonlinear colorbarz --nonlinearzwhether use nonlinear colorbar.z--outdiroutdirzoutput directory. z --outfileoutfilezoutput file name. ) argparseArgumentParserRawTextHelpFormatterEXAMPLE add_argumentstradd_argument_groupfloatint)parserZ shp_optionZ gps_optionZcolorbar_optionrEI/home/centos/operations/rsmas_insar/sources/MimtPy/mimtpy/plot_geotiff.py create_parser@s:   rGcCst}|j|d}|S)N)args)rG parse_args)iargsrDinpsrErErFcmd_line_parsezs rLF?cs\|\}}|d|g}|s ||g}ttt|tt|t|dfdd|D}|S)z.Get auto figure size based on input data shapeg?rcsg|]}|qSrErE.0i fig_scaleratiorErF sz$auto_figure_size..)minmin_figsize_singlemax_figsize_singlemaxmax_figsize_height)shape disp_cbarrSlengthwidthZ plot_shapefig_sizerErQrFauto_figure_sizes    r_cCs.t}t||D]\}}t|}|||<q|S)zdefine fault color)dictzip fcolor_table)fcolors shp_faults fault_colorsr#r!colorrErErFr#s  r#cCs.t}t||D]\}}t|}|||<q|S)zdefind fault linestyle)r`raflinestyle_table)Zfstylesrdfault_linestylesr$r!stylerErErF flinestyles  rjcCsft|d}||s0t|j|j|j}|}n|||j}t |j |j |j ||j|jgg}|S)zprocess single gps site)site)ropenutenu2losZvel_eZvel_nZvel_uZread_gps_los_velocityvel_losnparrayrksite_latsite_lon)gpsnamer3gps_objrolos_vel site_inverErErF cgps_processs  "rxc Cst||d}td|||sLt|j|j|j}|j }t ||}n| |\}}| ||j }t ||j}t ||j}t|j|j|j|||gg} | S)zprocess ngps sites)rkdata_dirzprocess {} site)rprintformatrlrmrndis_edis_ndis_udates dis_velocityread_gps_los_displacementget_gps_los_velocityvelocityrprqrkrrrs) rtr3gps_dirrudis_losrrvZe_velZn_velrwrErErF ngps_processs     rcCsHdd|D}t|dkr>t|}ttj||d}ntj}|S)zcalculate velocitycSsg|]}t|dqS)z%Y%m%d)dtstrftimerNrErErFrTsz dis_velocity..r8r)lenrZ"get_design_matrix4average_velocityrpdotlinalgpinvnan)rdis date_listAvelrErErFrs   rc Cs||jddkr"t|j\}}}n |jddkrBt|j\}}}tttt|gt|gft|gf}|S)z$search gps sites based on given SNWErNGPSr) r*gps search_gpsr+cgpsrp transposevstackrq)rK site_names site_lats site_lonssites_baseinforErErFsearch_gpssitess 6rc& CsRt|jd}|d}t|}|jdkrDt|}t|}n|jd}|jd}t j j }t |}t j dd|d\} } | } td|jrtjj|d| tjdddt|t|d|d d } ntjj|d| |||d d |jr|jd} |jr t| |j}td |jd | dd|jrt}|jD]}t| |||<q4t|j|}t|j|}td|jD]$}||j||| ||ddqt|dk rbt|dddft}t|dddft}t|dddft}t|dddft}|j d}| j!||||d |d}|j"d}dt#|d}| j$|dd|d|dddd | j%dddd d d d d!|jr*| &d"d#d$d g}t j j'tjdddt|t|d|d%}|(g|)t|t|| j*||d&d'd(}|j+j%dd)|,t-t|d*dd+t|gnLt.| j/d,d-d-d.}t0jj1|d/|d/d0}t0j*j2|||d1}|j+j%dd)d2d3d4d5} t3j45|jdd65d7d}!td8|!|!d9kr|j6d:| d;n|!dd5}"| 7d?|"| 8d@|"| 9| :}#dAdB|#D|j;ddC}$|j.z.pngi,tight)dpi bbox_inches)>rasteriorlrreadrprZr7nanminnanmaxpltcmjetr_subplotsrzZ nonlinearplotshowcolors SymLogNormrr% geopandas read_filer!r`r#rjr$listastyperBr5quiverr6r@ quiverkey tick_paramsadd_axesScalarMappable set_arrayset_climcolorbarr set_ticksrqr append_axesmpl Normalize ColorbarBaseospathsplit set_label set_xlabel set_ylabelget_xticklabelsget_yticklabelsr:r9savefig)&rK site_infovelrasterZ raster_datarZZ raster_minZ raster_maxrZ figure_sizefigaxesax1imrZ shp_refpoirdr!rerhlonlatZ x_componentZ y_componentZ quiver_scaleh1Z U_parametersZu_labelZcax1Zsm1cbarcaxrfont2filetypefont1labelsZfig_nameZ fig_outputrErErF plot_geotiffs            "    (       rc CsPt|}|jrD|j}|jrt|}t|jddg}t|dddft |jdD]f\}}|j ddkr|j }t t |||}n|j ddkrtt ||}|dddf||ddf<qZtj|dddf|jdd|fdd} nH|j}|j ddkr|j }t |||} n|j ddkr8tt ||} t|| nt|dS)Nrrrrr)axis)rLZ gps_puttonr3searchrrpzerosrZraaranger*r4rr@rx concatenatereshaper2r) iagrsrKZgeodatarZ sites_velrtrPr4Zsite_velrrErErFmainjs,*. r __main__)N)FrM)N)N)0r;rrZ rasterio.plot matplotlibrmpl_toolkits.axes_grid1rmatplotlib.pyplotpyplotrmatplotlib.colorsrrnumpyrpmintpy.objectsrrrmintpy.objects.gpsobjectsrr mintpy.utilsrrmZmimtpy.objects.cgpsrrrVrWrYr>rbrgrGrLr_r#rjrxrrrrr __name__rErErErFsH         :