StimBuff: Add skeleton common impl with rate limiting
This commit is contained in:
@@ -6,6 +6,13 @@
|
||||
#include <cstdint>
|
||||
#include <atomic>
|
||||
#include <mutex>
|
||||
#include <functional>
|
||||
#include <iostream>
|
||||
#include <config.h>
|
||||
#include <boost/asio/io_service.hpp>
|
||||
#include <boost/asio/deadline_timer.hpp>
|
||||
#include <spinLock.h>
|
||||
#include <asynchronousBridge.h>
|
||||
#include "stimFrame.h"
|
||||
#include "deviceAttachmentSpec.h"
|
||||
|
||||
@@ -40,8 +47,11 @@ public:
|
||||
|
||||
public:
|
||||
explicit StimulusBuffer(
|
||||
const device::DeviceAttachmentSpec& deviceAttachmentSpec)
|
||||
: deviceAttachmentSpec(deviceAttachmentSpec)
|
||||
const device::DeviceAttachmentSpec& deviceAttachmentSpec,
|
||||
boost::asio::io_service& ioService)
|
||||
: deviceAttachmentSpec(deviceAttachmentSpec),
|
||||
ioService(ioService),
|
||||
shouldContinue(false), timer(ioService)
|
||||
{}
|
||||
|
||||
~StimulusBuffer() = default;
|
||||
@@ -52,9 +62,107 @@ public:
|
||||
StimulusBuffer(StimulusBuffer&&) = default;
|
||||
StimulusBuffer& operator=(StimulusBuffer&&) = default;
|
||||
|
||||
// Control methods
|
||||
void start()
|
||||
{
|
||||
shouldContinue.store(true);
|
||||
scheduleNextTimeout();
|
||||
}
|
||||
|
||||
void stop()
|
||||
{
|
||||
shouldContinue.store(false);
|
||||
|
||||
// Set up a timeout bridge using the io_service
|
||||
boost::asio::deadline_timer delayTimer(ioService);
|
||||
AsynchronousBridge bridge(ioService);
|
||||
|
||||
// Set up the delay for CONFIG_STIMBUFF_FRAME_PERIOD_MS to let in-flight
|
||||
// operation finish
|
||||
delayTimer.expires_from_now(
|
||||
boost::posix_time::milliseconds(CONFIG_STIMBUFF_FRAME_PERIOD_MS));
|
||||
|
||||
delayTimer.async_wait(
|
||||
[&bridge](const boost::system::error_code& error)
|
||||
{
|
||||
(void)error;
|
||||
|
||||
// Always signal complete, whether timeout expired or was cancelled
|
||||
bridge.setAsyncOperationComplete();
|
||||
});
|
||||
|
||||
bridge.waitForAsyncOperationCompleteOrIoServiceStopped();
|
||||
|
||||
// After delay, cancel timer and perform cleanup
|
||||
timer.cancel();
|
||||
}
|
||||
|
||||
public:
|
||||
device::DeviceAttachmentSpec deviceAttachmentSpec;
|
||||
std::vector<StimFrame> frames_;
|
||||
|
||||
protected:
|
||||
SpinLock frameAssemblyRateLimiter;
|
||||
|
||||
private:
|
||||
boost::asio::io_service& ioService;
|
||||
std::atomic<bool> shouldContinue;
|
||||
boost::asio::deadline_timer timer;
|
||||
|
||||
void scheduleNextTimeout(int delayMs = CONFIG_STIMBUFF_FRAME_PERIOD_MS)
|
||||
{
|
||||
if (!shouldContinue.load())
|
||||
{ return; }
|
||||
|
||||
// Schedule the next timeout using the provided delay
|
||||
timer.expires_from_now(
|
||||
boost::posix_time::milliseconds(delayMs));
|
||||
|
||||
timer.async_wait(
|
||||
std::bind(
|
||||
&StimulusBuffer::onTimeout, this, std::placeholders::_1));
|
||||
}
|
||||
|
||||
void onTimeout(const boost::system::error_code& error)
|
||||
{
|
||||
// Timer was cancelled, which is expected when stopping
|
||||
if (error == boost::asio::error::operation_aborted) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (error)
|
||||
{
|
||||
std::cerr << "StimulusBuffer: Timer error: " << error.message()
|
||||
<< std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
if (!shouldContinue.load())
|
||||
{ return; }
|
||||
|
||||
/** EXPLANATION:
|
||||
* We need to ensure that there's only ever one stimframe being produced
|
||||
* during any CONFIG_STIMBUFF_FRAME_PERIOD_MS period. To guarantee this, we
|
||||
* use a spinlock.
|
||||
*
|
||||
* When a new frame is to be produced, the async producer will first acquire
|
||||
* the frameAssemblyLimiter spinlock. This way, when the next timeout is
|
||||
* fired it can check whether its predecessor stimframe has finished being
|
||||
* produced. If the preceding stimframe is still being produced, then we'll
|
||||
* sleep for CONFIG_STIMBUFF_FRAME_RETRY_DELAY_MS ms before trying again.
|
||||
*/
|
||||
int nextWakeupDelayMs;
|
||||
if (frameAssemblyRateLimiter.tryAcquire())
|
||||
{ nextWakeupDelayMs = CONFIG_STIMBUFF_FRAME_PERIOD_MS; }
|
||||
else
|
||||
{ nextWakeupDelayMs = CONFIG_STIMBUFF_FRAME_RETRY_DELAY_MS; }
|
||||
|
||||
// Placeholder handler (empty for now)
|
||||
// Note: The lock should be released when frame production completes
|
||||
|
||||
// Schedule next timeout with the pre-determined duration
|
||||
scheduleNextTimeout(nextWakeupDelayMs);
|
||||
}
|
||||
};
|
||||
|
||||
} // namespace stim_buff
|
||||
|
||||
Reference in New Issue
Block a user