o 2h{@s|dZddlZddlmZddlmZddlmZddlm Z ddl m Z dd Z d d Z     dd d Z dddZdS)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. )valuecSg|] }|dur |ndqSN.0dimr r ^/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/plugins/mesh/summary_v2.py 6z"_write_summary..N json_config)tagtensorstepr) tfconvert_to_tensorshapeas_listrcreate_summary_metadatasummarywriteget_instance_name) name descriptionr content_type componentsrrrtensor_metadatar r r_write_summarys$   r$cCsd}|dur tj|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 tjjddp=tjj} d} | |d|d|D]} | oYt||| j| j| ||} qJWd| S1sfwY| 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. cSg|] }|jdur|qSNdatar rr r rrzmesh..cSg|]}|jqSr r!r.r r rr summary_scopeNT mesh_summary)values)r)r MeshTensorrMeshPluginDataVERTEXrfloat32FACEint32COLORuint8get_components_bitmaskgetattrr experimentalr3r$r-r!) rverticesfacescolorsr(rr rtensorsr"r3 all_successrr r rmeshPsF        rFc Cst|}g}t|tjjtjt|tjjtj t|tjj tj g}dd|D}t dd|D} |D]2} | j j} dd| D} tj| j | jd} tj|d| j| | ||d} t|| j}||| | fq6t}|D]\}} } |jj|| | dqo|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. cSr*r+r,r.r r rrr/zmesh_pb..cSr0r r1r.r r rrr2cSrr r r r r rrr)dtypeNr)rrr)r)rr6rr7r8rr9r:r;r<r=r>r-rrmake_tensor_proto data_typerr!rappendrSummaryradd)rrArBrCr(r r summariesrDr"rr tensor_protosummary_metadata instance_tagrr r rmesh_pbsN      rQ)NNNNN)NNNN)__doc__r&tensorboard.compatrrtensorboard.compat.protortensorboard.plugins.meshrrtensorboard.utilrr$r)rFrQr r r rs"     +  J