Improve platform management and allow energy simulation

1 files changed, 112 insertions, 0 deletions

diff --git a/src/simulator.cc b/src/simulator.cc
new file mode 100644
index 0000000..b1461d5
--- /dev/null
+++ b/src/simulator.cc
@@ -0,0 +1,112 @@
+#include "simgrid/s4u.hpp"
+#include <simgrid/s4u/Mailbox.hpp>
+#include <string>
+#include <sstream>
+#include "inputs.hpp"
+#include "simgrid/s4u/Host.hpp"
+#include "simgrid/plugins/energy.h"
+#include "xbt/log.h"
+
+#define PLATFORM_FILE "platform.xml"
+#define TURN_OFF() simgrid::s4u::this_actor::get_host()->set_pstate(0);
+#define TURN_ON() simgrid::s4u::this_actor::get_host()->set_pstate(1);
+
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(simulator, "[DAO]");
+
+typedef unsigned int u32;
+typedef struct{
+    double hint;
+    bool isHint;
+} Payload;
+
+/// @brief Observation unit code
+static void obs_node(std::vector<std::string> args);
+
+
+int main(int argc, char **argv) {
+    
+    // Build engine
+    sg_host_energy_plugin_init();
+    simgrid::s4u::Engine engine(&argc, argv);
+    Inputs::GeneratePlatform(PLATFORM_FILE);
+    engine.load_platform(PLATFORM_FILE);
+
+    XBT_INFO("-------------------------------------------------");
+    XBT_INFO("Sarting loosely coupled data dissemination experiments");
+    XBT_INFO("-------------------------------------------------");
+
+    u32 nObsUnit=simgrid::s4u::Engine::get_instance()->get_host_count();
+    for(u32 i=0;i<nObsUnit;i++){
+        std::vector<std::string> args;
+        std::ostringstream ss;
+        ss<< "on" <<i;
+        simgrid::s4u::Actor::create("ON", simgrid::s4u::Host::by_name(ss.str()), obs_node, args);
+    }
+
+    // Setup/Run simulation
+    engine.run();
+
+    XBT_INFO("Simulation took %fs", simgrid::s4u::Engine::get_clock());
+
+    return (0);
+}
+
+
+static void obs_node(std::vector<std::string> args) {
+    std::string selfName = simgrid::s4u::this_actor::get_host()->get_name();
+    simgrid::s4u::this_actor::get_host()->turn_on();
+    Inputs i(selfName);
+    simgrid::s4u::Mailbox *m = simgrid::s4u::Mailbox::by_name("medium");
+    XBT_INFO("Deploying observation node %s",selfName.c_str());
+
+    double wake_interval=i.wake_interval;
+    double wake_duration=i.wake_duration;
+    double startup_delay=i.startup_delay;
+    int max_attempts=i.max_attempts;
+    bool isSender=i.is_sender;
+
+    // Starting node
+    u32 effectiveAttemps=0;
+    double effective_wake_duration=wake_duration;
+    double effective_wake_interval=wake_interval;
+    simgrid::s4u::this_actor::sleep_for(startup_delay);
+    Payload p;
+    for(u32 i=0;i<max_attempts;i++){
+        XBT_INFO("%s is spleeping",selfName.c_str());
+        simgrid::s4u::this_actor::sleep_for(effective_wake_interval);
+        XBT_INFO("%s wakes up",selfName.c_str());
+
+        try
+        {
+            if(isSender){
+                p.isHint=false;
+                p.hint=500;
+                m->put(&p,0,effective_wake_duration);
+                XBT_INFO("%s send data successfully",selfName.c_str());
+                effective_wake_interval=wake_interval;   
+            }
+            else {
+                Payload *p=m->get<Payload>(effective_wake_duration);
+                if(p->isHint){
+                    XBT_INFO("%s received and hint of %f",selfName.c_str(),p->hint);
+                    effective_wake_interval=p->hint;
+                }
+                else{
+                    XBT_INFO("%s received data successfully",selfName.c_str());
+                }
+            }
+        }
+        catch (...)
+        {
+                if(isSender)
+                    XBT_INFO("%s failed to send data",selfName.c_str());
+                else
+                    XBT_INFO("%s failed to receive data",selfName.c_str());
+        }
+        effectiveAttemps++;
+    }
+
+    // Done
+    XBT_INFO("Observation node %s finished (attemps:%d)",selfName.c_str(),effectiveAttemps);
+}
\ No newline at end of file