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