o td?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_optionrEH/home/exouser/operations/rsmas_insar/tools/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 z$auto_figure_size..)minmin_figsize_singlemax_figsize_singlemaxmax_figsize_height)shape disp_cbarrSlengthwidthZ plot_shapefig_sizerErQrFauto_figure_sizes    r`cC.t}t||D] \}}t|}|||<q|S)zdefine fault color)dictzip fcolor_table)fcolors shp_faults fault_colorsr#r!colorrErErFr#  r#cCra)zdefind fault linestyle)rbrcflinestyle_table)Zfstylesrffault_linestylesr$r!stylerErErF flinestylerirmcCsft|d}||st|j|j|j}|}n|||j}t |j |j |j ||j|jgg}|S)zprocess single gps site)site)ropenutenu2losZvel_eZvel_nZvel_uZread_gps_los_velocityvel_losnparrayrnsite_latsite_lon)gpsnamer3gps_objrrlos_vel site_inverErErF cgps_processs  "r{c Cst||d}td|||s&t|j|j|j}|j }t ||}n| |\}}| ||j }t ||j}t ||j}t|j|j|j|||gg} | S)zprocess ngps sites)rndata_dirzprocess {} site)rprintformatrorprqdis_edis_ndis_udates dis_velocityread_gps_los_displacementget_gps_los_velocityvelocityrsrtrnrurv) rwr3gps_dirrxdis_losrryZe_velZn_velrzrErErF ngps_processs     rcCsJdd|D}t|dkr t|}ttj||d}|Stj}|S)zcalculate velocitycSsg|]}t|dqS)z%Y%m%d)dtstrftimerNrErErFrTrUz dis_velocity..r8r)lenrZ"get_design_matrix4average_velocityrsdotlinalgpinvnan)rdis date_listAvelrErErFrs  rc Cs||jddkrt|j\}}}n|jddkr!t|j\}}}tttt|gt|gft|gf}|S)z$search gps sites based on given SNWErNGPSr) r*gps search_gpsr+cgpsrs transposevstackrt)rK site_names site_lats site_lonssites_baseinforErErFsearch_gpssitess 6rc& CsFt|jd} |d}t|}|jdkr#t|}t|}n |jd}|jd}t j j }t |}t j dd|d\} } | } td|jrctjj|d| tjdddt|t|d|d d } n tjj|d| |||d d |jr|jd} |jrt| |j}td |jd | dd|jrt}|jD] }t| |||<qt|j|}t|j|}td|jD]}||j||| ||ddq|dur-t|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| &gd"}t j j'tjdddt|t|d|d#}|(g|)t|t|| j*||d$d%d&}|j+j%dd'|,t-t|d(dd)t|gn&t.| j/d*d+d+d,}t0jj1|d-|d-d.}t0j*j2|||d/}|j+j%dd'd0d1d2d3} t3j45|jdd45d5d}!td6|!|!d7kr|j6d8| d9n |!d:kr|j6d;| d9d0d1d|"| 9| :}#d?d@|#D|j;ddA}$|j.z.pngi,tight)dpi bbox_inches)>rasteriororreadrsr[r7nanminnanmaxpltcmjetr`subplotsr}Z nonlinearplotshowcolors SymLogNormrr% geopandas read_filer!rbr#rmr$listastyperBr5quiverr6r@ quiverkey tick_paramsadd_axesScalarMappable set_arrayset_climcolorbarr set_ticksrtr append_axesmpl Normalize ColorbarBaseospathsplit set_label set_xlabel set_ylabelget_xticklabelsget_yticklabelsr:r9savefig)&rK site_infovelrasterZ raster_datar[Z raster_minZ raster_maxr figure_sizefigaxesax1imrZ shp_refpoirfr!rgrklonlatZ x_componentZ y_componentZ quiver_scaleh1Z U_parametersZu_labelZcax1Zsm1cbarcaxrfont2filetypefont1labelsZfig_nameZ fig_outputrErErF plot_geotiffs                  (       rc CsLt|}|jr|j}|jrwt|}t|jddg}t|dddft |jdD]3\}}|j ddkrC|j }t t |||}n|j ddkrQtt ||}|dddf||ddf<q,tj|dddf|jdd|fdd} n"|j}|j ddkr|j }t |||} n|j ddkrtt ||} t|| dSt|dS)Nrrrrr)axis)rLZ gps_puttonr3searchrrszerosr[rcaranger*r4rr@r{ concatenatereshaper2r) iagrsrKZgeodatarZ sites_velrwrPr4Zsite_velrrErErFmainjs,*. r__main__)N)FrM)0r;rrZ rasterio.plot matplotlibrmpl_toolkits.axes_grid1rmatplotlib.pyplotpyplotrmatplotlib.colorsrrnumpyrsmintpy.objectsrrrmintpy.objects.gpsobjectsrr mintpy.utilsrrpZmimtpy.objects.cgpsrrrWrXrZr>rdrjrGrLr`r#rmr{rrrrr__name__rErErErFsL          :