coupling/MotionELCCoupledSolver.h

Go to the documentation of this file.

// -*- C++ -*-

#ifndef MOTION_ELC_COUPLED_SOLVER_H
#define MOTION_ELC_COUPLED_SOLVER_H

#include "Point.h"
#include "MotionCoupledSolver.h"
#include "elc/LagrangianComm.h"

template <class DataType,int dim>
class MotionELCCoupledSolver : public MotionCoupledSolver<DataType,dim> {
  typedef MotionCoupledSolver<DataType,dim> base;
public:
  typedef elc::LagrangianComm<dim,DataType> elc_lagcomm_type;

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

  ~MotionELCCoupledSolver() {
    if (_elcLag) delete _elcLag;
  }

  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());
    base::_bolog = _olog;
#endif
    MKlog = _olog;

#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 SendBoundaryData() {
    int elNumNodes = base::NumNodes();
    int elNumElements = base::NumElements();
    const int *elGlobalNodeIDs = base::NodeIDs();
    const int *elConnectivity = base::Connections();
    const DataType *elCoordinates = base::Coordinates();
    const DataType *elVelocities = base::Velocities();
 
#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() << "*** MotionELCCoupledSolver::SendBoundaryData() " ).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();

#ifdef DEBUG_PRINT_ELC
    ( log() << " done.\n" ).flush();
#endif
  }

  virtual void PostReceiveBoundaryData() {
    DataType *elPressures = base::Pressures();
    int elNumNodes = base::NumNodes(); 

#ifdef DEBUG_PRINT_ELC
    ( log() << "*** MotionELCCoupledSolver::PostReceiveBoundaryData()" ).flush();
#endif
        
    _elcLag->receivePressure(elNumNodes, reinterpret_cast<void*>(elPressures));

#ifdef DEBUG_PRINT_ELC
    ( log() << " done.\n" ).flush();
#endif
  }

  virtual void WaitReceiveBoundaryData() {
    DataType *elPressures = base::Pressures();
    int elNumNodes = base::NumNodes();

#ifdef DEBUG_PRINT_ELC
    ( log() << "*** MotionELCCoupledSolver::WaitReceiveBoundaryData()" ).flush();
#endif
        
    _elcLag->waitForPressure();

    // Necessary if solid domain larger than fluid domain
    for (register int n=0; n<elNumNodes; n++) 
      if (elPressures[n] == std::numeric_limits<DataType>::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
  }

  virtual void sendBoundaryReceivePressure() {
    int elNumNodes = base::NumNodes();
    int elNumElements = base::NumElements();
    DataType *elPressures = base::Pressures();
    const int* elGlobalNodeIDs = base::NodeIDs();
    const int* elConnectivity = base::Connections();
    const DataType* elCoordinates = base::Coordinates();
    const DataType* elVelocities = base::Velocities();
#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() << "*** MotionELCCoupledSolver::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<DataType>::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

---