Homec4science
Diffusion tamaas (ci-fix-static-checks) Browse

tamaas/src/model/field_container.hh
a2d2aa3c75bfci-fix-static-checks

Last Change
Raw File

field_container.hh
View Options

/*
* SPDX-License-Indentifier: AGPL-3.0-or-later
*
* Copyright (©) 2016-2022 EPFL (École Polytechnique Fédérale de Lausanne),
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
* Copyright (©) 2020-2022 Lucas Frérot
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published
* by the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#ifndef FIELD_CONTAINER_HH
#define FIELD_CONTAINER_HH
/* -------------------------------------------------------------------------- */
#include "grid.hh"
#include <algorithm>
#include <boost/variant.hpp>
#include <memory>
#include <string>
#include <unordered_map>
/* -------------------------------------------------------------------------- */
namespace tamaas {
/* -------------------------------------------------------------------------- */
struct FieldContainer {
// Useful type names
using Key = std::string;
template <class T>
using GridBasePtr = std::shared_ptr<GridBase<T>>;
// TODO C++17: switch to std::variant
using Value =
boost::variant<GridBasePtr<Real>, GridBasePtr<UInt>, GridBasePtr<Int>,
GridBasePtr<Complex>, GridBasePtr<bool>>;
using FieldsMap = std::unordered_map<Key, Value>;
/// Destructor
virtual ~FieldContainer() = default;
/// Access field pointer in const context
const Value& at(const Key& name) const { return fields_.at(name); }
/// Access/insert new pointer
Value& operator[](const Key& name) { return fields_[name]; }
/// Access field of given type (non-const)
template <class T>
auto& field(const Key& name) {
return *boost::get<GridBasePtr<T>>(at(name));
}
/// Access field of given type (const)
template <class T>
const auto& field(const Key& name) const {
return *boost::get<GridBasePtr<T>>(at(name));
}
/// Return field keys
decltype(auto) fields() const {
std::vector<Key> keys;
std::transform(fields_.cbegin(), fields_.cend(), std::back_inserter(keys),
[](auto&& pair) { return pair.first; });
return keys;
}
/// Return pointer map to fields
const auto& fields_map() const { return fields_; }
private:
FieldsMap fields_;
};
} // namespace tamaas
#endif
c4science · Help