audio-drain/drain/EndPointWrite.cpp

/** @file
 * @author Edouard DUPIN 
 * @copyright 2011, Edouard DUPIN, all right reserved
 * @license APACHE v2.0 (see license file)
 */
#include "debug.h"
#include <drain/EndPointWrite.h>


#undef __class__
#define __class__ "EndPointWrite"

drain::EndPointWrite::EndPointWrite() :
  m_function(nullptr) {
	
}

void drain::EndPointWrite::init() {
	drain::EndPoint::init();
	m_type = "EndPoint";
}

std11::shared_ptr<drain::EndPointWrite> drain::EndPointWrite::create() {
	std11::shared_ptr<drain::EndPointWrite> tmp(new drain::EndPointWrite());
	tmp->init();
	return tmp;
}

void drain::EndPointWrite::configurationChange() {
	drain::EndPoint::configurationChange();
	m_needProcess = true;
}


bool drain::EndPointWrite::process(std11::chrono::system_clock::time_point& _time,
                                   void* _input,
                                   size_t _inputNbChunk,
                                   void*& _output,
                                   size_t& _outputNbChunk){
	drain::AutoLogInOut tmpLog("EndPointWrite");
	//DRAIN_INFO("                              nb Sample in buffer : " << m_tmpData.size());
	if (m_function != nullptr) {
		if (m_tmpData.size() <= 20000) {
			m_function(_time, _inputNbChunk, m_output.getFormat(), m_output.getFrequency(), m_output.getMap());
		}
	}
	// resize output buffer:
	//DRAIN_INFO("    resize : " << (int32_t)m_formatSize << "*" << (int32_t)_inputNbChunk << "*" << (int32_t)m_outputMap.size());
	m_outputData.resize(m_formatSize*_inputNbChunk*m_output.getMap().size(), 0);
	// set output pointer:
	_outputNbChunk = m_outputData.size()/(m_formatSize*m_output.getMap().size());
	_output = &m_outputData[0];
	std11::unique_lock<std11::mutex> lock(m_mutex);
	// check if data in the tmpBuffer
	if (m_tmpData.size() == 0) {
		DRAIN_WARNING("No data in the user buffer (write null data ... " << _outputNbChunk << " chunks)");
		// just send no data ...
		return true;
	}
	DRAIN_INFO("Write " << _outputNbChunk << " chunks");
	// check if we have enought data:
	int32_t nbChunkToCopy = std::min(_inputNbChunk, m_tmpData.size()/(m_output.getMap().size()*m_formatSize));
	
	DRAIN_INFO("      " << nbChunkToCopy << " chunks ==> " << nbChunkToCopy*m_output.getMap().size()*m_formatSize << " Byte sizeBuffer=" << m_tmpData.size());
	// copy data to the output:
	memcpy(_output, &m_tmpData[0], nbChunkToCopy*m_output.getMap().size()*m_formatSize);
	// remove old data:
	m_tmpData.erase(m_tmpData.begin(), m_tmpData.begin() + nbChunkToCopy*m_output.getMap().size()*m_formatSize);
	//DRAIN_INFO("                              nb Sample in buffer : " << m_tmpData.size());
	return true;
}

void drain::EndPointWrite::write(const void* _value, size_t _nbChunk) {
	std11::unique_lock<std11::mutex> lock(m_mutex);
	DRAIN_INFO("[ASYNC] Write data : " << _nbChunk << " chunks"
	              << " ==> " << _nbChunk*m_output.getMap().size() << " samples"
	              << " formatSize=" << int32_t(m_formatSize)
	              << " bufferSize=" << m_tmpData.size());
	const int8_t* value = static_cast<const int8_t*>(_value);
	for (size_t iii=0; iii<_nbChunk*m_formatSize*m_output.getMap().size(); ++iii) {
		m_tmpData.push_back(*value++);
	}
}