fei_GraphReducer.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 _fei_GraphReducer_hpp_
#define _fei_GraphReducer_hpp_
 
#include <fei_iosfwd.hpp>
#include <fei_SharedPtr.hpp>
#include <fei_Graph.hpp>
#include <fei_CommUtils.hpp>
#include <fei_Reducer.hpp>
#include <snl_fei_RaggedTable_specialize.hpp>
 
namespace fei {
 
  class GraphReducer : public fei::Graph {
  public:
    GraphReducer(fei::SharedPtr<fei::Reducer> reducer,
                 fei::SharedPtr<fei::Graph> target);
 
    virtual ~GraphReducer();
 
    int addIndices(int row,
       int len,
       const int* indices);
 
    int addSymmetricIndices(int numIndices,
          int* indices,
          bool diagonal=false);
 
    int gatherFromOverlap();
 
    table_type* getLocalGraph()
      {
  return( target_->getLocalGraph());
      }
 
    std::vector<remote_table_type*>& getRemoteGraph()
      {
  return( target_->getRemoteGraph());
      }
 
    int writeLocalGraph(FEI_OSTREAM& os,
      bool debug=false,
      bool prefixLinesWithPoundSign=true);
 
    int writeRemoteGraph(FEI_OSTREAM& os);
 
  private:
    fei::SharedPtr<fei::Reducer> reducer_;
    fei::SharedPtr<fei::Graph> target_;
  };//class GraphReducer
 
} //namespace fei
 
#endif // _fei_GraphReducer_hpp_