Moved Qt Stuff into separate file

display_utils.py

-import sys
-from PyQt5.QtWidgets import QWidget, QApplication, QTreeWidget, QTreeWidgetItem, QHBoxLayout
-from pystencils.astnodes import Block, LoopOverCoordinate, KernelFunction
-
 
 def toDot(expr, graphStyle={}):
     """Show a sympy or pystencils AST as dot graph"""
@@ -48,15 +44,6 @@ def showCode(ast):
     return CodeDisplay(ast)
 
 
-def debugGUI(ast):
-    app = QApplication.instance()
-    if app is None:
-        app = QApplication(sys.argv)
-    else:
-        print('QApplication instance already exists: %s' % str(app))
-    ex = DebugTree()
-    ex.insert_ast(ast)
-    app.exec_()
 
 # ----------------- Embedding of animations as videos in IPython notebooks ---------------------------------------------
 
@@ -188,39 +175,3 @@ def makeSurfacePlotAnimation(runFunction, frames=90, interval=30):
     return animation.FuncAnimation(fig, updatefig, interval=interval, frames=frames, blit=False)
 
 
-# -------------------- DEBUG GUI ------------------------
-
-
-class DebugTree(QWidget):
-    def __init__(self):
-        super().__init__()
-        self.initUI()
-
-    def initUI(self):
-        self.tree = QTreeWidget(self)
-        self.tree.setColumnCount(1)
-        self.tree.setHeaderLabel('repr')
-
-        hbox = QHBoxLayout()
-        hbox.stretch(1)
-        hbox.addWidget(self.tree)
-
-        self.setWindowTitle('Debug')
-        self.setLayout(hbox)
-        self.show()
-
-    def insert_ast(self, node, parent=None):
-        if parent is None:
-            parent = self.tree
-        if isinstance(node, Block):  # Blocks are represented with the tree structure
-            item = parent
-        else:
-            item = QTreeWidgetItem(parent)
-            item.setText(0, repr(node))
-
-        if node.func in [LoopOverCoordinate, KernelFunction]:
-            self.tree.expandItem(item)
-
-        for child in node.args:
-            self.insert_ast(child, item)
-

qtgui.py

+import sys
+from PyQt5.QtWidgets import QWidget, QApplication, QTreeWidget, QTreeWidgetItem, QHBoxLayout
+from pystencils.astnodes import Block, LoopOverCoordinate, KernelFunction
+
+
+def debugGUI(ast):
+    app = QApplication.instance()
+    if app is None:
+        app = QApplication(sys.argv)
+    else:
+        print('QApplication instance already exists: %s' % str(app))
+    ex = DebugTree()
+    ex.insert_ast(ast)
+    app.exec_()
+
+
+class DebugTree(QWidget):
+    def __init__(self):
+        super().__init__()
+        self.initUI()
+
+    def initUI(self):
+        self.tree = QTreeWidget(self)
+        self.tree.setColumnCount(1)
+        self.tree.setHeaderLabel('repr')
+
+        hbox = QHBoxLayout()
+        hbox.stretch(1)
+        hbox.addWidget(self.tree)
+
+        self.setWindowTitle('Debug')
+        self.setLayout(hbox)
+        self.show()
+
+    def insert_ast(self, node, parent=None):
+        if parent is None:
+            parent = self.tree
+        if isinstance(node, Block):  # Blocks are represented with the tree structure
+            item = parent
+        else:
+            item = QTreeWidgetItem(parent)
+            item.setText(0, repr(node))
+
+        if node.func in [LoopOverCoordinate, KernelFunction]:
+            self.tree.expandItem(item)
+
+        for child in node.args:
+            self.insert_ast(child, item)