fei_Graph_Impl.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_Graph_Impl_hpp_
#define _fei_Graph_Impl_hpp_

#include <fei_iosfwd.hpp>
#include <fei_SharedPtr.hpp>
#include <fei_Graph.hpp>
#include <fei_mpi.h>
#include <fei_EqnComm.hpp>

namespace fei {

  class Graph_Impl : public fei::Graph {
  public:
    Graph_Impl(MPI_Comm comm, int firstLocalRow, int lastLocalRow);

    virtual ~Graph_Impl();

    int addIndices(int row,
       int len,
       const int* indices);

    int addSymmetricIndices(int numIndices,
          int* indices,
          bool diagonal=false);

    int gatherFromOverlap();

    table_type* getLocalGraph()
      {
  return( localGraphData_ );
      }

    std::vector<remote_table_type*>& getRemoteGraph()
      {
  return( remoteGraphData_);
      }

    int writeLocalGraph(FEI_OSTREAM& os,
      bool debug=false,
      bool prefixLinesWithPoundSign=true);

    int writeRemoteGraph(FEI_OSTREAM& os);

    int getNumLocalRows();

    int getNumLocalNonzeros() const;

    int getLocalRowLength(int row);

  private:
    void addDiagonals(int numIndices, int* indices);

    table_type* localGraphData_;
    std::vector<remote_table_type*> remoteGraphData_;
    fei::SharedPtr<fei::EqnComm> eqnComm_;

    int firstLocalRow_, lastLocalRow_;
    int localProc_, numProcs_;
    MPI_Comm comm_;
  };//class Graph_Impl

} //namespace fei

#endif // _fei_Graph_Impl_hpp_