def getValue():
nz = CPlot.getSelectedZone()
l = CPlot.getActivePointIndex()
if l == []: return
field = CPlot.getState('scalarField')
if field == -1: return
ind = CPlot.getActivePointIndex()
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
varNames = C.getVarNames(z)[0]
if len(varNames) > field + 3:
var = varNames[field + 3]
val = C.getValue(z, var, ind[0])
VARS[0].set(str(val))
else:
CTK.TXT.insert('START',
'Field %d not found. Use scalar mode.\n' % field)
CTK.TXT.insert('START', 'Error: ', 'Error')
def refineCells():
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
W = WIDGETS['refine']
if CTK.__BUSY__ == False:
CPlot.unselectAllZones()
CTK.__BUSY__ = True
TTK.sunkButton(W)
CPlot.setState(cursor=1)
while CTK.__BUSY__:
l = []
while l == []:
nz = CPlot.getSelectedZone()
l = CPlot.getActivePointIndex()
CPlot.unselectAllZones()
time.sleep(CPlot.__timeStep__)
W.update()
if CTK.__BUSY__ == False: break
if CTK.__BUSY__:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
CTK.saveTree()
z = CTK.t[2][nob][2][noz]
C._initVars(z, 'centers:__tag__', 0)
C.setValue(z, 'centers:__tag__', l[1], 1)
try:
z = P.refine(z, '__tag__')
CTK.replace(CTK.t, nob, noz, z)
except:
pass
CTK.TKTREE.updateApp()
CPlot.render()
CTK.__BUSY__ = False
TTK.raiseButton(W)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
TTK.raiseButton(W)
CPlot.setState(cursor=0)
def sculpt():
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
TOOLS = createTools()
#CTK.display(TOOLS)
bbox = G.bbox(CTK.t)
size = max(bbox[3] - bbox[0], bbox[4] - bbox[1], bbox[5] - bbox[2])
CPlot.unselectAllZones()
w = WIDGETS['sculpt']
if CTK.__BUSY__ == False:
CTK.__BUSY__ = True
TTK.sunkButton(w)
CPlot.setState(cursor=1)
while CTK.__BUSY__:
l = []
while l == []:
nz = CPlot.getSelectedZone()
l = CPlot.getActivePointIndex()
time.sleep(CPlot.__timeStep__)
w.update()
if (CTK.__BUSY__ == False): break
if CTK.__BUSY__:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
CTK.saveTree()
depth = 0.5 * WIDGETS['depth'].get() / 100.
depth = size * depth
width = 0.5 * WIDGETS['width'].get() / 100.
width = size * width
brushType = VARS[2].get()
z = CTK.t[2][nob][2][noz]
posCam = CPlot.getState('posCam')
posEye = CPlot.getState('posEye')
vect = (posEye[0] - posCam[0], posEye[1] - posCam[1],
posEye[2] - posCam[2])
if brushType == 'Deform':
click = CPlot.getActivePoint()
point = (click[0], click[1], click[2])
z = T.deformPoint(z, point, vect, depth, width)
CTK.replace(CTK.t, nob, noz, z)
elif brushType == 'Sphere':
click = CPlot.getActivePoint()
center = (click[0], click[1], click[2])
s = D.sphere(center, depth, N=10)
s = C.convertArray2Tetra(s)
s = G.close(s)
z = C.convertArray2Tetra(z)
z = G.close(z)
z = XOR.booleanMinus(z, s)
CTK.replace(CTK.t, nob, noz, z)
elif brushType == 'Cube':
click = CPlot.getActivePoint()
center = (click[0], click[1], click[2])
s = D.sphere(center, depth, N=20)
s = C.convertArray2Tetra(s)
s = G.close(s)
z = C.convertArray2Tetra(z)
z = G.close(z)
z = XOR.booleanMinus(z, s)
CTK.replace(CTK.t, nob, noz, z)
CTK.TKTREE.updateApp()
CPlot.unselectAllZones()
CPlot.render()
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
def paint():
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
# get patined var
field = CPlot.getState('scalarField')
if field == -1:
CTK.TXT.insert('START', 'Scalar field is not set.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CPlot.unselectAllZones()
CTK.saveTree()
w = WIDGETS['paint']
if not CTK.__BUSY__:
CTK.__BUSY__ = True
TTK.sunkButton(w)
CPlot.setState(cursor=1)
while CTK.__BUSY__ == True:
l = []
while l == []:
nz = CPlot.getSelectedZone()
l = CPlot.getActivePointIndex()
#time.sleep(CPlot.__timeStep__)
w.update()
if not CTK.__BUSY__: break
if CTK.__BUSY__:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
CTK.saveTree()
value = float(WIDGETS['value'].get())
#width = float(WIDGETS['width'].get())
#brushType = VARS[2].get()
z = CTK.t[2][nob][2][noz]
posCam = CPlot.getState('posCam')
posEye = CPlot.getState('posEye')
vect = (posEye[0] - posCam[0], posEye[1] - posCam[1],
posEye[2] - posCam[2])
click = CPlot.getActivePoint()
point = (click[0], click[1], click[2])
ind = CPlot.getActivePointIndex()
#hook = C.createHook
varNames = C.getVarNames(z)[0]
if len(varNames) > field + 3:
var = varNames[field + 3]
C.setValue(z, var, ind[0], value)
else:
CTK.TXT.insert(
'START',
'Field %d not found. Use scalar mode.\n' % field)
CTK.TXT.insert('START', 'Error: ', 'Error')
CTK.replace(CTK.t, nob, noz, z)
CTK.TKTREE.updateApp()
CPlot.unselectAllZones()
CPlot.render()
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
def display1D(event=None):
if CTK.t == []: return
# Get slot
try:
slot = int(VARS[5].get())
except:
slot = 0
# Get grid size
try:
gridSize = VARS[1].get()
grids = gridSize.split(';')
if (len(grids) == 1): gridSize = (int(grids[0]), 1)
else: gridSize = (int(grids[0]), int(grids[1]))
except:
gridSize = (1, 1)
CPlot.setState(gridSize=gridSize)
# Get grid pos
try:
gridPos = VARS[2].get()
grids = gridPos.split(';')
if (len(grids) == 1): gridPos = (int(grids[0]), 1)
else: gridPos = (int(grids[0]), int(grids[1]))
except:
gridPos = (0, 0)
# Recupere la direction pour la coupe ou 'Elements'
dir = VARS[0].get()
if dir == 'None':
CPlot.display1D([], slot=slot)
return # clear
# Recupere le pt pour la coupe ou les elements 1D
if dir == 'Elements': # elements -> recupere les elements
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
points = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
selected = CTK.t[2][nob][0] + '/' + z[0]
points.append(selected)
elif (dir == 'I' or dir == 'J'
or dir == 'K'): # indice -> recupere les indices + la zone
if (CTK.__MAINTREE__ <= 0):
CTK.TXT.insert('START', 'Fail on a temporary tree.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
nz = CPlot.getSelectedZone()
if (nz == -1):
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
points = []
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
selected = CTK.t[2][nob][0] + '/' + z[0]
index = CPlot.getActivePointIndex()
points = (selected, index)
else: # les coupes -> recupere les coord du pt
point = CPlot.getActivePoint()
if point == []: point = (0., 0., 0.)
# Recupere les variables a afficher
var1 = VARS[3].get()
var1 = var1.replace('centers:', '')
var2 = VARS[4].get()
var2 = var2.replace('centers:', '')
# Recupere les zones actives
actives = []
zones = Internal.getZones(CTK.t)
if CTK.__MAINTREE__ == 1:
nzs = CPlot.getActiveZones()
for nz in nzs:
actives.append(zones[nz])
else:
actives = zones
if actives == []: return
if (dir == 'X (Y)'):
elts = P.isoSurfMC(actives, 'CoordinateY', point[1])
if elts != []:
elts2 = P.isoSurfMC(elts, 'CoordinateZ', point[2])
if (elts2 != []): elts = elts2
elif (dir == 'Y (X)'):
elts = P.isoSurfMC(actives, 'CoordinateX', point[0])
if elts != []:
elts2 = P.isoSurfMC(elts, 'CoordinateZ', point[2])
if (elts2 != []): elts = elts2
elif (dir == 'Z (X)'):
elts = P.isoSurfMC(actives, 'CoordinateX', point[0])
if (elts != []):
elts2 = P.isoSurfMC(elts, 'CoordinateY', point[1])
if (elts2 != []): elts = elts2
elif (dir == 'X (Z)'):
elts = P.isoSurfMC(actives, 'CoordinateZ', point[2])
if elts != []:
elts2 = P.isoSurfMC(elts, 'CoordinateY', point[1])
if (elts2 != []): elts = elts2
elif (dir == 'Y (Z)'):
elts = P.isoSurfMC(actives, 'CoordinateZ', point[2])
if elts != []:
elts2 = P.isoSurfMC(elts, 'CoordinateX', point[0])
if (elts2 != []): elts = elts2
elif (dir == 'Z (Y)'):
elts = P.isoSurfMC(actives, 'CoordinateY', point[1])
if (elts != []):
elts2 = P.isoSurfMC(elts, 'CoordinateX', point[0])
if (elts2 != []): elts = elts2
elif (dir == 'I'):
v = points[0]
ind = points[1]
v = v.lstrip()
v = v.rstrip()
sname = v.split('/', 1)
bases = Internal.getNodesFromName1(CTK.t, sname[0])
elts = []
if bases != []:
zones = Internal.getNodesFromType1(bases[0], 'Zone_t')
for z in zones:
if (z[0] == sname[1]):
try:
zp = C.center2Node(z, Internal.__FlowSolutionCenters__)
zp = T.subzone(zp, (1, ind[3], ind[4]),
(-1, ind[3], ind[4]))
elts.append(zp)
except:
pass
elif (dir == 'J'):
v = points[0]
ind = points[1]
v = v.lstrip()
v = v.rstrip()
sname = v.split('/', 1)
bases = Internal.getNodesFromName1(CTK.t, sname[0])
elts = []
if bases != []:
zones = Internal.getNodesFromType1(bases[0], 'Zone_t')
for z in zones:
if (z[0] == sname[1]):
try:
zp = C.center2Node(z, Internal.__FlowSolutionCenters__)
zp = T.subzone(zp, (ind[2], 1, ind[4]),
(ind[2], -1, ind[4]))
elts.append(zp)
except:
pass
elif (dir == 'K'):
v = points[0]
ind = points[1]
v = v.lstrip()
v = v.rstrip()
sname = v.split('/', 1)
bases = Internal.getNodesFromName1(CTK.t, sname[0])
elts = []
if bases != []:
zones = Internal.getNodesFromType1(bases[0], 'Zone_t')
for z in zones:
if (z[0] == sname[1]):
try:
zp = C.center2Node(z, Internal.__FlowSolutionCenters__)
zp = T.subzone(zp, (ind[2], ind[3], 1),
(ind[2], ind[3], -1))
elts.append(zp)
except:
pass
elif (dir == 'Elements'):
elts = []
for v in points:
v = v.lstrip()
v = v.rstrip()
sname = v.split('/', 1)
bases = Internal.getNodesFromName1(CTK.t, sname[0])
if (bases != []):
zones = Internal.getNodesFromType1(bases[0], 'Zone_t')
for z in zones:
if (z[0] == sname[1]): elts.append(z)
if elts == []:
CTK.TXT.insert('START', 'Nothing to display.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
try:
elts = D.getCurvilinearAbscissa(elts)
except:
pass
# Fit first axis
pos = WIDGETS['rangePos'].get() / 50. - 1.
zoom = WIDGETS['rangeZoom'].get() / 120.
minv1 = C.getMinValue(elts, var1)
maxv1 = C.getMaxValue(elts, var1)
if (maxv1 - minv1 < 1.e-6):
maxv1 += 5.e-7
minv1 -= 5.e-7
# active point localisation
nz = CPlot.getSelectedZone()
if (nz != -1):
ind = CPlot.getActivePointIndex()
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
f1 = C.getValue(z, var1, ind[0])
try:
r1min = (f1 - minv1) * zoom + minv1 + pos * (1. - zoom) * (maxv1 -
minv1)
r1max = (f1 - maxv1) * zoom + maxv1 + pos * (1. - zoom) * (maxv1 -
minv1)
except: # var1 not found in z, le cherche dans elts
xf1 = C.getValue(z, 'CoordinateX', ind[0])
yf1 = C.getValue(z, 'CoordinateY', ind[0])
zf1 = C.getValue(z, 'CoordinateZ', ind[0])
f1 = minv1 + 0.5 * (maxv1 - minv1)
r1min = 0.5 * (maxv1 - minv1) * zoom + minv1 + pos * (
1. - zoom) * (maxv1 - minv1)
r1max = -0.5 * (maxv1 - minv1) * zoom + maxv1 + pos * (
1. - zoom) * (maxv1 - minv1)
else:
f1 = minv1 + 0.5 * (maxv1 - minv1)
r1min = 0.5 * (maxv1 - minv1) * zoom + minv1 + pos * (1. - zoom) * (
maxv1 - minv1)
r1max = -0.5 * (maxv1 - minv1) * zoom + maxv1 + pos * (1. - zoom) * (
maxv1 - minv1)
# Fit second axis
p = P.selectCells(
elts,
'({%s} < %20.16g) & ({%s} > %20.16g)' % (var1, r1max, var1, r1min))
minv2 = C.getMinValue(p, var2)
maxv2 = C.getMaxValue(p, var2)
# display
CPlot.display1D(p,
slot=slot,
bgBlend=0.,
gridPos=gridPos,
var1=var1,
var2=var2,
r1=(r1min, r1max),
r2=(minv2, maxv2))
CTK.TXT.insert('START', 'Plot displayed.\n')
