import Generator.PyTree as G
import Connector.PyTree as X
a = G.cylinder((0,0,0),1.,3.,360,0,1,(200,30,3)); a[0] = 'cylindre1'
a = C.addBC2Zone(a, 'wall1', 'BCWall', 'jmin')
a = C.addBC2Zone(a, 'ov1', 'BCOverlap', 'jmax')
b = G.cylinder((4,0,0),1.,3.,360,0,1,(200,30,3)); b[0] = 'cylindre2'
b = C.addBC2Zone(b, 'wall1', 'BCWall', 'jmin')
b = C.addBC2Zone(b, 'ov1', 'BCOverlap', 'jmax')
c = G.cart((-5.,-7.5,0), (15./200,15./200,1), (200,200,3))
t = C.newPyTree(['Corps1','Corps2','Bgrd'])
t[2][1][2].append(a); t[2][2][2].append(b); t[2][3][2].append(c)
t = X.connectMatch(t, dim=2)
t = C.fillEmptyBCWith(t,'nref','BCFarfield', dim=3)
t = X.applyBCOverlaps(t, depth=2)
t = X.setInterpolations(t, storage = 'direct')
C._addVars(t,'centers:Density')
C._addVars(t,'centers:MomentumX')
C._addVars(t,'centers:MomentumY')
C._addVars(t,'centers:MomentumZ')
C._addVars(t,'centers:StagnationEnergy')
for i in range(len(t[2])):
C._initVars(t[2][i], 'centers:Density', float(i+1))
C._initVars(t[2][i], 'centers:MomentumX', float(i+1))
C._initVars(t[2][i], 'centers:MomentumY', float(i+1))
C._initVars(t[2][i], 'centers:MomentumZ', float(i+1))
C._initVars(t[2][i], 'centers:StagnationEnergy', float(i+1))
t = X.chimeraTransfer(t, storage='direct', variables=['centers:Density','centers:MomentumX','centers:MomentumY','centers:MomentumZ','centers:StagnationEnergy'])
C.convertPyTree2File(t, 'out.cgns')
a = G.cylinder((0, 0, 0), 1., 3., 360, 0, 1, (200, 30, 3))
a[0] = 'cylindre1'
a = C.addBC2Zone(a, 'wall1', 'BCWall', 'jmin')
a = C.addBC2Zone(a, 'ov1', 'BCOverlap', 'jmax')
b = G.cylinder((4, 0, 0), 1., 3., 360, 0, 1, (200, 30, 3))
b[0] = 'cylindre2'
b = C.addBC2Zone(b, 'wall1', 'BCWall', 'jmin')
b = C.addBC2Zone(b, 'ov1', 'BCOverlap', 'jmax')
c = G.cart((-5., -7.5, 0), (15. / 200, 15. / 200, 1), (200, 200, 3))
t = C.newPyTree(['Corps1', 'Corps2', 'Bgrd'])
t[2][1][2].append(a)
t[2][2][2].append(b)
t[2][3][2].append(c)
t = X.connectMatch(t, dim=3)
t = C.fillEmptyBCWith(t, 'nref', 'BCFarfield', dim=3)
t = X.applyBCOverlaps(t, depth=2)
t = C.initVars(t, '{F}={CoordinateX}')
t = C.node2Center(t, ['F'])
t = C.initVars(t, '{centers:G}=1.')
t1 = X.setInterpolations(t, loc='cell', storage='direct')
t1 = X.chimeraTransfer(t1,
storage='direct',
variables=['F', 'centers:F', 'centers:G'])
test.testT(t1, 1)
t1 = X.setInterpolations(t, storage='inverse')
t1, res = X.chimeraTransfer(t1,
storage='inverse',
variables=['F', 'centers:F'])
test.testT(t1, 2)
C._addBC2Zone(a, 'ov1', 'BCOverlap', 'jmax')
b = G.cylinder((4, 0, 0), 1., 3., 360, 0, 1, (200, 30, 3))
b[0] = 'cylindre2'
C._addBC2Zone(b, 'wall1', 'BCWall', 'jmin')
C._addBC2Zone(b, 'ov1', 'BCOverlap', 'jmax')
c = G.cart((-5., -7.5, 0), (15. / 200, 15. / 200, 1), (200, 200, 3))
t = C.newPyTree(['Corps1', a, 'Corps2', b, 'Bgrd', c])
t = X.connectMatch(t, dim=3)
C._fillEmptyBCWith(t, 'nref', 'BCFarfield', dim=3)
t = X.applyBCOverlaps(t, depth=2)
C._initVars(t, '{F}={CoordinateX}')
t = C.node2Center(t, ['F'])
C._initVars(t, 'centers:G', 1.)
t1 = X.setInterpolations(t, loc='cell', storage='direct')
t1 = X.chimeraTransfer(t1,
storage='direct',
variables=['F', 'centers:F', 'centers:G'])
test.testT(t1, 1)
t1 = X.setInterpolations(t, storage='inverse')
t2, res = X.chimeraTransfer(t1,
storage='inverse',
variables=['F', 'centers:F'],
mesh='standard')
test.testT(t2, 2)
t2, res = X.chimeraTransfer(t1,
storage='inverse',
variables=['F', 'centers:F'],
mesh='extended')
test.testT(t2, 3)
