Branch data Line data Source code
1 : : // *****************************************************************************
2 : : /*!
3 : : \file src/Control/Walker/Options/InitPolicy.hpp
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 Differential equation initialization policy options for walker
9 : : \details Differential equation initialization policy options for walker
10 : : */
11 : : // *****************************************************************************
12 : : #ifndef InitPolicyOptions_h
13 : : #define InitPolicyOptions_h
14 : :
15 : : #include <brigand/sequences/list.hpp>
16 : : #include <brigand/algorithms/for_each.hpp>
17 : :
18 : : #include "Toggle.hpp"
19 : : #include "Keywords.hpp"
20 : : #include "PUPUtil.hpp"
21 : :
22 : : namespace walker {
23 : : namespace ctr {
24 : :
25 : : //! Differential equation initializion policy types
26 : : enum class InitPolicyType : uint8_t { RAW=0,
27 : : ZERO,
28 : : JOINTDELTA,
29 : : JOINTGAUSSIAN,
30 : : JOINTCORRGAUSSIAN,
31 : : JOINTBETA,
32 : : JOINTGAMMA,
33 : : JOINTDIRICHLET };
34 : :
35 : : //! Pack/Unpack InitPolicyType: forward overload to generic enum class packer
36 : 422 : inline void operator|( PUP::er& p, InitPolicyType& e ) { PUP::pup( p, e ); }
37 : :
38 : : //! InitPolicy options: outsource searches to base templated on enum type
39 : : class InitPolicy : public tk::Toggle< InitPolicyType > {
40 : :
41 : : public:
42 : : //! Valid expected choices to make them also available at compile-time
43 : : using keywords = brigand::list< kw::raw
44 : : , kw::zero
45 : : , kw::jointdelta
46 : : , kw::jointgaussian
47 : : , kw::jointcorrgaussian
48 : : , kw::jointbeta
49 : : , kw::jointgamma
50 : : , kw::jointdirichlet
51 : : >;
52 : :
53 : : //! \brief Options constructor
54 : : //! \details Simply initialize in-line and pass associations to base, which
55 : : //! will handle client interactions
56 : 311 : explicit InitPolicy() :
57 : : tk::Toggle< InitPolicyType >(
58 : : //! Group, i.e., options, name
59 : : "Initialization Policy",
60 : : //! Enums -> names
61 [ + - ]: 311 : { { InitPolicyType::RAW, kw::raw::name() },
62 [ + - ]: 622 : { InitPolicyType::ZERO, kw::zero::name() },
63 [ + - ]: 622 : { InitPolicyType::JOINTDELTA, kw::jointdelta::name() },
64 [ + - ]: 622 : { InitPolicyType::JOINTGAUSSIAN, kw::jointgaussian::name() },
65 [ + - ]: 622 : { InitPolicyType::JOINTCORRGAUSSIAN, kw::jointcorrgaussian::name() },
66 [ + - ]: 622 : { InitPolicyType::JOINTBETA, kw::jointbeta::name() },
67 [ + - ]: 622 : { InitPolicyType::JOINTGAMMA, kw::jointgamma::name() },
68 [ + - ]: 622 : { InitPolicyType::JOINTDIRICHLET, kw::jointdirichlet::name() } },
69 : : //! keywords -> Enums
70 : 0 : { { kw::raw::string(), InitPolicyType::RAW },
71 [ + - ]: 622 : { kw::zero::string(), InitPolicyType::ZERO },
72 [ + - ]: 622 : { kw::jointdelta::string(), InitPolicyType::JOINTDELTA },
73 [ + - ]: 622 : { kw::jointgaussian::string(), InitPolicyType::JOINTGAUSSIAN },
74 [ + - ]: 622 : { kw::jointcorrgaussian::string(),
75 : 0 : InitPolicyType::JOINTCORRGAUSSIAN },
76 [ + - ]: 622 : { kw::jointbeta::string(), InitPolicyType::JOINTBETA },
77 [ + - ]: 622 : { kw::jointgamma::string(), InitPolicyType::JOINTGAMMA },
78 [ + - ][ + - ]: 15239 : { kw::jointdirichlet::string(), InitPolicyType::JOINTDIRICHLET } } )
[ + - ][ + - ]
[ + - ][ + + ]
[ + + ][ + - ]
[ + + ][ - - ]
[ - - ][ - - ]
79 : : {
80 : 311 : brigand::for_each< keywords >( assertPolicyCodes() );
81 : 311 : }
82 : :
83 : : //! \brief Return policy code based on Enum
84 : : //! \param[in] p Enum value of the option requested
85 : : //! \return Policy code of the option
86 : 19344 : const std::string& code( InitPolicyType p ) const {
87 : : using tk::operator<<;
88 [ + - ]: 19344 : auto it = policy.find( p );
89 [ - + ][ - - ]: 19344 : Assert( it != end(policy),
[ - - ][ - - ]
[ - - ][ - - ]
90 : : std::string("Cannot find policy code for physics \"") << p <<
91 : : "\"" );
92 : 19344 : return it->second;
93 : : }
94 : :
95 : : private:
96 : : //! Function object for ensuring the existence of policy codes
97 : : struct assertPolicyCodes {
98 : : //! \brief Function call operator templated on the type to assert the
99 : : //! existence of a policy code
100 : 2488 : template< typename U > void operator()( brigand::type_<U> ) {
101 : : static_assert( tk::HasTypedef_code_v< typename U::info >,
102 : : "Policy code undefined for keyword" );
103 : 2488 : }
104 : : };
105 : :
106 : : //! Enums -> policy code
107 : : std::map< InitPolicyType, std::string > policy {
108 [ + - ][ + - ]: 622 : { InitPolicyType::RAW, *kw::raw::code() }
109 [ + - ]: 622 : , { InitPolicyType::ZERO, *kw::zero::code() }
110 [ + - ]: 622 : , { InitPolicyType::JOINTDELTA, *kw::jointdelta::code() }
111 [ + - ]: 622 : , { InitPolicyType::JOINTGAUSSIAN, *kw::jointgaussian::code() }
112 [ + - ]: 622 : , { InitPolicyType::JOINTCORRGAUSSIAN, *kw::jointcorrgaussian::code() }
113 [ + - ]: 622 : , { InitPolicyType::JOINTBETA, *kw::jointbeta::code() }
114 [ + - ]: 622 : , { InitPolicyType::JOINTGAMMA, *kw::jointgamma::code() }
115 [ + - ]: 622 : , { InitPolicyType::JOINTDIRICHLET, *kw::jointdirichlet::code() }
116 : 311 : };
117 : : };
118 : :
119 : : } // ctr::
120 : : } // walker::
121 : :
122 : : #endif // InitPolicyOptions_h
|
