* under the terms of the license (GNU LGPL) which comes with this package. */
#include "src/surf/network_ib.hpp"
+#include "simgrid/kernel/routing/NetPoint.hpp"
#include "simgrid/sg_config.hpp"
+#include "src/kernel/EngineImpl.hpp"
#include "src/surf/HostImpl.hpp"
#include "src/surf/xml/platf.hpp"
#include "surf/surf.hpp"
using simgrid::kernel::resource::IBNode;
using simgrid::kernel::resource::NetworkIBModel;
- static int id=0;
-
- ((NetworkIBModel*)surf_network_model)->active_nodes.emplace(host.get_name(), IBNode(id));
+ static int id = 0;
+ auto* ibModel = static_cast<NetworkIBModel*>(host.get_netpoint()->get_englobing_zone()->get_network_model().get());
+ ibModel->active_nodes.emplace(host.get_name(), IBNode(id));
id++;
}
simgrid::kernel::resource::Action::State /*previous*/)
{
using simgrid::kernel::resource::IBNode;
- using simgrid::kernel::resource::NetworkIBModel;
if (action.get_state() != simgrid::kernel::resource::Action::State::FINISHED)
return;
- std::pair<IBNode*, IBNode*> pair = ((NetworkIBModel*)surf_network_model)->active_comms[&action];
+ auto* ibModel = static_cast<simgrid::kernel::resource::NetworkIBModel*>(action.get_model());
+ std::pair<IBNode*, IBNode*> pair = ibModel->active_comms[&action];
XBT_DEBUG("IB callback - action %p finished", &action);
- ((NetworkIBModel*)surf_network_model)->updateIBfactors(&action, pair.first, pair.second, 1);
+ ibModel->updateIBfactors(&action, pair.first, pair.second, 1);
- ((NetworkIBModel*)surf_network_model)->active_comms.erase(&action);
+ ibModel->active_comms.erase(&action);
}
static void IB_action_init_callback(simgrid::kernel::resource::NetworkAction& action)
{
- auto* ibModel = static_cast<simgrid::kernel::resource::NetworkIBModel*>(surf_network_model);
+ auto* ibModel = static_cast<simgrid::kernel::resource::NetworkIBModel*>(action.get_model());
auto* act_src = &ibModel->active_nodes.at(action.get_src().get_name());
auto* act_dst = &ibModel->active_nodes.at(action.get_dst().get_name());
/* } */
void surf_network_model_init_IB()
{
- xbt_assert(surf_network_model == nullptr, "Cannot set the network model twice");
+ auto net_model = std::make_shared<simgrid::kernel::resource::NetworkIBModel>("Network_IB");
+ simgrid::kernel::EngineImpl::get_instance()->add_model(net_model);
+ simgrid::s4u::Engine::get_instance()->get_netzone_root()->get_impl()->set_network_model(net_model);
- surf_network_model = new simgrid::kernel::resource::NetworkIBModel();
simgrid::s4u::Link::on_communication_state_change.connect(IB_action_state_changed_callback);
simgrid::s4u::Link::on_communicate.connect(IB_action_init_callback);
simgrid::s4u::Host::on_creation.connect(IB_create_host_callback);
namespace kernel {
namespace resource {
-NetworkIBModel::NetworkIBModel() : NetworkSmpiModel()
+NetworkIBModel::NetworkIBModel(const std::string& name) : NetworkSmpiModel(name)
{
- /* Do not add this into all_existing_models: our ancestor already does so */
-
std::string IB_factors_string = config::get_value<std::string>("smpi/IB-penalty-factors");
std::vector<std::string> radical_elements;
boost::split(radical_elements, IB_factors_string, boost::is_any_of(";"));