LinearSystem.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_LinearSystem_hpp
#define stk_linsys_LinearSystem_hpp

#include <stk_linsys/FeiBaseIncludes.hpp>
#include <stk_linsys/DofMapper.hpp>
#include <stk_linsys/LinearSystemInterface.hpp>

#include <Teuchos_ParameterList.hpp>

namespace stk_classic {
namespace linsys {

class LinearSystem : public LinearSystemInterface {
 public:
  LinearSystem(MPI_Comm comm, fei::SharedPtr<fei::Factory> factory);

  virtual ~LinearSystem();

  void set_parameters(Teuchos::ParameterList& paramlist);

  void synchronize_mappings_and_structure();

  void create_fei_LinearSystem();

  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;
  DofMapper m_dof_mapper;
  fei::SharedPtr<fei::MatrixGraph> m_fei_mgraph;

  fei::SharedPtr<fei::LinearSystem> m_fei_linearsystem;
  fei::ParameterSet m_param_set;
};//class LinearSystem

}//namespace linsys
}//namespace stk_classic

#endif