# - Calcul d'isosurfaces a la paroi en distribue -
import Converter.PyTree as C
import Distributor2.PyTree as Distributor2
import Converter.Mpi as Cmpi
import Transform.PyTree as T
import Post.PyTree as P
rank = Cmpi.rank
size = Cmpi.size
# lecture du squelette
a = Cmpi.convertFile2SkeletonTree('cart.cgns')
# equilibrage
(a, dic) = Distributor2.distribute(a, NProc=size, algorithm='fast', useCom=0)
# load des zones locales dans le squelette
a = Cmpi.readZones(a, 'cart.cgns', proc=rank)
# Passage en arbre partiel
a = Cmpi.convert2PartialTree(a)
# Calcul de l'iso surface
isos = P.isoSurfMC(a, 'F', 0.3)
isos = Cmpi.setProc(isos, rank) # affecte les isos au proc local
t = C.newPyTree(['Base'])
t[2][1][2] += isos
# Reconstruit l'arbre complet a l'ecriture
Cmpi.convertPyTree2File(t, 'out.cgns')
# - setProc (pyTree) -
import Converter.PyTree as C
import Converter.Internal as Internal
import Converter.Mpi as Cmpi
import Distributor2.PyTree as Distributor2
import Generator.PyTree as G
import KCore.test as test
a = G.cart((0, 0, 0), (1, 1, 1), (10, 10, 10))
b = G.cart((12, 0, 0), (1, 1, 1), (10, 10, 10))
t = C.newPyTree(['Base', a, b])
t = Cmpi.setProc(t, 1)
zones = Internal.getNodesFromType(t, 'Zone_t')
for z in zones:
print z[0] + ' -> ' + str(Cmpi.getProc(z))
