Teko_ALOperator.hpp

/*
 * Author: Zhen Wang
 * Email: wangz@ornl.gov
 *        zhen.wang@alum.emory.edu
 */
 
#ifndef __Teko_ALOperator_hpp__
#define __Teko_ALOperator_hpp__
 
#include "Teko_BlockedTpetraOperator.hpp"
#include "Teko_Utilities.hpp"
#include "Teko_ConfigDefs.hpp"
 
namespace Teko::NS {
 
class ALOperator : public Teko::TpetraHelpers::BlockedTpetraOperator {
 public:
  ALOperator(const std::vector<std::vector<GO> >& vars,
             const Teuchos::RCP<Tpetra::Operator<ST, LO, GO, NT> >& content,
             LinearOp pressureMassMatrix, double gamma = 0.05, const std::string& label = "<ANYM>");
 
  ALOperator(const std::vector<std::vector<GO> >& vars,
             const Teuchos::RCP<Tpetra::Operator<ST, LO, GO, NT> >& content, double gamma = 0.05,
             const std::string& label = "<ANYM>");
 
  virtual ~ALOperator() {}
 
  void setPressureMassMatrix(LinearOp pressureMassMatrix);
 
  const LinearOp& getPressureMassMatrix() const { return pressureMassMatrix_; }
 
  void setGamma(double gamma);
 
  const double& getGamma() const { return gamma_; }
 
  void augmentRHS(const Tpetra::MultiVector<ST, LO, GO, NT>& b,
                  Tpetra::MultiVector<ST, LO, GO, NT>& bAugmented);
 
  int getNumberOfBlockRows() const { return numBlockRows_; }
 
  virtual void RebuildOps() { BuildALOperator(); }
 
  const Teuchos::RCP<const Tpetra::Operator<ST, LO, GO, NT> > GetBlock(int i, int j) const;
 
 protected:
  Teuchos::RCP<Thyra::LinearOpBase<ST> > alOperator_;
 
  Teuchos::RCP<Thyra::LinearOpBase<ST> > alOperatorRhs_;
 
  LinearOp pressureMassMatrix_;
  LinearOp invPressureMassMatrix_;
 
  double gamma_;
 
  int dim_;
 
  int numBlockRows_;
 
  void checkDim(const std::vector<std::vector<GO> >& vars);
 
  void BuildALOperator();
};
 
}  // end namespace Teko::NS
 
#endif /* __Teko_ALOperator_hpp__ */