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__ */