Line data Source code
1 0 : // Distributed under the MIT License.
2 : // See LICENSE.txt for details.
3 :
4 : #pragma once
5 :
6 : #include <cstddef>
7 :
8 : #include "DataStructures/DataBox/PrefixHelpers.hpp"
9 : #include "Domain/Creators/Factory1D.hpp"
10 : #include "Domain/Creators/Factory2D.hpp"
11 : #include "Domain/Creators/Factory3D.hpp"
12 : #include "Domain/Tags.hpp"
13 : #include "Elliptic/Actions/RunEventsAndTriggers.hpp"
14 : #include "Elliptic/BoundaryConditions/BoundaryCondition.hpp"
15 : #include "Elliptic/DiscontinuousGalerkin/DgElementArray.hpp"
16 : #include "Elliptic/Executables/Solver.hpp"
17 : #include "Elliptic/Systems/Elasticity/Actions/InitializeConstitutiveRelation.hpp"
18 : #include "Elliptic/Systems/Elasticity/BoundaryConditions/Factory.hpp"
19 : #include "Elliptic/Systems/Elasticity/FirstOrderSystem.hpp"
20 : #include "Elliptic/Systems/Elasticity/Tags.hpp"
21 : #include "Elliptic/Triggers/Factory.hpp"
22 : #include "IO/Observer/Actions/RegisterEvents.hpp"
23 : #include "IO/Observer/Helpers.hpp"
24 : #include "IO/Observer/ObserverComponent.hpp"
25 : #include "Options/Protocols/FactoryCreation.hpp"
26 : #include "Options/String.hpp"
27 : #include "Parallel/Phase.hpp"
28 : #include "Parallel/PhaseControl/VisitAndReturn.hpp"
29 : #include "Parallel/PhaseDependentActionList.hpp"
30 : #include "Parallel/Protocols/RegistrationMetavariables.hpp"
31 : #include "Parallel/Reduction.hpp"
32 : #include "ParallelAlgorithms/Actions/TerminatePhase.hpp"
33 : #include "ParallelAlgorithms/Amr/Actions/SendAmrDiagnostics.hpp"
34 : #include "ParallelAlgorithms/Amr/Criteria/Factory.hpp"
35 : #include "ParallelAlgorithms/Amr/Protocols/AmrMetavariables.hpp"
36 : #include "ParallelAlgorithms/Events/Completion.hpp"
37 : #include "ParallelAlgorithms/Events/Factory.hpp"
38 : #include "ParallelAlgorithms/Events/Tags.hpp"
39 : #include "ParallelAlgorithms/EventsAndTriggers/Event.hpp"
40 : #include "ParallelAlgorithms/EventsAndTriggers/Trigger.hpp"
41 : #include "ParallelAlgorithms/LinearSolver/Multigrid/ElementsAllocator.hpp"
42 : #include "ParallelAlgorithms/LinearSolver/Multigrid/Tags.hpp"
43 : #include "PointwiseFunctions/AnalyticSolutions/Elasticity/Factory.hpp"
44 : #include "PointwiseFunctions/Elasticity/ConstitutiveRelations/Factory.hpp"
45 : #include "PointwiseFunctions/Elasticity/PotentialEnergy.hpp"
46 : #include "PointwiseFunctions/Elasticity/Strain.hpp"
47 : #include "PointwiseFunctions/Elasticity/Stress.hpp"
48 : #include "PointwiseFunctions/InitialDataUtilities/AnalyticSolution.hpp"
49 : #include "PointwiseFunctions/InitialDataUtilities/Background.hpp"
50 : #include "PointwiseFunctions/InitialDataUtilities/InitialGuess.hpp"
51 : #include "PointwiseFunctions/InitialDataUtilities/NumericData.hpp"
52 : #include "Utilities/ProtocolHelpers.hpp"
53 : #include "Utilities/TMPL.hpp"
54 :
55 : /// \cond
56 : namespace PUP {
57 : class er;
58 : } // namespace PUP
59 :
60 : template <size_t Dim>
61 : struct Metavariables {
62 : static constexpr Options::String help{
63 : "Find the solution to a linear elasticity problem."};
64 :
65 : static constexpr size_t volume_dim = Dim;
66 : using system = Elasticity::FirstOrderSystem<Dim>;
67 : using solver = elliptic::Solver<Metavariables, Dim, system>;
68 :
69 : using analytic_solution_fields = typename system::primal_fields;
70 : using error_compute = ::Tags::ErrorsCompute<analytic_solution_fields>;
71 : using error_tags = db::wrap_tags_in<Tags::Error, analytic_solution_fields>;
72 : using observe_fields = tmpl::append<
73 : analytic_solution_fields, error_tags, typename solver::observe_fields,
74 : tmpl::list<Elasticity::Tags::StrainCompute<volume_dim>,
75 : Elasticity::Tags::StressCompute<volume_dim>,
76 : Elasticity::Tags::PotentialEnergyDensityCompute<volume_dim>,
77 : domain::Tags::Coordinates<volume_dim, Frame::Inertial>>>;
78 : using observer_compute_tags =
79 : tmpl::list<::Events::Tags::ObserverMeshCompute<volume_dim>,
80 : ::Events::Tags::ObserverDetInvJacobianCompute<
81 : Frame::ElementLogical, Frame::Inertial>,
82 : error_compute>;
83 :
84 : // Collect all items to store in the cache.
85 : using const_global_cache_tags = tmpl::list<>;
86 :
87 : // Just a name in the input file
88 : struct ObserveNormsPerLayer {};
89 :
90 : struct factory_creation
91 : : tt::ConformsTo<Options::protocols::FactoryCreation> {
92 : using factory_classes = tmpl::map<
93 : tmpl::pair<DomainCreator<volume_dim>, domain_creators<volume_dim>>,
94 : tmpl::pair<elliptic::BoundaryConditions::BoundaryCondition<volume_dim>,
95 : Elasticity::BoundaryConditions::standard_boundary_conditions<
96 : system>>,
97 : tmpl::pair<elliptic::analytic_data::Background,
98 : tmpl::push_back<Elasticity::Solutions::
99 : all_analytic_solutions<volume_dim>,
100 : elliptic::analytic_data::NumericData>>,
101 : tmpl::pair<elliptic::analytic_data::InitialGuess,
102 : tmpl::push_back<Elasticity::Solutions::
103 : all_analytic_solutions<volume_dim>,
104 : elliptic::analytic_data::NumericData>>,
105 : tmpl::pair<elliptic::analytic_data::AnalyticSolution,
106 : Elasticity::Solutions::all_analytic_solutions<volume_dim>>,
107 : tmpl::pair<
108 : Elasticity::ConstitutiveRelations::ConstitutiveRelation<volume_dim>,
109 : Elasticity::ConstitutiveRelations::standard_constitutive_relations<
110 : volume_dim>>,
111 : tmpl::pair<::amr::Criterion,
112 : ::amr::Criteria::standard_criteria<
113 : volume_dim,
114 : tmpl::list<Elasticity::Tags::Displacement<volume_dim>>>>,
115 : tmpl::pair<
116 : Event,
117 : tmpl::flatten<tmpl::list<
118 : Events::Completion,
119 : dg::Events::field_observations<volume_dim, observe_fields,
120 : observer_compute_tags,
121 : ::amr::Tags::IsFinestGrid>,
122 : // Observation per material layer
123 : ::Events::ObserveNorms<observe_fields, observer_compute_tags,
124 : Elasticity::Tags::MaterialLayerName,
125 : ObserveNormsPerLayer>>>>,
126 : tmpl::pair<Trigger, elliptic::Triggers::all_triggers<
127 : ::amr::OptionTags::AmrGroup>>,
128 : tmpl::pair<
129 : PhaseChange,
130 : tmpl::list<
131 : // Phases for AMR
132 : PhaseControl::VisitAndReturn<
133 : Parallel::Phase::EvaluateAmrCriteria>,
134 : PhaseControl::VisitAndReturn<Parallel::Phase::AdjustDomain>,
135 : PhaseControl::VisitAndReturn<Parallel::Phase::UpdateSections>,
136 : PhaseControl::VisitAndReturn<Parallel::Phase::CheckDomain>>>>;
137 : };
138 :
139 : // Collect all reduction tags for observers
140 : using observed_reduction_data_tags =
141 : observers::collect_reduction_data_tags<tmpl::flatten<tmpl::list<
142 : tmpl::at<typename factory_creation::factory_classes, Event>,
143 : solver>>>;
144 :
145 : using initialization_actions =
146 : tmpl::push_back<typename solver::initialization_actions,
147 : Elasticity::Actions::InitializeConstitutiveRelation<Dim>,
148 : Parallel::Actions::TerminatePhase>;
149 :
150 : using register_actions =
151 : tmpl::push_back<typename solver::register_actions,
152 : observers::Actions::RegisterEventsWithObservers>;
153 :
154 : using solve_actions = typename solver::template solve_actions<tmpl::list<>>;
155 :
156 : using dg_element_array = elliptic::DgElementArray<
157 : Metavariables,
158 : tmpl::list<Parallel::PhaseActions<Parallel::Phase::Initialization,
159 : initialization_actions>,
160 : Parallel::PhaseActions<
161 : Parallel::Phase::Register,
162 : tmpl::push_back<register_actions,
163 : Parallel::Actions::TerminatePhase>>,
164 : Parallel::PhaseActions<
165 : Parallel::Phase::Restart,
166 : tmpl::push_back<register_actions,
167 : Parallel::Actions::TerminatePhase>>,
168 : Parallel::PhaseActions<Parallel::Phase::Solve, solve_actions>,
169 : Parallel::PhaseActions<
170 : Parallel::Phase::CheckDomain,
171 : tmpl::list<::amr::Actions::SendAmrDiagnostics,
172 : Parallel::Actions::TerminatePhase>>>,
173 : LinearSolver::multigrid::ElementsAllocator<
174 : volume_dim, typename solver::multigrid::options_group>>;
175 :
176 : struct amr : tt::ConformsTo<::amr::protocols::AmrMetavariables> {
177 : using element_array = dg_element_array;
178 : using projectors = tmpl::push_back<
179 : typename solver::amr_projectors,
180 : Elasticity::Actions::InitializeConstitutiveRelation<Dim>>;
181 : static constexpr bool keep_coarse_grids = true;
182 : static constexpr bool p_refine_only_in_event = false;
183 : };
184 :
185 : struct registration
186 : : tt::ConformsTo<Parallel::protocols::RegistrationMetavariables> {
187 : using element_registrars =
188 : tmpl::map<tmpl::pair<dg_element_array, register_actions>>;
189 : };
190 :
191 : // Specify all parallel components that will execute actions at some point.
192 : using component_list = tmpl::flatten<
193 : tmpl::list<dg_element_array, typename solver::component_list,
194 : observers::Observer<Metavariables>,
195 : observers::ObserverWriter<Metavariables>>>;
196 :
197 : static constexpr std::array<Parallel::Phase, 6> default_phase_order{
198 : {Parallel::Phase::Initialization, Parallel::Phase::Register,
199 : Parallel::Phase::UpdateSections, Parallel::Phase::CheckDomain,
200 : Parallel::Phase::Solve, Parallel::Phase::Exit}};
201 :
202 : // NOLINTNEXTLINE(google-runtime-references)
203 : void pup(PUP::er& /*p*/) {}
204 : };
205 : /// \endcond
|
