From 4fb41ac1ae9a2a705c454587ba39267b75e02f83 Mon Sep 17 00:00:00 2001
From: Mihail <321mig123@gmail.com>
Date: Sun, 13 Nov 2022 12:28:47 +0500
Subject: [PATCH] space bar preoload

---
 nndt/space2/space_preloader.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/nndt/space2/space_preloader.py b/nndt/space2/space_preloader.py
index 59a4997..9215597 100644
--- a/nndt/space2/space_preloader.py
+++ b/nndt/space2/space_preloader.py
@@ -43,15 +43,18 @@ def __init__(
     def preload(self, space: Space, verbose=True):
 
         # Stage 1. Initialization of FileSources
-        for node in PostOrderIter(space):
+        iter_tmp = [node for node in PostOrderIter(space)]
+        if verbose:
+            iter_tmp = tqdm(iter_tmp, desc="Initialization of FileSources")
+
+        for node in iter_tmp:
             if isinstance(node, FileSource):
                 self._init_FileSource(node)
 
         # Stage 2. Initialization of Object3D
+        iter_tmp = [node for node in PostOrderIter(space)]
         if verbose:
-            iter_tmp = tqdm(PostOrderIter(space))
-        else:
-            iter_tmp = PostOrderIter(space)
+            iter_tmp = tqdm(iter_tmp, desc="Initialization of Object3D")
 
         for node in iter_tmp:
             if isinstance(node, Object3D):
@@ -64,7 +67,11 @@ def preload(self, space: Space, verbose=True):
                             node2._loader.unload_data()
 
         # Stage 3. Initialization of Group
-        for node in PostOrderIter(space):
+        iter_tmp = [node for node in PostOrderIter(space)]
+        if verbose:
+            iter_tmp = tqdm(iter_tmp, desc="Initialization of Group")
+
+        for node in iter_tmp:
             if isinstance(node, Group):
                 self._init_Group(node)