Teko_LU2x2Strategy.hpp

// @HEADER
// *****************************************************************************
//      Teko: A package for block and physics based preconditioning
//
// Copyright 2010 NTESS and the Teko contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef __Teko_LU2x2Strategy_hpp__
#define __Teko_LU2x2Strategy_hpp__

#include "Teuchos_RCP.hpp"
#include "Thyra_LinearOpBase.hpp"

#include "Teko_Utilities.hpp"
#include "Teko_InverseFactory.hpp"
#include "Teko_BlockPreconditionerFactory.hpp"

namespace Teko {

class LU2x2Strategy {
 public:
  virtual ~LU2x2Strategy() {}

  virtual const Teko::LinearOp getHatInvA00(const Teko::BlockedLinearOp& A,
                                            BlockPreconditionerState& state) const = 0;

  virtual const Teko::LinearOp getTildeInvA00(const Teko::BlockedLinearOp& A,
                                              BlockPreconditionerState& state) const = 0;

  virtual const Teko::LinearOp getInvS(const Teko::BlockedLinearOp& A,
                                       BlockPreconditionerState& state) const = 0;

  virtual void initializeFromParameterList(const Teuchos::ParameterList& /* settings */,
                                           const InverseLibrary& /* invLib */) {}

  virtual Teuchos::RCP<Teuchos::ParameterList> getRequestedParameters() const {
    return Teuchos::null;
  }

  virtual bool updateRequestedParameters(const Teuchos::ParameterList& /* pl */) { return true; }

  void setRequestHandler(const Teuchos::RCP<RequestHandler>& rh) { requestHandler_ = rh; }

  Teuchos::RCP<RequestHandler> getRequestHandler() const { return requestHandler_; }

 private:
  Teuchos::RCP<RequestHandler> requestHandler_;
};

class StaticLU2x2Strategy : public LU2x2Strategy {
 public:
  StaticLU2x2Strategy(const Teko::LinearOp& hInvA00, const Teko::LinearOp& tInvA00,
                      const Teko::LinearOp& invS)
      : hatInvA00_(hInvA00), tildeInvA00_(tInvA00), invS_(invS) {}

  virtual ~StaticLU2x2Strategy() {}


  virtual const Teko::LinearOp getHatInvA00(const Teko::BlockedLinearOp& /* A */,
                                            BlockPreconditionerState& /* state */) const {
    return hatInvA00_;
  }

  virtual const Teko::LinearOp getTildeInvA00(const Teko::BlockedLinearOp& /* A */,
                                              BlockPreconditionerState& /* state */) const {
    return tildeInvA00_;
  }

  virtual const Teko::LinearOp getInvS(const Teko::BlockedLinearOp& /* A */,
                                       BlockPreconditionerState& /* state */) const {
    return invS_;
  }


 protected:
  const Teko::LinearOp hatInvA00_;   
  const Teko::LinearOp tildeInvA00_; 
  const Teko::LinearOp invS_;        
 private:
  // hide me!
  StaticLU2x2Strategy() {}
};

}  // end namespace Teko

#endif