diff options
Diffstat (limited to 'source/blender/simulation/intern/simulation_collect_influences.hh')
-rw-r--r-- | source/blender/simulation/intern/simulation_collect_influences.hh | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/source/blender/simulation/intern/simulation_collect_influences.hh b/source/blender/simulation/intern/simulation_collect_influences.hh new file mode 100644 index 00000000000..8673a308b04 --- /dev/null +++ b/source/blender/simulation/intern/simulation_collect_influences.hh @@ -0,0 +1,65 @@ +/* + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + */ + +#pragma once + +#include "NOD_derived_node_tree.hh" + +#include "BLI_resource_collector.hh" + +#include "simulation_solver_influences.hh" + +namespace blender::sim { + +class RequiredStates { + private: + Map<std::string, const char *> state_type_by_state_name_; + + public: + void add(std::string state_name, const char *state_type) + { + BLI_assert(state_type != nullptr); + const char *type_name = state_type_by_state_name_.lookup_default(state_name, nullptr); + if (type_name != nullptr) { + if (!STREQ(state_type, type_name)) { + std::cout << "Warning: Tried to have two different states with the same name.\n"; + std::cout << " Name: " << state_name << "\n"; + std::cout << " Type 1: " << state_type << "\n"; + std::cout << " Type 2: " << type_name << "\n"; + } + return; + } + + state_type_by_state_name_.add(std::move(state_name), state_type); + } + + const Map<std::string, const char *> &states() const + { + return state_type_by_state_name_; + } + + bool is_required(StringRef state_name, StringRef state_type) const + { + return state_type_by_state_name_.lookup_default_as(state_name, "") == state_type; + } +}; + +void collect_simulation_influences(Simulation &simulation, + ResourceCollector &resources, + SimulationInfluences &r_influences, + RequiredStates &r_required_states); + +} // namespace blender::sim |