a dT@sdZddlZddlmZddlmZddlmZdZdZ ej ddGd d d Z d d Z d dZ ddZdddZddZdS)z+Internal information about the mesh plugin.N)Any) summary_pb2)plugin_data_pb2ZmeshT)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_typeZ data_typeN) __name__ __module__ __qualname____doc__r__annotations__rMeshPluginData ContentTyperrj/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/tensorboard/plugins/mesh/metadata.pyrs   rcCs2d}|D]$}|tjjkr td|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)Z content_types componentsrrrrget_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)namerrrrget_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.)versionrrrshape json_config)Z plugin_namecontent) display_nameZsummary_descriptionZ plugin_data) lenrstrrrrZSerializeToStringrZSummaryMetadataZ PluginData PLUGIN_NAME) rr!rrr descriptionrZmesh_plugin_datar rrrcreate_summary_metadataPs(  r&cCsHt|tstdtj|}|jdkrDttjjtjj 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 TypeErrorrrZ FromStringrrZVERTEXZFACEZCOLOR)r resultrrrparse_plugin_metadatas   r+)NN)r Z dataclassestypingrZtensorboard.compat.protorZtensorboard.plugins.meshrr$rZ dataclassrrrrr&r+rrrrs     1