Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
attempt of a function searching all zones of a given type
authorMartin Quinson <martin.quinson@loria.fr>
Fri, 23 Feb 2018 20:45:29 +0000 (21:45 +0100)
committerMartin Quinson <martin.quinson@loria.fr>
Fri, 23 Feb 2018 20:46:48 +0000 (21:46 +0100)
This is related to #253, but it's probably not sufficient. We'd need
an example of use, at least. I'm just pushing to make it public.

include/simgrid/s4u/Engine.hpp
src/s4u/s4u_engine.cpp
src/s4u/s4u_netzone.cpp

index b9c7707..c9d2318 100644 (file)
@@ -84,6 +84,9 @@ public:
   /** @brief Retrieve the netzone of the given name (or nullptr if not found) */
   simgrid::s4u::NetZone* getNetzoneByNameOrNull(const char* name);
 
+  /** @brief Retrieves all netzones of the same type than the subtype of the whereto vector */
+  template <class T> void getNetzoneByType(std::vector<T*> * whereto);
+
   /** @brief Retrieve the netcard of the given name (or nullptr if not found) */
   simgrid::kernel::routing::NetPoint* getNetpointByNameOrNull(std::string name);
   void getNetpointList(std::vector<simgrid::kernel::routing::NetPoint*> * list);
index f991b12..9913599 100644 (file)
@@ -154,6 +154,21 @@ NetZone* Engine::getNetzoneByNameOrNull(const char* name)
   return netzoneByNameRecursive(getNetRoot(), name);
 }
 
+template <class T> void netzoneByTypeRecursive(s4u::NetZone* current, std::vector<T*>* whereto)
+{
+  for (auto const& elem : *(current->getChildren())) {
+    netzoneByTypeRecursive(elem, whereto);
+    if (elem == dynamic_cast<T*>(elem))
+      whereto->push_back(elem);
+  }
+}
+
+/** @brief Retrieve all existing NetZones of the subtype of the whereto vector */
+template <class T> void Engine::getNetzoneByType(std::vector<T*>* whereto)
+{
+  netzoneByTypeRecursive(getNetRoot(), whereto);
+}
+
 /** @brief Retrieve the netpoint of the given name (or nullptr if not found) */
 simgrid::kernel::routing::NetPoint* Engine::getNetpointByNameOrNull(std::string name)
 {
index 7a6f5f2..22889ef 100644 (file)
@@ -59,10 +59,15 @@ void NetZone::setProperty(const char* key, const char* value)
   });
 }
 
+/** @brief Returns the list of direct children (no grand-children)
+ *
+ * This function returns the internal copy of the childrens, not a copy. Don't mess with it!
+ */
 std::vector<NetZone*>* NetZone::getChildren()
 {
   return children_;
 }
+
 const char* NetZone::getCname() const
 {
   return name_.c_str();