B QaW@sDddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z mZmZyddlmZWnek redYnXddlZddlmZddlmZmZmZmZmZmZ dd lm!Z!d Z"d d Z#d'd dZ$d(ddZ%d)ddZ&ddZ'ddZ(d*ddZ)dd Z*d!d"Z+d+d#d$Z,e-d%kr@e,ej.d&ddS),N)etree)ZipFile)pyplotcolorscolorbarticker)KML_ElementMakerzCan not import pykml!)timeseriesKeyNames) arg_group attributeptimereadfileutilsplot)subsetawexample: save_kmz.py geo/geo_velocity.h5 save_kmz.py geo/geo_velocity.h5 -u cm --wrap --wrap-range -3 7 save_kmz.py geo/geo_timeseries_ERA5_ramp_demErr.h5 20101120 save_kmz.py geo/geo_timeseries_ERA5_demErr.h5 20200505_20200517 save_kmz.py geo/geo_ifgramStack.h5 20101120_20110220 save_kmz.py geo/geo_geometryRadar.h5 height --cbar-label Elevation # save full resolution velocity in radar-coordinates # for ISCE products (with lookup table in radar coordinates) ONLY # use --sub-x/y to reduce file size for a smooth Google Earth experience save_kmz.py velocity.h5 --sub-x 300 800 --sub-y 1000 1500 --step 1 c Cstjdtjtd}|jddd|jdddd |jd d d d dd|jddddd|jddddd|jddddddd|jdd d!d d"d|jd#d$td%d&d'|jd(d)d*d+d,td-d.|jd/d0d1d2d|jd3d4d5d6d7d8d9|jd:dd;d<|jd=d>td+d?tjtjgd,d@dA| dB}|jdCdDdEtdFdGdH|jdIdJdKtd+dLdM|jdNdOdPdQdRdSgdPdTdU|jdVdWdXdYdZd[|jd\d]dEtd^d_| d`}|jdadbdcddd|jdedfdgdhdid[|jdjdkdEtd%dldH|jdmdndodpdqd[t |}|S)rNz1Generate Google Earth KMZ file with raster image.) descriptionformatter_classepilogfilez(file to be converted, in geo coordinate.)helpdset?z?date of timeseries, or date12 of interferograms to be converted)nargsrz-mz--mask mask_fileFILEzmask file for display)destmetavarrz --zero-mask zero_mask store_truezMask pixels with zero value.)ractionrz-oz--outputoutfilez3output file base name. Extension is fixed with .kmz)rrz--kkz --keep-kmlz--keep-kml-file keep_kml_filezJDo not remove KML and data/resource files after compressing into KMZ file.z-gz--geom geom_filezDgeometry file with lat/lon. [required for file in radar coordinates]z--stepstepzOoutput one point per {step} pixels, to reduce file size (default: %(default)s).)rtypedefaultrz-vz--vlimvlim)MINMAXzY/value limits for plotting.)rrrr%rz-u disp_unitUNITzunit for display.z-cz--cmz --colormapcolormapjetz#Colormap for plotting. Default: jet)rr&rz--wrapz(re-wrap data to display data in fringes.)rrz --wrap-range wrap_rangegz5range of one cycle after wrapping, default: [-pi, pi])rr%rr&rrFigurez--dpifig_dpiNUMiXz(Figure DPI (dots per inch). Default: 600)rrr%r&rz --figsizefig_size)WIDLENz(Figure size in inches - width and length)rrr%rrz --cbar-loccbar_locz lower leftz lower rightz upper leftz upper rightz8Location of colorbar in the screen. Default: lower left.)rchoicesr&rz --cbar-label cbar_labelZLABELzMean LOS velocityz*Colorbar label. Default: Mean LOS velocity)rrr&rz--cbar-bin-num cbar_bin_numz+Colorbar bin number (default: %(default)s).)rrr%rzReference Pixelz --noreferencedisp_ref_pixel store_falsezdo not show reference pointz --ref-colorref_marker_colorZCOLORkzmarker color of reference pointz --ref-sizeref_marker_sizez6marker size of reference point (default: %(default)s).z --ref-marker ref_markerZSYMBOLsz marker symbol of reference point) argparseArgumentParserRawTextHelpFormatterEXAMPLE add_argumentintfloatnppiadd_argument_groupr add_subset_argument)parserfigrefrOE/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/save_kmz.py create_parser8sT           rQcCst}|j|d}tjtj|j|_t |j}d| krddg}|j sdt j ||jdd|_ |j rxtj|j std|d|js|d d gtkrtd |d td |S) N)argsY_FIRSTlatitude longituderadar)work_dircoordzNo geometry file with z in radar coord found! FILE_TYPE ifgramStackz/No date/date12 input. It's required for {} fileAgg)rQ parse_argsospathabspathdirnamerrWr read_attributekeysr"utget_geometry_fileisfileFileNotFoundErrorrr Exceptionformatpltswitch_backend)iargsrLinpsatrZ geom_ds_listrOrOrPcmd_line_parseqs"     rncm/yearr.g ףp= ?g @Mean LOS velocityc Cstj|d\}} tj||d} tj| t|| dd} | jd||dd|rft j |d| _ | | j jd dd |jd |jd td ||j|d|dd|S)N)figsize)vminvmaxvertical)cmapnorm orientationz{} [{}] )fontsize)nbinsboth)which labelsizewhitegffffff?writingtighti,) bbox_inches facecolordpi)risubplotsr Normalizer ColorbarBaseget_cmap set_labelrhr MaxNLocatorlocator update_ticksax tick_paramspatch set_facecolor set_alphaprintsavefig get_facecolor) out_filersrtunitrvrrr{labelrMcaxrwcbarrOrOrP plot_colorbars   r lower leftc Cst|||||||d}|ddkr0d\}} n|ddkrHd\}} |ddkrbd\} } n|dd krzd\} } ttd ttd tj |t d tj | |d d dtj | | d d dtjdddddtdtdtd} | S)N)rrsrtrrvr{rrlower)0rupper)1rleftrightrz{}g?fraction)xyxunitsyunitsrZ250pixel)rsplitKMLZ ScreenOverlaynameIconhrefrhr]r^basenameZviewBoundScaleZ overlayXYZscreenXYsizerotationZ visibilityopen) cbar_filersrtrrvlocr{roysyoxsx cbar_overlayrOrOrPgenerate_cbar_elements8    rcCs>|||}tj|d|d|d|dgdddd}|S)a Get color name in hex format. Parameters: v : float, number of interest colormap : matplotlib.colors.Colormap instance norm : matplotlib.colors.Normalize instance Returns: c_hex : color name in hex format r(rrT) keep_alphaN)rto_hex)vr-rwrgbaZc_hexrOrOrP get_hex_colors .rcCsttttt||j|jtdt t |t tj t |ddddtj t |ddddtj t |dddtj t |dd dtj t |d d |jd dtt|d |}|S)a]Create an KMZ Placemark element. Parameters: lat/lon - float, latitude / longitude in degrees row/col - int, row / column number val - float, value icon_file - str, path of the icon file inps - Namespace object Returns: placemark - KMZ.Placemark() object g?z.6fu˚ZLatitude)rZ Longitudez.0fZRowColumnz.2f Value,)rZ PlacemarkStyle IconStylecolorrr-rwscalerrZ ExtendedDataDatavaluer+Point coordinates)latlonrowcolval icon_filerl placemarkrOrOrPcreate_placemark_elements2   rFc Cs~|dkr gn|}|dkrgn|}tj|}d|}d|}t}||td|t|d } | t j |dd dWdQRX|rtj tjtjd } x<|D]4} tj | tj| } t| |td | qWtjt} t|t|dP}xH|g||D]6} | tj| |st| td | qWWdQRXt| td ||S) zWrite KML and KMZ files. Parameters: out_file_base - str, output file name without extension kml_doc - KML.Document() object res_files - list of str, rel path of resource files Nz{}.kmlz{}.kmzzwriting wT) pretty_printutf8datazcopy {} to the local directoryz remove {}zmerged all files to {})r]r^r`rhrkmlappendrrwritertostringdecodejoinmintpy__file__rshutilcopy2r_getcwdchdirrrelpathremove) out_file_basekml_doc data_files res_filesr!rWZkml_filekmz_filerfZres_dirfnamesrc_fileZrun_dirfzrOrOrPwrite_kmz_files6      $      rcCsHt|\}}}}td|jsset_xlimset_ylimr]r^splitextrhr1rrDocumentrZ GroundOverlayrrrZ altitudeModeZ LatLonBoxnorthstreastsouthwestrrr+r6r9r8rr!)rmetarrlr r r r rMrrxrywidthlengthrZ data_png_filerZimg_nameZ img_overlayrrrrOrOrPwrite_kmz_overlayCsn                rc CsVtj|d}d}d}tj|d|jdd}tj|d|jdd} t} tdd |} t | |j d|j d |j |j |j|j|jd } | | t|d d} t|d d}d |krDtdt|d t|d}}|||f}| ||f}t|||||| d||d}d|_d|jj_| |tj|||f<ttd}td |j|j\}}tj |d}xt!d||jD]}xpt!d||jD]^}|||f}t"|s|||f}| ||f}t|||||| |||d}||qW|j#|d d d|d d|dqW|$| |t%|| ||g|j&d}|S)aGenerate Google Earth Placemark KMZ file for data in RADAR coordinates. Parameters: data - 2D np.array in int/float, data matrix to write meta - dict, containing the following attributes: WIDTH/LENGTH : required, file size X/Y_FIRST/STEP : required, for lat/lon spatial converage REF_X/Y : optional, column/row number of reference pixel geom_file - str, path of the geometry file with latitude/longitude datasets out_file - string, output file name inps - Namespace, optional, input options for display Returns: kmz_file - string, output KMZ filename rzshaded_dot.pngzstar.pngrT) datasetNameboxrUz0plot and add colorbar as a ScreenOverlay elementz {}_cbar.pngr)rsrtrrvrr{r SUBSET_XMIN SUBSET_YMINrz"add reference point as a star iconrg)rrrrrrrlZReferencePointg?rz4generating point element with step size of {} pixels)maxValuezrow=/)everysuffix)rr!)'r]r^rr readpix_boxrrrrhrr'r+r-r6r9r8rrFrrbrrrrrrHnanZFolderr#shaper progressBarrangeisnanupdatecloserr!)rrrr"rlrZdot_fileZ star_filelatslonsrrrxminyminrrZrlatZrlonZ ref_pointZ data_folderrrprog_barrrrrrrrrOrOrPwrite_kmz_placemarks              *  r*cCst|}t|jd}|dkrF|jrFd|jkrF|jd\|_|_nd|_tj|j|jd}tt ||d|_ t | |j |_ ddt|dt|df}td|td |j t||j }tj|j|j|j d d}|dkr|jr|tj|j|j|j d d8}tj|j|j|j|j d d}|dk r`td |jtj||dk<|jr~td tj||dk<~tj|||j|j|jd\}|_|_|_|jr|j|_d}|jst |\}|_t!||j"|_"t#|j"|j"|_"t$j%|jd|jdd|_&tj'|j|jt |d|_(|j)sBd|j(|_)t*j+,|j)|_)d|-krtt.|||j)|dnt/|||j)|j0|d|j)S)NrY timeseries_)rrrrzdata coverage in y/x: {}zsubset coverage in y/x: {})rr)rrrz.masking out pixels with zero value in file: {}z"masking out pixels with zero value)metadatar+wrapr/r)rsrt) datasetNames inps_dictz{}.kmzrS)rrrl)rrr"rl)1rnr rarrrref_datersubset_input_dict2boxvarsrrc coordinatecheck_box_within_data_coveragerFrrhattrupdate_attribute4subsetrr read_maskrrHrrscale_data4disp_unit_and_rewrapr+r.r/ disp_scaler''auto_adjust_colormap_lut_and_disp_limitauto_colormap_namer- ColormapExtrrrwauto_figure_title fig_titler r]r^r_rbrr*r")rkrlr=rmdata_boxrmaskcmap_lutrOrOrPmain sj  rD__main__r)N)ror.rpNrq)ror.rNrq)NNF)N)/r]sysrrAnumpyrHlxmlrzipfiler matplotlibrrirrrZ pykml.factoryrr ImportErrorrmintpy.objectsr mintpy.utilsr r r7r r rrcrrrrDrQrnrrrrrrr*rD__name__argvrOrOrOrP s<     9   (6 3^i W