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