Branch data Line data Source code
1 : : // *****************************************************************************
2 : : /*!
3 : : \file src/DiffEq/Dirichlet/ConfigureGeneralizedDirichlet.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 generlized Dirichlet SDE
9 : : \details Register and compile configuration on the generlized 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 "ConfigureGeneralizedDirichlet.hpp"
28 : : #include "GeneralizedDirichlet.hpp"
29 : : #include "GeneralizedDirichletCoeffPolicy.hpp"
30 : :
31 : : namespace walker {
32 : :
33 : : void
34 : 323 : registerGenDir( DiffEqFactory& f, std::set< ctr::DiffEqType >& t )
35 : : // *****************************************************************************
36 : : // Register generlized 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 GenDirPolicies =
43 : : tk::cartesian_product< InitPolicies, GeneralizedDirichletCoeffPolicies >;
44 : : // Register SDE for all combinations of policies
45 : : brigand::for_each< GenDirPolicies >(
46 : 323 : registerDiffEq< GeneralizedDirichlet >( f, ctr::DiffEqType::GENDIR, t ) );
47 : 323 : }
48 : :
49 : : std::vector< std::pair< std::string, std::string > >
50 : 4 : infoGenDir( std::map< ctr::DiffEqType, tk::ctr::ncomp_t >& cnt )
51 : : // *****************************************************************************
52 : : // Return information on the generalized 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 [ + - ]: 4 : auto c = ++cnt[ ctr::DiffEqType::GENDIR ]; // count eqs
58 : : --c; // used to index vectors starting with 0
59 : :
60 : : std::vector< std::pair< std::string, std::string > > nfo;
61 : :
62 [ + - ][ + - ]: 8 : nfo.emplace_back( ctr::DiffEq().name( ctr::DiffEqType::GENDIR ), "" );
63 : :
64 : 4 : nfo.emplace_back( "start offset in particle array", std::to_string(
65 [ + - ]: 4 : g_inputdeck.get< tag::component >().offset< tag::gendir >(c) ) );
66 [ + - ]: 4 : auto ncomp = g_inputdeck.get< tag::component >().get< tag::gendir >()[c];
67 [ + - ]: 4 : nfo.emplace_back( "number of components", std::to_string( ncomp ) );
68 : :
69 [ + - ]: 4 : nfo.emplace_back( "kind", "stochastic" );
70 [ + - ]: 4 : nfo.emplace_back( "dependent variable", std::string( 1,
71 [ + - ][ + - ]: 4 : g_inputdeck.get< tag::param, tag::gendir, tag::depvar >()[c] ) );
72 [ + - ]: 4 : nfo.emplace_back( "initialization policy", ctr::InitPolicy().name(
73 [ + - ]: 8 : g_inputdeck.get< tag::param, tag::gendir, tag::initpolicy >()[c] ) );
74 [ + - ]: 4 : nfo.emplace_back( "coefficients policy", ctr::CoeffPolicy().name(
75 [ + - ]: 8 : g_inputdeck.get< tag::param, tag::gendir, tag::coeffpolicy >()[c] ) );
76 [ + - ][ - + ]: 8 : nfo.emplace_back( "random number generator", tk::ctr::RNG().name(
77 [ + - ]: 8 : g_inputdeck.get< tag::param, tag::gendir, tag::rng >()[c] ) );
78 : : nfo.emplace_back(
79 [ + - ][ + - ]: 8 : "coeff b [" + std::to_string( ncomp ) + "]",
[ - + ][ - + ]
[ - + ][ - - ]
[ - - ][ - - ]
80 [ + - ][ + - ]: 12 : parameters( g_inputdeck.get< tag::param, tag::gendir, tag::b >().at(c) ) );
[ - + ]
81 : : nfo.emplace_back(
82 [ + - ][ + - ]: 8 : "coeff S [" + std::to_string( ncomp ) + "]",
[ - + ][ - + ]
[ - + ][ - - ]
[ - - ][ - - ]
83 [ + - ][ + - ]: 12 : parameters( g_inputdeck.get< tag::param, tag::gendir, tag::S >().at(c) ) );
[ - + ]
84 : : nfo.emplace_back(
85 [ + - ][ + - ]: 8 : "coeff kappa [" + std::to_string( ncomp ) + "]",
[ - + ][ - + ]
[ - + ][ - - ]
[ - - ][ - - ]
86 [ + - ][ - + ]: 8 : parameters( g_inputdeck.get< tag::param, tag::gendir, tag::kappa >().at(c) )
87 [ + - ]: 4 : );
88 : : nfo.emplace_back(
89 [ + - ][ + - ]: 8 : "coeff c [" + std::to_string( ncomp*(ncomp-1)/2 ) + "]",
[ + - ][ - + ]
[ - + ][ - + ]
[ - - ][ - - ]
[ - - ]
90 [ + - ][ + - ]: 12 : parameters( g_inputdeck.get< tag::param, tag::gendir, tag::c >().at(c) ) );
[ - + ]
91 [ + - ]: 4 : spikes( nfo, g_inputdeck.get< tag::param, tag::gendir, tag::init,
92 : : tag::spike >().at(c) );
93 [ + - ]: 4 : betapdfs( nfo, g_inputdeck.get< tag::param, tag::gendir, tag::init,
94 : : tag::betapdf >().at(c) );
95 : :
96 : 4 : return nfo;
97 : : }
98 : :
99 : : } // walker::
|
