o 2hT@sdZddlZddlmZddlmZddlmZdZdZ ej ddGd d d Z d d Z d dZ ddZ  dddZddZdS)z+Internal information about the mesh plugin.N)Any) summary_pb2)plugin_data_pb2meshT)frozenc@s.eZdZUdZeed<ejjed<eed<dS) 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 ContentTyperr\/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/plugins/mesh/metadata.pyrs   rcCs2d}|D]}|tjjkrtd|d|>B}q|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 rrrget_components_bitmask2s  rcCstS)z%Returns current verions of the proto.)_PROTO_VERSIONrrrrget_current_versionBsrcCsd|tjj|fS)zKReturns a unique instance name for a given summary related to the mesh.z%s_%s)rrrName)namer rrrget_instance_nameGs rc CsXt|dkrtdt|tjt|||||d}|}tj||tjj t |ddS)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.)versionrr rshape json_config) plugin_namecontent) display_namesummary_description plugin_data) lenrstrrrrSerializeToStringrSummaryMetadata PluginData PLUGIN_NAME) rr%r rr! descriptionr"mesh_plugin_datar$rrrcreate_summary_metadataPs(  r0cCsHt|ts tdtj|}|jdkr"ttjjtjj tjj g|_|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$resultrrrparse_plugin_metadatas   r9)NN)r dataclassestypingrtensorboard.compat.protortensorboard.plugins.meshrr-r dataclassrrrrr0r9rrrrs      1