cannam@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
cannam@0: 
cannam@0: #ifdef HAVE_PORTAUDIO
cannam@0: 
cannam@0: #include "AudioPortAudioSource.h"
cannam@0: #include "AudioCallbackRecordTarget.h"
cannam@0: 
cannam@0: #include <iostream>
cannam@0: #include <cassert>
cannam@0: #include <cmath>
cannam@0: 
cannam@0: //#define DEBUG_AUDIO_PORT_AUDIO_SOURCE 1
cannam@0: 
cannam@0: AudioPortAudioSource::AudioPortAudioSource(AudioCallbackRecordTarget *target) :
cannam@0:     AudioCallbackRecordSource(target),
cannam@0:     m_stream(0),
cannam@0:     m_bufferSize(0),
cannam@0:     m_sampleRate(0),
cannam@0:     m_latency(0)
cannam@0: {
cannam@0:     PaError err;
cannam@0: 
cannam@0:     err = Pa_Initialize();
cannam@0:     if (err != paNoError) {
cannam@0: 	std::cerr << "ERROR: AudioPortAudioSource: Failed to initialize PortAudio" << std::endl;
cannam@0: 	return;
cannam@0:     }
cannam@0: 
cannam@0:     m_bufferSize = 1024;
cannam@0:     m_sampleRate = 44100;
cannam@0:     if (m_target && (m_target->getPreferredSampleRate() != 0)) {
cannam@0: 	m_sampleRate = m_target->getPreferredSampleRate();
cannam@0:     }
cannam@0: 
cannam@0:     m_latency = Pa_GetMinNumBuffers(m_bufferSize, m_sampleRate) * m_bufferSize;
cannam@0: 
cannam@0:     err = Pa_OpenDefaultStream(&m_stream, m_target->getChannelCount(),
cannam@0:                                0, paFloat32,
cannam@0: 			       m_sampleRate, m_bufferSize, 0,
cannam@0: 			       processStatic, this);
cannam@0: 
cannam@0:     if (err != paNoError) {
cannam@0: 	std::cerr << "ERROR: AudioPortAudioSource: Failed to open PortAudio stream" << std::endl;
cannam@0: 	m_stream = 0;
cannam@0: 	Pa_Terminate();
cannam@0: 	return;
cannam@0:     }
cannam@0: 
cannam@0:     err = Pa_StartStream(m_stream);
cannam@0: 
cannam@0:     if (err != paNoError) {
cannam@0: 	std::cerr << "ERROR: AudioPortAudioSource: Failed to start PortAudio stream" << std::endl;
cannam@0: 	Pa_CloseStream(m_stream);
cannam@0: 	m_stream = 0;
cannam@0: 	Pa_Terminate();
cannam@0: 	return;
cannam@0:     }
cannam@0: 
cannam@0:     if (m_target) {
cannam@0: 	std::cerr << "AudioPortAudioSource: block size " << m_bufferSize << std::endl;
cannam@0: 	m_target->setSourceBlockSize(m_bufferSize);
cannam@0: 	m_target->setSourceSampleRate(m_sampleRate);
cannam@0: 	m_target->setSourceRecordLatency(m_latency);
cannam@0:     }
cannam@0: }
cannam@0: 
cannam@0: AudioPortAudioSource::~AudioPortAudioSource()
cannam@0: {
cannam@0:     if (m_stream) {
cannam@0: 	PaError err;
cannam@0: 	err = Pa_CloseStream(m_stream);
cannam@0: 	if (err != paNoError) {
cannam@0: 	    std::cerr << "ERROR: AudioPortAudioSource: Failed to close PortAudio stream" << std::endl;
cannam@0: 	}
cannam@0: 	Pa_Terminate();
cannam@0:     }
cannam@0: }
cannam@0: 
cannam@0: bool
cannam@0: AudioPortAudioSource::isOK() const
cannam@0: {
cannam@0:     return (m_stream != 0);
cannam@0: }
cannam@0: 
cannam@0: int
cannam@0: AudioPortAudioSource::processStatic(void *input, void *output,
cannam@0: 				    unsigned long nframes,
cannam@0: 				    PaTimestamp outTime, void *data)
cannam@0: {
cannam@0:     return ((AudioPortAudioSource *)data)->process(input, output,
cannam@0: 						   nframes, outTime);
cannam@0: }
cannam@0: 
cannam@0: int
cannam@0: AudioPortAudioSource::process(void *inputBuffer,
cannam@0:                               void *outputBuffer,
cannam@0: 			      unsigned long nframes,
cannam@0: 			      PaTimestamp)
cannam@0: {
cannam@0: #ifdef DEBUG_AUDIO_PORT_AUDIO_SOURCE    
cannam@0:     std::cout << "AudioPortAudioSource::process(" << nframes << ")" << std::endl;
cannam@0: #endif
cannam@0: 
cannam@0:     if (!m_target) return 0;
cannam@0: 
cannam@0:     float *input = (float *)inputBuffer;
cannam@0: 
cannam@0:     assert(nframes <= m_bufferSize);
cannam@0: 
cannam@0:     static float **tmpbuf = 0;
cannam@0:     static size_t tmpbufch = 0;
cannam@0:     static size_t tmpbufsz = 0;
cannam@0: 
cannam@0:     size_t channels = m_target->getChannelCount();
cannam@0: 
cannam@0:     if (!tmpbuf || tmpbufch != channels || tmpbufsz < m_bufferSize) {
cannam@0: 
cannam@0: 	if (tmpbuf) {
cannam@0: 	    for (size_t i = 0; i < tmpbufch; ++i) {
cannam@0: 		delete[] tmpbuf[i];
cannam@0: 	    }
cannam@0: 	    delete[] tmpbuf;
cannam@0: 	}
cannam@0: 
cannam@0: 	tmpbufch = channels;
cannam@0: 	tmpbufsz = m_bufferSize;
cannam@0: 	tmpbuf = new float *[tmpbufch];
cannam@0: 
cannam@0: 	for (size_t i = 0; i < tmpbufch; ++i) {
cannam@0: 	    tmpbuf[i] = new float[tmpbufsz];
cannam@0: 	}
cannam@0:     }
cannam@0: 	
cannam@0:     float peakLeft = 0.0, peakRight = 0.0;
cannam@0: 
cannam@0:     for (size_t ch = 0; ch < 2; ++ch) {
cannam@0: 	
cannam@0: 	float peak = 0.0;
cannam@0: 
cannam@0: 	if (ch < channels) {
cannam@0: 
cannam@0: 	    // PortAudio samples are interleaved
cannam@0: 	    for (size_t i = 0; i < nframes; ++i) {
cannam@0: 		tmpbuf[ch][i] = input[i * 2 + ch];
cannam@0: 		float sample = fabsf(input[i * 2 + ch]);
cannam@0: 		if (sample > peak) peak = sample;
cannam@0: 	    }
cannam@0: 
cannam@0: 	} else if (ch == 1 && channels == 1) {
cannam@0: 
cannam@0: 	    for (size_t i = 0; i < nframes; ++i) {
cannam@0: 		tmpbuf[0][i] = input[i * 2 + ch];
cannam@0: 		float sample = fabsf(input[i * 2 + ch]);
cannam@0: 		if (sample > peak) peak = sample;
cannam@0: 	    }
cannam@0: 
cannam@0: 	} else {
cannam@0: 	    for (size_t i = 0; i < nframes; ++i) {
cannam@0: 		tmpbuf[ch][i] = 0;
cannam@0: 	    }
cannam@0: 	}
cannam@0: 
cannam@0: 	if (ch == 0) peakLeft = peak;
cannam@0: 	if (ch > 0 || channels == 1) peakRight = peak;
cannam@0:     }
cannam@0: 
cannam@0:     m_target->putSamples(nframes, tmpbuf);
cannam@0:     m_target->setInputLevels(peakLeft, peakRight);
cannam@0: 
cannam@0:     return 0;
cannam@0: }
cannam@0: 
cannam@0: #endif /* HAVE_PORTAUDIO */
cannam@0: