1VhTdZddlZddlmZddlmZddlmZdZdZ ejdGd d Z d Z d Z d Z ddZdZy)z+Internal information about the mesh plugin.N)Any) summary_pb2)plugin_data_pb2meshT)frozencXeZdZUdZeed<ejjed<eed<y) MeshTensoraA mesh tensor. Attributes: data: Tensor of shape `[dim_1, ..., dim_n, 3]` representing the mesh data of one of the following: - 3D coordinates of vertices - Indices of vertices within each triangle - Colors for each vertex content_type: Type of the mesh plugin data content. data_type: Data type of the elements in the tensor. data content_type data_typeN) __name__ __module__ __qualname____doc__r__annotations__rMeshPluginData ContentTypeQ/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/plugins/mesh/metadata.pyr r s%  I!00<<<Nrr cxd}|D]2}|tjjk(r td|d|zz}4|S)zCreates bitmask for all existing components of the summary. Args: content_type: list of plugin_data_pb2.MeshPluginData.ContentType, representing all components related to the summary. Returns: bitmask based on passed tensors. rz.Cannot include UNDEFINED content type in mask.)rr UNDEFINED ValueError) content_types componentsr s rget_components_bitmaskr2sOJ%6 ?99CC CMN N1 #45 6 rctS)z%Returns current verions of the proto.)_PROTO_VERSIONrrrget_current_versionr Bs rc^|dtjjj|S)zKReturns a unique instance name for a given summary related to the mesh._)rrrName)namer s rget_instance_namer%Gs+ &&2277 E rc(t|dk7rtdt|ztjt |||||}|j }tj||tjjt|S)aCreates summary metadata which defined at MeshPluginData proto. Arguments: name: Original merged (summaries of different types) summary name. display_name: The display name used in TensorBoard. content_type: Value from MeshPluginData.ContentType enum describing data. components: Bitmask representing present parts (vertices, colors, etc.) that belong to the summary. shape: list of dimensions sizes of the tensor. description: The description to show in TensorBoard. json_config: A string, JSON-serialized dictionary of ThreeJS classes configuration. Returns: A `summary_pb2.SummaryMetadata` protobuf object. z2Tensor shape should be of shape BxNx3, but got %s.)versionr$r rshape json_config) plugin_namecontent) display_namesummary_description plugin_data) lenrstrrrr SerializeToStringrSummaryMetadata PluginData PLUGIN_NAME) r$r-r rr) descriptionr*mesh_plugin_datar,s rcreate_summary_metadatar8Ps6 5zQ @3u: M  '55#% ! 002G  & &!'//::#W;  rcNt|ts tdtjj |}|j dk(r[ttjjtjjtjjg|_|S)a"Parse summary metadata to a Python object. Arguments: content: The `content` field of a `SummaryMetadata` proto corresponding to the mesh plugin. Returns: A `MeshPluginData` protobuf object. Raises: Error if the version of the plugin is not supported. zContent type must be bytes.r) isinstancebytes TypeErrorrr FromStringrrVERTEXFACECOLOR)r,results rparse_plugin_metadatarBs gu %566  + + 6 6w ?F A2..55..33..44   Mr)NN)r dataclassestypingrtensorboard.compat.protortensorboard.plugins.meshrr5r dataclassr rr r%r8rBrrrrHsj204 d#$$   .br