1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
import netgen
from . import libngguipy
from . import libngpy
def StartGUI():
from tkinter import Tk
from netgen import config
import sys, os
try:
# the GUI tries to load ngsolve.tcl (which loads ngsolve shared libraries)
# BUT might fail to load dependencies (like mkl), these are handled by the
# ngsolve __init__.py script, so import ngsolve from python already here
import ngsolve
except:
pass
global win
win = Tk()
win.tk.eval('lappend ::auto_path ' + netgen._netgen_lib_dir)
win.tk.eval('lappend ::auto_path ' + netgen._netgen_bin_dir)
# load with absolute path to avoid issues on MacOS
win.tk.eval('load "'+netgen._netgen_lib_dir.replace('\\','/')+'/libnggui[info sharedlibextension]" gui')
if config.is_python_package and 'darwin' in sys.platform:
# libngsolve and other libraries are installed into netgen python dir to keep relative installation paths, but tcl won't find them there automatically
netgen_dir = os.path.abspath(os.path.dirname(netgen.__file__))
win.tk.eval(f'set netgen_library_dir {netgen_dir}')
win.tk.eval( netgen.libngpy._meshing._ngscript)
try:
from IPython import get_ipython
ipython = get_ipython()
ipython.magic('gui tk')
except:
pass
def _Redraw(*args, **kwargs):
if libngpy._meshing._Redraw(*args, **kwargs):
import netgen
import tkinter
cnt = 0
while(win.tk.dooneevent(tkinter._tkinter.DONT_WAIT) and cnt < 100):
cnt += 1
netgen._Redraw = _Redraw
_Redraw(blocking=True)
def Snapshot(w,h, filename=None):
netgen.Redraw(blocking=True)
import numpy
image = netgen.libngguipy.Snapshot(w, h)
image = numpy.array(image, dtype=numpy.uint8).reshape(h, w, 3)
image = image[::-1,:,:]
if filename:
import PIL.Image
im = PIL.Image.fromarray(image)
im.save(filename)
return image
def run_pyfile(filename):
with open(filename) as f:
exec(f.read(), {})
if __name__ == "__main__":
import sys, threading
StartGUI()
if(len(sys.argv) > 1):
if(sys.argv[1].endswith(".py")):
t = threading.Thread(target=run_pyfile, args=(sys.argv[1],),
daemon=True)
t.start()
win.mainloop()
else:
if not netgen.libngpy._meshing._netgen_executable_started:
import os
if not "NETGEN_DOCUMENTATION_RST_FORMAT" in os.environ:
StartGUI()
|
