ngsolve 6.2.2506.post74.dev0__cp314-cp314-macosx_10_15_universal2.whl

ngsolve/internal.py

@@ -0,0 +1,89 @@
+#######################################################
+# Visualization settings
+
+import ngsolve.solve
+import os
+
+# list of possible attributes -- necessary to provide autocompletion
+visoptions_variables = ['usetexture', 'invcolor', 'imaginary', 'lineartexture', 'numtexturecols', 'showclipsolution', 'showsurfacesolution', 'drawfieldlines', 'drawpointcurves', 'numfieldlines', 'fieldlinesrandomstart', 'fieldlinesstartarea', 'fieldlinesstartareap1x', 'fieldlinesstartareap1y', 'fieldlinesstartareap1z', 'fieldlinesstartareap2x', 'fieldlinesstartareap2y', 'fieldlinesstartareap2z', 'fieldlinesstartface', 'fieldlinesfilename', 'fieldlinestolerance', 'fieldlinesrktype', 'fieldlineslength', 'fieldlinesmaxpoints', 'fieldlinesthickness', 'fieldlinesvecfunction', 'fieldlinesphase', 'fieldlinesonlyonephase', 'lineplotfile', 'lineplotsource', 'lineplotusingx', 'lineplotusingy', 'lineplotautoscale', 'lineplotxmin', 'lineplotxmax', 'lineplotymin', 'lineplotymax', 'lineplotcurrentnum', 'lineplotinfos', 'lineplotselected', 'lineplotselector', 'lineplotcolor', 'lineplotsizex', 'lineplotsizey', 'lineplotselectedeval', 'lineplotdatadescr', 'lineplotxcoordselector', 'lineplotycoordselector', 'evaluatefilenames', 'evaluatefiledescriptions', 'clipsolution', 'scalfunction', 'vecfunction', 'evaluate', 'gridsize', 'xoffset', 'yoffset', 'autoscale', 'redrawperiodic', 'logscale', 'mminval', 'mmaxval', 'isolines', 'isosurf', 'subdivisions', 'numiso', 'autoredraw', 'autoredrawtime', 'simulationtime', 'multidimcomponent', 'deformation', 'scaledeform1', 'scaledeform2']
+
+viewoptions_variables = ['specpointvlen', 'colormeshsize', 'whitebackground', 'drawcoordinatecross', 'drawcolorbar', 'drawnetgenlogo', 'stereo', 'shrink', 'drawfilledtrigs', 'drawedges', 'drawbadels', 'centerpoint', 'drawelement', 'drawoutline', 'drawtets', 'drawtetsdomain', 'drawprisms', 'drawpyramids', 'drawhexes', 'drawidentified', 'drawpointnumbers', 'drawedgenumbers', 'drawfacenumbers', 'drawelementnumbers', 'drawdomainsurf', 'drawededges', 'drawedpoints', 'drawedpointnrs', 'drawedtangents', 'drawededgenrs', 'drawmetispartition', 'drawcurveproj', 'drawcurveprojedge', 'usecentercoords', 'centerx', 'centery', 'centerz', 'drawspecpoint', 'specpointx', 'specpointy', 'specpointz']
+
+clipping_variables = ['nx', 'ny', 'nz', 'dist', 'dist2', 'enable', 'onlydomain', 'notdomain']
+
+class TclVariables:
+    def __init__(self, name, update_cmd, attributes = []):
+        object.__setattr__(self, '_name', name)
+        object.__setattr__(self, '_update_cmd', update_cmd)
+        object.__setattr__(self, '_attributes', attributes)
+
+    # set corresponding variable in tcl and call update_cmd
+    def __setattr__(self, attribute_name, value):
+        if not attribute_name in self._attributes:
+            raise KeyError()
+        tcl_string = 'set '+self._name+'.'+attribute_name+' '+str(value)+'; '+self._update_cmd+';\n'
+        ngsolve.solve.Tcl_Eval(tcl_string)
+        ngsolve.Redraw()
+
+    # return list of possible attributes - for autocompletion
+    def __dir__(self):
+        return list(self.__dict__.keys()) + self._attributes
+
+    # rlcomplete checks existence of attribute with this function
+    def __getattr__(self, name):
+        if name in self.__dict__:
+            return self.__dict__[name]
+
+        if name in self._attributes:
+            return True
+        raise Exception()
+    def add_group( self, name, attributes=[] ):
+        self.__dict__[name] = TclVariables(self._name +'.'+name, self._update_cmd, attributes)
+
+
+visoptions = TclVariables('::visoptions', 'Ng_Vis_Set parameters', visoptions_variables)
+viewoptions = TclVariables('::viewoptions', 'Ng_SetVisParameters', viewoptions_variables)
+
+# add subgroups to viewoptions
+viewoptions.add_group( 'clipping', clipping_variables )
+viewoptions.add_group( 'light', ['amb', 'diff', 'spec', 'locviewer'] )
+viewoptions.add_group( 'mat', ['shininess', 'transp'] )
+
+def VideoStart(filename):
+    ngsolve.solve.Tcl_Eval("Ng_VideoClip .ndraw init " + filename+';\n')
+
+def VideoAddFrame():
+    ngsolve.solve.Tcl_Eval("Ng_VideoClip .ndraw addframe;\n")
+
+def VideoFinalize():
+    ngsolve.solve.Tcl_Eval("Ng_VideoClip .ndraw finalize;\n")
+
+def SnapShot(filename):
+    tmp_filename = filename.lower()
+    needs_conversion = not tmp_filename.endswith(".ppm")
+    if needs_conversion:
+        tmp_filename += ".ppm"
+    ngsolve.solve.Tcl_Eval("Ng_SnapShot .ndraw {};\n".format(tmp_filename))
+    ngsolve.Redraw(True)
+
+    if needs_conversion:
+        import time
+        from PIL import Image
+        t0 = time.time()
+        while not os.path.exists(tmp_filename) and time.time()-t0 < 5:
+            time.sleep(0.1)
+        im = Image.open(tmp_filename)
+        im.save(filename)
+        os.remove(tmp_filename)
+
+def Move(dx, dy):
+    ngsolve.solve.Tcl_Eval("Ng_MouseMove 0 0 {} {} move; redraw;\n".format(dx, -dy))
+
+def Rotate(dx, dy):
+    ngsolve.solve.Tcl_Eval("Ng_MouseMove 0 0 {} {} rotate; redraw;\n".format(dx, -dy))
+
+def Zoom(z):
+    ngsolve.solve.Tcl_Eval("Ng_MouseMove 0 0 0 {} zoom; redraw;\n".format(-z))
+
+def Center():
+    ngsolve.solve.Tcl_Eval("Ng_Center; redraw;\n")