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()