a d=ic@sfdZddlZddlmZddlmZeddZdZdZ dd Z d d Z d d Z dddZ ddZdS)z+Internal information about the mesh plugin.N) summary_pb2)plugin_data_pb2 MeshTensor)data content_type data_typeZmeshcCs2d}|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.)rMeshPluginData UNDEFINED ValueError)Z content_types componentsrr r/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorboard/plugins/mesh/metadata.pyget_components_bitmask!s  rcCstS)z%Returns current verions of the proto.)_PROTO_VERSIONr r r rget_current_version1srcCsd|tjj|fS)zKReturns a unique instance name for a given summary related to the mesh.z%s_%s)rr Z ContentTypeName)namerr r rget_instance_name6s 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.)versionrrr shape json_config)Z plugin_namecontent) display_nameZsummary_descriptionZ plugin_data) lenr strrr rZSerializeToStringrZSummaryMetadataZ PluginData PLUGIN_NAME) rrrr r descriptionrZmesh_plugin_datarr r rcreate_summary_metadata?s(  rcCsHt|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 TypeErrorrr Z FromStringr rZVERTEXZFACEZCOLOR)rresultr r rparse_plugin_metadataps   r$)NN)__doc__ collectionsZtensorboard.compat.protorZtensorboard.plugins.meshr namedtuplerrrrrrrr$r r r rs   1