U bb6X@sDddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z mZmZzddlmZWnek 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)subsetaexample: 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 # to generate placemarks for the file in radar coordinates, the corresponding # geometry file with latitude & longitude in radar coordinates are required, # such as provided by ISCE + MintPy workflow 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)rNz[Generate Google Earth KMZ file (overlay / placemarks for files in geo / radar coordinates).) descriptionformatter_classepilogfilezfile to be converted, in geo or radar coordinate. Note: for files in radar-coordinate, the corresponding lookup table in radar-coordinate (as provided by ISCE) is required.)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--stepstepzroutput one point per {step} pixels, to reduce file size (default: %(default)s). For file in radar-coordinate ONLY.)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_loc lower leftz lower rightz upper leftz upper rightz8Location of colorbar in the screen. Default: lower left.)rchoicesr&rz --cbar-label cbar_labelZLABELMean 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)parserfigrefrQE/home/centos/operations/rsmas_insar/sources/MintPy/mintpy/save_kmz.py create_parser8sr          rScCst}|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)rS parse_argsospathabspathdirnamerrYr read_attributekeysr"utget_geometry_fileisfileFileNotFoundErrorrr Exceptionformatpltswitch_backend)iargsrNinpsatrZ geom_ds_listrQrQrRcmd_line_parsets$    rpcm/yearr.g ףp= ?g @r:c Cstj|d\}} tj||d} tj| t|| dd} | jd||dd|rft j |d| _ | | j jd dd |jd |jd td ||j|d|dd|S)N)figsizevminvmaxvertical)cmapnorm orientationz{} [{}] )fontsize)nbinsboth)which labelsizewhitegffffff?writingtighti,) bbox_inches facecolordpi)rksubplotsr Normalizer ColorbarBaseget_cmap set_labelrjr MaxNLocatorlocator update_ticksax tick_paramspatch set_facecolor set_alphaprintsavefig get_facecolor) out_filerurvunitrxrsr}labelrOcaxrycbarrQrQrR plot_colorbars   rr7c Cst|||||||d}|ddkr0d\}} n|ddkrHd\}} |ddkrbd\} } n|dd krzd\} } ttd ttd tj |t d tj | |d d dtj | | d d dtjdddddtdtdtd} | S)N)rrurvrrxr}rrlower)0rupper)1rleftrightrz{}g?fraction)xyxunitsyunitsrZ250pixel)rsplitKMLZ ScreenOverlaynameIconhrefrjr_r`basenameZviewBoundScaleZ overlayXYZscreenXYsizerotationZ visibilityopen) cbar_filerurvrrxlocr}roysyoxsx cbar_overlayrQrQrRgenerate_cbar_elements>   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-ryrgbaZc_hexrQrQrR 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˚Latitude)r Longitudez.0fZRowColumnz.2f Value,)rZ PlacemarkStyle IconStylecolorrr-ryscalerrZ ExtendedDataDatavaluer+Point coordinates)latlonrowcolval icon_filern placemarkrQrQrRcreate_placemark_elementsF    )rFc Csv|dkr gn|}|dkrgn|}tj|}d|}d|}t}||td|t|d } | t j |dd dW5QRX|rtj tjtjd } |D]4} tj | tj| } t| |td | qtjt} t|t|dL}|g||D]6} | tj| |st| td | qW5QRXt| 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{}.kml{}.kmzzwriting wT) pretty_printutf8datazcopy {} to the local directoryz remove {}zmerged all files to {})r_r`rbrjrkmlappendrrwritertostringdecodejoinmintpy__file__rshutilcopy2ragetcwdchdirrrelpathremove) out_file_basekml_doc data_files res_filesr!rYZkml_filekmz_filerfZres_dirfnamesrc_fileZrun_dirfzrQrQrRwrite_kmz_files6      $     rcCsHt|\}}}}td|jsr@set_xlimset_ylimr_r`splitextrjr1rrDocumentrZ GroundOverlayrrrZ altitudeModeZ LatLonBoxnorthstreastsouthwestrrr+r6r;r9rr!)rmetarrnrrrrrOrrxrywidthlengthrZ data_png_filerZimg_nameZ img_overlayrrrrQrQrRwrite_kmz_overlayFs~             rc CsNtj|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}t!d||jD]}t!d||jD]^}|||f}t"|s|||f}| ||f}t|||||| |||d}||q|j#|d d d|d d|dq|$| |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.pngrV datasetNameboxrWz0plot and add colorbar as a ScreenOverlay elementrrr SUBSET_XMIN SUBSET_YMINrz"add reference point as a star iconrr)rrrrrrrnZReferencePointrrz4generating point element with step size of {} pixels)maxValuezrow=/)everysuffix)rr!)'r_r`rr readpix_boxrrrrjrr'r+r-r6r;r9rrHr rdrrrrrrJnanZFolderr#shaper progressBarrangeisnanupdatecloserr!)rrrr"rnrZdot_fileZ star_filelatslonsrrrxminyminrrZrlatZrlonZ ref_pointZ data_folderrrprog_barrrrrrrrrQrQrRwrite_kmz_placemarks               ( r3cCst|}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)Nr[ timeseries_)rrrrzdata coverage in y/x: {}zsubset coverage in y/x: {}r)rrrz.masking out pixels with zero value in file: {}z"masking out pixels with zero value)metadatar+wrapr/rrt) datasetNames inps_dictrrU)rrrn)rrr"rn)1rpr rcrrrref_datersubset_input_dict2boxvarsr&re coordinatecheck_box_within_data_coveragerHrrjattrupdate_attribute4subsetr%r read_maskrrJr'rscale_data4disp_unit_and_rewrapr+r7r/ disp_scaler''auto_adjust_colormap_lut_and_disp_limitauto_colormap_namer- ColormapExtrrryauto_figure_title fig_titler r_r`rardrr3r")rmrnr?rodata_boxrmaskcmap_lutrQrQrRmain sx  rM__main__r)N)rqr.rrNr:)rqr.r7Nr:)NNF)N)/r_sysrrCnumpyrJlxmlrzipfiler matplotlibrrkrrrZ pykml.factoryrr ImportErrorrmintpy.objectsr mintpy.utilsr r r@r r rrerrrrFrSrprrrrrrr3rM__name__argvrQrQrQrR sD     <   )6 3^i W