Branch data Line data Source code
1 : : // *****************************************************************************
2 : : /*!
3 : : \file src/DiffEq/Dirichlet/ConfigureDirichlet.cpp
4 : : \copyright 2012-2015 J. Bakosi,
5 : : 2016-2018 Los Alamos National Security, LLC.,
6 : : 2019-2021 Triad National Security, LLC.
7 : : All rights reserved. See the LICENSE file for details.
8 : : \brief Register and compile configuration on the Dirichlet SDE
9 : : \details Register and compile configuration on the Dirichlet SDE.
10 : : */
11 : : // *****************************************************************************
12 : :
13 : : #include <set>
14 : : #include <map>
15 : : #include <vector>
16 : : #include <string>
17 : : #include <utility>
18 : :
19 : : #include <brigand/algorithms/for_each.hpp>
20 : :
21 : : #include "Tags.hpp"
22 : : #include "CartesianProduct.hpp"
23 : : #include "DiffEqFactory.hpp"
24 : : #include "Walker/Options/DiffEq.hpp"
25 : : #include "Walker/Options/InitPolicy.hpp"
26 : :
27 : : #include "ConfigureDirichlet.hpp"
28 : : #include "Dirichlet.hpp"
29 : : #include "DirichletCoeffPolicy.hpp"
30 : :
31 : : namespace walker {
32 : :
33 : : void
34 : 323 : registerDirichlet( DiffEqFactory& f, std::set< ctr::DiffEqType >& t )
35 : : // *****************************************************************************
36 : : // Register Dirichlet SDE into DiffEq factory
37 : : //! \param[in,out] f Differential equation factory to register to
38 : : //! \param[in,out] t Counters for equation types registered
39 : : // *****************************************************************************
40 : : {
41 : : // Construct vector of vectors for all possible policies for SDE
42 : : using DirPolicies =
43 : : tk::cartesian_product< InitPolicies, DirichletCoeffPolicies >;
44 : : // Register SDE for all combinations of policies
45 : : brigand::for_each< DirPolicies >(
46 : 323 : registerDiffEq< Dirichlet >( f, ctr::DiffEqType::DIRICHLET, t ) );
47 : 323 : }
48 : :
49 : : std::vector< std::pair< std::string, std::string > >
50 : 8 : infoDirichlet( std::map< ctr::DiffEqType, tk::ctr::ncomp_t >& cnt )
51 : : // *****************************************************************************
52 : : // Return information on the Dirichlet SDE
53 : : //! \param[inout] cnt std::map of counters for all differential equation types
54 : : //! \return vector of string pairs describing the SDE configuration
55 : : // *****************************************************************************
56 : : {
57 [ + - ]: 8 : auto c = ++cnt[ ctr::DiffEqType::DIRICHLET ]; // count eqs
58 : : --c; // used to index vectors starting with 0
59 : :
60 : : std::vector< std::pair< std::string, std::string > > nfo;
61 : :
62 [ + - ][ + - ]: 24 : nfo.emplace_back( ctr::DiffEq().name( ctr::DiffEqType::DIRICHLET ), "" );
63 : :
64 : 8 : nfo.emplace_back( "start offset in particle array", std::to_string(
65 [ + - ]: 8 : g_inputdeck.get< tag::component >().offset< tag::dirichlet >(c) ) );
66 [ + - ]: 8 : auto ncomp = g_inputdeck.get< tag::component >().get< tag::dirichlet >()[c];
67 [ + - ]: 8 : nfo.emplace_back( "number of components", std::to_string( ncomp ) );
68 : :
69 [ + - ]: 8 : nfo.emplace_back( "kind", "stochastic" );
70 [ + - ]: 8 : nfo.emplace_back( "dependent variable", std::string( 1,
71 [ + - ][ + - ]: 8 : g_inputdeck.get< tag::param, tag::dirichlet, tag::depvar >()[c] ) );
72 : :
73 : : auto init =
74 [ + - ]: 8 : g_inputdeck.get< tag::param, tag::dirichlet, tag::initpolicy >()[c];
75 [ + - ][ + - ]: 16 : nfo.emplace_back( "initialization policy", ctr::InitPolicy().name( init ) );
76 : :
77 [ + - ]: 8 : nfo.emplace_back( "coefficients policy", ctr::CoeffPolicy().name(
78 [ + - ]: 16 : g_inputdeck.get< tag::param, tag::dirichlet, tag::coeffpolicy >()[c] ) );
79 [ + - ][ - + ]: 16 : nfo.emplace_back( "random number generator", tk::ctr::RNG().name(
80 [ + - ]: 16 : g_inputdeck.get< tag::param, tag::dirichlet, tag::rng >()[c] ) );
81 : : nfo.emplace_back(
82 [ + - ][ + - ]: 16 : "coeff b [" + std::to_string( ncomp ) + "]",
[ - + ][ - + ]
[ - + ][ - - ]
[ - - ][ - - ]
83 [ + - ][ - + ]: 16 : parameters( g_inputdeck.get< tag::param, tag::dirichlet, tag::b >().at(c) )
84 [ + - ]: 8 : );
85 : : nfo.emplace_back(
86 [ + - ][ + - ]: 16 : "coeff S [" + std::to_string( ncomp ) + "]",
[ - + ][ - + ]
[ - + ][ - - ]
[ - - ][ - - ]
87 [ + - ][ - + ]: 16 : parameters( g_inputdeck.get< tag::param, tag::dirichlet, tag::S >().at(c) )
88 [ + - ]: 8 : );
89 : : nfo.emplace_back(
90 [ + - ][ + - ]: 16 : "coeff kappa [" + std::to_string( ncomp ) + "]",
[ - + ][ - + ]
[ - + ][ - - ]
[ - - ][ - - ]
91 [ + - ]: 8 : parameters(
92 : : g_inputdeck.get< tag::param, tag::dirichlet, tag::kappa >().at(c) )
93 [ + - ]: 8 : );
94 : :
95 [ - + ]: 8 : if (init == ctr::InitPolicyType::JOINTCORRGAUSSIAN) {
96 : : nfo.emplace_back(
97 [ - - ][ - - ]: 0 : "coeff mean [" + std::to_string( ncomp ) + "]",
[ - - ][ - - ]
[ - - ][ - - ]
[ - - ][ - - ]
98 [ - - ][ - - ]: 0 : parameters( g_inputdeck.get< tag::param, tag::dirichlet, tag::init,
99 [ - - ]: 0 : tag::mean >().at(c) ) );
100 : : auto n = std::to_string( ncomp );
101 : : nfo.emplace_back(
102 [ - - ][ - - ]: 0 : "coeff cov [" + n + '(' + n + "+1)/2="
[ - - ][ - - ]
[ - - ][ - - ]
[ - - ][ - - ]
[ - - ][ - - ]
[ - - ][ - - ]
103 [ - - ][ - - ]: 0 : + std::to_string( ncomp*(ncomp+1)/2 ) + "]",
[ - - ][ - - ]
[ - - ][ - - ]
[ - - ][ - - ]
[ - - ]
104 [ - - ][ - - ]: 0 : parameters( g_inputdeck.get< tag::param, tag::dirichlet, tag::init,
[ - - ]
105 : : tag::cov >().at(c) )
106 [ - - ]: 0 : );
107 : : }
108 : :
109 : 8 : return nfo;
110 : : }
111 : :
112 : : } // walker::
|
