/* 
 * File:   runSimpleMQStateMachine.h
 * Author: winckler
 *
 * Created on July 2, 2015, 2:07 PM
 */

#ifndef RUNSIMPLEMQSTATEMACHINE_H
#define RUNSIMPLEMQSTATEMACHINE_H

/// std
#include <iostream>
#include <type_traits>
#include <string>

/// boost
#include "boost/program_options.hpp"

/// FairRoot - FairMQ
#include "FairMQLogger.h"
#include "FairMQParser.h"
#include "FairMQProgOptions.h"

// template function that take any device, 
// and run a simple MQ state machine configured from a JSON file
template<typename TMQDevice>
inline int runStateMachine(TMQDevice& device, FairMQProgOptions& config)
{
    device.CatchSignals();
    std::string id = config.GetValue<std::string>("id");
    int ioThreads = config.GetValue<int>("io-threads");

    device.fChannels = config.GetFairMQMap();

    device.SetProperty(TMQDevice::Id, id);
    device.SetProperty(TMQDevice::NumIoThreads, ioThreads);

    LOG(INFO) << "PID: " << getpid();

    device.SetTransport(config.GetValue<std::string>("transport"));

    device.ChangeState(TMQDevice::INIT_DEVICE);
    device.WaitForEndOfState(TMQDevice::INIT_DEVICE);

    device.ChangeState(TMQDevice::INIT_TASK);
    device.WaitForEndOfState(TMQDevice::INIT_TASK);

    device.ChangeState(TMQDevice::RUN);
    device.InteractiveStateLoop();

    return 0;
}

template<typename TMQDevice>
inline int runNonInteractiveStateMachine(TMQDevice& device, FairMQProgOptions& config)
{
    device.CatchSignals();
    std::string id = config.GetValue<std::string>("id");
    int ioThreads = config.GetValue<int>("io-threads");

    device.fChannels = config.GetFairMQMap();

    device.SetProperty(TMQDevice::Id, id);
    device.SetProperty(TMQDevice::NumIoThreads, ioThreads);

    LOG(INFO) << "PID: " << getpid();

    device.SetTransport(config.GetValue<std::string>("transport"));

    device.ChangeState(TMQDevice::INIT_DEVICE);
    device.WaitForEndOfState(TMQDevice::INIT_DEVICE);

    device.ChangeState(TMQDevice::INIT_TASK);
    device.WaitForEndOfState(TMQDevice::INIT_TASK);

    device.ChangeState(TMQDevice::RUN);
    device.WaitForEndOfState(TMQDevice::RUN);

    device.ChangeState(TMQDevice::RESET_TASK);
    device.WaitForEndOfState(TMQDevice::RESET_TASK);

    device.ChangeState(TMQDevice::RESET_DEVICE);
    device.WaitForEndOfState(TMQDevice::RESET_DEVICE);

    device.ChangeState(TMQDevice::END);

    return 0;
}

#endif /* RUNSIMPLEMQSTATEMACHINE_H */