nonlinear_solve.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 <vector>
#include "smith/differentiable_numerics/field_state.hpp"
#include "smith/differentiable_numerics/dirichlet_boundary_conditions.hpp"
#include "smith/physics/common.hpp"
 
namespace smith {
 
class WeakForm;
class NonlinearBlockSolverBase;
class BoundaryConditionManager;
class DirichletBoundaryConditions;
 
static constexpr size_t invalid_block_index = std::numeric_limits<size_t>::max() - 1;
 
// with unknowns (with respect to the solver) being a, and b.
// r1 has unknowns a,b in the ‘slots’ 0, 1
// r2 has unknowns a,b, in the ‘slots’ 3,0
std::vector<FieldState> block_solve(const std::vector<WeakForm*>& residual_evals,
                                    const std::vector<std::vector<size_t>> block_indices, const FieldState& shape_disp,
                                    const std::vector<std::vector<FieldState>>& states,
                                    const std::vector<std::vector<FieldState>>& params, const TimeInfo& time_info,
                                    const NonlinearBlockSolverBase* solver,
                                    const std::vector<const BoundaryConditionManager*>& bc_managers);
 
inline FieldState solve(const WeakForm& residual_eval, const FieldState& shape_disp,
                        const std::vector<FieldState>& states, const std::vector<FieldState>& params,
                        const TimeInfo& time_info, const NonlinearBlockSolverBase& solver,
                        const DirichletBoundaryConditions& bcs, size_t unknown_state_index = 0)
{
  std::vector<const BoundaryConditionManager*> bc_managers{&bcs.getBoundaryConditionManager()};
  auto solutions = block_solve({const_cast<WeakForm*>(&residual_eval)}, {{unknown_state_index}}, shape_disp, {states},
                               {params}, time_info, &solver, bc_managers);
  return solutions[0];
}
 
inline FieldState solve([[maybe_unused]] const FieldState& initial_guess, const FieldState& shape_disp,
                        const std::vector<FieldState>& weak_form_inputs, const TimeInfo& time_info,
                        const WeakForm& residual_eval, const NonlinearBlockSolverBase& solver,
                        const DirichletBoundaryConditions& bcs, size_t unknown_state_index = 0)
{
  return solve(residual_eval, shape_disp, weak_form_inputs, {}, time_info, solver, bcs, unknown_state_index);
}
 
}  // namespace smith