src/broadcastmeshbuilder.cpp

Go to the documentation of this file.

/***************************************************************************
                          broadcastmeshbuilder.cpp  -  description
                             -------------------
    begin                : Mon Dec 15 2003
    copyright            : (C) 2003 by Roman Geus
    email                : roman.geus@psi.ch
***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#include <new>
#include "broadcastmeshbuilder.h"

BroadcastMeshBuilder::BroadcastMeshBuilder(const Epetra_Comm* comm, MeshBuilder* builder)
    : _builder(builder), _comm(comm), _bc_face_counter(0)
{
}

BroadcastMeshBuilder::~BroadcastMeshBuilder() {
}

void BroadcastMeshBuilder::receiver() {
    init_coord(0);
    finalize_coord();
    init_tet(0);
    finalize_tet();
    init_bc(0);
    finalize_bc(0);
}

void BroadcastMeshBuilder::init_coord(int nof_node) {
    // broadcast nof_node
    _comm->Broadcast(&nof_node, 1, 0);
    _nof_node = nof_node;
    // call builder
    _builder->init_coord(_nof_node);
    // allocate buffer
    _coord = new double[3*nof_node];
}

void BroadcastMeshBuilder::set_coord(int i, double x, double y, double z) {
    // this function will only be called on the master processor
    _coord[3*i + 0] = x;
    _coord[3*i + 1] = y;
    _coord[3*i + 2] = z;
}

void BroadcastMeshBuilder::finalize_coord() {
    // broadcast coord data
    _comm->Broadcast(_coord, 3*_nof_node, 0);
    // call builder
    for (int i = 0; i < _nof_node; i ++) {
        _builder->set_coord(i, _coord[3*i + 0], _coord[3*i + 1], _coord[3*i + 2]);
    }
    _builder->finalize_coord();
    // delete _coord buffer
    delete[] _coord;
}

void BroadcastMeshBuilder::init_tet(int nof_tet) {
    // broadcast nof_tet
    _comm->Broadcast(&nof_tet, 1, 0);
    _nof_tet = nof_tet;
    // allocate _tet_node buffer
    _tet_node = new int[4*_nof_tet];
    _material = new int[_nof_tet];
    // call builder
    _builder->init_tet(_nof_tet);
}

void BroadcastMeshBuilder::set_tet(int t, int id0, int id1, int id2, int id3, int material) {
    // this function will only be called on the master processor
    _tet_node[4*t + 0] = id0;
    _tet_node[4*t + 1] = id1;
    _tet_node[4*t + 2] = id2;
    _tet_node[4*t + 3] = id3;
    _material[t] = material;
}

void BroadcastMeshBuilder::finalize_tet() {
    // broadcast tet_node data
    _comm->Broadcast(_tet_node, 4*_nof_tet, 0);
    // call builder
    for (int t = 0; t < _nof_tet; t ++) {
        _builder->set_tet(t, _tet_node[4*t + 0], _tet_node[4*t + 1],
                          _tet_node[4*t + 2], _tet_node[4*t + 3], _material[t]);
    }
    _builder->finalize_tet();
    // delete _tet_node buffer
    delete[] _tet_node;
}

void BroadcastMeshBuilder::init_bc(int nof_bc_face) {
    // broadcast nof_bc_face
    _comm->Broadcast(&nof_bc_face, 1, 0);
    _nof_bc_face = nof_bc_face;
    // allocate _bc_face buffer
    _bc_face = new int[4*_nof_bc_face];
    // call builder
    _builder->init_bc(_nof_bc_face);
}

void BroadcastMeshBuilder::set_bc(int id0, int id1, int id2, int bc_id) {
    // this function will only be called on the master processor
    _bc_face[_bc_face_counter ++] = id0;
    _bc_face[_bc_face_counter ++] = id1;
    _bc_face[_bc_face_counter ++] = id2;
    _bc_face[_bc_face_counter ++] = bc_id;
}

void BroadcastMeshBuilder::finalize_bc(int nof_sym) {
    // broadcast _bc_face data
    _comm->Broadcast(_bc_face, 4*_nof_bc_face, 0);
    // call builder
    int k = 0;
    for (int t = 0; t < _nof_bc_face; t ++) {
        _builder->set_bc(_bc_face[k], _bc_face[k+1],
                         _bc_face[k+2], _bc_face[k+3]);
        k += 4;
    }
    // delete _bc_face buffer
    delete[] _bc_face;
    // broadcast nof_sym
    _comm->Broadcast(&nof_sym, 1, 0);
    _nof_sym = nof_sym;
    // call builder
    _builder->finalize_bc(_nof_sym);
}

---