1Vh+dZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z dd lmZd ZGd d e j"Zy) z&TensorBoard 3D mesh visualizer plugin.N)wrappers) http_util)provider) base_plugin)metadata)plugin_data_pb2) plugin_utildceZdZdZej ZdZdZdZ dZ e jjdZdZdZd Zd Zd Zd ZddZe jjdZe jjdZy ) MeshPluginz0A plugin that serves 3D visualization of meshes.c|j|_|jxsij|jt |_tjdd|_ y)zInstantiates a MeshPlugin via TensorBoard core. Args: context: A base_plugin.TBContext instance. A magic container that TensorBoard uses to make objects available to the plugin. meshr) data_kindlatest_known_versionN) data_provider_data_providersampling_hintsget plugin_name_DEFAULT_DOWNSAMPLING_downsample_tor _MetadataVersionChecker_version_checker)selfcontexts T/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/plugins/mesh/mesh_plugin.py__init__zMeshPlugin.__init__#sW&33&55;@@   3 !, C C!"! c |jj||tjt j |g|g}|||j }tj|S)z4Gets the `MeshPluginData` proto for an instance tag.runstags experiment_idrrun_tag_filter)r list_tensorsr PLUGIN_NAMEr RunTagFilterplugin_contentparse_plugin_metadata)rctx experimentrun instance_tagresultscontents r_instance_tag_metadataz!MeshPlugin._instance_tag_metadata3sj%%22 $ ,,#00U, 3 #,|,;;--g66rc>|j||||jS)z2Gets the user-facing tag name for an instance tag.)r1name)rr+r,r-r.s r_tagzMeshPlugin._tag@s#** S, $ rcP|jj||tjt j |g}|j |ijDcgc]3\}}|tj|jjk(r|5c}}Scc}}w)z2Gets the instance tag names for a user-facing tag.)r!r#) rr&rr'rr(ritemsr*r)r3)rr+r,r-tagindexr.tss r_instance_tagszMeshPlugin._instance_tagsFs##00 $ ,,#00se< 1 ',iiR&8&>&>&@ "rh44R5F5FGLLL    s&8B"ctj|j}tj|j}|jj ||t j}t}|jD]\}}t||<|jD]\}} t j| j} |jj| j||sM|j||||} |j!||||} d| j"di||| <t%j&||dS)aaA route (HTTP handler) that returns a response with tags. Args: request: The werkzeug.Request object. Returns: A response that contains a JSON object. The keys of the object are all the runs. Each run is mapped to a (potentially empty) list of all tags that are relevant to this plugin. )r$rsamplesrapplication/json)r renvironr$rr&rr'dictr6r*r)rokversionr4r1shaperRespond) rrequestr+r,all_runsresponser-r"r. metadatummdr7metas r _serve_tagszMeshPlugin._serve_tagsTs9!!'//2 ..w? &&33 $ ,,4 6!) @IC FHSM+/::< @' i33I4L4LM,,// CNiiZlC22S, '0A%? c" @ @  (4FGGrcJ|j|j|jdS)zGets all routes offered by the plugin. This method is called by TensorBoard when retrieving all the routes offered by the plugin. Returns: A dictionary mapping URL path to route that handles it. )z/tagsz/meshesz/data)rJ_serve_mesh_metadata_serve_mesh_datars rget_plugin_appszMeshPlugin.get_plugin_apps}s)%%00**  rcy)NFrNs r is_activezMeshPlugin.is_activesrc.tjdS)Nzmesh-dashboard) element_name)rFrontendMetadatarNs rfrontend_metadatazMeshPlugin.frontend_metadatas++9IJJrc<|j|jS)z0Returns a single sample from a batch of samples.)numpytolist)r tensor_datumsamples r _get_samplezMeshPlugin._get_samples!!&)0022rcN|j|j|||t|dS)aLConverts a TensorDatum into a JSON-compatible response. Args: event: TensorDatum object containing data in proto format. content_type: enum plugin_data_pb2.MeshPluginData.ContentType value, representing content type in TensorDatum. components: Bitmask representing all parts (vertices, colors, etc.) that belong to the summary. data_shape: list of dimensions sizes of the tensor. config: rendering scene configuration as dictionary. Returns: Dictionary of transformed metadata. ) wall_timestep content_type componentsconfig data_shape)r^r_list)reventr`rarcrbs r_get_tensor_metadatazMeshPlugin._get_tensor_metadatas-$JJ($z*   rc*|j||}|S)z6Convert a TensorDatum into a JSON-compatible response.)r\)rrer[datas r_get_tensor_datazMeshPlugin._get_tensor_datasv. rNc tj|j}tj|j}|jj d}|jj d}g}|j ||||D]}|jj||tjtj|g|g|j||} |j||||} || D cgc]} | | fc} z }|$|D cgc]} | dj|k(s| }} |St!|d}|Scc} wcc} w)z0Collects list of tensor events based on request.r-r7r )r$rr% downsamplec |djS)Nrl)r_) tensor_datas rz3MeshPlugin._collect_tensor_events..s{1~7J7Jr)key)r rr>r$argsrr:r read_tensorsrr'rr(rr1r_sorted) rrDr_r+r,r-r7 tensor_eventsr.tensorsrItensorres r_collect_tensor_eventsz!MeshPlugin._collect_tensor_eventssm!!'//2 ..w? llu%llu%  //ZcJ DL))66($00'44l^  ..7 !G..ZlD 7CtVnC CM D  #0E!HMMT4IM ##JMDs< EE ,E ct|jjdd}|j||}|jjd} tj j j|}t|jjdd}|Dcgc]&\}}|j|k(r|j||(}}}tj jtj tj j"tj$tj j&tj(i|} tj*|| }|j-d j/}tj||d S#t$rtj|dddcYSwxYwcc}}w) aUA route that returns data for particular summary of specified type. Data can represent vertices coordinates, vertices indices in faces, vertices colors and so on. Each mesh may have different combination of abovementioned data and each type/part of mesh summary must be served as separate roundtrip to the server. Args: request: werkzeug.Request containing content_type as a name of enum plugin_data_pb2.MeshPluginData.ContentType. Returns: werkzeug.Response either float32 or int32 data in binary format. r_gr`zBad content_typez text/plainir[r)dtype arraybuffer)floatrqrrwrMeshPluginData ContentTypeValue ValueErrorrrCintr`riVERTEXnpfloat32FACEint32COLORuint8arrayreshapetobytes) rrDr_rtr`r[rIrvrFnp_types rrMzMeshPlugin._serve_mesh_datas W\\%%fc2333GTB ||''7  *99EEKKLW\\%%h23!. f  L0  ! !&& 1    * * 1 12::  * * / /  * * 0 0"((    88HG4##B'//1  (MBB- $$+\3    s)F'+G!GGc |j|}|Dcgc]B\}}|j||j|j|j|j D}}}t j||dScc}}w)a]A route that returns the mesh metadata associated with a tag. Metadata consists of wall time, type of elements in tensor, scene configuration and so on. Args: request: The werkzeug.Request object. Returns: A JSON list of mesh data associated with the run and tag combination. r=)rwrfr`rarB json_configrrC)rrDrtrIrvrFs rrLzMeshPlugin._serve_mesh_metadata s33G< !.  f  % %!!          (4FGG  sAA8)N)__name__ __module__ __qualname____doc__rr'rrr1r4r:rRequest applicationrJrOrRrVr\rfrirwrMrLrQrrr r s:&&K 7  !!&H"&HP "K3 4 !F!!,C",C\!!H"Hrr )rrXrwerkzeugrtensorboard.backendrtensorboard.datartensorboard.pluginsrtensorboard.plugins.meshrr tensorboardr rTBPluginr rQrrrs>-)%+-4#FH%%FHr