Branch data Line data Source code
1 : : // *****************************************************************************
2 : : /*!
3 : : \file src/Control/Options/PartitioningAlgorithm.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 Mesh partitioning algorithm options
9 : : \details Mesh partitioning algorithm options
10 : : */
11 : : // *****************************************************************************
12 : : #ifndef InciterPartitioningAlgorithmOptions_h
13 : : #define InciterPartitioningAlgorithmOptions_h
14 : :
15 : : #include <map>
16 : :
17 : : #include <brigand/sequences/list.hpp>
18 : :
19 : : #include "Toggle.hpp"
20 : : #include "Keywords.hpp"
21 : : #include "PUPUtil.hpp"
22 : :
23 : : namespace tk {
24 : : namespace ctr {
25 : :
26 : : //! Mesh partitioning algorithm types
27 : : enum class PartitioningAlgorithmType : uint8_t { RCB,
28 : : RIB,
29 : : HSFC,
30 : : MJ,
31 : : PHG };
32 : :
33 : : //! \brief Pack/Unpack PartitioningAlgorithmType: forward overload to generic
34 : : //! enum class packer
35 : 1798 : inline void operator|( PUP::er& p, PartitioningAlgorithmType& e )
36 : 1798 : { PUP::pup( p, e ); }
37 : :
38 : : //! \brief PartitioningAlgorithm options: outsource searches to base templated
39 : : //! on enum type
40 : : class PartitioningAlgorithm : public tk::Toggle< PartitioningAlgorithmType > {
41 : :
42 : : public:
43 : : using ParamType = std::string;
44 : :
45 : : //! Valid expected choices to make them also available at compile-time
46 : : using keywords = brigand::list< kw::rcb
47 : : , kw::rib
48 : : , kw::hsfc
49 : : , kw::mj
50 : : , kw::phg
51 : : >;
52 : :
53 : : //! \brief Options constructor
54 : : //! \details Simply initialize in-line and pass associations to base, which
55 : : //! will handle client interactions
56 : 1012 : explicit PartitioningAlgorithm() :
57 : : tk::Toggle< PartitioningAlgorithmType >(
58 : : //! Group, i.e., options, name
59 : : "Mesh partitioning algorithm",
60 : : //! Enums -> names
61 [ + - ]: 1012 : { { PartitioningAlgorithmType::RCB, kw::rcb::name() },
62 [ + - ]: 2024 : { PartitioningAlgorithmType::RIB, kw::rib::name() },
63 [ + - ]: 2024 : { PartitioningAlgorithmType::HSFC, kw::hsfc::name() },
64 [ + - ]: 2024 : { PartitioningAlgorithmType::MJ, kw::mj::name() },
65 [ + - ]: 2024 : { PartitioningAlgorithmType::PHG, kw::phg::name() } },
66 : : //! keywords -> Enums
67 : 0 : { { kw::rcb::string(), PartitioningAlgorithmType::RCB },
68 [ + - ]: 2024 : { kw::rib::string(), PartitioningAlgorithmType::RIB },
69 [ + - ]: 2024 : { kw::hsfc::string(), PartitioningAlgorithmType::HSFC },
70 [ + - ]: 2024 : { kw::mj::string(), PartitioningAlgorithmType::MJ },
71 [ + - ][ + - ]: 26312 : { kw::phg::string(), PartitioningAlgorithmType::PHG } } ) {}
[ + - ][ + - ]
[ + - ][ + + ]
[ + + ][ + - ]
[ + + ][ - - ]
[ - - ][ - - ]
72 : :
73 : : //! \brief Return parameter based on Enum
74 : : //! \details Here 'parameter' is the library-specific identifier of the
75 : : //! option, i.e., as the library identifies the given option
76 : : //! \param[in] m Enum value of the option requested
77 : : //! \return Library-specific parameter of the option
78 : 666 : const ParamType& param( PartitioningAlgorithmType m ) const {
79 : : using tk::operator<<;
80 [ + - ]: 666 : auto it = method.find( m );
81 [ - + ][ - - ]: 666 : Assert( it != end(method),
[ - - ][ - - ]
[ - - ][ - - ]
82 : : std::string("Cannot find parameter for PartitioningAlgorithm \"")
83 : : << m << "\"" );
84 : 666 : return it->second;
85 : : }
86 : :
87 : : // Return true if partitioning algorithm is geometric
88 : : //! \param[in] m Enum value of the option requested
89 : : //! \return True if partitioning algorithm is geometric, false if it is not
90 : : bool geometric( PartitioningAlgorithmType m ) const {
91 : : if ( m == PartitioningAlgorithmType::RCB ||
92 : : m == PartitioningAlgorithmType::RIB ||
93 : : m == PartitioningAlgorithmType::HSFC ||
94 : : m == PartitioningAlgorithmType::MJ )
95 : : return true;
96 : : else
97 : : return false;
98 : : }
99 : :
100 : : private:
101 : : //! Enums -> Zoltan partitioning algorithm parameters
102 : : std::map< PartitioningAlgorithmType, ParamType > method {
103 [ + - ]: 1012 : { PartitioningAlgorithmType::RCB, "rcb" },
104 [ + - ]: 1012 : { PartitioningAlgorithmType::RIB, "rib" },
105 [ + - ]: 1012 : { PartitioningAlgorithmType::HSFC, "hsfc" },
106 [ + - ]: 1012 : { PartitioningAlgorithmType::MJ, "multijagged" },
107 [ + - ]: 1012 : { PartitioningAlgorithmType::PHG, "phg" }
108 : : };
109 : : };
110 : :
111 : : } // ctr::
112 : : } // tk::
113 : :
114 : : #endif // InciterPartitioningAlgorithmOptions_h
|