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