FairMQ: Move static and interactive control modes to plugin

NOT YET FINISHED
2025-10-15 09:31:45 +00:00 · 2017-09-14 02:10:00 +02:00
parent 2af3ae99eb
commit 10f67e4c72
9 changed files with 223 additions and 113 deletions
--- a/fairmq/plugins/Control.cxx
+++ b/fairmq/plugins/Control.cxx
@@ -0,0 +1,137 @@
+/********************************************************************************
+ *    Copyright (C) 2017 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
+ *                                                                              *
+ *              This software is distributed under the terms of the             *
+ *              GNU Lesser General Public Licence (LGPL) version 3,             *
+ *                  copied verbatim in the file "LICENSE"                       *
+ ********************************************************************************/
+
+#include "Control.h"
+#include <chrono>
+#include <thread>
+
+
+using namespace std;
+
+namespace fair
+{
+namespace mq
+{
+namespace plugins
+{
+
+Control::Control(
+    const string name,
+    const Plugin::Version version,
+    const string maintainer,
+    const string homepage,
+    PluginServices* pluginServices)
+: Plugin(name, version, maintainer, homepage, pluginServices)
+{
+    try
+    {
+        TakeDeviceControl();
+
+        auto control = GetProperty<string>("control");
+
+        if(control == "static")
+        {
+            LOG(DEBUG) << "Running builtin controller: static";
+            thread t(&Control::StaticMode, this);
+            t.detach();
+        }
+        else if(control == "interactive")
+        {
+            LOG(DEBUG) << "Running builtin controller: interactive";
+            thread t(&Control::InteractiveMode, this);
+            t.detach();
+        }
+        else
+        {
+            LOG(ERROR) << "Unrecognized control mode '" << control << "' requested via command line. "
+                << "Ignoring and falling back to interactive control mode.";
+            thread t(&Control::InteractiveMode, this);
+            t.detach();
+        }
+    }
+    catch(PluginServices::DeviceControlError& e)
+    {
+        LOG(DEBUG) << e.what();
+    }   
+}
+
+auto ControlPluginProgramOptions() -> Plugin::ProgOptions
+{
+    auto plugin_options = boost::program_options::options_description{"Control (builtin) Plugin"};
+    plugin_options.add_options()
+        ("ctrlmode", boost::program_options::value<string>(), "Control mode, 'static' or 'interactive'");
+        // should rename to --control and remove control from device options ?
+    return plugin_options;
+}
+    
+auto Control::InteractiveMode() -> void
+{
+    LOG(ERROR) << "NOT YET IMPLEMENTED";
+    ReleaseDeviceControl();
+}
+
+auto Control::WaitForNextState() -> DeviceState
+{
+    unique_lock<mutex> lock{fEventsMutex};
+    while(fEvents.empty())
+    {
+        fNewEvent.wait(lock);
+    }
+    // lock.lock();
+    auto result = fEvents.front();
+    fEvents.pop();
+    return result;
+}
+
+auto Control::StaticMode() -> void
+{
+    clock_t cStart = clock();
+    auto tStart = chrono::high_resolution_clock::now();
+
+    SubscribeToDeviceStateChange(
+        [&](DeviceState newState){
+            {
+                lock_guard<mutex> lock{fEventsMutex};
+                fEvents.push(newState);
+            }
+            fNewEvent.notify_one();
+        }
+    );
+    
+    ChangeDeviceState(DeviceStateTransition::InitDevice);
+    while(WaitForNextState() != DeviceState::DeviceReady) {};
+    
+    clock_t cEnd = std::clock();
+    auto tEnd = chrono::high_resolution_clock::now();
+
+    LOG(DEBUG) << "Init time (CPU) : " << fixed << setprecision(2) << 1000.0 * (cEnd - cStart) / CLOCKS_PER_SEC << " ms";
+    LOG(DEBUG) << "Init time (Wall): " << chrono::duration<double, milli>(tEnd - tStart).count() << " ms";
+    
+    ChangeDeviceState(DeviceStateTransition::InitTask);
+    while(WaitForNextState() != DeviceState::Ready) {};
+    ChangeDeviceState(DeviceStateTransition::Run);
+    // WaitForNextState();
+    // ChangeDeviceState(DeviceStateTransition::ResetTask);
+    // WaitForNextState();
+    // WaitForNextState();
+    // ChangeDeviceState(DeviceStateTransition::ResetDevice);
+    // WaitForNextState();
+    // WaitForNextState();
+    // ChangeDeviceState(DeviceStateTransition::End);
+    while(WaitForNextState() != DeviceState::Exiting) {};
+    LOG(WARN) << "1";
+
+    UnsubscribeFromDeviceStateChange();
+    LOG(WARN) << "2";
+    ReleaseDeviceControl();
+    LOG(WARN) << "3";
+}
+
+} /* namespace plugins */
+} /* namespace mq */
+} /* namespace fair */