Teko_ModALPreconditionerFactory.hpp

/*
 * Author: Zhen Wang
 * Email: wangz@ornl.gov
 *        zhen.wang@alum.emory.edu
 */
 
#ifndef __Teko_ModALPreconditionerFactory_hpp__
#define __Teko_ModALPreconditionerFactory_hpp__
 
#include "Teko_BlockPreconditionerFactory.hpp"
#include "Teko_InvModALStrategy.hpp"
 
namespace Teko
{
 
namespace NS
{

class ModALPrecondState : public BlockPreconditionerState
{
public:
   ModALPrecondState();
 
   LinearOp pressureMassMatrix_;
   LinearOp invPressureMassMatrix_;
   ModifiableLinearOp B1tMpB1_, B2tMpB2_, B3tMpB3_;
   ModifiableLinearOp A11p_, A22p_, A33p_;
   ModifiableLinearOp invA11p_, invA22p_, invA33p_, invS_;
   ModifiableLinearOp S_;
 
   double gamma_;
   bool isStabilized_;
};

 
class ModALPreconditionerFactory : public BlockPreconditionerFactory
{
public:
 
   ModALPreconditionerFactory();
 
   ModALPreconditionerFactory(const Teuchos::RCP<InverseFactory> & factory);
 
   ModALPreconditionerFactory(const Teuchos::RCP<InverseFactory> & invFactoryA,
         const Teuchos::RCP<InverseFactory> & invFactoryS);
 
   ModALPreconditionerFactory(const Teuchos::RCP<InverseFactory> & factory,
         LinearOp & pressureMassMatrix);
 
   ModALPreconditionerFactory(const Teuchos::RCP<InverseFactory> & invFactoryA,
         const Teuchos::RCP<InverseFactory> & invFactoryS,
         LinearOp & pressureMassMatrix);
 
   ModALPreconditionerFactory(const Teuchos::RCP<InvModALStrategy> & strategy);

   void
   setPressureMassMatrix(const LinearOp & pressureMassMatrix)
   {
      invOpsStrategy_->setPressureMassMatrix(pressureMassMatrix);
   }

   void
   setGamma(double gamma)
   {
      invOpsStrategy_->setGamma(gamma);
   }

   virtual LinearOp
   buildPreconditionerOperator(BlockedLinearOp & alOp,
         BlockPreconditionerState & state) const;

   virtual Teuchos::RCP<PreconditionerState>
   buildPreconditionerState() const
   {
      return Teuchos::rcp(new ModALPrecondState());
   }
 
protected:
   using BlockPreconditionerFactory::buildPreconditionerOperator;
 
   Teuchos::RCP<InvModALStrategy> invOpsStrategy_;
 
   bool isSymmetric_;
};
 
} // end namespace NS
 
} // end namespace Teko
 
#endif /* __Teko_ModALPreconditionerFactory_hpp__ */