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_