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 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 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 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 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 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 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 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 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 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 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 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 moveSelection():
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
baseName = VARS[0].get()
if baseName == 'New Base':
C.registerBaseNames(CTK.t)
baseName = C.getBaseName('NewBase')
CTK.t = C.addBase2PyTree(CTK.t, baseName, 3)
base = Internal.getNodesFromName1(CTK.t, baseName)
if base == []:
C.registerBaseNames(CTK.t)
baseName = C.getBaseName(baseName)
CTK.t = C.addBase2PyTree(CTK.t, baseName, 3)
base = Internal.getNodesFromName1(CTK.t, baseName)
base = base[0]
CTK.saveTree()
Z = []
deletedZoneNames = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
zone = CTK.t[2][nob][2][noz]
deletedZoneNames.append(CTK.t[2][nob][0] + Internal.SEP1 +
CTK.t[2][nob][2][noz][0])
Z.append(zone)
CTK.t = CPlot.deleteSelection(CTK.t, CTK.Nb, CTK.Nz, nzs)
CPlot.delete(deletedZoneNames)
nob = C.getNobOfBase(base, CTK.t)
for i in Z:
CTK.add(CTK.t, nob, -1, i)
CTK.TXT.insert('START', 'Selection moved to %s.\n' % baseName)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def createPoint(event=None):
point = CTK.varsFromWidget(VARS[0].get(), type=1)
if len(point) != 3:
CTK.TXT.insert('START', 'Point coords are incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
CTK.t = C.addBase2PyTree(CTK.t, 'POINTS', 1)
base = Internal.getNodesFromName1(CTK.t, 'POINTS')
base = base[0]
nob = C.getNobOfBase(base, CTK.t)
a = D.point((point[0], point[1], point[2]))
CTK.add(CTK.t, nob, -1, a)
CTK.TXT.insert('START', 'Point ' + VARS[0].get() + ' created.\n')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def extractIsoSurf(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
field = VARS[0].get()
value = VARS[1].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 = []
try:
iso = P.isoSurfMC(z, field, value)
isos += iso
except Exception as e:
Panels.displayErrors([0,str(e)], header='Error: isoSurf')
if isos == []:
CTK.TXT.insert('START', 'isoSurf failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
else:
CTK.TXT.insert('START', 'isoSurf of '+field+'='
+str(value)+' computed.\n')
for i in isos: i[0] = C.getZoneName(i[0]) # unique name
CTK.t = C.addBase2PyTree(CTK.t, 'SURFACES', 2)
base = Internal.getNodeFromName1(CTK.t, 'SURFACES')
nob = C.getNobOfBase(base, CTK.t)
for i in isos: CTK.add(CTK.t, nob, -1, i)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def TFI():
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]
dim = Internal.getZoneDim(z)
if dim[3] == 'BAR':
zp = C.convertBAR2Struct(z)
zones.append(zp)
else:
zones.append(z)
try:
CTK.saveTree()
mesh = G.TFI(zones)
if surf != []: mesh = T.projectOrthoSmooth(mesh, surf)
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, mesh)
CTK.TXT.insert('START', 'TFI mesh created.\n')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
except Exception as e:
Panels.displayErrors([0, str(e)], header='Error: TFI')
CTK.TXT.insert('START', 'TFI mesh failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
def cpBlock():
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()
CTK.t = C.addBase2PyTree(CTK.t, 'COPY', 3)
base = Internal.getNodesFromName1(CTK.t, 'COPY')[0]
gnob = C.getNobOfBase(base, CTK.t)
newFamilyZoneNames = set()
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = Internal.copyRef(CTK.t[2][nob][2][noz])
z[0] = C.getZoneName(z[0] + '.dup')
CTK.add(CTK.t, gnob, -1, z)
# Create new Family Name for Family Zones
nbdup = z[0].split('.dup')[-1]
nodes = Internal.getNodesFromType1(z, 'FamilyName_t')
for f in nodes:
newFamilyZoneName = Internal.getValue(f) + '.dup' + nbdup
f[1] = newFamilyZoneName
if newFamilyZoneName not in newFamilyZoneNames:
newFamilyZoneNames.add(newFamilyZoneName)
CTK.TXT.insert(
'START', CTK.t[2][nob][0] + '/' + CTK.t[2][nob][2][noz][0] +
' duplicated.\n')
# Create new Family_t node for each Tag Zone
for f in newFamilyZoneNames:
Internal.newFamily(name=f, parent=base)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def exteriorFaces():
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()
CTK.t = C.addBase2PyTree(CTK.t, 'CONTOURS', 1)
p = Internal.getNodesFromName1(CTK.t, 'CONTOURS')
gnob = C.getNobOfBase(p[0], CTK.t)
fail = False
errors = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
try:
ext = P.exteriorFaces(CTK.t[2][nob][2][noz])
ext = T.splitConnexity(ext)
for i in ext:
CTK.add(CTK.t, gnob, -1, i)
except TypeError as e: # type d'element non reconnu
fail = True
errors += [0, str(e)]
except ValueError: # empty set
pass
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
if not fail: CTK.TXT.insert('START', 'Exterior faces done.\n')
else:
Panels.displayErrors(errors, header='Error: exteriorFaces')
CTK.TXT.insert('START',
'Exterior faces fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
CPlot.render()
def convert2Hexa():
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()
list = []
fail = False
errors = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
dim = Internal.getZoneDim(z)
try:
if (dim[0] == 'Unstructured' and dim[3] == 'TRI'):
a, b = C.convertTri2Quad(z)
CTK.replace(CTK.t, nob, noz, a)
CTK.add(CTK.t, nob, -1, b)
else:
a = C.convertArray2Hexa(z)
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 hexa.\n')
else:
Panels.displayErrors(errors, header='Error: convert2Hexa')
CTK.TXT.insert('START',
'Hexa 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 extract(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
BCtype = VARS[0].get()
if BCtype not in Internal.KNOWNBCS: BCtype = 'FamilySpecified:'+BCtype
CTK.saveTree()
CTK.t = C.addBase2PyTree(CTK.t, 'EXTRACTED_BC', 2)
base = Internal.getNodeFromName1(CTK.t, 'EXTRACTED_BC')
nzs = CPlot.getSelectedZones()
if CTK.__MAINTREE__ <= 0 or nzs == []: zones = CTK.t
else:
zones = []
for nz in nzs:
nob = CTK.Nb[nz]+1; noz = CTK.Nz[nz]
zones.append(CTK.t[2][nob][2][noz])
if BCtype != 'FamilySpecified:-All BC-':
Z = C.extractBCOfType(zones, BCtype, topTree=CTK.t)
for i in Z: Internal._createChild(i, 'BCType', 'UserDefined_t', BCtype)
else:
(Zp, BCNames, BCTypes) = C.getBCs(zones)
Z = []
for i in Zp: Z += i
for i, z in enumerate(Z): Internal._createChild(z, 'BCType', 'UserDefined_t', BCTypes[i])
nob = C.getNobOfBase(base, CTK.t)
for i in Z:
i[0] = C.getZoneName(i[0])
CTK.add(CTK.t, nob, -1, i)
#C._fillMissingVariables(CTK.t) # a cause du BC data set
CTK.TXT.insert('START', 'BCs of type %s extracted.\n'%BCtype)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
if CTK.TKPLOTXY is not None: CTK.TKPLOTXY.updateApp()
def breakElts():
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]
z = CTK.t[2][nob][2][noz]
try:
zones = T.breakElements(z)
if (len(zones) > 0): CTK.replace(CTK.t, nob, noz, zones[0])
for zz in zones[1:]:
CTK.add(CTK.t, nob, -1, zz)
except Exception as e:
fail = True
errors += [0, str(e)]
if not fail:
CTK.TXT.insert('START', 'Zones converted to basic elements.\n')
else:
Panels.displayErrors(errors, header='Error: breakElts')
CTK.TXT.insert('START', 'Break elts 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 tetraMesher():
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
if VARS[0].get() == 'netgen': algo = 0
else: algo = 1
CTK.saveTree()
out = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
out.append(CTK.t[2][nob][2][noz])
try:
mesh = G.tetraMesher(out, algo=algo, grading=0.3)
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, mesh)
CTK.TXT.insert('START', 'Tetra mesh created.\n')
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
except Exception as e:
Panels.displayErrors([0, str(e)], header='Error: TetraMesher')
CTK.TXT.insert('START', 'Tetra mesh failed.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
def setCanvas(event=None):
dir = VARS[0].get()
CTK.saveTree()
deleteCanvasBase()
CTK.t = C.addBase2PyTree(CTK.t, 'CANVAS', 2)
size = CANVASSIZE
if dir == 'View':
a = G.cart((-size, 0, -size), (2 * size, 2 * size, 2 * size),
(2, 1, 2))
a = T.translate(a, (XC, YC, ZC))
a = T.rotate(a, (XC, YC, ZC), ((1, 0, 0), (0, 1, 0), (0, 0, 1)),
((-DIRX, -DIRY, -DIRZ), (LX, LY, LZ), (UX, UY, UZ)))
VARS[1].set(str(XC) + ';' + str(YC) + ';' + str(ZC))
elif dir == 'YZ':
a = G.cart((0, -size, -size), (2 * size, 2 * size, 2 * size),
(1, 2, 2))
a = T.translate(a, (XC, YC, ZC))
VARS[1].set(str(XC))
elif dir == 'XZ':
a = G.cart((-size, 0, -size), (2 * size, 2 * size, 2 * size),
(2, 1, 2))
a = T.translate(a, (XC, YC, ZC))
VARS[1].set(str(YC))
else:
a = G.cart((-size, -size, 0), (2 * size, 2 * size, 2 * size),
(2, 2, 1))
a = T.translate(a, (XC, YC, ZC))
VARS[1].set(str(ZC))
nodes = Internal.getNodesFromName1(CTK.t, 'CANVAS')
base = nodes[0]
nob = C.getNobOfBase(base, CTK.t)
CTK.add(CTK.t, nob, -1, a)
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def drawRectangle(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 left/lower corner...\n')
w = WIDGETS['draw']
prev = []; second = []
if (CTK.__BUSY__ == False):
CTK.__BUSY__ = True
TTK.sunkButton(w)
CPlot.setState(cursor=1)
while (CTK.__BUSY__ == True):
CPlot.unselectAllZones()
CTK.saveTree()
surfaces = getSurfaces()
l = []
while (l == []):
l = CPlot.getActivePoint()
time.sleep(CPlot.__timeStep__)
w.update()
if (CTK.__BUSY__ == False): break
if (CTK.__BUSY__ == True):
if (prev == []):
prev = l
CTK.TXT.insert('START', 'Click right/up corner...\n')
elif (prev != l):
e1,e2 = getVectorsFromCanvas()
e1n = Vector.norm(e1)
e2n = Vector.norm(e2)
if (e2n > e1n): e1 = e2
P1 = l; P2 = prev
P1P2 = Vector.sub(P2, P1)
P1P2n = Vector.norm(P1P2)
Q = Vector.norm(Vector.cross(e1, P1P2))
L = math.sqrt( P1P2n*P1P2n - Q*Q )
sign = Vector.dot(e1, P1P2)
if (sign > 0): e1 = Vector.mul(L, e1)
else: e1 = Vector.mul(-L, e1)
P3 = Vector.add(P1, e1)
P4 = Vector.sub(P2, e1)
l1 = D.line(P1, P3, npts)
l2 = D.line(P3, P2, npts)
l3 = D.line(P2, P4, npts)
l4 = D.line(P4, P1, npts)
rect = T.join([l1,l2,l3,l4])
if (surfaces != []):
rect = T.projectOrthoSmooth(rect, surfaces)
CTK.add(CTK.t, nob, -1, rect)
CTK.TXT.insert('START', 'Rectangle created.\n')
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
#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(w)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
def drawArc(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')
w = WIDGETS['draw']
prev = []; second = []
if CTK.__BUSY__ == False:
CTK.__BUSY__ = True
TTK.sunkButton(w)
CPlot.setState(cursor=1)
while CTK.__BUSY__:
CPlot.unselectAllZones()
CTK.saveTree()
surfaces = getSurfaces()
l = []
while (l == []):
l = CPlot.getActivePoint()
time.sleep(CPlot.__timeStep__)
w.update()
if (CTK.__BUSY__ == False): break
if (CTK.__BUSY__ == True):
if (prev == []):
prev = l
CTK.TXT.insert('START', 'Click second point...\n')
elif (second == [] and prev != l):
second = l
CTK.TXT.insert('START', 'Click third point...\n')
elif (prev != l and second != l):
x1 = l[0]; y1 = l[1]; z1 = l[2]
x2 = prev[0]; y2 = prev[1]; z2 = prev[2]
x3 = second[0]; y3 = second[1]; z3 = second[2]
xa = x2 - x1; ya = y2 - y1; za = z2 - z1
xb = x3 - x1; yb = y3 - y1; zb = z3 - z1
xc = x3 - x2; yc = y3 - y2; zc = z3 - z2
a2 = xa*xa + ya*ya + za*za
b2 = xb*xb + yb*yb + zb*zb
c2 = xc*xc + yc*yc + zc*zc
A = 2*b2*c2 + 2*c2*a2 + 2*a2*b2 - a2*a2 - b2*b2 - c2*c2
R = math.sqrt( a2*b2*c2 / A )
nx = ya*zb - za*yb
ny = za*xb - xa*zb
nz = xa*yb - ya*xb
tx = ya*nz - za*ny
ty = za*nx - xa*nz
tz = xa*ny - ya*nx
norm = tx*tx + ty*ty + tz*tz
normi = 1./math.sqrt(norm)
tx = tx*normi; ty = ty*normi; tz = tz*normi;
alpha = R*R - (xa*xa+ya*ya+za*za)*0.25
alpha = math.sqrt(alpha)
center = [0,0,0]
center[0] = 0.5*(x1+x2) + alpha*tx
center[1] = 0.5*(y1+y2) + alpha*ty
center[2] = 0.5*(z1+z2) + alpha*tz
dx3 = center[0]-x3; dy3 = center[1]-y3; dz3 = center[2]-z3
l = dx3*dx3 + dy3*dy3 + dz3*dz3
if (abs(l - R*R) > 1.e-10):
center[0] = 0.5*(x1+x2) - alpha*tx
center[1] = 0.5*(y1+y2) - alpha*ty
center[2] = 0.5*(z1+z2) - alpha*tz
dx3 = center[0]-x3; dy3 = center[1]-y3; dz3 = center[2]-z3
l = dx3*dx3 + dy3*dy3 + dz3*dz3
e1 = [x1-center[0], y1-center[1], z1-center[2]]
e2 = [x2-center[0], y2-center[1], z2-center[2]]
e3 = Vector.cross(e1, e2)
e4 = Vector.cross(e1, e3)
# Images des pts dans le plan xyz
pt1 = D.point((x1,y1,z1))
pt2 = D.point((x2,y2,z2))
pt3 = D.point((x3,y3,z3))
pt1 = T.rotate(pt1,
(center[0], center[1], center[2]),
(e1, e4, e3),
((1,0,0), (0,1,0), (0,0,1)) )
pt2 = T.rotate(pt2,
(center[0], center[1], center[2]),
(e1, e4, e3),
((1,0,0), (0,1,0), (0,0,1)))
pt3 = T.rotate(pt3,
(center[0], center[1], center[2]),
(e1, e4, e3),
((1,0,0), (0,1,0), (0,0,1)))
xp1 = C.getValue(pt1, 'CoordinateX', 0)
yp1 = C.getValue(pt1, 'CoordinateY', 0)
zp1 = C.getValue(pt1, 'CoordinateZ', 0)
xp2 = C.getValue(pt2, 'CoordinateX', 0)
yp2 = C.getValue(pt2, 'CoordinateY', 0)
zp2 = C.getValue(pt2, 'CoordinateZ', 0)
xp3 = C.getValue(pt3, 'CoordinateX', 0)
yp3 = C.getValue(pt3, 'CoordinateY', 0)
zp3 = C.getValue(pt3, 'CoordinateZ', 0)
dx1 = (xp1-center[0])/R; dy1 = (yp1-center[1])/R
if dx1 > 1.: dx1 = 1.
if dx1 < -1.: dx1 = -1.
if dy1 > 0: teta1 = math.acos(dx1)
else: teta1 = 2*math.pi - math.acos(dx1)
teta1 = teta1*180./math.pi; teta1 = 360.
dx2 = (xp2-center[0])/R; dy2 = (yp2-center[1])/R
if dx2 > 1.: dx2 = 1.
if dx2 < -1.: dx2 = -1.
if dy2 > 0: teta2 = math.acos(dx2)
else: teta2 = 2*math.pi - math.acos(dx2)
teta2 = teta2*180./math.pi
dx3 = (xp3-center[0])/R; dy3 = (yp3-center[1])/R
if dx3 > 1.: dx3 = 1.
if dx3 < -1.: dx3 = -1.
if dy3 > 0: teta3 = math.acos(dx3)
else: teta3 = 2*math.pi - math.acos(dx3)
teta3 = teta3*180./math.pi
if teta3 > teta2: teta1 = 360.
else: teta1 = 0.
circle = D.circle((center[0],center[1],center[2]), R,
tetas=teta2, tetae=teta1, N=npts)
circle = T.rotate(circle,
(center[0], center[1], center[2]),
((1,0,0), (0,1,0), (0,0,1)),
(e1, e4, e3))
if surfaces != []:
circle = T.projectOrthoSmooth(circle, surfaces)
CTK.add(CTK.t, nob, -1, circle)
CTK.TXT.insert('START', 'Circle created.\n')
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
#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(w)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
def addLayers():
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
h = CTK.varsFromWidget(VARS[0].get(), type=1)
if len(h) != 1:
CTK.TXT.insert('START', 'Layer height is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
h = h[0]
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]
smooth = CTK.varsFromWidget(VARS[2].get(), type=2)
if len(smooth) != 1:
CTK.TXT.insert('START', 'Smoothing iterations number is incorrect.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
smooth = smooth[0]
nzs = CPlot.getSelectedZones()
if nzs == []:
CTK.TXT.insert('START', 'Selection is empty.\n')
CTK.TXT.insert('START', 'Error: ', 'Error')
return
CTK.saveTree()
zlist = []
for nz in nzs:
nob = CTK.Nb[nz] + 1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
zlist.append(z)
d = G.cart((0, 0, 0), (h, 1, 1), (N + 1, 1, 1))
fail = False
errors = []
try:
zlist = G.addNormalLayers(zlist, d, niter=smooth)
except Exception as e:
fail = True
errors += [0, str(e)]
for z in zlist:
z[0] = C.getZoneName(z[0]) # unique name
CTK.t = C.addBase2PyTree(CTK.t, 'MESHES')
base = Internal.getNodeFromName1(CTK.t, 'MESHES')
if not fail:
nob = C.getNobOfBase(base, CTK.t)
for i in zlist:
CTK.add(CTK.t, nob, -1, i)
CTK.TXT.insert('START', 'Normal layers added.\n')
else:
Panels.displayErrors(errors, header='Error: addNormalLayers')
CTK.TXT.insert('START',
'Add normal layers 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 drawFreeHand():
global CURRENTZONE; global CURRENTPOLYLINE; global ALLZONES
w = WIDGETS['draw']
prev = []; first = []
if CTK.__BUSY__ == False:
CPlot.unselectAllZones()
CTK.saveTree()
CTK.__BUSY__ = True
TTK.sunkButton(w)
CPlot.setState(cursor=1)
buttonState = 0
while CTK.__BUSY__ == True:
if prev == []: # first point
l = []
while l == []:
l = CPlot.getActivePoint()
if l != []: prev = l; first = l
time.sleep(CPlot.__timeStep__)
w.update()
if CTK.__BUSY__ == False: break
else: # next points
diff = -1.
while (diff < 1.e-10):
(buttonState,x,y,z) = CPlot.getMouseState()
l = (x,y,z)
diff = Vector.norm2(Vector.sub(l,prev))
diff1 = Vector.norm2(Vector.sub(l,first))
if (diff1 < 1.e-10): l = first
if (buttonState == 5): break
time.sleep(CPlot.__timeStep__)
w.update()
if (CTK.__BUSY__ == False): break
prev = l
CPlot.unselectAllZones()
if (buttonState == 5): # button released
ALLZONES.append(CURRENTZONE)
CURRENTZONE = None; prev = []; first = []
CURRENTPOLYLINE = []
CTK.TKTREE.updateApp()
if (CTK.__BUSY__ == True and buttonState != 5):
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)
CPlot.render()
buttonState = 0
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
surfaces = getSurfaces()
if (surfaces != []):
if (CURRENTZONE != None): ALLZONES += [CURRENTZONE]
for s in ALLZONES:
ret = Internal.getParentOfNode(CTK.t, s)
nob = C.getNobOfBase(ret[0], CTK.t)
a = T.projectOrthoSmooth(s, surfaces)
noz = ret[1]
CTK.replace(CTK.t, nob, noz, a)
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
CURRENTZONE = None; ALLZONES = []
CURRENTPOLYLINE = []
TTK.raiseButton(w)
CPlot.setState(cursor=0)
def generate(event=None):
CTK.saveTree()
N = CTK.varsFromWidget(VARS[0].get(), type=2)
if len(N) != 1:
CTK.TXT.insert('START', 'NPts is incorrect.\n')
return
N = N[0]
eltType = VARS[1].get()
surfType = VARS[2].get()
if surfType == 'Sphere':
s = D.sphere6((0, 0, 0), 0.5, N=N)
xc = 0
yc = 0
zc = 0
elif surfType == 'Plane':
h = 1. / (N - 1)
s1 = G.cart((-0.5, -0.5, -0.5), (h, h, h), (N, 1, N))
s = [s1]
xc = 0
yc = 0
zc = 0
elif surfType == 'Cube':
h = 1. / (N - 1)
s1 = G.cart((-0.5, -0.5, -0.5), (h, h, h), (N, N, 1))
s1 = T.reorder(s1, (-1, 2, 3))
s2 = G.cart((-0.5, -0.5, 0.5), (h, h, h), (N, N, 1))
s3 = G.cart((-0.5, -0.5, -0.5), (h, h, h), (N, 1, N))
s4 = G.cart((-0.5, 0.5, -0.5), (h, h, h), (N, 1, N))
s4 = T.reorder(s4, (-1, 2, 3))
s5 = G.cart((-0.5, -0.5, -0.5), (h, h, h), (1, N, N))
s5 = T.reorder(s5, (1, -2, 3))
s6 = G.cart((0.5, -0.5, -0.5), (h, h, h), (1, N, N))
s = [s1, s2, s3, s4, s5, s6]
xc = 0
yc = 0
zc = 0
elif surfType == 'Tetra':
m1 = meshTri([0, 0, 0], [1, 0, 0], [0, 1, 0], N=N)
m1 = T.reorder(m1, (-1, 2, 3))
m2 = meshTri([0, 0, 0], [1, 0, 0], [0, 0, 1], N=N)
m3 = meshTri([0, 0, 0], [0, 1, 0], [0, 0, 1], N=N)
m3 = T.reorder(m3, (-1, 2, 3))
m4 = meshTri([1, 0, 0], [0, 1, 0], [0, 0, 1], N=N)
s = m1 + m2 + m3 + m4
xc = 0.5
yc = 0.5
zc = 0.5
elif surfType == 'Pyramid':
h = 1. / (2 * N - 2)
m0 = G.cart((-0.5, -0.5, -0.5), (h, h, h), (2 * N - 1, 2 * N - 1, 1))
m0 = T.reorder(m0, (-1, 2, 3))
m1 = meshTri([-0.5, -0.5, -0.5], [0.5, -0.5, -0.5], [0, 0, 0.5], N=N)
m2 = meshTri([-0.5, -0.5, -0.5], [-0.5, 0.5, -0.5], [0, 0, 0.5], N=N)
m2 = T.reorder(m2, (-1, 2, 3))
m3 = meshTri([-0.5, 0.5, -0.5], [0.5, 0.5, -0.5], [0, 0, 0.5], N=N)
m3 = T.reorder(m3, (-1, 2, 3))
m4 = meshTri([0.5, -0.5, -0.5], [0.5, 0.5, -0.5], [0, 0, 0.5], N=N)
s = [m0] + m1 + m2 + m3 + m4
xc = 0.
yc = 0.
zc = 0.
elif surfType == 'Cylinder':
m0 = meshCircle((0, 0, -0.5), 0.5, N)
m1 = meshCircle((0, 0, 0.5), 0.5, N)
m1 = T.reorder(m1, (-1, 2, 3))
m2 = D.circle((0, 0, -0.5),
0.5,
tetas=-45,
tetae=-45 + 360,
N=4 * N - 3)
l = D.line((0, 0, -0.5), (0, 0, 0.5), N=N)
m2 = D.lineDrive(m2, l)
s = m0 + m1 + [m2]
xc = 0.
yc = 0.
zc = 0.
elif surfType == 'Cone':
s = [D.cone((0., 0, 0), 1, 0.1, 1, N=N)]
(xc, yc, zc) = G.barycenter(s)
else: # Geom parametrics surfaces
formula = base[surfType]
if formula.replace('{u}', '') == formula: # curve
s = D.curve(base[surfType], N)
else:
s = D.surface(base[surfType], N)
(xc, yc, zc) = G.barycenter(s)
s = [s]
if eltType == 'TRI':
s = C.convertArray2Tetra(s)
s = T.join(s)
s = G.close(s)
elif eltType == 'QUAD':
s = C.convertArray2Hexa(s)
s = T.join(s)
s = G.close(s)
posCam = CPlot.getState('posCam')
posEye = CPlot.getState('posEye')
dirCam = CPlot.getState('dirCam')
s = T.translate(s, (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 < 1.e-10: 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)
s = T.homothety(s, (posEye[0], posEye[1], posEye[2]), 0.5 * ll)
ux = dirCam[1] * lz - dirCam[2] * ly
uy = dirCam[2] * lx - dirCam[0] * lz
uz = dirCam[0] * ly - dirCam[1] * lx
s = T.rotate(s, (posEye[0], posEye[1], posEye[2]),
((1, 0, 0), (0, 1, 0), (0, 0, 1)),
((-ux, -uy, -uz), (lx, ly, lz), dirCam))
CTK.t = C.addBase2PyTree(CTK.t, 'SURFACES', 2)
b = Internal.getNodeFromName1(CTK.t, 'SURFACES')
if eltType == 'TRI' or eltType == 'QUAD':
nob = C.getNobOfBase(b, CTK.t)
CTK.add(CTK.t, nob, -1, s)
else:
nob = C.getNobOfBase(b, CTK.t)
if CP.__slot__ is None:
CTK.t[2][nob][2] += s
CTK.display(CTK.t)
else:
for i in s:
CTK.add(CTK.t, nob, -1, i)
#C._fillMissingVariables(CTK.t)
CTK.TXT.insert('START', 'Surface created.\n')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
def walkOut():
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
# Constraints
name = VARS[0].get()
names = name.split(';')
constraints = []
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]): constraints.append(z)
# surfaces
name = VARS[1].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[2].get(), type=1); dhloc = dhloc[0]
N = CTK.varsFromWidget(VARS[3].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[4].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)
contour = T.reorder(contour,(-1,2,3))
contours.append(contour)
# surfaceWalk
zlist = []
fail = False; errors = []
for c in contours:
try:
z = G.surfaceWalk(surfaces, c, dh, constraints=constraints,
niter=nit, check=1)
zlist.append(z)
except Exception as e:
fail = True; errors += [0,str(e)]
# Ajout dans la base SURFACES
CTK.t = C.addBase2PyTree(CTK.t, 'SURFACES')
bases = Internal.getNodesFromName1(CTK.t, 'SURFACES')
nob = C.getNobOfBase(bases[0], CTK.t)
for i in zlist: CTK.add(CTK.t, nob, -1, i)
#C._fillMissingVariables(CTK.t)
if not fail: CTK.TXT.insert('START', 'Surface walk done.\n')
else:
Panels.displayErrors(errors, header='Error: surfaceWalk')
CTK.TXT.insert('START', 'Surface walk fails for at least one zone.\n')
CTK.TXT.insert('START', 'Warning: ', 'Warning')
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CTK.display(CTK.t)
