def moveNodeUp():
if CTK.t == []: return
node = CTK.TKTREE.getCurrentSelectedNode()
if node[3] == 'CGNSTree_t': return # Tree node can not move
(p, c) = Internal.getParentOfNode(CTK.t, node)
if c == 0: return # already first
if node[3] == 'Zone_t': # optimise
z1 = p[2][c - 1]
z2 = p[2][c]
if z1[3] != 'Zone_t':
temp = p[2][c - 1]
p[2][c - 1] = p[2][c]
p[2][c] = temp
CTK.TKTREE.updateApp()
return
(nob1, noz1) = C.getNobNozOfZone(z1, CTK.t)
(nob2, noz2) = C.getNobNozOfZone(z2, CTK.t)
CTK.replace(CTK.t, nob1, noz1, z2)
CTK.replace(CTK.t, nob2, noz2, z1)
CTK.TKTREE.updateApp()
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CPlot.render()
elif node[3] == 'CGNSBase_t': # non optimise
temp = p[2][c - 1]
p[2][c - 1] = p[2][c]
p[2][c] = temp
CTK.TKTREE.updateApp()
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.display(CTK.t)
else: # n'impacte pas CPlot
temp = p[2][c - 1]
p[2][c - 1] = p[2][c]
p[2][c] = temp
CTK.TKTREE.updateApp()
def drawCubic(npts):
global CURRENTZONE; global CURRENTPOLYLINE
if (CTK.t == []): return
w = WIDGETS['draw']
if (CTK.__BUSY__ == False):
CPlot.unselectAllZones()
CTK.saveTree()
CTK.__BUSY__ = True
TTK.sunkButton(w)
CPlot.setState(cursor=1)
while (CTK.__BUSY__ == True):
l = []
while (l == []):
l = CPlot.getActivePoint()
CPlot.unselectAllZones()
time.sleep(CPlot.__timeStep__)
w.update()
if (CTK.__BUSY__ == False): break
if (CTK.__BUSY__ == True):
CURRENTPOLYLINE.append((l[0],l[1],l[2]))
if (CURRENTZONE == None):
CTK.t = C.addBase2PyTree(CTK.t, 'CONTOURS', 1)
base = Internal.getNodeFromName1(CTK.t, 'CONTOURS')
nob = C.getNobOfBase(base, CTK.t)
a = D.polyline(CURRENTPOLYLINE)
CURRENTZONE = a
CTK.add(CTK.t, nob, -1, a)
ret = Internal.getParentOfNode(CTK.t, CURRENTZONE)
noz = ret[1]
else:
a = D.polyline(CURRENTPOLYLINE)
CURRENTZONE = a
CTK.replace(CTK.t, nob, noz, a)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
ret = Internal.getParentOfNode(CTK.t, CURRENTZONE)
a = D.polyline(CURRENTPOLYLINE)
d = G.cart( (0,0,0), (1./(npts-1),1,1), (npts,1,1) )
a = G.map(a, d)
surfaces = getSurfaces()
if (surfaces != []): a = T.projectOrthoSmooth(a, surfaces)
nob = C.getNobOfBase(ret[0], CTK.t)
CTK.replace(CTK.t, nob, ret[1], a)
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
CURRENTZONE = None
CURRENTPOLYLINE = []
TTK.raiseButton(w)
CPlot.setState(cursor=0)
def selectCells(event=None):
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
formula = VARS[0].get()
strict = VARS[1].get()
strict = int(strict)
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
z = P.selectCells(z, formula, strict=strict)
CTK.replace(CTK.t, nob, noz, z)
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TXT.insert('START', 'Cells selected.\n')
CTK.TKTREE.updateApp()
CPlot.render()
def streamRibbon():
if CTK.t == []: return
npts = CTK.varsFromWidget(VARS[0].get(), type=2)
if len(npts) != 1:
CTK.TXT.insert('START', 'Number of points in stream incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
npts = npts[0]
v1 = VARS[1].get()
v2 = VARS[2].get()
v3 = VARS[3].get()
CTK.TXT.insert('START', 'Click to select starting point...\n')
l = []
while (l == []):
l = CPlot.getActivePoint()
time.sleep(0.1)
print('Ribbon: starting point %d' % l)
CTK.saveTree()
CTK.t = C.addBase2PyTree(CTK.t, 'STREAMS', 2)
b = Internal.getNodesFromName1(CTK.t, 'STREAMS')
nob = C.getNobOfBase(b[0], CTK.t)
try:
stream = P.streamRibbon(CTK.t, (l[0], l[1], l[2]), (0, 0, 0.01),
[v1, v2, v3],
N=npts)
CTK.add(CTK.t, nob, -1, stream)
CTK.TXT.insert('START', 'Stream ribbon created.\n')
except Exception as e:
Panels.displayErrors([0, str(e)], header='Error: streamRibbon')
CTK.TXT.insert('START', 'Stream ribbon fails.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CTK.display(CTK.t)
def pointedHat():
if (CTK.t == []): return
if (CTK.__MAINTREE__ <= 0):
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
args = VARS[3].get()
args = args.split(';')
if (len(args) != 3): return
x0 = float(args[0])
y0 = float(args[1])
z0 = float(args[2])
if (nzs == []):
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
else:
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
CTK.t[2][nob][2][noz] = G.pointedHat(CTK.t[2][nob][2][noz],
(x0, y0, z0))
CTK.TXT.insert('START', 'Pointed hat created.\n')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CTK.display(CTK.t)
return
def convert2Tetra():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
fail = False
errors = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
try:
a = C.convertArray2Tetra(CTK.t[2][nob][2][noz])
CTK.replace(CTK.t, nob, noz, a)
except Exception as e:
fail = True
errors += [0, str(e)]
if not fail: CTK.TXT.insert('START', 'Zones converted to tetra.\n')
else:
Panels.displayErrors(errors, header='Error: convert2Tetra')
CTK.TXT.insert('START',
'Tetra conversion fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def adapt(event=None):
if CTK.t == []: return
import Converter.elsAProfile
CTK.t = Converter.elsAProfile.convert2elsAxdt(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CTK.TXT.insert('START', 'Tree adapted for elsA.CGNS.\n')
def extractBodies():
pref = 'BODY#'
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
baseName = CTK.t[2][nob][0]
bodyName = pref + baseName
CTK.t = C.addBase2PyTree(CTK.t, bodyName, cellDim=2)
nodes = Internal.getNodesFromName(CTK.t, bodyName)
p = Internal.getParentOfNode(CTK.t, nodes[0])
walls = C.extractBCOfType(z, 'BCWall')
walls += C.extractBCOfType(z, 'BCWallInviscid')
walls += C.extractBCOfType(z, 'BCWallViscous')
CTK.t[2][p[1]][2] = CTK.t[2][p[1]][2] + walls
CTK.TXT.insert('START', 'Walls extracted.\n')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CTK.display(CTK.t)
def setBackground(event=None):
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
type = VARS[0].get()
CTK.saveTree()
if type == 'None':
deleteBackgroundBase()
else:
deleteBackgroundBase()
CTK.t = C.addBase2PyTree(CTK.t, 'BACKGROUND', 2)
if type == 'Half-Box': B = createBox(1)
elif type == 'Box': B = createBox(0)
elif type == 'Z-Half-Box': B = createBox(1, 1)
elif type == 'Z-Box': B = createBox(0, 1)
elif type == 'Z-Ellipse': B = createZEllipse()
elif type == 'Z-Plane': B = createZPlane()
elif type == 'Z-Square-Ground': B = createGround()
base = Internal.getNodesFromName1(CTK.t, 'BACKGROUND')[0]
nob = C.getNobOfBase(base, CTK.t)
for b in B:
CTK.add(CTK.t, nob, -1, b)
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def extendSurf():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
# surfaces
name = VARS[0].get()
names = name.split(';')
surfaces = []
for v in names:
v = v.lstrip()
v = v.rstrip()
sname = v.split('/', 1)
bases = Internal.getNodesFromName1(CTK.t, sname[0])
if bases != []:
nodes = Internal.getNodesFromType1(bases[0], 'Zone_t')
for z in nodes:
if z[0] == sname[1]: surfaces.append(z)
# - Hauteur de chaque maille -
dhloc = CTK.varsFromWidget(VARS[1].get(), type=1)
dhloc = dhloc[0]
N = CTK.varsFromWidget(VARS[2].get(), type=2)
N = N[0]
dh = G.cart((0., 0., 0.), (dhloc, 1., 1.), (N + 1, 1, 1))
# - nb d'iterations de lissage -
nit = CTK.varsFromWidget(VARS[3].get(), type=2)
nit = nit[0]
# - contour -
nzs = CPlot.getSelectedZones()
if (nzs == []):
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
contours = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
contour = C.convertBAR2Struct(z)
contours.append(contour)
# Extension
zlist = G.buildExtension(contours, surfaces, dh, niter=nit)
c = 0
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
CTK.t[2][nob][2][noz] = zlist[c]
c += 1
CTK.TXT.insert('START', 'Surface extension done.\n')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CTK.display(CTK.t)
def addkplanes():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
N = CTK.varsFromWidget(VARS[1].get(), type=2)
if len(N) != 1:
CTK.TXT.insert('START', 'Number of layers is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
N = N[0]
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
fail = False
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
#try:
z = T.addkplane(CTK.t[2][nob][2][noz], N=N)
CTK.replace(CTK.t, nob, noz, z)
#except Exception as e: fail = True
if not fail: CTK.TXT.insert('START', 'K planes added.\n')
else:
CTK.TXT.insert('START', 'add K planes failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def outCore():
if (CTK.t == []): return
if (CTK.__MAINTREE__ <= 0):
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if (nzs == []):
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
ooc = Internal.getNodesFromName1(z, 'OutOfCore')
if (len(ooc) == 0):
# Save zone
base = CTK.t[2][nob]
name = '.' + base[0] + '#' + z[0] + '.cgns'
t = C.newPyTree(['Base'])
t[2][1][2].append(z)
C.convertPyTree2File(t, name)
# Replace zone
bb = G.BB(z)
bb[2].append(['OutOfCore', numpy.array([1]), [], \
'UserDefinedData_t'])
bb = CPlot.addRender2Zone(bb, blending=0.2)
CTK.replace(CTK.t, nob, noz, bb)
CTK.saveTree() # il est apres pour forcer le flush
CTK.TXT.insert('START', 'Selected zones out of core.\n')
CTK.t = C.fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def inCore():
if CTK.t == []: return
if (CTK.__MAINTREE__ <= 0):
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
ooc = Internal.getNodesFromName1(z, 'OutOfCore')
if (len(ooc) != 0):
# Read zone
base = CTK.t[2][nob]
name = '.' + base[0] + '#' + z[0] + '.cgns'
t = C.convertFile2PyTree(name)
CTK.replace(CTK.t, nob, noz, t[2][1][2][0])
CTK.saveTree() # il est apres pour forcer le flush
CTK.TXT.insert('START', 'Selected zones in core.\n')
CTK.t = C.fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def rmBlock():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
deletedZoneNames = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
deletedZoneNames.append(CTK.t[2][nob][0] + Internal.SEP1 +
CTK.t[2][nob][2][noz][0])
CTK.saveTree()
CTK.t = CPlot.deleteSelection(CTK.t, CTK.Nb, CTK.Nz, nzs)
CTK.TXT.insert('START', 'Selected zones deleted.\n')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.delete(deletedZoneNames)
CPlot.render()
def TRITFI():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if len(nzs) == 0:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
surf = getSurfaces()
zones = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
z = C.convertBAR2Struct(z)
zones.append(z)
if len(zones) != 3:
CTK.TXT.insert('START', 'TRI TFI takes 3 contours.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
coords1 = C.getFields(Internal.__GridCoordinates__, zones[0])[0]
coords2 = C.getFields(Internal.__GridCoordinates__, zones[1])[0]
coords3 = C.getFields(Internal.__GridCoordinates__, zones[2])[0]
[m1, m2, m3] = trimesh(coords1, coords2, coords3)
if (m1 == 0):
CTK.TXT.insert('START', m2 + '\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
m1 = C.convertArrays2ZoneNode('TFI1', [m1])
m2 = C.convertArrays2ZoneNode('TFI2', [m2])
m3 = C.convertArrays2ZoneNode('TFI3', [m3])
if surf != []:
m1 = T.projectOrthoSmooth(m1, surf)
m2 = T.projectOrthoSmooth(m2, surf)
m3 = T.projectOrthoSmooth(m3, surf)
CTK.saveTree()
CTK.t = C.addBase2PyTree(CTK.t, 'MESHES')
bases = Internal.getNodesFromName1(CTK.t, 'MESHES')
nob = C.getNobOfBase(bases[0], CTK.t)
for i in [m1, m2, m3]:
CTK.add(CTK.t, nob, -1, i)
CTK.TXT.insert('START', 'TRI-TFI mesh created.\n')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def importFile(event=None):
if CTK.t == []: return
s = VARS[4].get()
s = s.split(';')
try:
t1 = []
for filename in s:
if filename != '':
t2 = C.convertFile2PyTree(filename)
# Fusion des bases de t et t2
if t1 == []: t1 = t2
else: t1 = C.mergeTrees(t1, t2)
except:
CTK.TXT.insert('START', 'Import failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
if t1 == []:
CTK.TXT.insert('START', 'Import failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
nzs = CPlot.getSelectedZones()
# Essaie de trouver une methode adaptee
method = 1 # match by geom
import sets
zoneNames = sets.Set(C.getZoneNames(CTK.t, prefixByBase=False))
zoneNames1 = sets.Set(C.getZoneNames(t1, prefixByBase=False))
inter = zoneNames & zoneNames1
linter = len(inter) * 1.
comp = min(len(zoneNames), len(zoneNames1)) * 1.
if linter / comp > 0.9: method = 0 # try match by name (safer)
if CTK.__MAINTREE__ <= 0 or nzs == []:
CTK.t = P.importVariables(t1, CTK.t, method=method)
else:
zones = C.newPyTree(['Base'])
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
zone = CTK.t[2][nob][2][noz]
zones[2][1][2].append(zone)
zones = P.importVariables(t1, zones, method=method)
c = 0
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
CTK.t[2][nob][2][noz] = zones[2][1][2][c]
c += 1
CTK.TXT.insert('START', 'Variable file %s imported.\n' % filename)
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CTK.display(CTK.t)
if CTK.TKPLOTXY is not None: CTK.TKPLOTXY.updateApp()
def uniformize(event=None):
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
type = VARS[2].get()
density = -1; npts = 2; factor = -1
if type == 'Density':
density = CTK.varsFromWidget(VARS[0].get(), 1)
if len(density) != 1:
CTK.TXT.insert('START', 'Invalid points density.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
density = density[0]
elif type == 'Npts':
npts = CTK.varsFromWidget(VARS[0].get(), 2)
if len(npts) != 1:
CTK.TXT.insert('START', 'Invalid number of points.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
npts = npts[0]
elif type == 'Factor':
factor = CTK.varsFromWidget(VARS[0].get(), 1)
if len(factor) != 1:
CTK.TXT.insert('START', 'Invalid number factor.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
CTK.saveTree()
# Get first selected zone
nz = nzs[0]
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
zone = CTK.t[2][nob][2][noz]
dim = Internal.getZoneDim(zone)
if dim[0] == 'Structured':
if dim[2] != 1 and dim[3] != 1:
fail = apply3D(density, npts, factor, ntype=0)
elif dim[2] != 1 and dim[3] == 1:
fail = apply2D(density, npts, factor, ntype=0)
else: fail = uniformize1D(density, npts, factor)
else: fail = uniformize1D(density, npts, factor) # all zones
if not fail:
CTK.TXT.insert('START', 'Uniformize successfull.\n')
else:
CTK.TXT.insert('START', 'Uniformize edge fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def union():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if len(nzs) < 2:
CTK.TXT.insert('START', 'Please, select two or more surfaces.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
tol = CTK.varsFromWidget(VARS[0].get(), type=1)
if len(tol) != 1:
CTK.TXT.insert('START', 'Tolerance is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
tol = tol[0]
CTK.saveTree()
zlist = []
deletedZoneNames = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
deletedZoneNames.append(CTK.t[2][nob][0] + Internal.SEP1 +
CTK.t[2][nob][2][noz][0])
z = CTK.t[2][nob][2][noz]
zlist.append(z)
try:
j = XOR.booleanUnion(zlist[0], zlist[1], tol=tol)
except Exception as e:
Panels.displayErrors([0, str(e)], header='Error: union')
CTK.TXT.insert('START', 'Union failed\n')
return
for nz in range(len(zlist) - 2):
try:
j = XOR.booleanUnion(j, zlist[nz + 2], tol=tol)
except Exception as e:
Panels.displayErrors([0, str(e)], header='Error: union')
CTK.TXT.insert('START', 'Union failed.\n')
return
CTK.t = CPlot.deleteSelection(CTK.t, CTK.Nb, CTK.Nz, nzs)
CPlot.delete(deletedZoneNames)
CTK.add(CTK.t, CTK.Nb[0] + 1, -1, j)
CTK.TXT.insert('START', 'Union performed.\n')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def createText(event=None):
CTK.saveTree()
text = VARS[0].get()
type = VARS[1].get()
font = VARS[3].get()
smoothness = VARS[2].get()
smooth = 0
if smoothness == 'Regular': smooth = 0
elif smoothness == 'Smooth': smooth = 2
elif smoothness == 'Very smooth': smooth = 4
CTK.t = C.addBase2PyTree(CTK.t, 'TEXT', 3)
nodes = Internal.getNodesFromName1(CTK.t, 'TEXT')
base = nodes[0]
if type == '3D': a = D.text3D(text, smooth=smooth, font=font)
elif type == '2D': a = D.text2D(text, smooth=smooth, font=font)
elif type == '1D':
a = D.text1D(text, smooth=smooth, font=font)
a = C.convertArray2Tetra(a)
a = T.join(a)
# Modification de l'angle, de la position et de la taille du texte
# en fonction du point de vue
posCam = CPlot.getState('posCam')
posEye = CPlot.getState('posEye')
dirCam = CPlot.getState('dirCam')
BB = G.bbox(a)
xc = 0.5 * (BB[3] + BB[0])
yc = 0.5 * (BB[4] + BB[1])
zc = 0.5 * (BB[5] + BB[2])
a = T.translate(a, (posEye[0] - xc, posEye[1] - yc, posEye[2] - zc))
lx = posEye[0] - posCam[0]
ly = posEye[1] - posCam[1]
lz = posEye[2] - posCam[2]
if (lx * lx + ly * ly + lz * lz == 0.): lx = -1
if (dirCam[0] * dirCam[0] + dirCam[1] * dirCam[1] +
dirCam[2] * dirCam[2] == 0.):
dirCam = (0, 0, 1)
ll = math.sqrt(lx * lx + ly * ly + lz * lz)
a = T.homothety(a, (posEye[0], posEye[1], posEye[2]), 0.01 * ll)
ux = dirCam[1] * lz - dirCam[2] * ly
uy = dirCam[2] * lx - dirCam[0] * lz
uz = dirCam[0] * ly - dirCam[1] * lx
a = T.rotate(a, (posEye[0], posEye[1], posEye[2]),
((1, 0, 0), (0, 1, 0), (0, 0, 1)),
((-ux, -uy, -uz), dirCam, (lx, ly, lz)))
nob = C.getNobOfBase(base, CTK.t)
CTK.add(CTK.t, nob, -1, a)
#C._fillMissingVariables(CTK.t)
CTK.TXT.insert('START', 'Text created.\n')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def extrudeWCurve(mode=0):
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
# - curve -
name = VARS[3].get()
names = name.split(';')
curve = []
for v in names:
v = v.lstrip()
v = v.rstrip()
sname = v.split('/', 1)
bases = Internal.getNodesFromName1(CTK.t, sname[0])
if bases != []:
nodes = Internal.getNodesFromType1(bases[0], 'Zone_t')
for z in nodes:
if z[0] == sname[1]: curve.append(z)
if len(curve) == 0:
CTK.TXT.insert('START', 'Curve is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
fail = False
errors = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
try:
if mode == 0: z = D.lineDrive(z, curve)
else: z = D.orthoDrive(z, curve)
CTK.replace(CTK.t, nob, noz, z)
except Exception as e:
fail = True
errors += [0, str(e)]
if not fail:
CTK.TXT.insert('START', 'Mesh extruded with curve(s).\n')
else:
Panels.displayErrors(errors, header='Error: extrusion')
CTK.TXT.insert('START', 'Extrusion fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def extractIsoLine():
if (CTK.t == []): return
if (CTK.__MAINTREE__ <= 0):
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
field = VARS[0].get()
value = VARS[2].get()
try:
value = float(value)
except:
value = 1.
nzs = CPlot.getSelectedZones()
CTK.saveTree()
if nzs == []:
z = Internal.getZones(CTK.t)
else:
z = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z.append(CTK.t[2][nob][2][noz])
isos = []
fail = False
errors = []
for zone in z:
try:
i = P.isoLine(zone, field, value)
isos.append(i)
except Exception as e:
fail = True
errors += [0, str(e)]
CTK.t = C.addBase2PyTree(CTK.t, 'CONTOURS', 1)
bases = Internal.getNodesFromName1(CTK.t, 'CONTOURS')
nob = C.getNobOfBase(bases[0], CTK.t)
for i in isos:
CTK.add(CTK.t, nob, -1, i)
if (fail == False):
CTK.TXT.insert('START', 'Isolines extracted.\n')
else:
Panels.displayErrors(errors, header='Error: Isolines')
CTK.TXT.insert('START', 'Isolines fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def extract():
if CTK.t == []: return
type = VARS[0].get()
if CTK.__MAINTREE__ == 1:
CTK.__MAINACTIVEZONES__ = CPlot.getActiveZones()
active = []
zones = Internal.getZones(CTK.t)
for z in CTK.__MAINACTIVEZONES__:
active.append(CTK.t[2][CTK.Nb[z] + 1][2][CTK.Nz[z]])
Z = None
if type == 'cellN=-99999':
Z = selectWithFormula(active, '{cellN} == -99999')
elif type == 'cellN=1':
Z = selectWithFormula(active, '{cellN} == 1')
elif type == 'cellN=0':
Z = selectWithFormula(active, '{cellN} == 0')
elif type == 'cellN=2':
Z = selectWithFormula(active, '{cellN} == 2')
elif type == 'cellN<0':
Z = selectWithFormula(active, '{cellN}<0')
elif type == '0<cellN<1':
Z = selectWithFormula(active, '({cellN}>0) & ({cellN}<1)')
elif type == 'Interpolated points':
Z = X.extractChimeraInfo(zones, type='interpolated', loc='centers')
if Z == []: Z = None
elif type == 'Extrapolated points':
Z = X.extractChimeraInfo(zones, type='extrapolated', loc='centers')
if Z == []: Z = None
elif type == 'Orphan points':
Z = X.extractChimeraInfo(zones, type='orphan', loc='centers')
if Z == []: Z = None
elif type == 'cf>1':
Z = X.extractChimeraInfo(zones, type='cf>1', loc='centers')
if Z == []: Z = None
if Z is not None:
CTK.TXT.insert('START', 'Filter ' + type + ' extracted.\n')
C._addBase2PyTree(CTK.t, 'EXTRACT')
b = Internal.getNodesFromName1(CTK.t, 'EXTRACT')
base = b[0]
base[2] += Z
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
#C._fillMissingVariables(CTK.t)
CTK.TKTREE.updateApp()
CTK.display(CTK.t)
else:
CTK.TXT.insert('START', 'Nothing extracted.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
def splitAndDistribute(event=None):
global STATS
if CTK.t == []: return
try:
NProc = int(VARS[0].get())
except:
CTK.TXT.insert('START', 'distribute: NProc is invalid.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
try:
comSpeed = float(VARS[1].get())
except:
CTK.TXT.insert('START', 'distribute: ComSpeed is invalid.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
algo = VARS[2].get()
useCom = VARS[3].get()
level = int(VARS[5].get())
CTK.saveTree()
try:
#CTK.t = T.splitNParts(CTK.t, NProc, multigrid=level)
CTK.t = T.splitSize(CTK.t, N=0, multigrid=level, R=NProc, type=2)
# no need to check inconsistant match (they have been deleted)
node = Internal.getNodeFromName(CTK.t, 'EquationDimension')
if node is not None:
ndim = Internal.getValue(node)
# Manque le reglage de la tol
else:
CTK.TXT.insert(
'START', 'EquationDimension not found (tkState). Using 3D.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
ndim = 3
CTK.t = X.connectMatch(CTK.t, dim=ndim)
CTK.display(CTK.t)
except Exception as e:
Panels.displayErrors([0, str(e)], header='Error: distribute/split')
CTK.TXT.insert('START', 'splitSize fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.t, STATS = D.distribute(CTK.t,
NProc,
perfo=(1., 0., comSpeed),
useCom=useCom,
algorithm=algo)
CTK.TXT.insert('START', 'Blocks split and distributed.\n')
CTK.TKTREE.updateApp()
updateStats()
def streamSurface():
if (CTK.t == []): return
if (CTK.__MAINTREE__ <= 0):
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
npts = CTK.varsFromWidget(VARS[0].get(), type=2)
if (len(npts) != 1):
CTK.TXT.insert('START', 'Number of points in stream incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
npts = npts[0]
nzs = CPlot.getSelectedZones()
if (nzs == []):
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
v1 = VARS[1].get()
v2 = VARS[2].get()
v3 = VARS[3].get()
streams = []
fail = False
errors = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
z = C.convertArray2Tetra(z)
try:
stream = P.streamSurf(CTK.t, z, [v1, v2, v3], N=npts)
streams.append(stream)
except Exception as e:
fail = True
errors += [0, str(e)]
CTK.saveTree()
CTK.t = C.addBase2PyTree(CTK.t, 'STREAMS', 2)
b = Internal.getNodesFromName1(CTK.t, 'STREAMS')
nob = C.getNobOfBase(b[0], CTK.t)
for i in streams:
CTK.add(CTK.t, nob, -1, i)
if (fail == False):
CTK.TXT.insert('START', 'Stream surface created.\n')
else:
Panels.displayErrors(errors, header='Error: streamSurf')
CTK.TXT.insert('START', 'Sream surface fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def MONO2TFI():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if len(nzs) == 0:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
surf = getSurfaces()
zones = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
z = C.convertBAR2Struct(z)
zones.append(z)
if len(zones) != 2:
CTK.TXT.insert('START', 'MONO2 TFI takes 2 contours.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
coords1 = C.getFields(Internal.__GridCoordinates__, zones[0])[0]
coords2 = C.getFields(Internal.__GridCoordinates__, zones[1])[0]
[m] = mono2mesh(coords1, coords2)
if isinstance(m, str):
CTK.TXT.insert('START', m + '\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
m = C.convertArrays2ZoneNode('TFI', [m])
if surf != []: m = T.projectOrthoSmooth(m, surf)
CTK.saveTree()
CTK.t = C.addBase2PyTree(CTK.t, 'MESHES')
bases = Internal.getNodesFromName1(CTK.t, 'MESHES')
nob = C.getNobOfBase(bases[0], CTK.t)
CTK.add(CTK.t, nob, -1, m)
CTK.TXT.insert('START', 'HO-TFI mesh created.\n')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def remap():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
a = []
for nz in nzs:
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
a.append(z)
# density
density = CTK.varsFromWidget(VARS[0].get(), type=1)
if len(density) != 1:
CTK.TXT.insert('START', 'Density is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
density = density[0]
# offset
offset = CTK.varsFromWidget(VARS[1].get(), type=1)
if len(offset) != 1:
CTK.TXT.insert('START', 'Offset is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
offset = offset[0]
CTK.saveTree()
if VARS[2].get() == '0': iso = withCart(a, offset, density)
else: iso = withOctree(a, offset, density)
if iso != []:
nob = CTK.Nb[nzs[0]]+1
for i in iso: CTK.add(CTK.t, nob, -1, i)
#C._fillMissingVariables(CTK.t)
CTK.TXT.insert('START', 'Surface filtered and offset (offset=%g).\n'%offset)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
else:
CTK.TXT.insert('START', 'Surface filter failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
def generatePLM():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if len(nzs) == 0:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
hf = CTK.varsFromWidget(VARS[1].get(), type=1)
if (len(hf) != 1):
CTK.TXT.insert('START', 'First cell height is incorrect.\n')
return
hf = hf[0]
h = CTK.varsFromWidget(VARS[0].get(), type=1)
if (len(h) != 1):
CTK.TXT.insert('START', 'Mesh height is incorrect.\n')
return
h = h[0]
density = CTK.varsFromWidget(VARS[2].get(), type=1)
if (len(density) != 1):
CTK.TXT.insert('START', 'Grid point density is incorrect.\n')
return
density = density[0]
try:
CTK.saveTree()
CTK.t = C.addBase2PyTree(CTK.t, 'MESHES')
bases = Internal.getNodesFromName1(CTK.t, 'MESHES')
gnob = C.getNobOfBase(bases[0], CTK.t)
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
B = G.polyLineMesher(z, h, hf, density)
for i in B[0]:
CTK.add(CTK.t, gnob, -1, i)
CTK.TXT.insert('START', 'PLM mesh created.\n')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
except Exception, e:
Panels.displayErrors([0, str(e)], header='Error: PLM')
CTK.TXT.insert('START', 'PLM mesh failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
def difference2():
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nzs = CPlot.getSelectedZones()
if len(nzs) != 2:
CTK.TXT.insert('START', 'Please, select two surfaces.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
tol = CTK.varsFromWidget(VARS[0].get(), type=1)
if len(tol) != 1:
CTK.TXT.insert('START', 'Tolerance is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
tol = tol[0]
CTK.saveTree()
deletedZoneNames = []
nz = nzs[0]
nob1 = CTK.Nb[nz] + 1
noz1 = CTK.Nz[nz]
deletedZoneNames.append(CTK.t[2][nob1][0] + Internal.SEP1 +
CTK.t[2][nob1][2][noz1][0])
z1 = CTK.t[2][nob1][2][noz1]
nz = nzs[1]
nob2 = CTK.Nb[nz] + 1
noz2 = CTK.Nz[nz]
deletedZoneNames.append(CTK.t[2][nob2][0] + Internal.SEP1 +
CTK.t[2][nob2][2][noz2][0])
z2 = CTK.t[2][nob2][2][noz2]
try:
j = XOR.booleanMinus(z2, z1, tol=tol)
CTK.t = CPlot.deleteSelection(CTK.t, CTK.Nb, CTK.Nz, nzs)
CPlot.delete(deletedZoneNames)
CTK.add(CTK.t, nob1, -1, j)
CTK.TXT.insert('START', 'Difference performed.\n')
except Exception as e:
Panels.displayErrors([0, str(e)], header='Error: difference')
CTK.TXT.insert('START', 'Difference failed.\n')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def drawLine(npts):
CTK.t = C.addBase2PyTree(CTK.t, 'CONTOURS', 1)
nodes = Internal.getNodesFromName1(CTK.t, 'CONTOURS')
nob = C.getNobOfBase(nodes[0], CTK.t)
CTK.TXT.insert('START', 'Click first point...\n')
prev = []
if CTK.__BUSY__ == False:
CTK.__BUSY__ = True
TTK.sunkButton(WIDGETS['draw'])
CPlot.setState(cursor=1)
while CTK.__BUSY__:
CPlot.unselectAllZones()
CTK.saveTree()
surfaces = getSurfaces()
l = []
while l == []:
l = CPlot.getActivePoint()
time.sleep(CPlot.__timeStep__)
WIDGETS['draw'].update()
if CTK.__BUSY__ == False: break
if CTK.__BUSY__:
if prev == []:
prev = l
CTK.TXT.insert('START', 'Click second point...\n')
elif (prev != l):
line = D.line(prev, l, npts)
if surfaces != []: line = T.projectOrthoSmooth(line, surfaces)
CTK.add(CTK.t, nob, -1, line)
CTK.TXT.insert('START', 'Line created.\n')
CTK.__BUSY__ = False
TTK.raiseButton(WIDGETS['draw'])
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
CPlot.setState(cursor=0)
prev = []
return
CTK.__BUSY__ = False
TTK.raiseButton(WIDGETS['draw'])
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
TTK.raiseButton(WIDGETS['draw'])
CPlot.setState(cursor=0)
return
def smooth(event=None):
if CTK.t == []: return
if CTK.__MAINTREE__ <= 0:
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
eps = CTK.varsFromWidget(VARS[7].get(), 1)
if len(eps) != 1:
CTK.TXT.insert('START', 'Invalid smoother power.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
eps = eps[0]
niter = CTK.varsFromWidget(VARS[6].get(), 2)
if len(niter) != 1:
CTK.TXT.insert('START', 'Invalid number of smoother iterations.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
niter = niter[0]
CTK.saveTree()
# Get first selected zone
nz = nzs[0]
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
zone = CTK.t[2][nob][2][noz]
dim = Internal.getZoneDim(zone)
if dim[0] == 'Structured':
if dim[2] != 1 and dim[3] != 1:
fail = apply3D(1., niter, eps, ntype=4)
elif dim[2] != 1 and dim[3] == 1:
fail = apply2D(1., niter, eps, ntype=4)
else: fail = smooth1D(niter, eps)
else: fail = smooth1D(niter, eps) # all zones
if not fail:
CTK.TXT.insert('START', 'Smooth successfull.\n')
else:
CTK.TXT.insert('START', 'Smooth fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
