RateDrop.cpp

Go to the documentation of this file.

 
#include "rogue/Directives.h"
 
#include "rogue/interfaces/stream/RateDrop.h"
 
#include <stdint.h>
#include <sys/time.h>
 
#include <cstdio>
#include <memory>
 
#include "rogue/Logging.h"
#include "rogue/interfaces/stream/Frame.h"
#include "rogue/interfaces/stream/Master.h"
#include "rogue/interfaces/stream/Slave.h"
 
namespace ris = rogue::interfaces::stream;
 
#ifndef NO_PYTHON
    #include <boost/python.hpp>
namespace bp = boost::python;
#endif
 
ris::RateDropPtr ris::RateDrop::create(bool period, double value) {
    ris::RateDropPtr p = std::make_shared<ris::RateDrop>(period, value);
    return (p);
}
 
void ris::RateDrop::setup_python() {
#ifndef NO_PYTHON
    bp::class_<ris::RateDrop, ris::RateDropPtr, bp::bases<ris::Master, ris::Slave>, boost::noncopyable>(
        "RateDrop",
        bp::init<bool, double>());
#endif
}
 
ris::RateDrop::RateDrop(bool period, double value) : ris::Master(), ris::Slave() {
    struct timeval currTime;
    uint32_t per;
 
    if ((!period) || value == 0) {
        periodFlag_ = false;
        dropCount_  = static_cast<uint32_t>(value);
        dropTarget_ = static_cast<uint32_t>(value);
    } else {
        periodFlag_ = true;
 
        per = static_cast<uint32_t>(value * 1e6);
 
        div_t divResult     = div(per, 1000000);
        timePeriod_.tv_sec  = divResult.quot;
        timePeriod_.tv_usec = divResult.rem;
 
        gettimeofday(&currTime, NULL);
        timeradd(&currTime, &timePeriod_, &nextPeriod_);
    }
}
 
ris::RateDrop::~RateDrop() {}
 
void ris::RateDrop::acceptFrame(ris::FramePtr frame) {
    struct timeval currTime;
 
    // Dropping based upon frame count, if countPeriod_ is zero we never drop
    if (!periodFlag_) {
        if (dropCount_++ == dropTarget_) {
            sendFrame(frame);
            dropCount_ = 0;
        }
 
        // Dropping based upon time
    } else {
        gettimeofday(&currTime, NULL);
 
        if (timercmp(&currTime, &(nextPeriod_), >)) {
            sendFrame(frame);
            timeradd(&currTime, &timePeriod_, &nextPeriod_);
        }
    }
}