/*
 Itay Marom
 Cisco Systems, Inc.
*/

/*
Copyright (c) 2015-2015 Cisco Systems, Inc.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

#include <trex_dp_port_events.h>
#include <trex_stateless_messaging.h>
#include <sstream>
#include <os_time.h>
#include <trex_stateless.h>

/**
 * port events
 */
TrexDpPortEvents::TrexDpPortEvents(TrexStatelessPort *port) {
    m_port = port;
    m_event_id_counter = EVENT_ID_INVALID;
}

TrexDpPortEvent *
TrexDpPortEvents::lookup(int event_id) {
    auto search = m_events.find(event_id);

    if (search != m_events.end()) {
        return search->second;
    } else {
        return NULL;
    }
}

/**
 * mark the next allowed event 
 * all other events will be disabled 
 * 
 */
int
TrexDpPortEvents::create_event(TrexDpPortEvent *event, int timeout_ms) {
    /* allocate ID for event */
    int event_id = ++m_event_id_counter;

    /* init and add */
    event->init(m_port, event_id, timeout_ms);
    m_events[event_id] = event;

    return event_id;
}

void 
TrexDpPortEvents::destroy_event(int event_id) {
    TrexDpPortEvent *event = lookup(event_id);
    if (!event) {
        /* cannot find event */
        throw TrexException("internal error - cannot find event");
    }

    m_events.erase(event_id);
    delete event;
}

class DPBarrier : public TrexDpPortEvent {
protected:
    virtual void on_event() {
        /* do nothing */
    }
};

void
TrexDpPortEvents::barrier() {

    /* simulator will be stuck here forever */
    #ifdef TREX_SIM
    return;
    #endif
     
    int barrier_id = create_event(new DPBarrier());

    TrexStatelessCpToDpMsgBase *barrier_msg = new TrexStatelessDpBarrier(m_port->m_port_id, barrier_id);
    m_port->send_message_to_all_dp(barrier_msg);

    get_stateless_obj()->get_platform_api()->flush_dp_messages();
    while (lookup(barrier_id) != NULL) {
        delay(1);
        get_stateless_obj()->get_platform_api()->flush_dp_messages();
    }
}

/**
 * handle an event
 * 
 */
void 
TrexDpPortEvents::on_core_reporting_in(int event_id, int thread_id) {
    TrexDpPortEvent *event = lookup(event_id);
    /* event might have been deleted */
    if (!event) {
        return;
    }

    bool done = event->on_core_reporting_in(thread_id);

    if (done) {
        destroy_event(event_id);
    }
}


/***************************
 * event
 * 
 **************************/
TrexDpPortEvent::TrexDpPortEvent() {
    m_port = NULL;
    m_event_id = -1;
}

void 
TrexDpPortEvent::init(TrexStatelessPort *port, int event_id, int timeout_ms) {
    m_port = port;
    m_event_id = event_id;

    /* do we have a timeout ? */
    if (timeout_ms > 0) {
        m_expire_limit_ms = os_get_time_msec() + timeout_ms;
    } else {
        m_expire_limit_ms = -1;
    }

    /* prepare the signal array */
    m_pending_cnt = 0;
    for (int core_id : m_port->get_core_id_list()) {
        m_signal[core_id] = false;
        m_pending_cnt++;
    }
}

bool
TrexDpPortEvent::on_core_reporting_in(int thread_id) {
    /* mark sure no double signal */
    if (m_signal.at(thread_id)) {
        std::stringstream err;
        err << "double signal detected on event id: " << m_event_id;
        throw TrexException(err.str());

    }

    /* mark */
    m_signal.at(thread_id) = true;
    m_pending_cnt--;

    /* event occured */
    if (m_pending_cnt == 0) {
        on_event();
        return true;
    } else {
        return false;
    }
}