snl_fei_SubdMsgHandler.hpp

/*--------------------------------------------------------------------*/
/*    Copyright 2005 Sandia Corporation.                              */
/*    Under the terms of Contract DE-AC04-94AL85000, there is a       */
/*    non-exclusive license for use of this work by or on behalf      */
/*    of the U.S. Government.  Export of this program may require     */
/*    a license from the United States Government.                    */
/*--------------------------------------------------------------------*/

#ifndef _snl_fei_SubdMsgHandler_hpp_
#define _snl_fei_SubdMsgHandler_hpp_

#include <fei_macros.hpp>
#include <fei_CommUtils.hpp>
#include <fei_fwd.hpp>

#include <vector>

namespace fei {
  template<typename T> class SharedIDs;
}

namespace snl_fei {
  class SubdMsgHandler : public fei::MessageHandler<int> {
  public:
    SubdMsgHandler(RecordCollection* recordCollection,
       fei::SharedIDs<int>* sharedIDTable,
       fei::SharedIDs<int>* subdomainIDTable);
    virtual ~SubdMsgHandler();

    std::vector<int>& getSendProcs();

    std::vector<int>& getRecvProcs();

    int getSendMessageLength(int destProc, int& messageLength);

    int getSendMessage(int destProc, std::vector<int>& message);

    int processRecvMessage(int srcProc, std::vector<int>& message);

    void setSendPattern(fei::comm_map* pattern)
      { sendPattern_ = pattern; }

    void setRecvPattern(fei::comm_map* pattern)
      { recvPattern_ = pattern; }

  private:
    fei::comm_map* sendPattern_;
    fei::comm_map* recvPattern_;
    RecordCollection* recordCollection_;
    fei::SharedIDs<int>* sharedIDTable_;
    fei::SharedIDs<int>* subdomainIDTable_;

    std::vector<int> sendProcs_;
    std::vector<int> recvProcs_;
  };//class SubdMsgHandler
}//namespace snl_fei

#endif // _snl_fei_SubdMsgHandler_hpp_