Added buffer code
parent
46248d5ee6
commit
4a5e9d41ea
|
@ -0,0 +1,86 @@
|
|||
#include "buffer.hpp"
|
||||
|
||||
/*
|
||||
A Buffer consists of three parts:
|
||||
an incoming vector, a rollingbuffer, and an output vector.
|
||||
|
||||
Note that the RollingBuffer class does most of the work:
|
||||
its `get` and `put` methods move the values in our
|
||||
input and output vectors.
|
||||
*/
|
||||
|
||||
Buffer::Buffer(
|
||||
const char *pipe_file,
|
||||
size_t buffer_size,
|
||||
size_t output_size
|
||||
):
|
||||
buffer_size(buffer_size),
|
||||
output_size(output_size),
|
||||
pipe_file(pipe_file)
|
||||
{
|
||||
|
||||
sample_con_rate = 5;
|
||||
sample_con_rate_up_ctr = 0;
|
||||
sample_con_rate_dn_ctr = 0;
|
||||
|
||||
incoming.resize(buffer_size);
|
||||
rolling_buffer.resize(buffer_size);
|
||||
|
||||
output.resize(output_size);
|
||||
}
|
||||
|
||||
|
||||
void Buffer::update() {
|
||||
// TODO:
|
||||
// Disable and enable FIFO here to get rid of
|
||||
// the difference between audio and visualization.
|
||||
|
||||
int fd = open(pipe_file, O_RDONLY);
|
||||
|
||||
ssize_t bytes_read = read(
|
||||
fd,
|
||||
incoming.data(),
|
||||
sizeof(int16_t) * incoming.size()
|
||||
);
|
||||
|
||||
const auto begin = incoming.begin();
|
||||
const auto end = incoming.begin() + bytes_read/sizeof(int16_t);
|
||||
|
||||
// Autoscale here
|
||||
|
||||
rolling_buffer.put(begin, end);
|
||||
|
||||
// 44100 samples per second / fps = samples per frame
|
||||
// 60 fps
|
||||
// *2 if stereo
|
||||
size_t requested_samples = (44100 / 60) * pow(1.1, sample_con_rate);
|
||||
|
||||
size_t new_samples = rolling_buffer.get(
|
||||
requested_samples,
|
||||
output
|
||||
);
|
||||
|
||||
if (new_samples == 0) {
|
||||
printf("no new samples\n");
|
||||
return;
|
||||
}
|
||||
|
||||
// A crude way to adjust the amount of samples consumed from the buffer
|
||||
// depending on how fast the rendering is.
|
||||
if (rolling_buffer.size() > 0) {
|
||||
if (++sample_con_rate_up_ctr > 8) {
|
||||
sample_con_rate_up_ctr = 0;
|
||||
++sample_con_rate;
|
||||
}
|
||||
} else if (sample_con_rate > 0) {
|
||||
if (++sample_con_rate_dn_ctr > 4) {
|
||||
sample_con_rate_dn_ctr = 0;
|
||||
--sample_con_rate;
|
||||
}
|
||||
sample_con_rate_up_ctr = 0;
|
||||
}
|
||||
|
||||
if (fd >= 0) {
|
||||
close(fd);
|
||||
}
|
||||
}
|
|
@ -0,0 +1,39 @@
|
|||
#pragma once
|
||||
|
||||
#include <cstdint>
|
||||
#include <vector>
|
||||
#include <math.h>
|
||||
// For reading FIFO
|
||||
#include <fcntl.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "rollingbuffer.hpp"
|
||||
|
||||
|
||||
class Buffer {
|
||||
public:
|
||||
Buffer(
|
||||
const char *pipe_file,
|
||||
size_t buffer_size,
|
||||
size_t output_size
|
||||
);
|
||||
|
||||
void update();
|
||||
|
||||
const std::vector<int16_t>& get_output() const {
|
||||
return output;
|
||||
};
|
||||
|
||||
private:
|
||||
std::vector<int16_t> incoming;
|
||||
RollingBuffer<int16_t> rolling_buffer;
|
||||
std::vector<int16_t> output;
|
||||
|
||||
size_t buffer_size;
|
||||
size_t output_size;
|
||||
const char *pipe_file;
|
||||
|
||||
size_t sample_con_rate;
|
||||
size_t sample_con_rate_up_ctr;
|
||||
size_t sample_con_rate_dn_ctr;
|
||||
};
|
|
@ -0,0 +1,137 @@
|
|||
#pragma once
|
||||
|
||||
#include <stdexcept>
|
||||
#include <stdlib.h>
|
||||
#include <cstdint>
|
||||
#include <vector>
|
||||
|
||||
|
||||
/*
|
||||
This file defines a simple "RollingBuffer" class, based on the
|
||||
implementation of SampleBuffer in ncmpcpp.
|
||||
*/
|
||||
|
||||
|
||||
template<class T>
|
||||
struct RollingBuffer {
|
||||
public:
|
||||
typedef typename std::vector<T>::iterator Iterator;
|
||||
|
||||
RollingBuffer() : m_offset(0) { }
|
||||
|
||||
void put(Iterator begin, Iterator end);
|
||||
size_t get(size_t elems, std::vector<T> &dest);
|
||||
|
||||
void resize(size_t n);
|
||||
void clear();
|
||||
|
||||
size_t size() const;
|
||||
const std::vector<T> &buffer() const;
|
||||
|
||||
private:
|
||||
size_t m_offset;
|
||||
std::vector<T> m_buffer;
|
||||
};
|
||||
|
||||
|
||||
|
||||
template <class T>
|
||||
void RollingBuffer<T>::put(
|
||||
RollingBuffer::Iterator begin,
|
||||
RollingBuffer::Iterator end
|
||||
) {
|
||||
|
||||
// How many elements to put
|
||||
size_t elems = end - begin;
|
||||
if (elems > m_buffer.size()) {
|
||||
throw std::out_of_range("Size of the buffer is smaller than the amount of elements");
|
||||
}
|
||||
|
||||
// How much space we have for new elements
|
||||
size_t free_elems = m_buffer.size() - m_offset;
|
||||
|
||||
// If we don't have enough free space,
|
||||
// make more by moving the buffer forwards.
|
||||
if (elems > free_elems) {
|
||||
size_t to_remove = elems - free_elems;
|
||||
std::copy(
|
||||
m_buffer.begin() + to_remove,
|
||||
m_buffer.end() - free_elems,
|
||||
m_buffer.begin()
|
||||
);
|
||||
m_offset -= to_remove;
|
||||
}
|
||||
|
||||
// Add new elements to buffer
|
||||
std::copy(begin, end, m_buffer.begin() + m_offset);
|
||||
m_offset += elems;
|
||||
}
|
||||
|
||||
|
||||
|
||||
template <class T>
|
||||
size_t RollingBuffer<T>::get(
|
||||
size_t elems,
|
||||
std::vector<T> &dest
|
||||
) {
|
||||
// elems: how many samples to get
|
||||
// dest: where to put them
|
||||
|
||||
|
||||
// (m_offset == 0) => we have no data
|
||||
if (m_offset == 0) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
// If too many elements are requested,
|
||||
// give as much as we can.
|
||||
if (elems > m_offset) {
|
||||
elems = m_offset;
|
||||
}
|
||||
|
||||
if (elems >= dest.size()) {
|
||||
// If dest is smaller than request size,
|
||||
// discard earlier elements to fit.
|
||||
size_t elems_lost = elems - dest.size();
|
||||
std::copy(m_buffer.begin() + elems_lost, m_buffer.begin() + elems, dest.begin());
|
||||
} else {
|
||||
// otherwise, copy samples to the destination buffer.
|
||||
std::copy(dest.begin() + elems, dest.end(), dest.begin());
|
||||
std::copy(m_buffer.begin(), m_buffer.begin() + elems, dest.end() - elems);
|
||||
}
|
||||
|
||||
// Remove elements from the internal buffer.
|
||||
std::copy(m_buffer.begin() + elems, m_buffer.begin() + m_offset, m_buffer.begin());
|
||||
m_offset -= elems;
|
||||
|
||||
return elems;
|
||||
}
|
||||
|
||||
|
||||
|
||||
template <class T>
|
||||
void RollingBuffer<T>::resize(size_t n) {
|
||||
m_buffer.resize(n);
|
||||
clear();
|
||||
}
|
||||
|
||||
|
||||
|
||||
template <class T>
|
||||
void RollingBuffer<T>::clear() {
|
||||
m_offset = 0;
|
||||
}
|
||||
|
||||
|
||||
|
||||
template <class T>
|
||||
size_t RollingBuffer<T>::size() const {
|
||||
return m_offset;
|
||||
}
|
||||
|
||||
|
||||
|
||||
template <class T>
|
||||
const std::vector<T> &RollingBuffer<T>::buffer() const {
|
||||
return m_buffer;
|
||||
}
|
Loading…
Reference in New Issue