o eJ@sTdZddlZddlZddlmZddlmZddlmZeZ Gddde Z dS)zCLoader for event file data for an entire TensorBoard log directory.N)directory_watcher) io_wrapper) tb_loggingc@s0eZdZdZddZddZddZdd Zd S) LogdirLoaderaLoader for a root log directory, maintaining multiple DirectoryLoaders. This class takes a root log directory and a factory for DirectoryLoaders, and maintains one DirectoryLoader per "logdir subdirectory" of the root logdir. Note that this class is not thread-safe. cCs6|durtd|durtd||_||_i|_dS)aZConstructs a new LogdirLoader. Args: logdir: The root log directory to load from. directory_loader_factory: A factory for creating DirectoryLoaders. The factory should take a path and return a DirectoryLoader. Raises: ValueError: If logdir or directory_loader_factory are None. NzA logdir is requiredz&A directory loader factory is required) ValueError_logdir_directory_loader_factory_directory_loaders)selfZlogdirZdirectory_loader_factoryr RD:\Projects\ConvertPro\env\Lib\site-packages\tensorboard/uploader/logdir_loader.py__init__&s  zLogdirLoader.__init__cCstd|jt}t|jD]"}tj||j}| |||j vr2td|| ||j |<qt|j |}|rK|D] }td||j |=q>td|jdS)zFinds new runs within `logdir` and makes `DirectoryLoaders` for them. In addition, any existing `DirectoryLoader` whose run directory no longer exists will be deleted. zStarting logdir traversal of %sz&- Adding run for relative directory %sz(- Removing run for relative directory %szEnding logdir traversal of %sN) loggerinforsetrZGetLogdirSubdirectoriesospathrelpathaddr r)r Z runs_seensubdirrunZ stale_runsr r r synchronize_runs:s"      zLogdirLoader.synchronize_runscCsPt|j}tdt|t}t|D]}|j|}|| ||<q|S)aReturns tf.Event generators for each run's `DirectoryLoader`. Warning: the generators are stateful and consuming them will affect the results of any other existing generators for that run; calling code should ensure it takes events from only a single generator per run at a time. Returns: Dictionary containing an entry for each run, mapping the run name to a generator yielding tf.Event protobuf objects loaded from that run. z-Creating event loading generators for %d runs) listr rrlen collections OrderedDictsorted_wrap_loader_generatorLoad)r runsZ run_to_loaderrun_nameloaderr r r get_run_eventsRs   zLogdirLoader.get_run_eventsccs0z |D]}|VqWdStjyYdSw)zMWraps `DirectoryLoader` generator to swallow `DirectoryDeletedError`.N)rZDirectoryDeletedError)r Zloader_generatoritemr r r resz#LogdirLoader._wrap_loader_generatorN)__name__ __module__ __qualname____doc__r rr"rr r r r rs  r) r'rrZ$tensorboard.backend.event_processingrrZtensorboard.utilrZ get_loggerrobjectrr r r r s