optimize perf - part 2

[simgrid.git] / src / msg / msg_environment.cpp

diff --git a/src/msg/msg_environment.cpp b/src/msg/msg_environment.cpp
index 27a3673..65a7a8c 100644 (file)
--- a/src/msg/msg_environment.cpp
+++ b/src/msg/msg_environment.cpp
@@ -1,15 +1,11 @@
-/* Copyright (c) 2004-2015. The SimGrid Team.
- * All rights reserved.                                                     */
+/* Copyright (c) 2004-2016. The SimGrid Team. All rights reserved.          */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
-#include "msg_private.h"
-#include "xbt/sysdep.h"
-#include "xbt/log.h"
-
-#include "simgrid/s4u/As.hpp"
+#include "simgrid/s4u/NetZone.hpp"
 #include "simgrid/s4u/engine.hpp"
+#include "src/msg/msg_private.h"
 
 #if HAVE_LUA
 #include <lua.h>
@@ -17,6 +13,8 @@
 #include <lualib.h>
 #endif
 
+SG_BEGIN_DECL()
+
 /********************************* MSG **************************************/
 
 /** \ingroup msg_simulation
@@ -38,7 +36,7 @@ void MSG_create_environment(const char *file)
   SIMIX_create_environment(file);
 }
 
-void MSG_post_create_environment(void) {
+void MSG_post_create_environment() {
   xbt_lib_cursor_t cursor;
   void **data;
   char *name;
@@ -50,30 +48,45 @@ void MSG_post_create_environment(void) {
   }
 }
 
-msg_as_t MSG_environment_get_routing_root() {
-  return simgrid::s4u::Engine::instance()->rootAs();
+msg_netzone_t MSG_environment_get_routing_root()
+{
+  return simgrid::s4u::Engine::instance()->netRoot();
 }
 
-const char *MSG_environment_as_get_name(msg_as_t as) {
-  return as->name();
+const char* MSG_environment_as_get_name(msg_netzone_t netzone)
+{
+  return netzone->name();
+}
+
+msg_netzone_t MSG_environment_as_get_by_name(const char* name)
+{
+  return simgrid::s4u::Engine::instance()->netzoneByNameOrNull(name);
 }
 
-msg_as_t MSG_environment_as_get_by_name(const char * name) {
-  return simgrid::s4u::Engine::instance()->asByNameOrNull(name);
+xbt_dict_t MSG_environment_as_get_routing_sons(msg_netzone_t netzone)
+{
+  return netzone->children();
 }
 
-xbt_dict_t MSG_environment_as_get_routing_sons(msg_as_t as) {
-  return as->children();
+const char* MSG_environment_as_get_property_value(msg_netzone_t netzone, const char* name)
+{
+  return netzone->property(name);
 }
 
-const char *MSG_environment_as_get_property_value(msg_as_t as, const char *name)
+void MSG_environment_as_set_property_value(msg_netzone_t netzone, const char* name, char* value)
 {
-  xbt_dict_t dict = (xbt_dict_t) xbt_lib_get_or_null(as_router_lib, MSG_environment_as_get_name(as), ROUTING_PROP_ASR_LEVEL);
-  if (dict==nullptr)
-    return nullptr;
-  return (char*) xbt_dict_get_or_null(dict, name);
+  netzone->setProperty(name, value);
 }
 
-xbt_dynar_t MSG_environment_as_get_hosts(msg_as_t as) {
-  return as->hosts();
+xbt_dynar_t MSG_environment_as_get_hosts(msg_netzone_t netzone)
+{
+  xbt_dynar_t res = xbt_dynar_new(sizeof(sg_host_t), nullptr);
+
+  for (auto host : *netzone->hosts()) {
+    xbt_dynar_push(res, &host);
+  }
+
+  return res;
 }
+
+SG_END_DECL()