multiphysics_time_integrator.hpp

// 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 "smith/differentiable_numerics/state_advancer.hpp"
#include "smith/differentiable_numerics/field_state.hpp"
#include "smith/differentiable_numerics/nonlinear_block_solver.hpp"
#include "smith/numerics/solver_config.hpp"
#include "smith/differentiable_numerics/time_discretized_weak_form.hpp"
#include "smith/differentiable_numerics/time_integration_rule.hpp"
#include "smith/physics/mesh.hpp"
#include "smith/differentiable_numerics/field_store.hpp"
 
namespace smith {
 
class CoupledSystemSolver;
class DirichletBoundaryConditions;
class BoundaryConditionManager;
 
std::vector<FieldState> solve(const std::vector<std::shared_ptr<WeakForm>>& weak_forms, const FieldStore& field_store,
                              const CoupledSystemSolver* solver, const TimeInfo& time_info,
                              const std::vector<FieldState>& params = {});
 
class MultiphysicsTimeIntegrator : public StateAdvancer {
 public:
  MultiphysicsTimeIntegrator(std::shared_ptr<FieldStore> field_store,
                             const std::vector<std::shared_ptr<WeakForm>>& weak_forms,
                             std::shared_ptr<smith::CoupledSystemSolver> solver,
                             std::shared_ptr<WeakForm> cycle_zero_weak_form = nullptr,
                             std::shared_ptr<smith::CoupledSystemSolver> cycle_zero_solver = nullptr);
 
  std::pair<std::vector<FieldState>, std::vector<ReactionState>> advanceState(
      const TimeInfo& time_info, const FieldState& shape_disp, const std::vector<FieldState>& states,
      const std::vector<FieldState>& params) const override;
 
 private:
  std::shared_ptr<FieldStore> field_store_;
  std::vector<std::shared_ptr<WeakForm>> weak_forms_;
  std::shared_ptr<smith::CoupledSystemSolver> solver_;
  std::shared_ptr<WeakForm> cycle_zero_weak_form_;
  std::shared_ptr<smith::CoupledSystemSolver> cycle_zero_solver_;
};
 
}  // namespace smith