# - computeDiff (pyTree) -
import Converter.PyTree as C
import Post.PyTree as P
import Generator.PyTree as G
ni = 30
nj = 40
nk = 1
m = G.cart((0, 0, 0), (10. / (ni - 1), 10. / (nj - 1), 1), (ni, nj, nk))
m = C.initVars(m, '{Density}=({CoordinateX}>5)*1.')
m = P.computeDiff(m, 'Density')
C.convertPyTree2File(m, 'out.cgns')
def celln(y):
if y > 5.: return True
else: return False
#---------------------
# noeuds 2D structure
#---------------------
ni = 30
nj = 40
nk = 1
m = G.cart((0, 0, 0), (10. / (ni - 1), 10. / (nj - 1), 1), (ni, nj, nk))
m = C.initVars(m, 'Density', F, ['CoordinateX'])
m = P.computeDiff(m, 'Density')
test.testT(m, 1)
#
# Prise en compte du cellN
#
m = C.initVars(m, 'cellN', celln, ['CoordinateY'])
m = P.computeDiff(m, 'Density')
test.testT(m, 2)
#
# Prise en compte des raccords pour les champs aux centres des cellules
#
m1 = G.cart((0., 0., 0.), (10. / (ni - 1), 10. / (nj - 1), 1), (ni, nj, nk))
m2 = G.cart((10., 0., 0.), (10. / (ni - 1), 10. / (nj - 1), 1), (ni, nj, nk))
m1 = C.initVars(m1, 'Density', F2, ['CoordinateX'])
m2 = C.initVars(m2, 'Density', F2, ['CoordinateX'])
import KCore.test as test
def F(x, y):
return (x - 5) * (x - 5)
def celln(y):
if (y > 5.): return True
else: return False
#-------------------------------
# 2D structure + raccords match
#-------------------------------
ni = 10
nj = 10
nk = 1
a = G.cart((0, 0, 0), (1, 1, 1), (ni, nj, nk))
a = C.initVars(a, 'Density', F, ['CoordinateX', 'CoordinateY'])
b = G.cart((9, 0, 0), (1, 1, 1), (ni, nj, nk))
b = C.initVars(b, 'Density', F, ['CoordinateX', 'CoordinateY'])
a = C.addBC2Zone(a, 'match1', 'BCMatch', 'imax', b, 'imin', trirac=[1, 2])
b = C.addBC2Zone(b, 'match2', 'BCMatch', 'imin', a, 'imax', trirac=[1, 2])
t = C.newPyTree(['Base'])
t[2][1][2] += [a, b]
t = P.computeDiff(t, 'Density')
test.testT(t, 1)
