audio-river/test/testAEC.h

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

namespace river_test_aec {
	
	class Linker {
		private:
			std::shared_ptr<audio::river::Manager> m_manager;
			std::shared_ptr<audio::river::Interface> m_interfaceOut;
			std::shared_ptr<audio::river::Interface> m_interfaceIn;
			audio::drain::CircularBuffer m_buffer;
		public:
			Linker(std::shared_ptr<audio::river::Manager> _manager, const std::string& _input, const std::string& _output) :
			  m_manager(_manager) {
				//Set stereo output:
				std::vector<audio::channel> channelMap;
				if (false) { //"speaker" == _output) {
					channelMap.push_back(audio::channel_frontCenter);
				} else {
					channelMap.push_back(audio::channel_frontLeft);
					channelMap.push_back(audio::channel_frontRight);
				}
				m_buffer.setCapacity(std::chrono::milliseconds(2000), sizeof(int16_t)*channelMap.size(), 48000);
				
				m_interfaceOut = m_manager->createOutput(48000,
				                                         channelMap,
				                                         audio::format_int16,
				                                         _output);
				if(m_interfaceOut == nullptr) {
					TEST_ERROR("nullptr interface");
					return;
				}
				// set callback mode ...
				m_interfaceOut->setOutputCallback(std::bind(&Linker::onDataNeeded,
				                                              this,
				                                              std::placeholders::_1,
				                                              std::placeholders::_2,
				                                              std::placeholders::_3,
				                                              std::placeholders::_4,
				                                              std::placeholders::_5,
				                                              std::placeholders::_6));
				m_interfaceOut->addVolumeGroup("FLOW");
				if ("speaker" == _output) {
					m_interfaceOut->setParameter("volume", "FLOW", "0dB");
				}
				
				m_interfaceIn = m_manager->createInput(48000,
				                                       channelMap,
				                                       audio::format_int16,
				                                       _input);
				if(m_interfaceIn == nullptr) {
					TEST_ERROR("nullptr interface");
					return;
				}
				// set callback mode ...
				m_interfaceIn->setInputCallback(std::bind(&Linker::onDataReceived,
				                                            this,
				                                            std::placeholders::_1,
				                                            std::placeholders::_2,
				                                            std::placeholders::_3,
				                                            std::placeholders::_4,
				                                            std::placeholders::_5,
				                                            std::placeholders::_6));
				
			}
			void onDataNeeded(void* _data,
			                  const audio::Time& _time,
			                  size_t _nbChunk,
			                  enum audio::format _format,
			                  uint32_t _frequency,
			                  const std::vector<audio::channel>& _map) {
				if (_format != audio::format_int16) {
					TEST_ERROR("call wrong type ... (need int16_t)");
				}
				m_buffer.read(_data, _nbChunk);
			}
			void onDataReceived(const void* _data,
			                    const audio::Time& _time,
			                    size_t _nbChunk,
			                    enum audio::format _format,
			                    uint32_t _frequency,
			                    const std::vector<audio::channel>& _map) {
				if (_format != audio::format_int16) {
					TEST_ERROR("call wrong type ... (need int16_t)");
				}
				m_buffer.write(_data, _nbChunk);
			}
			void start() {
				if(m_interfaceIn == nullptr) {
					TEST_ERROR("nullptr interface");
					return;
				}
				if(m_interfaceOut == nullptr) {
					TEST_ERROR("nullptr interface");
					return;
				}
				m_interfaceOut->start();
				m_interfaceIn->start();
			}
			void stop() {
				if(m_interfaceIn == nullptr) {
					TEST_ERROR("nullptr interface");
					return;
				}
				if(m_interfaceOut == nullptr) {
					TEST_ERROR("nullptr interface");
					return;
				}
				m_manager->generateDotAll("activeProcess.dot");
				m_interfaceOut->stop();
				m_interfaceIn->stop();
			}
	};
	
	static const std::string configurationRiver =
		"{\n"
		"	speaker:{\n"
		"		io:'output',\n"
		"		map-on:{\n"
		"			interface:'auto',\n"
		"			name:'hw:0,0',\n"
		"			timestamp-mode:'trigered',\n"
		"		},\n"
		"		frequency:0,\n"
		"		channel-map:['front-left', 'front-right'],\n"
		"		type:'auto',\n"
		"		nb-chunk:1024,\n"
		"	},\n"
		"	microphone:{\n"
		"		io:'input',\n"
		"		map-on:{\n"
		"			interface:'auto',\n"
		"			name:'hw:0,0',\n"
		"			timestamp-mode:'trigered',\n"
		"		},\n"
		"		frequency:0,\n"
		"		channel-map:['front-left', 'front-right'],\n"
		"		type:'auto',\n"
		"		nb-chunk:1024\n"
		"	},\n"
		"	speaker-test:{\n"
		"		io:'output',\n"
		"		map-on:{\n"
		"			interface:'alsa',\n"
		"			name:'hw:2,0',\n"
		"			timestamp-mode:'trigered',\n"
		"		},\n"
		//"		group:'groupSynchro',\n"
		"		frequency:0,\n"
		"		channel-map:['front-left', 'front-right'],\n"
		"		type:'auto',\n"
		"		nb-chunk:1024\n"
		"	},\n"
		"	microphone-test:{\n"
		"		io:'input',\n"
		"		map-on:{\n"
		"			interface:'alsa',\n"
		"			name:'hw:2,0',\n"
		"			timestamp-mode:'trigered',\n"
		"		},\n"
		//"		group:'groupSynchro',\n"
		"		frequency:0,\n"
		"		channel-map:['front-center'],\n"
		"		type:'auto',\n"
		"		nb-chunk:1024\n"
		"	},\n"
		"	# virtual Nodes :\n"
		"	microphone-clean:{\n"
		"		io:'aec',\n"
		"		map-on-microphone:{\n"
		"			io:'input',\n"
		"			map-on:'microphone-test'\n"
		"		},\n"
		"		map-on-feedback:{\n"
		"			io:'feedback',\n"
		"			map-on:'speaker-test',\n"
		"		},\n"
		"		frequency:48000,\n"
		"		channel-map:[\n"
		"			'front-left', 'front-right'\n"
		//"			'front-center'\n"
		"		],\n"
		"		nb-chunk:1024,\n"
		"		type:'int16',\n"
		"		algo:'river-remover',\n"
		"		algo-mode:'cutter',\n"
		"		feedback-delay:10000,\n"
		"		mux-demux-type:'int16'\n"
		"	}\n"
		"}\n";
	
	TEST(TestUser, testAECManually) {
		audio::river::initString(configurationRiver);
		std::shared_ptr<audio::river::Manager> manager;
		manager = audio::river::Manager::create("testApplication");
		std::shared_ptr<Linker> processLink1 = std::make_shared<Linker>(manager, "microphone-clean", "speaker");
		std::shared_ptr<Linker> processLink2 = std::make_shared<Linker>(manager, "microphone", "speaker-test");
		processLink1->start();
		processLink2->start();
		sleep(30);
		processLink1->stop();
		processLink2->stop();
		
		processLink1.reset();
		processLink2.reset();
		manager.reset();
		audio::river::unInit();
	}
};