audio-river/sample/read/read.cpp

/** @file
 * @author Edouard DUPIN 
 * @copyright 2015, Edouard DUPIN, all right reserved
 * @license MPL v2.0 (see license file)
 */

//! [audio_river_sample_read_all]
//! [audio_river_sample_include]
#include <audio/river/river.hpp>
#include <audio/river/Manager.hpp>
#include <audio/river/Interface.hpp>
//! [audio_river_sample_include]
#include <etk/os/FSNode.hpp>
#include <etk/etk.hpp>
#include <ethread/Thread.hpp>
#include <ethread/tools.hpp>
#include <test-debug/debug.hpp>



//! [audio_river_sample_read_config_file]
static const etk::String configurationRiver =
	"{\n"
	"	microphone:{\n"
	"		io:'input',\n"
	"		map-on:{\n"
	"			interface:'auto',\n"
	"			name:'default',\n"
	//"			timestamp-mode:'trigered',\n"
	"		},\n"
	"		frequency:0,\n"
	"		channel-map:['front-left', 'front-right'],\n"
	"		type:'auto',\n"
	"		nb-chunk:1024\n"
	"	}\n"
	"}\n";
//! [audio_river_sample_read_config_file]


//! [audio_river_sample_callback_implement]
void onDataReceived(const void* _data,
                    const audio::Time& _time,
                    size_t _nbChunk,
                    enum audio::format _format,
                    uint32_t _frequency,
                    const etk::Vector<audio::channel>& _map,
                    etk::FSNode* _outputNode) {
	if (    _format != audio::format_int16
	     && _format != audio::format_float) {
		TEST_ERROR("Call wrong type ... (need int16_t.float)");
		return;
	}
	//! [audio_river_sample_callback_implement]
	if (_outputNode->fileIsOpen() == false) {
		if (_format != audio::format_int16) {
			// get the curent power of the signal.
			const int16_t* data = static_cast<const int16_t*>(_data);
			int64_t value = 0;
			for (size_t iii=0; iii<_nbChunk*_map.size(); ++iii) {
				value += etk::abs(data[iii]);
			}
			value /= (_nbChunk*_map.size());
			TEST_PRINT("Get data ... average=" << int32_t(value));
		} else {
			// get the curent power of the signal.
			const float* data = static_cast<const float*>(_data);
			float value = 0;
			for (size_t iii=0; iii<_nbChunk*_map.size(); ++iii) {
				value += etk::abs(data[iii]);
			}
			value /= (_nbChunk*_map.size());
			TEST_PRINT("Get data ... average=" << float(value));
		}
	} else {
		// just write data
		TEST_VERBOSE("Get data ... chunks=" << _nbChunk << " time=" << _time);
		_outputNode->fileWrite(_data, _map.size()*audio::getFormatBytes(_format), _nbChunk);
	}
}

int main(int _argc, const char **_argv) {
	//! [audio_river_sample_init]
	// the only one init for etk:
	etk::init(_argc, _argv);
	//! [audio_river_sample_init]
	// local parameter:
	etk::String configFile;
	etk::String ioName="microphone";
	etk::String outputFileName = "";
	for (int32_t iii=0; iii<_argc ; ++iii) {
		etk::String data = _argv[iii];
		if (    data == "-h"
		     || data == "--help") {
			TEST_PRINT("Help : ");
			TEST_PRINT("    --conf=xxx.json Input/output configuration");
			TEST_PRINT("    --io=xxx        name configuration input");
			TEST_PRINT("    --file=yyy.raw  File name to store data");
			exit(0);
		} else if (etk::start_with(data, "--conf=") == true) {
			configFile = etk::String(data.begin()+7, data.end());
			TEST_PRINT("Select config: " << configFile);
		} else if (etk::start_with(data, "--io=") == true) {
			ioName = etk::String(data.begin()+5, data.end());
			TEST_PRINT("Select io: " << ioName);
		} else if (etk::start_with(data, "--file=") == true) {
			outputFileName = etk::String(data.begin()+7, data.end());
			TEST_PRINT("Select output file name: " << outputFileName);
		}
	}
	// initialize river interface
	if (configFile == "") {
		audio::river::initString(configurationRiver);
	} else {
		audio::river::init(configFile);
	}
	//! [audio_river_sample_get_interface]
	// Create the River manager for tha application or part of the application.
	ememory::SharedPtr<audio::river::Manager> manager = audio::river::Manager::create("river_sample_read");
	//! [audio_river_sample_get_interface]
	//! [audio_river_sample_create_read_interface]
	// create interface:
	ememory::SharedPtr<audio::river::Interface> interface;
	//Get the generic input:
	interface = manager->createInput(48000,
	                                 etk::Vector<audio::channel>(),
	                                 audio::format_int16,
	                                 ioName);
	if(interface == null) {
		TEST_ERROR("null interface");
		return -1;
	}
	//! [audio_river_sample_create_read_interface]
	etk::FSNode outputNode;
	// open output file if needed:
	if (outputFileName != "") {
		outputNode.setName(outputFileName);
		outputNode.fileOpenWrite();
	}
	//! [audio_river_sample_set_callback]
	// set callback mode ...
	interface->setInputCallback([&](const void* _data,
	                                const audio::Time& _time,
	                                size_t _nbChunk,
	                                enum audio::format _format,
	                                uint32_t _frequency,
	                                const etk::Vector<audio::channel>& _map) {
	                                	onDataReceived(_data, _time, _nbChunk, _format, _frequency, _map, &outputNode);
	                                });
	//! [audio_river_sample_set_callback]
	//! [audio_river_sample_read_start_stop]
	// start the stream
	interface->start();
	// wait 10 second ...
	ethread::sleepMilliSeconds(1000*(10));
	// stop the stream
	interface->stop();
	//! [audio_river_sample_read_start_stop]
	//! [audio_river_sample_read_reset]
	// remove interface and manager.
	interface.reset();
	manager.reset();
	//! [audio_river_sample_read_reset]
	// close the output file
	if (outputFileName != "") {
		outputNode.fileClose();
	}
	return 0;
}


//! [audio_river_sample_read_all]