def smooth1D(niter, eps):
fail = False
nzs = CPlot.getSelectedZones()
for nz in nzs:
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
dims = Internal.getZoneDim(z)
try:
if dims[0] == 'Unstructured': a = C.convertBAR2Struct(z)
else: a = z
a = D.getCurvilinearAbscissa(a)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
bornes = P.exteriorFaces(distrib)
distrib = T.smooth(distrib, eps=eps, niter=niter,
fixedConstraints=[bornes])
b = G.map(a, distrib)
CTK.replace(CTK.t, nob, noz, b)
except Exception as e:
fail = True
Panels.displayErrors([0,str(e)], header='Error: smooth1D')
return fail
# - getCurvilinearAbscissa (pyTree)- import Converter.PyTree as C import Geom.PyTree as D a = D.line((0., 0., 0.), (1., 0., 0), 100) a = D.getCurvilinearAbscissa(a) C.convertPyTree2File(a, 'out.cgns')
def copyDistrib():
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()
# source edge
v = VARS[3].get()
v = v.split(';')
try: pt = eval(v[1])
except: pt = None
v = v[0]
v = v.lstrip(); v = v.rstrip()
sname = v.split('/', 1)
edge = []
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]: edge.append(z)
if edge == []:
CTK.TXT.insert('START', 'Invalid source edge.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
source = edge[0]
dimSource = Internal.getZoneDim(source)
if dimSource[0] == 'Unstructured':
CTK.TXT.insert('START', 'Invalid source edge.\n')
CTK.TXT.insert('START', 'Error: ', 'Error'); return
# 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)
# subzone de source eventuellement
if dimSource[4] == 2: # source est 2D
ind = CPlot.getActivePointIndex()
ni = dimSource[1]; nj = dimSource[2]
deltai = min(ind[2]-ni,ind[2]-1)
deltaj = min(ind[3]-nj,ind[3]-1)
if deltai < deltaj:
source = T.subzone(source, (1,pt[1],pt[2]),(ni,pt[1],pt[2]))
else:
source = T.subzone(source, (pt[0],1,pt[2]),(pt[0],nj,pt[2]))
elif dimSource[4] == 3: # source est 3D
ind = CPlot.getActivePointIndex()
ni = dimSource[1]; nj = dimSource[2]; nk = dimSource[3]
deltai = min(ind[2]-ni,ind[2]-1)
deltaj = min(ind[3]-nj,ind[3]-1)
deltak = min(ind[4]-nk,ind[4]-1)
if deltai < deltaj and deltai < deltak:
source = T.subzone(source, (1,pt[1],pt[2]),(ni,pt[1],pt[2]))
elif deltaj < deltai and deltaj < deltak:
source = T.subzone(source, (pt[0],1,pt[2]),(pt[0],nj,pt[2]))
else:
source = T.subzone(source, (pt[0],pt[1],1),(pt[0],pt[1],nk))
# Extrait la distribution en i
source = D.getCurvilinearAbscissa(source)
C._initVars(source, '{CoordinateX}={s}')
C._initVars(source, 'CoordinateY', 0)
C._initVars(source, 'CoordinateZ', 0)
source = C.rmVars(source, 's')
# Traitement
if dim[0] == 'Structured':
if dim[2] != 1 and dim[3] != 1:
fail = apply3D(1., 1, source, ntype=3)
elif dim[2] != 1 and dim[3] == 1:
fail = apply2D(1., 1, source, ntype=3)
else: fail = copyDistrib1D(source)
else: fail = copyDistrib1D(source) # all zones
if not fail:
CTK.TXT.insert('START', 'Distribution copy done.\n')
else:
CTK.TXT.insert('START', 'Distribution copy 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 stretch1D(h):
fail = False
nzs = CPlot.getSelectedZones()
nz = nzs[0]
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
z = CTK.t[2][nob][2][noz]
dims = Internal.getZoneDim(z)
try:
if dims[0] == 'Unstructured': a = C.convertBAR2Struct(z)
else: a = z
except Exception as e:
#print 'Error: stretch1D: %s.'%str(e)
Panels.displayErrors([0,str(e)], header='Error: stretch1D')
return True # Fail
ind = CPlot.getActivePointIndex()
if ind == []: return True # Fail
ind = ind[0]
l = D.getLength(a)
a = D.getCurvilinearAbscissa(a)
zp = D.getCurvilinearAbscissa(z)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
N = dims[1]
val = C.getValue(zp, 's', ind)
Xc = CPlot.getActivePoint()
valf = val
Pind = C.getValue(z, 'GridCoordinates', ind)
if ind < N-1: # cherche avec indp1
Pindp1 = C.getValue(z, 'GridCoordinates', ind+1)
v1 = Vector.sub(Pindp1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(zp, 's', ind+1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if ind > 0 and val == valf: # cherche avec indm1
Pindm1 = C.getValue(z, 'GridCoordinates', ind-1)
v1 = Vector.sub(Pindm1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(zp, 's', ind-1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if h < 0: # enforce point
distrib = G.enforcePoint(distrib, valf)
else: # enforce h
if val == 0: distrib = G.enforcePlusX(distrib, h/l, N/10, 1)
elif val == 1: distrib = G.enforceMoinsX(distrib, h/l, N/10, 1)
else: distrib = G.enforceX(distrib, valf, h/l, N/10, 1)
try:
a1 = G.map(a, distrib)
CTK.replace(CTK.t, nob, noz, a1)
except Exception as e:
fail = True
Panels.displayErrors([0,str(e)], header='Error: stretch1D')
return fail
def apply3D(density, npts, factor, ntype):
nzs = CPlot.getSelectedZones()
nz = nzs[0]
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
zone = CTK.t[2][nob][2][noz]
ret = getEdges3D(zone, 0.)
if ret is None: return True
(m, r, f, ue, uf, ind) = ret
out = []
# Applique la fonction sur m
i = m[0]
dims = Internal.getZoneDim(i)
np = dims[1]*dims[2]*dims[3]
if ntype == 0: # uniformize
if density > 0: npts = D.getLength(i)*density
if factor > 0: npts = np*factor[0]
npts = int(max(npts, 2))
distrib = G.cart((0,0,0), (1./(npts-1.),1,1), (npts,1,1))
b = G.map(i, distrib)
elif ntype == 1: # refine
if factor < 0: factor = (npts-1.)/(np-1)
else: npts = factor*(np-1)+1
b = G.refine(i, factor, 1)
elif ntype == 2: # stretch (factor=h)
h = factor
l = D.getLength(i)
a = D.getCurvilinearAbscissa(i)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
N = dims[1]
val = C.getValue(a, 's', ind)
Xc = CPlot.getActivePoint()
valf = val
Pind = C.getValue(i, 'GridCoordinates', ind)
if ind < N-1: # cherche avec indp1
Pindp1 = C.getValue(i, 'GridCoordinates', ind+1)
v1 = Vector.sub(Pindp1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(a, 's', ind+1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if ind > 0 and val == valf: # cherche avec indm1
Pindm1 = C.getValue(i, 'GridCoordinates', ind-1)
v1 = Vector.sub(Pindm1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(a, 's', ind-1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if h < 0: distrib = G.enforcePoint(distrib, valf)
else:
if val == 0: distrib = G.enforcePlusX(distrib, h/l, N/10, 1)
elif val == 1: distrib = G.enforceMoinsX(distrib, h/l, N/10, 1)
else: distrib = G.enforceX(distrib, valf, h/l, N/10, 1)
b = G.map(i, distrib)
elif ntype == 3:
source = factor
b = G.map(i, source, 1)
elif ntype == 4: # smooth (factor=eps, npts=niter)
niter = npts
eps = factor
a = D.getCurvilinearAbscissa(i)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
bornes = P.exteriorFaces(distrib)
distrib = T.smooth(distrib, eps=eps, niter=niter,
fixedConstraints=[bornes])
b = G.map(i, distrib, 1)
dimb = Internal.getZoneDim(b)
npts = dimb[1]
out.append(b)
# Raffine les edges si necessaires
if npts != np:
ret = getEdges3D(zone, 2.)
if ret is None: return True
(m, r, f, ue, uf, ind) = ret
for i in r:
dims = Internal.getZoneDim(i)
np = dims[1]*dims[2]*dims[3]
factor = (npts-1.)/(np-1) # npts de m
b = G.refine(i, factor, 1)
out.append(b)
# Garde les autres
out += ue
outf = []
# Rebuild les faces
for i in f:
# trouve les edges de la face
edges = P.exteriorFacesStructured(i)
match = []
for e in edges:
dime = Internal.getZoneDim(e)
np = dime[1]-1
P0 = C.getValue(e, Internal.__GridCoordinates__, 0)
P1 = C.getValue(e, Internal.__GridCoordinates__, np)
for ei in out: # retrouve les edges par leurs extremites
dimei = Internal.getZoneDim(ei)
npi = dimei[1]-1
Q0 = C.getValue(ei, Internal.__GridCoordinates__, 0)
Q1 = C.getValue(ei, Internal.__GridCoordinates__, npi)
t1 = Vector.norm2(Vector.sub(P0,Q0))
t2 = Vector.norm2(Vector.sub(P1,Q1))
if (t1 < 1.e-12 and t2 < 1.e-12): match.append(ei)
if len(match) == 4: # OK
fn = G.TFI(match)
# Projection du patch interieur
#dimsf = Internal.getZoneDim(fn)
#fns = T.subzone(fn, (2,2,1), (dimsf[1]-1,dimsf[2]-1,1))
#fns = T.projectOrtho(fns, [i])
#fn = T.patch(fn, fns, position=(2,2,1))
#fn = T.projectOrtho(fn, [i])
outf.append(fn)
else: return True
outf += uf
try:
b = G.TFI(outf)
CTK.replace(CTK.t, nob, noz, b)
return False
except Exception as e:
Panels.displayErrors([0,str(e)], header='Error: apply3D')
return True
def apply2D(density, npts, factor, ntype=0):
nzs = CPlot.getSelectedZones()
nz = nzs[0]
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
zone = CTK.t[2][nob][2][noz]
ret = getEdges2D(zone, 0.)
if ret is None: return True
(m, r, u, ind) = ret
out = []
# Applique la fonction sur m[0] (edge a modifier)
i = m[0]
dims = Internal.getZoneDim(i)
np = dims[1]*dims[2]*dims[3]
if ntype == 0: # uniformize
if density > 0: npts = D.getLength(i)*density
if factor > 0: npts = np*factor[0]
npts = int(max(npts, 2))
distrib = G.cart((0,0,0), (1./(npts-1.),1,1), (npts,1,1))
b = G.map(i, distrib)
elif ntype == 1: # refine
if factor < 0: factor = (npts-1.)/(np-1)
else: npts = factor*(np-1)+1
b = G.refine(i, factor, 1)
elif ntype == 2: # stretch (factor=h)
h = factor
l = D.getLength(i)
a = D.getCurvilinearAbscissa(i)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
N = dims[1]
val = C.getValue(a, 's', ind)
Xc = CPlot.getActivePoint()
valf = val
Pind = C.getValue(i, 'GridCoordinates', ind)
if ind < N-1: # cherche avec indp1
Pindp1 = C.getValue(i, 'GridCoordinates', ind+1)
v1 = Vector.sub(Pindp1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(a, 's', ind+1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if ind > 0 and val == valf: # cherche avec indm1
Pindm1 = C.getValue(i, 'GridCoordinates', ind-1)
v1 = Vector.sub(Pindm1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(a, 's', ind-1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if h < 0: distrib = G.enforcePoint(distrib, valf)
else:
if val == 0: distrib = G.enforcePlusX(distrib, h/l, N/10, 1)
elif val == 1: distrib = G.enforceMoinsX(distrib, h/l, N/10, 1)
else: distrib = G.enforceX(distrib, valf, h/l, N/10, 1)
b = G.map(i, distrib)
elif ntype == 3: # copyDistrib (factor=source=edge pour l'instant)
source = factor
b = G.map(i, source, 1)
elif ntype == 4: # smooth (factor=eps, npts=niter)
niter = npts
eps = factor
a = D.getCurvilinearAbscissa(i)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
bornes = P.exteriorFaces(distrib)
distrib = T.smooth(distrib, eps=eps, niter=niter,
fixedConstraints=[bornes])
b = G.map(i, distrib, 1)
dimb = Internal.getZoneDim(b)
npts = dimb[1]
out.append(b)
# Raffine les edges si necessaires
if npts != np:
ret = getEdges2D(zone, 2.)
if ret is None: return True
(m, r, u, ind) = ret
for i in r:
dims = Internal.getZoneDim(i)
np = dims[1]*dims[2]*dims[3]
factor = (npts-1.)/(np-1) # npts de m
b = G.refine(i, factor, 1)
out.append(b)
# Garde les autres
out += u
#tp = C.newPyTree(['Base'])
#tp[2][1][2] += out
#C.convertPyTree2File(tp, 'edges.cgns')
# Rebuild
try:
b = G.TFI(out)
# Projection du patch interieur
#dimsb = Internal.getZoneDim(b)
#bs = T.subzone(b, (2,2,1), (dimsb[1]-1,dimsb[2]-1,1))
#bs = T.projectOrtho(bs, [zone])
#b = T.patch(b, bs, position=(2,2,1))
#tp = C.newPyTree(['Base'])
#tp[2][1][2] += [b, zone]
#C.convertPyTree2File(tp, 'face.cgns')
b = T.projectOrtho(b, [zone])
CTK.replace(CTK.t, nob, noz, b)
return False
except Exception as e:
Panels.displayErrors([0,str(e)], header='Error: apply2D')
return True
# - getCurvilinearAbscissa (pyTree)-
import Converter.PyTree as C
import Geom.PyTree as D
import KCore.test as test
a = D.line((0., 0., 0.), (1., 0., 0), 100)
a = D.getCurvilinearAbscissa(a)
t = C.newPyTree(['Base', 1])
t[2][1][2].append(a)
test.testT(t, 1)
# User definition of parametric curve
def f(t):
x = t
y = t * t + 1
z = 0.
return (x, y, z)
def g(t):
x = 2 * t
y = t * t + 4
z = 1.
return (x, y, z)
# test structure
a1 = D.curve(f)
b1 = D.getCurvilinearAbscissa(a1)
t = C.newPyTree(['Base', 1])
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')
