stdfunction_operator.hpp

Go to the documentation of this file.

// Copyright (c) 2019-2023, Lawrence Livermore National Security, LLC and
// other Serac Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include <functional>
 
#include "mfem.hpp"
 
namespace serac::mfem_ext {
 
class StdFunctionOperator : public mfem::Operator {
public:
  StdFunctionOperator(int n) : mfem::Operator(n) {}
 
  StdFunctionOperator(int h, int w) : mfem::Operator(h, w) {}
 
  StdFunctionOperator(int n, std::function<void(const mfem::Vector&, mfem::Vector&)> function)
      : mfem::Operator(n), function_(function)
  {
  }
 
  StdFunctionOperator(int h, int w, std::function<void(const mfem::Vector&, mfem::Vector&)> function)
      : mfem::Operator(h, w), function_(function)
  {
  }
 
  StdFunctionOperator(int n, std::function<void(const mfem::Vector&, mfem::Vector&)> function,
                      std::function<mfem::Operator&(const mfem::Vector&)> jacobian)
      : mfem::Operator(n), function_(function), jacobian_(jacobian)
  {
  }
 
  StdFunctionOperator(int h, int w, std::function<void(const mfem::Vector&, mfem::Vector&)> function,
                      std::function<mfem::Operator&(const mfem::Vector&)> jacobian)
      : mfem::Operator(h, w), function_(function), jacobian_(jacobian)
  {
  }
 
  void Mult(const mfem::Vector& k, mfem::Vector& y) const { function_(k, y); };
 
  mfem::Operator& GetGradient(const mfem::Vector& k) const { return jacobian_(k); };
 
private:
  std::function<void(const mfem::Vector&, mfem::Vector&)> function_;
 
  std::function<mfem::Operator&(const mfem::Vector&)> jacobian_;
};
 
}  // namespace serac::mfem_ext