/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             * 
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *  
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
/**
 * FairMQPollerZMQ.cxx
 *
 * @since 2014-01-23
 * @author A. Rybalchenko
 */

#include <zmq.h>

#include "FairMQPollerZMQ.h"
#include "FairMQLogger.h"

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

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

void FairMQPollerZMQ::Poll(int timeout)
{
    int rc = zmq_poll(items, fNumItems, timeout);
    if (rc < 0)
    {
        LOG(ERROR) << "polling failed, reason: " << zmq_strerror(errno);
    }
}

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

    return false;
}

bool FairMQPollerZMQ::CheckOutput(int index)
{
    if (items[index].revents & ZMQ_POLLOUT)
        return true;

    return false;
}

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