utest-mesh-slice.py

Go to the documentation of this file.

import sys
import coolfluid as cf
from math import pi

env = cf.Core.environment()
env.log_level = 1
#env.only_cpu0_writes = False

root = cf.Core.root()
source_domain = root.create_component('SourceDomain', 'cf3.mesh.Domain')
target_domain = root.create_component('TargetDomain', 'cf3.mesh.Domain')
source_mesh = source_domain.create_component('sourcemesh','cf3.mesh.Mesh')

h = 1.

div = 2

blocks = root.create_component('model', 'cf3.mesh.BlockMesh.BlockArrays')
grading = 20.
y_segs = 64/div
x_size = 4.*pi*h
z_size = 2.*pi*h
x_segs = 32/div
z_segs = 32/div
points = blocks.create_points(dimensions = 2, nb_points = 6)
points[0]  = [0, 0.]
points[1]  = [x_size, 0.]
points[2]  = [0., h]
points[3]  = [x_size, h]
points[4]  = [0.,2.*h]
points[5]  = [x_size, 2.*h]
block_nodes = blocks.create_blocks(2)
block_nodes[0] = [0, 1, 3, 2]
block_nodes[1] = [2, 3, 5, 4]
block_subdivs = blocks.create_block_subdivisions()
block_subdivs[0] = [x_segs, y_segs]
block_subdivs[1] = block_subdivs[0]
gradings = blocks.create_block_gradings()
gradings[0] = [1., 1., grading, grading]
gradings[1] = [1., 1., 1./grading, 1./grading]
left_patch = blocks.create_patch_nb_faces(name = 'left', nb_faces = 2)
left_patch[0] = [2, 0]
left_patch[1] = [4, 2]
bottom_patch = blocks.create_patch_nb_faces(name = 'bottom', nb_faces = 1)
bottom_patch[0] = [0, 1]
top_patch = blocks.create_patch_nb_faces(name = 'top', nb_faces = 1)
top_patch[0] = [5, 4]
right_patch = blocks.create_patch_nb_faces(name = 'right', nb_faces = 2)
right_patch[0] = [1, 3]
right_patch[1] = [3, 5]
blocks.extrude_blocks(positions=[z_size], nb_segments=[z_segs], gradings=[1.])
blocks.partition_blocks(nb_partitions = 2, direction = 1)
blocks.partition_blocks(nb_partitions = 2, direction = 0)
blocks.partition_blocks(nb_partitions = 2, direction = 2)
blocks.create_mesh(source_mesh.uri())

source_coords = source_mesh.geometry.coordinates
testfield = source_mesh.geometry.create_field(name = 'test', variables = 'testx,testy')
for i in range(len(source_coords)):
  testfield[i][0] = 1.+2.*source_coords[i][0]
  testfield[i][1] = 2.+3.*source_coords[i][1]
  
target_mesh = target_domain.create_component('targetmesh','cf3.mesh.Mesh')
generator = target_domain.create_component('Generator', 'cf3.mesh.GeneratePlane3D')
generator.mesh = target_mesh.uri()
generator.origin = [0., h, 0.]
generator.normal = [0., 1., 0.]
generator.size = [x_size, z_size]
generator.segments = [x_segs, z_segs]
generator.execute()

interpolator = root.create_component('Interpolator', 'cf3.mesh.actions.MeshInterpolator')
interpolator.source_mesh = source_mesh
interpolator.target_mesh = target_mesh
interpolator.execute()

tgt_coords = target_mesh.geometry.coordinates
tgt_test = target_mesh.geometry.test
tgt_ranks = target_mesh.geometry.children.rank
my_rank = cf.Core.rank()

for i in range(len(tgt_coords)):
  if tgt_ranks[i] == my_rank:
    print tgt_coords[i][0], tgt_coords[i][1], tgt_coords[i][2], tgt_test[i][0]