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