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