&Vf{vdZddlZddlmZddlmZddlmZddlm Z ddl m Z dZ d Z d d Z d d ZdS)zFMesh summaries and TensorFlow operations to create them. V2 versions N)tf2) summary_pb2)metadata)plugin_data_pb2) tensor_utilc tj|}|j}d|D}t j|d|||||}tjt j|||||S)aCreates a tensor summary with summary metadata. Args: name: A name for this summary. The summary tag used for TensorBoard will be this name prefixed by any active name scopes. description: Optional long-form description for this summary, as a constant `str`. Markdown is supported. Defaults to empty. tensor: Tensor to display in summary. content_type: Type of content inside the Tensor. components: Bitmask representing present parts (vertices, colors, etc.) that belong to the summary. json_config: A string, JSON-serialized dictionary of ThreeJS classes configuration. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. Returns: A boolean indicating if summary was saved successfully or not. )valuecg|]}||nd SN.0dims `/var/www/html/software/conda/lib/python3.11/site-packages/tensorboard/plugins/mesh/summary_v2.py z"_write_summary..6s = = =COSS = = =N json_config)tagtensorstepr) tfconvert_to_tensorshapeas_listrcreate_summary_metadatasummarywriteget_instance_name) name descriptionr content_type componentsrrrtensor_metadatas r_write_summaryr&s. ! / / /F L " "E = =u = = =E6   O :    &t\ : :    rc:d}|tj|d}|S)z6Parses and returns JSON string from python dictionary.z{}NT) sort_keys)jsondumps) config_dictrs r_get_json_configr,Hs'Kj=== rc t|}tj|tjjt jtj|tjjt j tj|tjj t j g}d|D}tj d|D} tt jjddpt jj} d} | |d|5|D]#} | ot#||| j| j| ||} $ dddn #1swxYwY| S)aWrites a TensorFlow mesh summary. Args: name: A name for this summary. The summary tag used for TensorBoard will be this name prefixed by any active name scopes. vertices: Tensor of shape `[dim_1, ..., dim_n, 3]` representing the 3D coordinates of vertices. faces: Tensor of shape `[dim_1, ..., dim_n, 3]` containing indices of vertices within each triangle. colors: Tensor of shape `[dim_1, ..., dim_n, 3]` containing colors for each vertex. config_dict: Dictionary with ThreeJS classes names and configuration. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. description: Optional long-form description for this summary, as a constant `str`. Markdown is supported. Defaults to empty. Returns: True if all components of the mesh were saved successfully and False otherwise. c g|] }|j | SNdatarrs rrzmesh..GGG&v{/Fv/F/F/Frcg|] }|j Sr r#r2s rrzmesh..333 333r summary_scopeNT mesh_summary)values)r,r MeshTensorrMeshPluginDataVERTEXrfloat32FACEint32COLORuint8get_components_bitmaskgetattrr experimentalr7r&r1r#) r!verticesfacescolorsr+rr"rtensorsr$r7 all_successrs rmeshrJPs>#;//K  o4;RZ    ?16    O28"(   GHGGGGGG0337333J  '$?? $ : #K t^G < < <    F%. #++KK                 s'D;;D?D?c $t|}g}tj|tjjt jtj|tjjt j tj|tjj t j g}d|D}tj d|D} |D]} | j j} d| D} tj| j | j} tj|d| j| | ||} tj|| j}||| | ft-j}|D]#\}} } |j|| | $|S)aCreate a mesh summary to save in pb format. Args: tag: String tag for the summary. vertices: numpy array of shape `[dim_1, ..., dim_n, 3]` representing the 3D coordinates of vertices. faces: numpy array of shape `[dim_1, ..., dim_n, 3]` containing indices of vertices within each triangle. colors: numpy array of shape `[dim_1, ..., dim_n, 3]` containing colors for each vertex. config_dict: Dictionary with ThreeJS classes names and configuration. description: Optional long-form description for this summary, as a constant `str`. Markdown is supported. Defaults to empty. Returns: Instance of tf.Summary class. c g|] }|j | Sr/r0r2s rrzmesh_pb..r3rcg|] }|j Sr r5r2s rrzmesh_pb..r6rcg|]}||nd Sr r rs rrzmesh_pb..s AAACRAAAr)dtypeNr)rrr)r,rr:rr;r<rr=r>r?r@rArBr1rrmake_tensor_proto data_typerr#r appendrSummaryr add)rrErFrGr+r"r summariesrHr$rr tensor_protosummary_metadata instance_tagrs rmesh_pbrYs(#;//KI o4;RZ    ?16    O28"(   GHGGGGGG0337333JII !AA5AAA"4 Kv/   $;       #    1#v7JKK ,(8,GHHHH!##G8A  4 &  '7       Nr)NNNNN)NNNN)__doc__r)tensorboard.compatrrtensorboard.compat.protortensorboard.plugins.meshrrtensorboard.utilrr&r,rJrYr rrr_s  ((((((000000------444444(((((((((V   FFFFTKO======r