ForkMaster/netgen
1
0
Fork 0
mirror of https://github.com/NGSolve/netgen.git synced 2024-11-11 16:49:16 +05:00
Code Issues Packages Projects Releases Wiki Activity

init_geom2d

Browse Source
This commit is contained in:
Matthias Hochsteger 2014-09-26 15:11:52 +00:00
parent ca9078c0d3
commit 46a6c155da
1 changed files with 42 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
python/init_geom2d.py
View File

@ -1,22 +1,16 @@
print ("Hello from init.py") try:
# import ngfem_cf # Linux
from libgeom2d.geom2d import *
except:
# Windows
from nglib.geom2d import *
def mydir(x=None):
if x==None:
return []
else:
return [i for i in dir(x) if not '__' in i]
def execfile(fname):
exec(open(fname).read())
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
def plotgeom(self): ############################################################################
if plotgeom.plot:
plt.close()
def plotgeom(self):
plt.close()
coords = self.PlotData() coords = self.PlotData()
for i in range(0,len(coords[2])): for i in range(0,len(coords[2])):
plt.plot(coords[2][i],coords[3][i],color='b') plt.plot(coords[2][i],coords[3][i],color='b')
@ -24,35 +18,48 @@ def plotgeom(self):
plt.axis('equal') plt.axis('equal')
plt.xlim(coords[0]) plt.xlim(coords[0])
plt.ylim(coords[1]) plt.ylim(coords[1])
plotgeom.plot = True
plt.show(block=False) plt.show(block=False)
plotgeom.plot = False SplineGeometry.Plot = plotgeom
del plotgeom
############################################################################
def plotpointindex(self,show = True): def plotpointindex(self,show = True):
try:
self._txt
except:
self._txt = list()
if show: if show:
if len(plotpointindex.txt) == 0: if len(self._txt) == 0:
pi = self.PointData() pi = self.PointData()
for i in range(0,len(pi[0])): for i in range(0,len(pi[0])):
plotpointindex.txt.append(plt.text(pi[0][i],pi[1][i],str(pi[2][i]))) self._txt.append(plt.text(pi[0][i],pi[1][i],str(pi[2][i])))
plotpointindex.txt.append(plt.plot(pi[0][i],pi[1][i],'ro')) self._txt.append(plt.plot(pi[0][i],pi[1][i],'ro'))
else: else:
pass pass
else: else:
for i in range(0,len(plotpointindex.txt)): for i in range(0,len(self._txt)):
try: try:
plotpointindex.txt[i].remove() self._txt[i].remove()
except: except:
plotpointindex.txt[i][0].remove() self._txt[i][0].remove()
plotpointindex.txt.clear() self._txt.clear()
#plt.draw() #plt.draw()
plt.show(block=False) plt.show(block=False)
plotpointindex.txt = list()
SplineGeometry.ShowPoints = plotpointindex
del plotpointindex
############################################################################
def plotdomainindex(self, show = True): def plotdomainindex(self, show = True):
try:
self._dom
except:
self._dom = list()
if show: if show:
if len(plotdomainindex.txt) == 0: if len(self._dom) == 0:
segdata = self.SegmentData() segdata = self.SegmentData()
for i in range(0,len(segdata[0])): for i in range(0,len(segdata[0])):
if segdata[0][i][2]: if segdata[0][i][2]:
@ -67,26 +74,21 @@ def plotdomainindex(self, show = True):
else: else:
vertr = 'bottom' vertr = 'bottom'
vertl = 'top' vertl = 'top'
plotdomainindex.txt.append(plt.text(segdata[0][i][0],segdata[0][i][1],str(segdata[2][i]),horizontalalignment=horl,verticalalignment=vertl)) self._dom.append(plt.text(segdata[0][i][0],segdata[0][i][1],str(segdata[2][i]),horizontalalignment=horl,verticalalignment=vertl))
plotdomainindex.txt.append(plt.text(segdata[1][i][0],segdata[1][i][1],str(segdata[3][i]),horizontalalignment=horr,verticalalignment=vertr)) self._dom.append(plt.text(segdata[1][i][0],segdata[1][i][1],str(segdata[3][i]),horizontalalignment=horr,verticalalignment=vertr))
else: else:
pass pass
else: else:
for i in range(0,len(plotdomainindex.txt)): for i in range(0,len(self._dom)):
plotdomainindex.txt[i].remove() self._dom[i].remove()
plotdomainindex.txt.clear() self._dom.clear()
#plt.draw() #plt.draw()
plt.show(block=False) plt.show(block=False)
plotdomainindex.txt = list()
from nglib.meshing import *
from nglib.geom2d import *
SplineGeometry.Plot = plotgeom
SplineGeometry.ShowPoints = plotpointindex
SplineGeometry.ShowDomains = plotdomainindex SplineGeometry.ShowDomains = plotdomainindex
del plotdomainindex
############################################################################
def Line(point_index1,point_index2): def Line(point_index1,point_index2):
return ["line",point_index1,point_index2] return ["line",point_index1,point_index2]
@ -94,25 +96,3 @@ def Line(point_index1,point_index2):
def Spline3(point_index1,point_index2,point_index3): def Spline3(point_index1,point_index2,point_index3):
return ["spline3",point_index1,point_index2,point_index3] return ["spline3",point_index1,point_index2,point_index3]
def startConsole():
import code
try:
import readline
import rlcompleter
readline.parse_and_bind("tab:complete") # autocomplete
except:
try:
import pyreadline as readline
import rlcompleter
readline.parse_and_bind("tab:complete") # autocomplete
except:
print('readline not found')
vars = globals()
vars.update(locals())
shell = code.InteractiveConsole(vars)
shell.interact()
startConsole()