default_function = code;
}
-void EngineImpl::add_model_ptask(resource::Model::Type type, resource::Model* model, bool is_default)
+void EngineImpl::add_model(std::shared_ptr<resource::Model> model, const std::vector<resource::Model*>& dependencies)
{
- if (is_default)
- models_by_type_[type].insert(models_by_type_[type].begin(), model);
- else
- models_by_type_[type].push_back(model);
-}
+ auto model_name = model->get_name();
+ xbt_assert(models_prio_.find(model_name) == models_prio_.end(),
+ "Model %s already exists, use model.set_name() to change its name", model_name.c_str());
-void EngineImpl::add_model(resource::Model::Type type, std::shared_ptr<resource::Model> model, bool is_default)
-{
- add_model_ptask(type, model.get(), is_default);
- models_.push_back(std::move(model));
-}
-
-resource::Model* EngineImpl::get_default_model(resource::Model::Type type)
-{
- if (models_by_type_[type].size() > 0)
- return models_by_type_[type][0];
- else
- return nullptr;
+ for (const auto dep : dependencies) {
+ xbt_assert(models_prio_.find(dep->get_name()) != models_prio_.end(),
+ "Model %s doesn't exists. Impossible to use it as dependency.", dep->get_name().c_str());
+ }
+ models_.push_back(model.get());
+ models_prio_[model_name] = std::move(model);
}
} // namespace kernel
