1Vh{tdZddlZddlmZddlmZddlmZddlm Z ddl m Z dZ d Z d d Z d d Zy)zFMesh summaries and TensorFlow operations to create them. V2 versions N)tf2) summary_pb2)metadata)plugin_data_pb2) tensor_utilc 8tj|}|jj}|Dcgc]}||nd }}t j |d|||||} tj jt j||||| Scc}w)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. )valueN json_config)tagtensorstepr) tfconvert_to_tensorshapeas_listrcreate_summary_metadatasummarywriteget_instance_name) name descriptionr content_type componentsr rrdimtensor_metadatas S/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/plugins/mesh/summary_v2.py_write_summaryrs. ! ! /F LL "E7< =COS + =E =66   O ::    & &t\ :    >s Bc<d}|tj|d}|S)z6Parses and returns JSON string from python dictionary.z{}T) sort_keys)jsondumps) config_dictr s r_get_json_configr%Hs#Kjj= c Xt|}tj|tjj t jtj|tjjt jtj|tjjt jg}|D cgc]} | j| }} tj|D cgc]} | jc} } tt j j"ddxst j j$} d} | |d|5|D]+} | xr%t'||| j| j| ||} - ddd| Scc} wcc} w#1swY| SxYw)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. N summary_scopeT mesh_summary)values)r%r MeshTensorrMeshPluginDataVERTEXrfloat32FACEint32COLORuint8dataget_components_bitmaskrgetattrr experimentalr(r) rverticesfacescolorsr$rrr tensorsrrr( all_successs rmeshr<Psx>#;/K  o44;;RZZ   ?1166   O2288"((  G%,G&v{{/FvGGG00+23  3J  ''$? $ :: # #K t^G <  F%. ##+K   /H 4  sFF1F1FF)c  t|}g}tj|tjj t jtj|tjjt jtj|tjjt jg}|D cgc]} | j| }} tj|D cgc]} | jc} } |D]} | jj} | D cgc]} | | nd } } t!j"| j| j$} tj&|d| j| | ||}tj(|| j}|j+||| ft-j.}|D]$\}}} |j0j3||| &|Scc} wcc} wcc} w)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. Nr )dtyper )r rr)r%rr+rr,r-rr.r/r0r1r2r3r4rrrmake_tensor_proto data_typerrappendrSummaryr add)r r7r8r9r$rr summariesr:rrrr tensor_protosummary_metadata instance_tagrs rmesh_pbrHs(#;/KI o44;;RZZ   ?1166   O2288"((  G%,G&v{{/FvGGG00+23  3JI !!;@ACR/AA"44 KKv// $;;        #  11#v7J7JK ,(8,GH!I$!!#G8A 4 &  '7     N7H3BsHH3H, H )NNNNN)NNNN)__doc__r"tensorboard.compatrrtensorboard.compat.protortensorboard.plugins.meshrrtensorboard.utilrrr%r<rHr&rrOsM  (0-4((V   FTKO=r&