Added test.cpp
parent
2332e2980a
commit
58fd2821b5
|
@ -0,0 +1,78 @@
|
|||
#include <stdio.h>
|
||||
#include <cstdint>
|
||||
#include <vector>
|
||||
|
||||
// For reading FIFO
|
||||
#include <fcntl.h>
|
||||
#include <unistd.h>
|
||||
// Math libs
|
||||
#include <math.h>
|
||||
#include <fftw3.h>
|
||||
|
||||
// Local files
|
||||
#include "utility/bitmap.hpp"
|
||||
#include "utility/buffer.hpp"
|
||||
#include "signal_processing/fft.hpp"
|
||||
|
||||
|
||||
// TODO:
|
||||
// stereo support (and maybe different bitrates?)
|
||||
// Optimization: don't copy filename in buffer?
|
||||
// understand consumption rate
|
||||
// understand BIN2HZ
|
||||
// understand values and sizes (DFT_TOTAL, DFT_NONZERO, etc)
|
||||
// note that wave and spectrum have different sizes
|
||||
//
|
||||
// MPD interface
|
||||
// hid interface to keyboard
|
||||
// beat detection
|
||||
|
||||
|
||||
|
||||
|
||||
void draw_spectrum_bitmap(
|
||||
const std::vector<size_t>& waveform,
|
||||
Bitmap& bitmap
|
||||
) {
|
||||
for (size_t x = 0; x < waveform.size(); x++) {
|
||||
for (size_t y = 0; y < waveform[x]; y++) {
|
||||
bitmap.setpixel(bitmap.get_height() - y - 1, x, 0xFF, 0x00, 0x00);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const size_t width = 300;
|
||||
const size_t height = 100;
|
||||
|
||||
|
||||
int main(int argc, char *argv[]) {
|
||||
|
||||
// buffer size for waveform:
|
||||
// (44100 / fps * 10), make 10 bigger for slower scrolling
|
||||
//
|
||||
// Double both buffer sizes if stereo
|
||||
|
||||
FFT_Visualizer fft = FFT_Visualizer(
|
||||
width, height,
|
||||
20, 20000
|
||||
);
|
||||
|
||||
std::vector<size_t> waveform;
|
||||
waveform.resize(width);
|
||||
|
||||
Buffer buf = Buffer(
|
||||
"/tmp/mpd.fifo",
|
||||
44100 / 2, // Keep 500ms of data in buffer
|
||||
fft.compute_buffer_output_size()
|
||||
);
|
||||
|
||||
Bitmap b = Bitmap(width, height);
|
||||
|
||||
while (1) {
|
||||
b.clear();
|
||||
buf.update();
|
||||
fft.update(buf);
|
||||
draw_spectrum_bitmap(fft.get_output(), b);
|
||||
b.save("/tmp/o.bmp");
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue