added support for displaying VPython output in a pyqt window

Author: Aaron Titus

vpython/no_notebook.py

@@ -1,4 +1,4 @@
-from .vpython import GlowWidget, baseObj, vector, canvas
+from .vpython import GlowWidget, baseObj, vector, canvas, _browsertype
 from ._notebook_helpers import _in_spyder, _undo_vpython_import_in_spyder
 
 from http.server import BaseHTTPRequestHandler, HTTPServer
@@ -13,6 +13,11 @@
 import txaio
 import copy
 import socket
+import PyQt5.QtCore
+import PyQt5.QtWebEngineWidgets
+from PyQt5.QtWidgets import QApplication
+import multiprocessing
+
 
 import signal
 from urllib.parse import unquote
@@ -231,10 +236,26 @@ def onClose(self, wasClean, code, reason):
     else:
         __server = HTTPServer(('', __HTTP_PORT), serveHTTP)
         # or webbrowser.open_new_tab()
-        _webbrowser.open('http://localhost:{}'.format(__HTTP_PORT))
+        if(_browsertype=='default'): #uses default browser
+            _webbrowser.open('http://localhost:{}'.format(__HTTP_PORT)) #uses default browser
+
 except:
     pass
 
+
+def start_Qapp(port):
+    # creates a python browser with PyQt5
+    # runs qtbrowser.py in a separate process
+    filepath=os.path.dirname(__file__)
+    filename=filepath+'/qtbrowser.py'
+    os.system('python '+filename+' http://localhost:{}'.format(port))
+
+
+#create a browser in its own process
+if(_browsertype=='pyqt'): #uses default browser
+    __m = multiprocessing.Process(target=start_Qapp, args=(__HTTP_PORT,))
+    __m.start()
+
 __w = threading.Thread(target=__server.serve_forever)
 __w.start()
 

vpython/vpython.py

@@ -24,7 +24,7 @@
            'event_return', 'extrusion', 'faces', 'frame', 'gcurve', 'gdots',
            'ghbars', 'gobj', 'graph', 'gvbars', 'helix', 'label',
            'local_light', 'menu', 'meta_canvas', 'points', 'pyramid',
-           'quad', 'radio', 'ring', 'simple_sphere', 'sleep', 'slider', 'sphere',
+           'quad', 'radio', 'ring', 'set_browser', 'simple_sphere', 'sleep', 'slider', 'sphere',
            'standardAttributes', 'text', 'textures', 'triangle', 'vertex',
            'wtext', 'winput', 'keysdown']
 
@@ -4118,3 +4118,12 @@ def keysdown():
     for k in keysdownlist: # return a copy of keysdownlist
         keys.append(k)
     return keys
+
+# global variable for type of web browser to display vpython
+_browsertype = 'default'
+def set_browser(type='default'):
+    global _browsertype
+    if type=='pyqt':
+        _browsertype='pyqt'
+    else:
+        _browsertype='default'