#ifndef _STIMULUS_BUFFER_H
#define _STIMULUS_BUFFER_H

#include <config.h>
#include <vector>
#include <memory>
#include <user/spMcRingBuffer.h>
#include "stimulusFrame.h"
#include "deviceAttachmentSpec.h"

namespace smo {
namespace stim_buff {

// Forward declaration
class StimulusProducer;

/**
 * StimulusBuffer manages a collection of stimulus frames and ring buffer.
 *
 * This buffer holds the actual frame storage and ring buffer for stimulus
 * data. It maintains a reference to its parent StimulusProducer.
 */
class StimulusBuffer
{
public:
	explicit StimulusBuffer(
		StimulusProducer& parent,
		const std::shared_ptr<device::DeviceAttachmentSpec>
			&deviceAttachmentSpec,
		int histbuffMs,
		const SpMcRingBuffer::InputEngineConstraints& ringBufferConstraints)
	: parent(parent),
	deviceAttachmentSpec(deviceAttachmentSpec),
	histbuffMs(histbuffMs),
	frames_(static_cast<size_t>(histbuffMs / CONFIG_STIMBUFF_FRAME_PERIOD_MS)),
	ringBufferConstraints(ringBufferConstraints),
	ringBuffer(
		static_cast<size_t>(histbuffMs / CONFIG_STIMBUFF_FRAME_PERIOD_MS),
		ringBufferConstraints)
	{}

	~StimulusBuffer() = default;

	// Non-copyable, movable
	StimulusBuffer(const StimulusBuffer&) = delete;
	StimulusBuffer& operator=(const StimulusBuffer&) = delete;
	StimulusBuffer(StimulusBuffer&&) = default;
	StimulusBuffer& operator=(StimulusBuffer&&) = default;

public:
	StimulusProducer& parent;
	std::shared_ptr<device::DeviceAttachmentSpec> deviceAttachmentSpec;
	int histbuffMs;
	std::vector<StimulusFrame> frames_;

protected:
	SpMcRingBuffer::InputEngineConstraints ringBufferConstraints;
	SpMcRingBuffer ringBuffer;
};

} // namespace stim_buff
} // namespace smo

#endif // _STIMULUS_BUFFER_H