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_BlockedEpetraOperator.hpp"
#include "Teko_Utilities.hpp"

namespace Teko {

namespace NS {

class ALOperator : public Teko::Epetra::BlockedEpetraOperator {
 public:
  ALOperator(const std::vector<std::vector<int> >& vars,
             const Teuchos::RCP<Epetra_Operator>& content, LinearOp pressureMassMatrix,
             double gamma = 0.05, const std::string& label = "<ANYM>");

  ALOperator(const std::vector<std::vector<int> >& vars,
             const Teuchos::RCP<Epetra_Operator>& content, double gamma = 0.05,
             const std::string& label = "<ANYM>");

  // Destructor
  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 Epetra_MultiVector& b, Epetra_MultiVector& bAugmented);

  int getNumberOfBlockRows() const { return numBlockRows_; }

  virtual void RebuildOps() { BuildALOperator(); }

  const Teuchos::RCP<const Epetra_Operator> GetBlock(int i, int j) const;

 protected:
  Teuchos::RCP<Thyra::LinearOpBase<double> > alOperator_;

  Teuchos::RCP<Thyra::LinearOpBase<double> > alOperatorRhs_;

  LinearOp pressureMassMatrix_;

  LinearOp invPressureMassMatrix_;

  double gamma_;

  int dim_;

  int numBlockRows_;

  void checkDim(const std::vector<std::vector<int> >& vars);

  void BuildALOperator();
};

}  // end namespace NS

}  // end namespace Teko

#endif /* __Teko_ALOperator_hpp__ */