/**
 * FairMQPollerNN.cxx
 *
 * @since 2014-01-23
 * @author A. Rybalchenko
 */

#include <nanomsg/nn.h>

#include "FairMQPollerNN.h"

FairMQPollerNN::FairMQPollerNN(const vector<FairMQSocket*>& inputs)
{
    fNumItems = inputs.size();
    items = new nn_pollfd[fNumItems];

    for (int i = 0; i < fNumItems; i++)
    {
        items[i].fd = inputs.at(i)->GetSocket(1);
        items[i].events = NN_POLLIN;
    }
}

void FairMQPollerNN::Poll(int timeout)
{
    nn_poll(items, fNumItems, timeout);
}

bool FairMQPollerNN::CheckInput(int index)
{
    if (items[index].revents & NN_POLLIN)
        return true;

    return false;
}

FairMQPollerNN::~FairMQPollerNN()
{
    if (items != NULL)
        delete[] items;
}