timestep_estimator.hpp

Go to the documentation of this file.

// Copyright (c) Lawrence Livermore National Security, LLC and
// other Smith Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include "gretl/double_state.hpp"
#include "smith/differentiable_numerics/field_state.hpp"
 
namespace smith {
 
class TimestepEstimator {
 public:
  virtual ~TimestepEstimator() {}
 
  virtual DoubleState dt(const FieldState& shape_disp, const std::vector<FieldState>& states,
                         const std::vector<FieldState>& params) const = 0;
};
 
class ConstantTimeStepEstimator : public TimestepEstimator {
 public:
  ConstantTimeStepEstimator(double dt) : dt_(dt) {}
 
  DoubleState dt([[maybe_unused]] const FieldState& shape_disp, [[maybe_unused]] const std::vector<FieldState>& states,
                 [[maybe_unused]] const std::vector<FieldState>& params) const override
  {
    double dt = dt_;
    DoubleState DT = gretl::create_state<double, double>(
        gretl::defaultInitializeZeroDual<double, double>(), [dt](FEFieldPtr) { return dt; },
        [](FEFieldPtr, double, FEDualPtr&, double) {}, shape_disp);
    return DT;
  }
 
  double dt_;  
};
 
}  // namespace smith