amroc/solids/beam/BeamELCCoupledSolver.h

Go to the documentation of this file.

// -*- C++ -*-

// Copyright (C) 2003-2007 California Institute of Technology
// Ralf Deiterding, ralf@cacr.caltech.edu

#ifndef BEAM_ELC_COUPLED_SOLVER_H
#define BEAM_ELC_COUPLED_SOLVER_H

template <int dim> class BeamCoupledSolver;

#include "BeamCoupledSolver2.h"
#include "elc/LagrangianComm.h"

template <int dim>
class BeamELCCoupledSolver : public BeamCoupledSolver<dim> {
  typedef BeamCoupledSolver<dim> base;
public:
  typedef elc::LagrangianComm<dim,double> elc_lagcomm_type;

  BeamELCCoupledSolver() : base(), _olog(0), _elcLag(0) {}

  virtual ~BeamELCCoupledSolver() {     
    if (_elcLag) delete _elcLag;
    if (_olog) delete _olog;
  }

  void SetupInterComm(MPI_Comm solidComm, int numFluidNodes, int firstFluidNode) {
    MPI_Comm_rank(solidComm, &myRank);
#ifdef DEBUG_PRINT
    std::ostringstream obuf;
    obuf << "S" << myRank << ".log" << static_cast<char>(0);
    oflog.open(obuf.str().c_str());     
    _olog = new std::ostream(oflog.rdbuf());
#endif

#ifdef DEBUG_PRINT_ELC
    ( log() << "*** LagrangianComm: " << numFluidNodes << "  " << firstFluidNode ).flush();
#endif  
    // instantiate an elc object for data exchange
    _elcLag = new elc_lagcomm_type(MPI_COMM_WORLD, solidComm, numFluidNodes,
                                   firstFluidNode, elc::GlobalIdentifiers);
#ifdef DEBUG_PRINT_ELC
    ( log() << " created.\n").flush();
#endif  
  }

  virtual void sendBoundaryReceivePressure() {

    int elNumNodes = base::NumNodes();
    int elNumElements = base::NumElements();
    double *elPressures = base::Pressures();
    const int* elGlobalNodeIDs = base::NodeIDs();
    const int* elConnectivity = base::Connections();
    const double* elCoordinates = base::Coordinates();
    const double* elVelocities = base::Velocities();
    base::UpdateBoundary();
#ifdef DEBUG_PRINT_ELC_VALUES
    for (register int n=0; n<elNumNodes; n++) 
      ( log() << "(" << elCoordinates[2*n] << "," 
        << elCoordinates[2*n+1] << ")  " ).flush();
    ( log() << std::endl ).flush();
#endif

#ifdef DEBUG_PRINT_ELC
    ( log() << "*** BeamCoupledSolver::sendBoundaryReceivePressure" ).flush();
#endif
        
    _elcLag->sendMesh(elNumNodes, reinterpret_cast<const void*>(elGlobalNodeIDs), 
                      reinterpret_cast<const void*>(elCoordinates), 
                      reinterpret_cast<const void*>(elVelocities), 
                      elNumElements, reinterpret_cast<const void*>(elConnectivity));
    _elcLag->waitForMesh();
    _elcLag->receivePressure(elNumNodes, reinterpret_cast<void*>(elPressures));
    _elcLag->waitForPressure();
    // Necessary if solid domain larger than fluid domain
    for (register int n=0; n<elNumNodes; n++) 
      if (elPressures[n] == std::numeric_limits<double>::max()) 
        elPressures[n] = 0.0;
#ifdef DEBUG_PRINT_ELC_VALUES
    for (register int n=0; n<elNumNodes; n++) 
      ( log() << elPressures[n] << "  " ).flush();
    ( log() << std::endl ).flush();
#endif
    
#ifdef DEBUG_PRINT_ELC
    ( log() << " done.\n" ).flush();
#endif
  }

  inline std::ostream& log() { return *_olog; }

protected:
  std::ostream* _olog;
  std::ofstream oflog; 
  int myRank;
  elc_lagcomm_type* _elcLag;
};

#endif

---