1Vh-dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZd Zd ZdZGdde j.Zy)zThe TensorBoard Custom Scalars plugin. This plugin lets the user create scalars plots with custom run-tag combinations by specifying regular expressions. See `http_api.md` in this directory for specifications of the routes for this plugin. N)wrappers) plugin_util) http_util)tf)provider) base_plugin) layout_pb2)metadata)scalars_plugin regex_valid tag_to_events<ceZdZdZej ZdZdZdZ dZ dZ e jjdZdZe jjd Zd Ze jjd Zd Zy )CustomScalarsPluginz%CustomScalars Plugin for TensorBoard.cj|j|_|j|_|j|_y)z}Instantiates ScalarsPlugin via TensorBoard core. Args: context: A base_plugin.TBContext instance. N)logdir_logdir data_provider_data_providerplugin_name_to_instance_plugin_name_to_instance)selfcontexts g/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/plugins/custom_scalar/custom_scalars_plugin.py__init__zCustomScalarsPlugin.__init__8s+ ~~ %33(/(G(G%cvtj|jvr|jtjSy)z~Tries to get the scalars plugin. Returns: The scalars plugin. Or None if it is not yet registered. N)scalars_metadata PLUGIN_NAMErrs r_get_scalars_pluginz'CustomScalarsPlugin._get_scalars_pluginBs3  ' '4+H+H H001A1M1MN NrcJ|j|j|jdS)N)z/download_dataz/layoutz/scalars)download_data_route layout_route scalars_router s rget_plugin_appsz#CustomScalarsPlugin.get_plugin_appsNs'"66((**  rcy)z?Plugin is active if there is a custom layout for the dashboard.Fr s r is_activezCustomScalarsPlugin.is_activeUsrc0tjddS)Nztf-custom-scalar-dashboardzCustom Scalars) element_nametab_name)rFrontendMetadatar s rfrontend_metadataz%CustomScalarsPlugin.frontend_metadataYs++5%  rctj|j}|jj d}|jj d}tj |j}|jj d} |j |||||\}}tj|||S#t$r,} tj|t| ddcYd} ~ Sd} ~ wwxYw)Nruntagformat text/plainrequestcontent content_typecode) rrenvironargsget experiment_iddownload_data_impl ValueErrorrRespondstr) rr6ctxr0r1 experimentresponse_formatbody mime_typees rr#z'CustomScalarsPlugin.download_data_route_s!!'//2llu%llu% ..w? !,,**84 "55S#z?OD)  $ :: $$A)   sC C5 !C0*C50C5ct|j}|s td|j|||||\}}||fS)a\Provides a response for downloading scalars data for a data series. Args: ctx: A tensorboard.context.RequestContext value. run: The run. tag: The specific tag. experiment: An experiment ID, as a possibly-empty `str`. response_format: A string. One of the values of the OutputFormat enum of the scalar plugin. Raises: ValueError: If the scalars plugin is not registered. Returns: 2 entities: - A JSON object response body. - A mime type (string) for the response. z\Failed to respond to request for /download_data. The scalars plugin is oddly not registered.)r!r? scalars_impl) rrBr0r1rCrDscalars_plugin_instancerErFs rr>z&CustomScalarsPlugin.download_data_implssV&#'":":"<&B 2>> c: iYrctj|j}|jj d}|jj d}tj |j}d} |j ||||}tj|||S#t$r,}tj|t|ddcYd}~Sd}~wwxYw)aGiven a tag regex and single run, return ScalarEvents. This route takes 2 GET params: run: A run string to find tags for. tag: A string that is a regex used to find matching tags. The response is a JSON object: { // Whether the regular expression is valid. Also false if empty. regexValid: boolean, // An object mapping tag name to a list of ScalarEvents. payload: Object, } r1r0application/jsonr3r4r5N) rrr:r;r<r=rIr?rr@rA) rr6rBtag_regex_stringr0rCrFrErGs rr%z!CustomScalarsPlugin.scalars_routes !!'//2"<<++E2llu% ..w? &  $$S#/?LD  $ :: $$A)   s8B## C,!C CCc t|stdtiiS tj|}|j j ||tjtj|g}d} ||}|r||j} | s tdtjj } |j#D cic],} |j%| r| | j'|| ||| d.}} tdtiS#tj$rtdtiicYSwxYw#t$ri}YwxYwcc} w)aGiven a tag regex and single run, return ScalarEvents. Args: ctx: A tensorboard.context.RequestContext value. run: A run string. tag_regex_string: A regular expression that captures portions of tags. Raises: ValueError: if the scalars plugin is not registered. Returns: A dictionary that is the JSON-able response. F)runs)r= plugin_namerun_tag_filterNzVFailed to respond to request for /scalars. The scalars plugin is oddly not registered.rT)_REGEX_VALID_PROPERTY_TAG_TO_EVENTS_PROPERTYrecompileerrorr list_scalarsrrr RunTagFilterKeyErrorr!r?r OutputFormatJSONkeysmatchrI) rrBr0rMrCregex run_to_data tag_to_datapayloadrJformr1s rrIz CustomScalarsPlugin.scalars_implsw &u'   JJ/0E))66 $(44#00se< 7   %c*K &*&>&>&@ #* F"..33D '++- ;;s# ,99c:tG "4 #W  Sxx %u'  " G  s)C=,D$<1D5=!D! D!$ D21D2ctj|j}tj|j}|j ||}t j ||dS)a@Fetches the custom layout specified by the config file in the logdir. If more than 1 run contains a layout, this method merges the layouts by merging charts within individual categories. If 2 categories with the same name are found, the charts within are merged. The merging is based on the order of the runs to which the layouts are written. The response is a JSON object mirroring properties of the Layout proto if a layout for any run is found. The response is an empty object if no layout could be found. rL)rrr:r= layout_implrr@)rr6rBrCrEs rr$z CustomScalarsPlugin.layout_routesT!!'//2 ..w? Z0  $0BCCrc i}d}|jj||tjt j tj gd}t|D]Q}||tj }|djj}tj} | jtjj||r| j D]} | j"|vr\|| j"j$j'| j$D cgc]} | || j"j$vr| !c} m|j j)j+| | || j"<0| }| j D]} | || j"<T|rt-j.|SiScc} w)N)tags)r=rPrQ downsampler)r read_tensorsr rrrXCONFIG_SUMMARY_TAGsortednumpyitemr LayoutParseFromStringrcompatas_bytescategorytitlechartextendadd MergeFromr proto_to_json) rrBrCtitle_to_category merged_layoutdatar0pointsr7 layout_protorrcs rrdzCustomScalarsPlugin.layout_impls ""// $ ,,#001120 $< AC#Yx::;FQioo**,G%,,.L  ( (););G)D E , 5 5EH~~)::*(..9??FF*2$%#$'8'H'N'N$O!"&..224>>xHr%rIr$rdr(rrrr3s/&&KH   !!;";&B!! ;" ;DD L!!D"D$1rr)rrTwerkzeugr tensorboardrtensorboard.backendrtensorboard.compatrtensorboard.datartensorboard.pluginsr!tensorboard.plugins.custom_scalarr r tensorboard.plugins.scalarrr rRrS_CONFIG_FILE_CHECK_THROTTLETBPluginrr(rrrsY #)!%+86C5&*!N+..Nr