stream class

git-svn-id: svn://elaine/murooma/trunk@67 d8a302eb-03bc-478d-80e4-98257eca68ef
This commit is contained in:
(no author) 2014-07-04 20:27:34 +00:00
parent 3141591be3
commit 5c4412562a
4 changed files with 198 additions and 110 deletions

View file

@ -18,6 +18,7 @@ struct ChunkT
{
int32_t tv_sec;
int32_t tv_usec;
int32_t idx;
int16_t payload[T];
};

View file

@ -14,21 +14,13 @@
#include <vector>
#include <algorithm>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <portaudio.h>
#include "chunk.h"
#include "doubleBuffer.h"
#include "timeUtils.h"
#include "stream.h"
DoubleBuffer<int> buffer(30000 / PLAYER_CHUNK_MS);
DoubleBuffer<int> shortBuffer(500 / PLAYER_CHUNK_MS);
std::deque<PlayerChunk*> chunks;
std::deque<int> timeDiffs;
std::mutex mtx;
std::mutex mutex;
std::condition_variable cv;
int bufferMs;
@ -44,9 +36,7 @@ void sleepMs(int ms)
}
int skip(0);
time_t lastUpdate(0);
Stream* stream;
/* This routine will be called by the PortAudio engine when audio is needed.
** It may called at interrupt level on some machines so don't do anything
@ -59,89 +49,21 @@ static int patestCallback( const void *inputBuffer, void *outputBuffer,
void *userData )
{
// std::cerr << "outputBufferDacTime: " << timeInfo->outputBufferDacTime*1000 << "\n";
std::deque<PlayerChunk*>* chunks = (std::deque<PlayerChunk*>*)userData;
Stream* stream = (Stream*)userData;
short* out = (short*)outputBuffer;
unsigned long i;
(void) timeInfo; /* Prevent unused variable warnings. */
(void) statusFlags;
(void) inputBuffer;
std::unique_lock<std::mutex> lck(mtx);
int age = 0;
int median = 0;
int shortMedian = 0;
PlayerChunk* chunk = NULL;
while (1)
{
if (chunks->empty())
cv.wait(lck);
mutex.lock();
chunk = chunks->front();
int chunkCount = chunks->size();
mutex.unlock();
age = getAge(*chunk) + timeInfo->outputBufferDacTime*1000 - bufferMs;
buffer.add(age);
shortBuffer.add(age);
time_t now = time(NULL);
std::vector<short> s = stream->getChunk(timeInfo->outputBufferDacTime, framesPerBuffer);
if (skip == 0)
for (size_t n=0; n<framesPerBuffer; n++)
{
if (now != lastUpdate)
{
lastUpdate = now;
median = buffer.median();
shortMedian = shortBuffer.median();
std::cerr << "age: " << getAge(*chunk) << "\t" << age << "\t" << shortMedian << "\t" << median << "\t" << buffer.size() << "\t" << chunkCount << "\t" << timeInfo->outputBufferDacTime*1000 << "\n";
*out++ = s[2*n];
*out++ = s[2*n+1];
}
if ((age > 500) || (age < -500))
skip = age / PLAYER_CHUNK_MS;
else if (shortBuffer.full() && ((shortMedian > 100) || (shortMedian < -100)))
skip = shortMedian / PLAYER_CHUNK_MS;
else if (buffer.full() && ((median > 10) || (median < -10)))
skip = median / PLAYER_CHUNK_MS;
}
if (skip != 0)
{
std::cerr << "age: " << getAge(*chunk) << "\t" << age << "\t" << shortMedian << "\t" << median << "\t" << buffer.size() << "\t" << timeInfo->outputBufferDacTime*1000 << "\n";
}
// bool silence = (age < -500) || (shortBuffer.full() && (shortMedian < -100)) || (buffer.full() && (median < -15));
if (skip > 0)
{
skip--;
chunks->pop_front();
delete chunk;
std::cerr << "packe too old, dropping\n";
buffer.clear();
shortBuffer.clear();
usleep(100);
}
else if (skip < 0)
{
skip++;
chunk = new PlayerChunk();
memset(&(chunk->payload[0]), 0, sizeof(int16_t)*PLAYER_CHUNK_SIZE);
// std::cerr << "age < bufferMs (" << age << " < " << bufferMs << "), playing silence\n";
buffer.clear();
shortBuffer.clear();
usleep(100);
break;
}
else
{
chunks->pop_front();
break;
}
}
for( i=0; i<framesPerBuffer; i++)
{
*out++ = chunk->payload[2*i];
*out++ = chunk->payload[2*i+1];
}
delete chunk;
// delete chunk;
return paContinue;
}
@ -159,7 +81,7 @@ static void StreamFinished( void* userData )
int initAudio()
{
PaStreamParameters outputParameters;
PaStream *stream;
PaStream *paStream;
PaError err;
printf("PortAudio Test: output sine wave. SR = %d, BufSize = %d\n", SAMPLE_RATE, FRAMES_PER_BUFFER);
@ -194,29 +116,29 @@ int initAudio()
outputParameters.hostApiSpecificStreamInfo = NULL;
err = Pa_OpenStream(
&stream,
&paStream,
NULL, /* no input */
&outputParameters,
SAMPLE_RATE,
FRAMES_PER_BUFFER,
paClipOff, /* we won't output out of range samples so don't bother clipping them */
patestCallback,
&chunks );
stream );
if( err != paNoError ) goto error;
err = Pa_SetStreamFinishedCallback( stream, &StreamFinished );
err = Pa_SetStreamFinishedCallback( paStream, &StreamFinished );
if( err != paNoError ) goto error;
err = Pa_StartStream( stream );
err = Pa_StartStream( paStream );
if( err != paNoError ) goto error;
// printf("Play for %d seconds.\n", NUM_SECONDS );
// Pa_Sleep( NUM_SECONDS * 1000 );
// err = Pa_StopStream( stream );
// err = Pa_StopStream( paStream );
// if( err != paNoError ) goto error;
// err = Pa_CloseStream( stream );
// err = Pa_CloseStream( paStream );
// if( err != paNoError ) goto error;
// Pa_Terminate();
@ -251,30 +173,20 @@ int main (int argc, char *argv[])
if (ret != 0)
std::cerr << "Unsuccessful in setting thread realtime prio" << std::endl;
*/
stream = new Stream();
initAudio();
Chunk* chunk;// = new Chunk();
while (1)
{
zmq::message_t update;
subscriber.recv(&update);
timeval now;
gettimeofday(&now, NULL);
std::cerr << "New chunk: " << chunkTime(*chunk) << "\t" << timeToStr(now) << "\t" << getAge(*chunk) << "\n";
// memcpy(chunk, update.data(), sizeof(Chunk));
chunk = (Chunk*)(update.data());
// timeval now;
// gettimeofday(&now, NULL);
// std::cerr << "New chunk: " << chunkTime(*chunk) << "\t" << timeToStr(now) << "\t" << getAge(*chunk) << "\n";
// std::cerr << chunk->tv_sec << "\t" << now.tv_sec << "\n";
for (size_t n=0; n<WIRE_CHUNK_MS/PLAYER_CHUNK_MS; ++n)
{
PlayerChunk* playerChunk = new PlayerChunk();
playerChunk->tv_sec = chunk->tv_sec;
playerChunk->tv_usec = chunk->tv_usec;
addMs(*playerChunk, n*PLAYER_CHUNK_MS);
memcpy(&(playerChunk->payload[0]), &chunk->payload[n*PLAYER_CHUNK_SIZE], sizeof(int16_t)*PLAYER_CHUNK_SIZE);
mutex.lock();
chunks.push_back(playerChunk);
mutex.unlock();
cv.notify_all();
}
stream->addChunk(chunk);
}
return 0;
}

View file

@ -58,6 +58,7 @@ int main () {
chunk->tv_sec = ts.tv_sec;
chunk->tv_usec = ts.tv_usec;
chunk->idx = 0;
zmq::message_t message(sizeof(Chunk));
memcpy(message.data(), chunk, sizeof(Chunk));
// snprintf ((char *) message.data(), size, "%05d %d", zipcode, c);

174
stream.h Normal file
View file

@ -0,0 +1,174 @@
#ifndef STREAM_H
#define STREAM_H
#include <deque>
#include <mutex>
#include <condition_variable>
#include <vector>
#include "doubleBuffer.h"
#include "timeUtils.h"
#include "chunk.h"
#include "timeUtils.h"
class Stream
{
public:
Stream() : lastUpdate(0), skip(0), idx(0)
{
pBuffer = new DoubleBuffer<int>(30000 / PLAYER_CHUNK_MS);
pShortBuffer = new DoubleBuffer<int>(5000 / PLAYER_CHUNK_MS);
pLock = new std::unique_lock<std::mutex>(mtx);
bufferMs = 500;
}
void addChunk(Chunk* chunk)
{
Chunk* c = new Chunk(*chunk);
mutex.lock();
chunks.push_back(c);
mutex.unlock();
cv.notify_all();
}
std::vector<short> getChunk(double outputBufferDacTime, unsigned long framesPerBuffer)
{
Chunk* chunk = NULL;
while (1)
{
if (chunks.empty())
cv.wait(*pLock);
int age(0);
int chunkCount(0);
mutex.lock();
do
{
chunk = chunks.front();
chunkCount = chunks.size();
int age = getAge(*chunk) + outputBufferDacTime*1000 - bufferMs;
if (age > WIRE_CHUNK_MS)
{
std::cerr << "age > WIRE_CHUNK_MS\n";
chunks.pop_front();
delete chunk;
chunk = NULL;
}
}
while (chunk == NULL);
mutex.unlock();
if (chunk == NULL)
{
std::cerr << "no chunks available\n";
continue;
}
std::cerr << "age: " << getAge(*chunk) << "\t" << age << "\t" << pBuffer->size() << "\t" << chunkCount << "\n";
pBuffer->add(age);
pShortBuffer->add(age);
time_t now = time(NULL);
if (skip == 0)
{
int age = 0;
int median = 0;
int shortMedian = 0;
if (now != lastUpdate)
{
lastUpdate = now;
median = pBuffer->median();
shortMedian = pShortBuffer->median();
std::cerr << "age: " << getAge(*chunk) << "\t" << age << "\t" << shortMedian << "\t" << median << "\t" << pBuffer->size() << "\t" << chunkCount << "\t" << outputBufferDacTime*1000 << "\n";
}
if ((age > 500) || (age < -500))
skip = age / PLAYER_CHUNK_MS;
else if (pShortBuffer->full() && ((shortMedian > 100) || (shortMedian < -100)))
skip = shortMedian / PLAYER_CHUNK_MS;
else if (pBuffer->full() && ((median > 10) || (median < -10)))
skip = median / PLAYER_CHUNK_MS;
}
if (skip != 0)
{
// std::cerr << "age: " << getAge(*chunk) << "\t" << age << "\t" << shortMedian << "\t" << median << "\t" << buffer->size() << "\t" << outputBufferDacTime*1000 << "\n";
}
// bool silence = (age < -500) || (shortBuffer.full() && (shortMedian < -100)) || (buffer.full() && (median < -15));
if (skip > 0)
{
skip--;
chunks.pop_front();
delete chunk;
std::cerr << "packe too old, dropping\n";
pBuffer->clear();
pShortBuffer->clear();
usleep(100);
}
else if (skip < 0)
{
skip++;
chunk = new Chunk();//PlayerChunk();
memset(&(chunk->payload[0]), 0, sizeof(int16_t)*PLAYER_CHUNK_SIZE);
// std::cerr << "age < bufferMs (" << age << " < " << bufferMs << "), playing silence\n";
pBuffer->clear();
pShortBuffer->clear();
usleep(100);
break;
}
else
{
chunks.pop_front();
break;
}
}
std::vector<short> v;
for (size_t n=0; n<framesPerBuffer; ++n)
{
v.push_back(0);
v.push_back(0);
}
return v;
// return chunk;
}
private:
std::deque<Chunk*> chunks;
std::mutex mtx;
std::mutex mutex;
std::unique_lock<std::mutex>* pLock;
std::condition_variable cv;
DoubleBuffer<int>* pBuffer;
DoubleBuffer<int>* pShortBuffer;
time_t lastUpdate;
int bufferMs;
int skip;
size_t idx;
};
// std::cerr << chunk->tv_sec << "\t" << now.tv_sec << "\n";
/* for (size_t n=0; n<WIRE_CHUNK_MS/PLAYER_CHUNK_MS; ++n)
{
PlayerChunk* playerChunk = new PlayerChunk();
playerChunk->tv_sec = chunk->tv_sec;
playerChunk->tv_usec = chunk->tv_usec;
addMs(*playerChunk, n*PLAYER_CHUNK_MS);
memcpy(&(playerChunk->payload[0]), &chunk->payload[n*PLAYER_CHUNK_SIZE], sizeof(int16_t)*PLAYER_CHUNK_SIZE);
mutex.lock();
chunks.push_back(playerChunk);
mutex.unlock();
cv.notify_all();
}
*/
#endif