[gui] [ipython] Add GUI.__del__ to close on variable dereference

python/taichi/lang/shell.py

@@ -15,7 +15,8 @@
 _env_enable_pybuf = os.environ.get('TI_ENABLE_PYBUF', '1')
 if not _env_enable_pybuf or int(_env_enable_pybuf):
     # When using in Jupyter / IDLE, the sys.stdout will be their wrapped ones.
-    pybuf_enabled = type(sys.stdout).__name__ != 'TextIOWrapper'
+    # While sys.__stdout__ should always be the raw console stdout.
+    pybuf_enabled = sys.stdout != sys.__stdout__
 
 from .core import taichi_lang_core
 taichi_lang_core.toggle_python_print_buffer(pybuf_enabled)

python/taichi/misc/gui.py

@@ -59,6 +59,9 @@ def __enter__(self):
     def __exit__(self, type, val, tb):
         self.close()
 
+    def __del__(self):
+        self.close()
+
     def close(self):
         self.core = None  # dereference to call GUI::~GUI()
 

taichi/gui/win32.cpp

@@ -223,6 +223,7 @@ GUI::~GUI() {
   if (show_gui) {
     std::free(data);
     DeleteDC(src);
+    DestroyWindow(hwnd);
     gui_from_hwnd.erase(hwnd);
   }
 }