U Vc{@spdZddlZddlmZddlmZddlmZddlm Z ddl m Z dd Z d d Z dd d ZdddZdS)zFMesh summaries and TensorFlow operations to create them. V2 versions N)tf2) summary_pb2)metadata)plugin_data_pb2) tensor_utilc CsXtj|d}|j}dd|D}tj|d|||||d}tjjt|||||dS)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. )valuecSsg|]}|dk r|ndqSN.0Zdimr r G/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/plugins/mesh/summary_v2.py 6sz"_write_summary..N json_config)tagtensorstepr) tfZconvert_to_tensorshapeZas_listrcreate_summary_metadatasummarywriteget_instance_name) name descriptionr content_type componentsrrrZtensor_metadatar r r _write_summarys$    rcCsd}|dk rtj|dd}|S)z6Parses and returns JSON string from python dictionary.z{}NT) sort_keys)jsondumps) config_dictrr r r _get_json_configHsr#c Cst|}t|tjjtjt|tjjtj t|tjj tj g}dd|D}t dd|D} t tjjddpztjj} d} | |d|d,|D] } | ot||| j| j| ||} qW5QRX| S) 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. cSsg|]}|jdk r|qSNdatar rr r r rs zmesh..cSsg|] }|jqSr rr'r r r rs summary_scopeNTZ mesh_summary)values)r#r MeshTensorrMeshPluginDataVERTEXrfloat32FACEint32COLORuint8get_components_bitmaskgetattrrZ experimentalr)rr&r) rverticesfacescolorsr"rrrtensorsrr)Z all_successrr r r meshPsJ   r9c Cst|}g}t|tjjtjt|tjjtj t|tjj tj g}dd|D}t dd|D} |D]d} | j j} dd| D} tj| j | jd} tj|d| j| | ||d} t|| j}||| | fqlt}|D]\}} } |jj|| | dq|S) 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. cSsg|]}|jdk r|qSr$r%r'r r r rs zmesh_pb..cSsg|] }|jqSr r(r'r r r rscSsg|]}|dk r|ndqSrr r r r r rs)ZdtypeNr)rrr)r#rr+rr,r-rr.r/r0r1r2r3r&rrZmake_tensor_protoZ data_typerrrappendrZSummaryradd)rr5r6r7r"rrZ summariesr8rrrZ tensor_protoZsummary_metadataZ instance_tagrr r r mesh_pbs`   r<)NNNNN)NNNN)__doc__r Ztensorboard.compatrrZtensorboard.compat.protorZtensorboard.plugins.meshrrZtensorboard.utilrrr#r9r<r r r r s&     +  J