AggregateLinearSystem.hpp

/*------------------------------------------------------------------------*/
/*                 Copyright 2010 Sandia Corporation.                     */
/*  Under 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 stk_linsys_AggregateLinearSystem_hpp
#define stk_linsys_AggregateLinearSystem_hpp
 
#include <stk_linsys/FeiBaseIncludes.hpp>
#include <stk_linsys/DofMapper.hpp>
#include <stk_linsys/LinearSystemInterface.hpp>
#include <stk_linsys/LinearSystem.hpp>
 
#include <Teuchos_ParameterList.hpp>
 
namespace stk_classic {
namespace linsys {
 
class AggregateLinearSystem : public LinearSystemInterface {
 public:
  AggregateLinearSystem(MPI_Comm comm, fei::SharedPtr<fei::Factory> factory, size_t num_matrices=1, size_t num_rhsvecs=1);
 
  virtual ~AggregateLinearSystem();
 
  void set_parameters(Teuchos::ParameterList& paramlist);
 
  void set_num_matrices_rhsvecs(size_t num_matrices, size_t num_rhsvecs);
 
  void synchronize_mappings_and_structure();
 
  void create_fei_LinearSystem();
 
  fei::SharedPtr<fei::Matrix> get_matrix(size_t index);
 
  fei::SharedPtr<fei::Vector> get_rhsvec(size_t index);
 
  void aggregate_system(const std::vector<double>& mat_scalars,
                        const std::vector<double>& rhs_scalars);
 
  void finalize_assembly();
 
  const DofMapper& get_DofMapper() const;
 
  DofMapper& get_DofMapper();
 
  void reset_to_zero();
 
  const fei::SharedPtr<fei::MatrixGraph> get_fei_MatrixGraph() const;
 
  fei::SharedPtr<fei::MatrixGraph> get_fei_MatrixGraph();
 
  const fei::SharedPtr<fei::LinearSystem> get_fei_LinearSystem() const;
 
  fei::SharedPtr<fei::LinearSystem> get_fei_LinearSystem();
 
  void write_files(const std::string& base_name) const;
 
  int solve(int & status, const Teuchos::ParameterList & params);
 
 private:
 
  fei::SharedPtr<fei::Factory> m_fei_factory;
  stk_classic::linsys::LinearSystem m_linear_system;
 
  std::vector<fei::SharedPtr<fei::Matrix> > m_matrices;
  std::vector<fei::SharedPtr<fei::Vector> > m_rhsvecs;
};//class AggregateLinearSystem
 
}//namespace linsys
}//namespace stk_classic
 
#endif