TactileBCI/src/UDPStimCodeSender/ovpCBoxAlgorithmUDPStimcodeSender.cpp

///-------------------------------------------------------------------------------------------------
/// 
/// \file CBoxAlgorithmUDPStimcodeSender.cpp
/// \brief Functions of Class UDP-StimCode-Sender
/// \author Tobias Baumann (TH Nuernberg).
/// \version 1.2.
/// \date Mon Oct 04 12:43:53 2021.
/// \copyright <a href="https://choosealicense.com/licenses/agpl-3.0/">GNU Affero General Public License v3.0</a>.
/// 
///-------------------------------------------------------------------------------------------------

//includes
#include "ovpCBoxAlgorithmUDPStimcodeSender.h"

using namespace OpenViBE;
using namespace /*OpenViBE::*/Kernel;
using namespace /*OpenViBE::*/Plugins;
using namespace /*OpenViBE::Plugins::*/Tactilebci;

bool CBoxAlgorithmUDPStimcodeSender::initialize()
{
	m_StimDecoder.initialize(*this, 0);
	m_StimEncoder.initialize(*this, 0);
	
	// retrieve box settings
	m_IP          = FSettingValueAutoCast(*this->getBoxAlgorithmContext(), 0);
	m_Port        = FSettingValueAutoCast(*this->getBoxAlgorithmContext(), 1);
	m_RowBase     = FSettingValueAutoCast(*this->getBoxAlgorithmContext(), 2);
	m_nTactilos    = FSettingValueAutoCast(*this->getBoxAlgorithmContext(), 3); 
	
	//set m_nTactilos to 2 if lower than 2
	if(m_nTactilos < 2)
	{
		m_nTactilos = 2;
	}
	//set m_nTactilos to MAX if greater than MAX_TACTILOS
	if(m_nTactilos > MAX_TACTILOS)
	{
		m_nTactilos = MAX_TACTILOS;
	}
	
	// connect UDP socket
	boost::asio::ip::udp::endpoint m_Endpoint(boost::asio::ip::address::from_string(m_IP), m_Port);
	m_Socket.connect(m_Endpoint);

	return true;
}
/*******************************************************************************/

bool CBoxAlgorithmUDPStimcodeSender::uninitialize()
{
	m_StimDecoder.uninitialize();
	m_StimEncoder.uninitialize();

	return true;
}
/*******************************************************************************/

bool CBoxAlgorithmUDPStimcodeSender::processInput(const size_t index)
{
	// some pre-processing code if needed...

	// ready to process !
	getBoxAlgorithmContext()->markAlgorithmAsReadyToProcess();

	return true;
}

/*******************************************************************************/


bool CBoxAlgorithmUDPStimcodeSender::process()
{
	
	// the static box context describes the box inputs, outputs, settings structures
	const IBox& staticBoxContext = this->getStaticBoxContext();
	// the dynamic box context describes the current state of the box inputs and outputs (i.e. the chunks)
	IBoxIO& boxContext       = this->getDynamicBoxContext();
	uint64_t StimulationID   = 0;
	uint64_t ChunkStartTime  = 0;
	uint64_t ChunkEndTime    = 0;
	uint64_t Size            = 0;
	const uint8_t*  Buffer   = nullptr;    

	//iterate over all chunk on input 0
	for (uint64_t i = 0; i < boxContext.getInputChunkCount(0); ++i)
	{
		// decode the chunk i
		m_StimDecoder.decode(i);
		if(m_StimDecoder.isBufferReceived())
		{
			//check received stimulations
			IStimulationSet* StimSet = m_StimDecoder.getOutputStimulationSet();
            for(uint64_t j=0; j<StimSet->getStimulationCount(); j++)
            {
                StimulationID   = StimSet->getStimulationIdentifier(j);
				
				if(StimulationID >= m_RowBase && StimulationID < (m_RowBase + m_nTactilos))
				{				
					this->getLogManager() << LogLevel_Debug << "send udp : [StimulusCode " << (StimulationID-m_RowBase+1) << "]\n";
					m_Socket.send(boost::asio::buffer(std::to_string(StimulationID-m_RowBase+1)));
				}
			}
		}
		// forward input chunks
		boxContext.getInputChunk(0, i, ChunkStartTime, ChunkEndTime, Size, Buffer);
		boxContext.appendOutputChunkData(0, Buffer, Size);
		boxContext.markOutputAsReadyToSend(0, ChunkStartTime, ChunkEndTime);
		boxContext.markInputAsDeprecated(0, i);
	}
	
	return true;
}