lbajardsilogic@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
lbajardsilogic@0: 
lbajardsilogic@0: /*
lbajardsilogic@0:     Sonic Visualiser
lbajardsilogic@0:     An audio file viewer and annotation editor.
lbajardsilogic@0:     Centre for Digital Music, Queen Mary, University of London.
lbajardsilogic@0:     This file copyright 2006 Chris Cannam.
lbajardsilogic@0:     
lbajardsilogic@0:     This program is free software; you can redistribute it and/or
lbajardsilogic@0:     modify it under the terms of the GNU General Public License as
lbajardsilogic@0:     published by the Free Software Foundation; either version 2 of the
lbajardsilogic@0:     License, or (at your option) any later version.  See the file
lbajardsilogic@0:     COPYING included with this distribution for more information.
lbajardsilogic@0: */
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef HAVE_PORTAUDIO
lbajardsilogic@0: 
lbajardsilogic@0: #include "AudioPortAudioTarget.h"
lbajardsilogic@0: #include "AudioCallbackPlaySource.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: #include <cassert>
lbajardsilogic@0: #include <cmath>
lbajardsilogic@0: 
lbajardsilogic@0: //#define DEBUG_AUDIO_PORT_AUDIO_TARGET 1
lbajardsilogic@0: 
lbajardsilogic@0: AudioPortAudioTarget::AudioPortAudioTarget(AudioCallbackPlaySource *source) :
lbajardsilogic@0:     AudioCallbackPlayTarget(source),
lbajardsilogic@0:     m_stream(0),
lbajardsilogic@0:     m_bufferSize(0),
lbajardsilogic@0:     m_sampleRate(0),
lbajardsilogic@0:     m_latency(0)
lbajardsilogic@0: {
lbajardsilogic@0:     PaError err;
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_AUDIO_PORT_AUDIO_TARGET
lbajardsilogic@0: #ifdef HAVE_PORTAUDIO_V18
lbajardsilogic@0:     std::cerr << "AudioPortAudioTarget: Initialising for PortAudio v18" << std::endl;
lbajardsilogic@0: #else
lbajardsilogic@0:     std::cerr << "AudioPortAudioTarget: Initialising for PortAudio v19" << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0:     err = Pa_Initialize();
lbajardsilogic@0:     if (err != paNoError) {
lbajardsilogic@0: 	std::cerr << "ERROR: AudioPortAudioTarget: Failed to initialize PortAudio: " << Pa_GetErrorText(err) << std::endl;
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     m_bufferSize = 1024;
lbajardsilogic@0:     m_sampleRate = 44100;
lbajardsilogic@0:     if (m_source && (m_source->getSourceSampleRate() != 0)) {
lbajardsilogic@0: 	m_sampleRate = m_source->getSourceSampleRate();
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef HAVE_PORTAUDIO_V18
lbajardsilogic@0:     m_latency = Pa_GetMinNumBuffers(m_bufferSize, m_sampleRate) * m_bufferSize;
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef HAVE_PORTAUDIO_V18
lbajardsilogic@0:     err = Pa_OpenDefaultStream(&m_stream, 0, 2, paFloat32,
lbajardsilogic@0: 			       m_sampleRate, m_bufferSize, 0,
lbajardsilogic@0: 			       processStatic, this);
lbajardsilogic@0: #else
lbajardsilogic@0:     err = Pa_OpenDefaultStream(&m_stream, 0, 2, paFloat32,
lbajardsilogic@0: 			       m_sampleRate, m_bufferSize,
lbajardsilogic@0: 			       processStatic, this);
lbajardsilogic@0: #endif    
lbajardsilogic@0: 
lbajardsilogic@0:     if (err != paNoError) {
lbajardsilogic@0: 	std::cerr << "ERROR: AudioPortAudioTarget: Failed to open PortAudio stream: " << Pa_GetErrorText(err) << std::endl;
lbajardsilogic@0: 	m_stream = 0;
lbajardsilogic@0: 	Pa_Terminate();
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0: #ifndef HAVE_PORTAUDIO_V18
lbajardsilogic@0:     const PaStreamInfo *info = Pa_GetStreamInfo(m_stream);
lbajardsilogic@0:     m_latency = int(info->outputLatency * m_sampleRate + 0.001);
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0:     std::cerr << "PortAudio latency = " << m_latency << " frames" << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:     err = Pa_StartStream(m_stream);
lbajardsilogic@0: 
lbajardsilogic@0:     if (err != paNoError) {
lbajardsilogic@0: 	std::cerr << "ERROR: AudioPortAudioTarget: Failed to start PortAudio stream: " << Pa_GetErrorText(err) << std::endl;
lbajardsilogic@0: 	Pa_CloseStream(m_stream);
lbajardsilogic@0: 	m_stream = 0;
lbajardsilogic@0: 	Pa_Terminate();
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_source) {
lbajardsilogic@0: 	std::cerr << "AudioPortAudioTarget: block size " << m_bufferSize << std::endl;
lbajardsilogic@0: 	m_source->setTargetBlockSize(m_bufferSize);
lbajardsilogic@0: 	m_source->setTargetSampleRate(m_sampleRate);
lbajardsilogic@0: 	m_source->setTargetPlayLatency(m_latency);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_PORT_AUDIO_TARGET
lbajardsilogic@0:     std::cerr << "AudioPortAudioTarget: initialised OK" << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: AudioPortAudioTarget::~AudioPortAudioTarget()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_stream) {
lbajardsilogic@0: 	PaError err;
lbajardsilogic@0: 	err = Pa_CloseStream(m_stream);
lbajardsilogic@0: 	if (err != paNoError) {
lbajardsilogic@0: 	    std::cerr << "ERROR: AudioPortAudioTarget: Failed to close PortAudio stream: " << Pa_GetErrorText(err) << std::endl;
lbajardsilogic@0: 	}
lbajardsilogic@0: 	err = Pa_Terminate();
lbajardsilogic@0:         if (err != paNoError) {
lbajardsilogic@0:             std::cerr << "ERROR: AudioPortAudioTarget: Failed to terminate PortAudio: " << Pa_GetErrorText(err) << std::endl;
lbajardsilogic@0: 	}   
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: AudioPortAudioTarget::isOK() const
lbajardsilogic@0: {
lbajardsilogic@0:     return (m_stream != 0);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef HAVE_PORTAUDIO_V18
lbajardsilogic@0: int
lbajardsilogic@0: AudioPortAudioTarget::processStatic(void *input, void *output,
lbajardsilogic@0: 				    unsigned long nframes,
lbajardsilogic@0: 				    PaTimestamp outTime, void *data)
lbajardsilogic@0: {
lbajardsilogic@0:     return ((AudioPortAudioTarget *)data)->process(input, output,
lbajardsilogic@0: 						   nframes, outTime);
lbajardsilogic@0: }
lbajardsilogic@0: #else
lbajardsilogic@0: int
lbajardsilogic@0: AudioPortAudioTarget::processStatic(const void *input, void *output,
lbajardsilogic@0:                                     unsigned long nframes,
lbajardsilogic@0:                                     const PaStreamCallbackTimeInfo *timeInfo,
lbajardsilogic@0:                                     PaStreamCallbackFlags flags, void *data)
lbajardsilogic@0: {
lbajardsilogic@0:     return ((AudioPortAudioTarget *)data)->process(input, output,
lbajardsilogic@0:                                                    nframes, timeInfo,
lbajardsilogic@0:                                                    flags);
lbajardsilogic@0: }
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: AudioPortAudioTarget::sourceModelReplaced()
lbajardsilogic@0: {
lbajardsilogic@0:     m_source->setTargetSampleRate(m_sampleRate);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef HAVE_PORTAUDIO_V18
lbajardsilogic@0: int
lbajardsilogic@0: AudioPortAudioTarget::process(void *inputBuffer, void *outputBuffer,
lbajardsilogic@0: 			      unsigned long nframes,
lbajardsilogic@0: 			      PaTimestamp)
lbajardsilogic@0: #else
lbajardsilogic@0: int
lbajardsilogic@0: AudioPortAudioTarget::process(const void *, void *outputBuffer,
lbajardsilogic@0:                               unsigned long nframes,
lbajardsilogic@0:                               const PaStreamCallbackTimeInfo *,
lbajardsilogic@0:                               PaStreamCallbackFlags)
lbajardsilogic@0: #endif
lbajardsilogic@0: {
lbajardsilogic@0: #ifdef DEBUG_AUDIO_PORT_AUDIO_TARGET    
lbajardsilogic@0:     std::cout << "AudioPortAudioTarget::process(" << nframes << ")" << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_source) return 0;
lbajardsilogic@0: 
lbajardsilogic@0:     float *output = (float *)outputBuffer;
lbajardsilogic@0: 
lbajardsilogic@0:     assert(nframes <= m_bufferSize);
lbajardsilogic@0: 
lbajardsilogic@0:     static float **tmpbuf = 0;
lbajardsilogic@0:     static size_t tmpbufch = 0;
lbajardsilogic@0:     static size_t tmpbufsz = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     size_t sourceChannels = m_source->getSourceChannelCount();
lbajardsilogic@0: 
lbajardsilogic@0:     // Because we offer pan, we always want at least 2 channels
lbajardsilogic@0:     if (sourceChannels < 2) sourceChannels = 2;
lbajardsilogic@0: 
lbajardsilogic@0:     if (!tmpbuf || tmpbufch != sourceChannels || int(tmpbufsz) < m_bufferSize) {
lbajardsilogic@0: 
lbajardsilogic@0: 	if (tmpbuf) {
lbajardsilogic@0: 	    for (size_t i = 0; i < tmpbufch; ++i) {
lbajardsilogic@0: 		delete[] tmpbuf[i];
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	    delete[] tmpbuf;
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	tmpbufch = sourceChannels;
lbajardsilogic@0: 	tmpbufsz = m_bufferSize;
lbajardsilogic@0: 	tmpbuf = new float *[tmpbufch];
lbajardsilogic@0: 
lbajardsilogic@0: 	for (size_t i = 0; i < tmpbufch; ++i) {
lbajardsilogic@0: 	    tmpbuf[i] = new float[tmpbufsz];
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: 	
lbajardsilogic@0:     size_t received = m_source->getSourceSamples(nframes, tmpbuf);
lbajardsilogic@0: 
lbajardsilogic@0:     float peakLeft = 0.0, peakRight = 0.0;
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t ch = 0; ch < 2; ++ch) {
lbajardsilogic@0: 	
lbajardsilogic@0: 	float peak = 0.0;
lbajardsilogic@0: 
lbajardsilogic@0: 	if (ch < sourceChannels) {
lbajardsilogic@0: 
lbajardsilogic@0: 	    // PortAudio samples are interleaved
lbajardsilogic@0: 	    for (size_t i = 0; i < nframes; ++i) {
lbajardsilogic@0:                 if (i < received) {
lbajardsilogic@0:                     output[i * 2 + ch] = tmpbuf[ch][i] * m_outputGain;
lbajardsilogic@0:                     float sample = fabsf(output[i * 2 + ch]);
lbajardsilogic@0:                     if (sample > peak) peak = sample;
lbajardsilogic@0:                 } else {
lbajardsilogic@0:                     output[i * 2 + ch] = 0;
lbajardsilogic@0:                 }
lbajardsilogic@0: 	    }
lbajardsilogic@0: 
lbajardsilogic@0: 	} else if (ch == 1 && sourceChannels == 1) {
lbajardsilogic@0: 
lbajardsilogic@0: 	    for (size_t i = 0; i < nframes; ++i) {
lbajardsilogic@0:                 if (i < received) {
lbajardsilogic@0:                     output[i * 2 + ch] = tmpbuf[0][i] * m_outputGain;
lbajardsilogic@0:                     float sample = fabsf(output[i * 2 + ch]);
lbajardsilogic@0:                     if (sample > peak) peak = sample;
lbajardsilogic@0:                 } else {
lbajardsilogic@0:                     output[i * 2 + ch] = 0;
lbajardsilogic@0:                 }
lbajardsilogic@0: 	    }
lbajardsilogic@0: 
lbajardsilogic@0: 	} else {
lbajardsilogic@0: 	    for (size_t i = 0; i < nframes; ++i) {
lbajardsilogic@0: 		output[i * 2 + ch] = 0;
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	if (ch == 0) peakLeft = peak;
lbajardsilogic@0: 	if (ch > 0 || sourceChannels == 1) peakRight = peak;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     m_source->setOutputLevels(peakLeft, peakRight);
lbajardsilogic@0: 
lbajardsilogic@0:     return 0;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: #endif /* HAVE_PORTAUDIO */
lbajardsilogic@0: