You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
392 lines
14 KiB
392 lines
14 KiB
3 years ago
|
#include "ResamplingSdReader.h"
|
||
|
|
#include "interpolation.h"
|
||
|
|
#include "waveheaderparser.h"
|
||
|
|
bool ResamplingSdReader::isUsingSPI = false;
|
||
|
|
|
||
|
|
// read n samples into each buffer (1 buffer per channel)
|
||
|
|
unsigned int ResamplingSdReader::read(void **buf, uint16_t nsamples) {
|
||
|
|
if (!_playing) return 0;
|
||
|
|
|
||
|
|
int16_t *index[_numChannels];
|
||
|
|
unsigned int count = 0;
|
||
|
|
for (int channel=0; channel < _numChannels; channel++) {
|
||
|
|
index[channel] = (int16_t*)buf[channel];
|
||
|
|
}
|
||
|
|
|
||
|
|
while (count < nsamples) {
|
||
|
|
|
||
|
|
for (int channel=0; channel < _numChannels; channel++) {
|
||
|
|
if (readNextValue(index[channel], channel)) {
|
||
|
|
if (channel == _numChannels - 1)
|
||
|
|
count++;
|
||
|
|
index[channel]++;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
// we have reached the end of the file
|
||
|
|
|
||
|
|
switch (_loopType) {
|
||
|
|
case looptype_repeat:
|
||
|
|
{
|
||
|
|
if (_playbackRate >= 0.0)
|
||
|
|
_bufferPosition = _loop_start;
|
||
|
|
else
|
||
|
|
_bufferPosition = _loop_finish / _numChannels - _numChannels;
|
||
|
|
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
case looptype_pingpong:
|
||
|
|
{
|
||
|
|
if (_playbackRate >= 0.0) {
|
||
|
|
_bufferPosition = _loop_finish / _numChannels - _numChannels;
|
||
|
|
//printf("switching to reverse playback...\n");
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
_bufferPosition = _header_offset;
|
||
|
|
//printf("switching to forward playback...\n");
|
||
|
|
}
|
||
|
|
_playbackRate = -_playbackRate;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
case looptype_none:
|
||
|
|
default:
|
||
|
|
{
|
||
|
|
//Serial.printf("end of loop...\n");
|
||
|
|
/* no looping - return the number of (resampled) bytes returned... */
|
||
|
|
_playing = false;
|
||
|
|
if (_sourceBuffer) _sourceBuffer->close();
|
||
|
|
_sourceBuffer = nullptr;
|
||
|
|
StopUsingSPI();
|
||
|
|
return count;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return count;
|
||
|
|
}
|
||
|
|
|
||
|
|
// read the sample value for given channel and store it at the location pointed to by the pointer 'value'
|
||
|
|
bool ResamplingSdReader::readNextValue(int16_t *value, uint16_t channel) {
|
||
|
|
|
||
|
|
if (_playbackRate >= 0 ) {
|
||
|
|
//forward playback
|
||
|
|
if (_bufferPosition >= _loop_finish )
|
||
|
|
return false;
|
||
|
|
|
||
|
|
} else if (_playbackRate < 0) {
|
||
|
|
// reverse playback
|
||
|
|
if (_bufferPosition < _header_offset)
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
newdigate::IndexableFile<128, 2> &sourceBuffer = (*_sourceBuffer);
|
||
|
|
|
||
|
|
int16_t result = sourceBuffer[_bufferPosition + channel];
|
||
|
|
if (_interpolationType == ResampleInterpolationType::resampleinterpolation_linear) {
|
||
|
|
|
||
|
|
double abs_remainder = abs(_remainder);
|
||
|
|
if (abs_remainder > 0.0) {
|
||
|
|
|
||
|
|
if (_playbackRate > 0) {
|
||
|
|
if (_remainder - _playbackRate < 0.0){
|
||
|
|
// we crossed over a whole number, make sure we update the samples for interpolation
|
||
|
|
|
||
|
|
if (_playbackRate > 1.0) {
|
||
|
|
// need to update last sample
|
||
|
|
_interpolationPoints[channel][1].y = sourceBuffer[_bufferPosition-_numChannels];
|
||
|
|
}
|
||
|
|
|
||
|
|
_interpolationPoints[channel][0].y = _interpolationPoints[channel][1].y;
|
||
|
|
_interpolationPoints[channel][1].y = result;
|
||
|
|
if (_numInterpolationPoints < 2)
|
||
|
|
_numInterpolationPoints++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (_playbackRate < 0) {
|
||
|
|
if (_remainder - _playbackRate > 0.0){
|
||
|
|
// we crossed over a whole number, make sure we update the samples for interpolation
|
||
|
|
|
||
|
|
if (_playbackRate < -1.0) {
|
||
|
|
// need to update last sample
|
||
|
|
_interpolationPoints[channel][1].y = sourceBuffer[_bufferPosition+_numChannels];
|
||
|
|
}
|
||
|
|
|
||
|
|
_interpolationPoints[channel][0].y = _interpolationPoints[channel][1].y;
|
||
|
|
_interpolationPoints[channel][1].y = result;
|
||
|
|
if (_numInterpolationPoints < 2)
|
||
|
|
_numInterpolationPoints++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (_numInterpolationPoints > 1) {
|
||
|
|
result = abs_remainder * _interpolationPoints[channel][1].y + (1.0 - abs_remainder) * _interpolationPoints[channel][0].y;
|
||
|
|
//Serial.printf("[%f]\n", interpolation);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
_interpolationPoints[channel][0].y = _interpolationPoints[channel][1].y;
|
||
|
|
_interpolationPoints[channel][1].y = result;
|
||
|
|
if (_numInterpolationPoints < 2)
|
||
|
|
_numInterpolationPoints++;
|
||
|
|
|
||
|
|
result =_interpolationPoints[channel][0].y;
|
||
|
|
//Serial.printf("%f\n", result);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (_interpolationType == ResampleInterpolationType::resampleinterpolation_quadratic) {
|
||
|
|
double abs_remainder = abs(_remainder);
|
||
|
|
if (abs_remainder > 0.0) {
|
||
|
|
if (_playbackRate > 0) {
|
||
|
|
if (_remainder - _playbackRate < 0.0){
|
||
|
|
// we crossed over a whole number, make sure we update the samples for interpolation
|
||
|
|
int numberOfSamplesToUpdate = - floor(_remainder - _playbackRate);
|
||
|
|
if (numberOfSamplesToUpdate > 4)
|
||
|
|
numberOfSamplesToUpdate = 4; // if playbackrate > 4, only need to pop last 4 samples
|
||
|
|
for (int i=numberOfSamplesToUpdate; i > 0; i--) {
|
||
|
|
_interpolationPoints[channel][0].y = _interpolationPoints[channel][1].y;
|
||
|
|
_interpolationPoints[channel][1].y = _interpolationPoints[channel][2].y;
|
||
|
|
_interpolationPoints[channel][2].y = _interpolationPoints[channel][3].y;
|
||
|
|
_interpolationPoints[channel][3].y = sourceBuffer[_bufferPosition-(i*_numChannels)+1+channel];
|
||
|
|
if (_numInterpolationPoints < 4) _numInterpolationPoints++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (_playbackRate < 0) {
|
||
|
|
if (_remainder - _playbackRate > 0.0){
|
||
|
|
// we crossed over a whole number, make sure we update the samples for interpolation
|
||
|
|
int numberOfSamplesToUpdate = ceil(_remainder - _playbackRate);
|
||
|
|
if (numberOfSamplesToUpdate > 4)
|
||
|
|
numberOfSamplesToUpdate = 4; // if playbackrate > 4, only need to pop last 4 samples
|
||
|
|
for (int i=numberOfSamplesToUpdate; i > 0; i--) {
|
||
|
|
_interpolationPoints[channel][0].y = _interpolationPoints[channel][1].y;
|
||
|
|
_interpolationPoints[channel][1].y = _interpolationPoints[channel][2].y;
|
||
|
|
_interpolationPoints[channel][2].y = _interpolationPoints[channel][3].y;
|
||
|
|
_interpolationPoints[channel][3].y = sourceBuffer[_bufferPosition+(i*_numChannels)-1+channel];
|
||
|
|
if (_numInterpolationPoints < 4) _numInterpolationPoints++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (_numInterpolationPoints >= 4) {
|
||
|
|
//int16_t interpolation = interpolate(_interpolationPoints, 1.0 + abs_remainder, 4);
|
||
|
|
int16_t interpolation
|
||
|
|
= fastinterpolate(
|
||
|
|
_interpolationPoints[channel][0].y,
|
||
|
|
_interpolationPoints[channel][1].y,
|
||
|
|
_interpolationPoints[channel][2].y,
|
||
|
|
_interpolationPoints[channel][3].y,
|
||
|
|
1.0 + abs_remainder);
|
||
|
|
result = interpolation;
|
||
|
|
//Serial.printf("[%f]\n", interpolation);
|
||
|
|
} else
|
||
|
|
result = 0;
|
||
|
|
} else {
|
||
|
|
_interpolationPoints[channel][0].y = _interpolationPoints[channel][1].y;
|
||
|
|
_interpolationPoints[channel][1].y = _interpolationPoints[channel][2].y;
|
||
|
|
_interpolationPoints[channel][2].y = _interpolationPoints[channel][3].y;
|
||
|
|
_interpolationPoints[channel][3].y = result;
|
||
|
|
if (_numInterpolationPoints < 4) {
|
||
|
|
_numInterpolationPoints++;
|
||
|
|
result = 0;
|
||
|
|
} else
|
||
|
|
result = _interpolationPoints[channel][1].y;
|
||
|
|
//Serial.printf("%f\n", result);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (channel == _numChannels - 1) {
|
||
|
|
_remainder += _playbackRate;
|
||
|
|
|
||
|
|
auto delta = static_cast<signed int>(_remainder);
|
||
|
|
_remainder -= static_cast<double>(delta);
|
||
|
|
_bufferPosition += (delta * _numChannels);
|
||
|
|
}
|
||
|
|
|
||
|
|
*value = result;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
void ResamplingSdReader::initializeInterpolationPoints(void) {
|
||
|
|
if (_numChannels < 0)
|
||
|
|
return;
|
||
|
|
|
||
|
|
deleteInterpolationPoints();
|
||
|
|
_interpolationPoints = new InterpolationData*[_numChannels];
|
||
|
|
for (int channel=0; channel < _numChannels; channel++) {
|
||
|
|
InterpolationData *interpolation = new InterpolationData[4];
|
||
|
|
interpolation[0].y = 0.0;
|
||
|
|
interpolation[1].y = 0.0;
|
||
|
|
interpolation[2].y = 0.0;
|
||
|
|
interpolation[3].y = 0.0;
|
||
|
|
_interpolationPoints[channel] = interpolation ;
|
||
|
|
}
|
||
|
|
_numInterpolationPointsChannels = _numChannels;
|
||
|
|
}
|
||
|
|
|
||
|
|
void ResamplingSdReader::deleteInterpolationPoints(void) {
|
||
|
|
if (!_interpolationPoints) return;
|
||
|
|
for (int i=0; i<_numInterpolationPointsChannels; i++) {
|
||
|
|
delete [] _interpolationPoints[i];
|
||
|
|
}
|
||
|
|
delete [] _interpolationPoints;
|
||
|
|
_interpolationPoints = nullptr;
|
||
|
|
_numInterpolationPointsChannels = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void ResamplingSdReader::begin(void)
|
||
|
|
{
|
||
|
|
if (_interpolationType != ResampleInterpolationType::resampleinterpolation_none) {
|
||
|
|
initializeInterpolationPoints();
|
||
|
|
}
|
||
|
|
_playing = false;
|
||
|
|
_bufferPosition = _header_offset;
|
||
|
|
_file_size = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ResamplingSdReader::playRaw(const char *filename, uint16_t numChannels) {
|
||
|
|
return play(filename, false, numChannels);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ResamplingSdReader::playWav(const char *filename) {
|
||
|
|
return play(filename, true);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ResamplingSdReader::play(const char *filename, bool isWave, uint16_t numChannelsIfRaw)
|
||
|
|
{
|
||
|
|
stop();
|
||
|
|
//system("echo ${PWD}");
|
||
|
|
if (!isWave) // if raw file, then hardcode the numChannels as per the parameter
|
||
|
|
setNumChannels(numChannelsIfRaw);
|
||
|
|
|
||
|
|
if (_sourceBuffer) {
|
||
|
|
//Serial.printf("closing %s\n", _filename);
|
||
|
|
__disable_irq();
|
||
|
|
_sourceBuffer->close();
|
||
|
|
__enable_irq();
|
||
|
|
}
|
||
|
|
if (_sourceBuffer) delete _sourceBuffer;
|
||
|
|
if (_filename) delete [] _filename;
|
||
|
|
_filename = new char[strlen(filename)+1] {0};
|
||
|
|
memcpy(_filename, filename, strlen(filename) + 1);
|
||
|
|
StartUsingSPI();
|
||
|
|
|
||
|
|
__disable_irq();
|
||
|
|
File file = SD.open(_filename);
|
||
|
|
__enable_irq();
|
||
|
|
|
||
|
|
if (!file) {
|
||
|
|
StopUsingSPI();
|
||
|
|
Serial.printf("Not able to open file: %s\n", _filename);
|
||
|
|
if (_filename) delete [] _filename;
|
||
|
|
_filename = nullptr;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
__disable_irq();
|
||
|
|
_file_size = file.size();
|
||
|
|
__enable_irq();
|
||
|
|
|
||
|
|
if (isWave) {
|
||
|
|
wav_header wav_header;
|
||
|
|
wav_data_header data_header;
|
||
|
|
|
||
|
|
WaveHeaderParser wavHeaderParser;
|
||
|
|
char buffer[36];
|
||
|
|
__disable_irq();
|
||
|
|
size_t bytesRead = file.read(buffer, 36);
|
||
|
|
|
||
|
|
__enable_irq();
|
||
|
|
wavHeaderParser.readWaveHeaderFromBuffer((const char *) buffer, wav_header);
|
||
|
|
if (wav_header.bit_depth != 16) {
|
||
|
|
Serial.printf("Needs 16 bit audio! Aborting.... (got %d)", wav_header.bit_depth);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
setNumChannels(wav_header.num_channels);
|
||
|
|
|
||
|
|
bytesRead = file.read(buffer, 8);
|
||
|
|
unsigned infoTagsSize;
|
||
|
|
if (!wavHeaderParser.readInfoTags((unsigned char *)buffer, 0, infoTagsSize))
|
||
|
|
{
|
||
|
|
Serial.println("Not able to read header! Aborting...");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
file.seek(36 + infoTagsSize);
|
||
|
|
bytesRead = file.read(buffer, 8);
|
||
|
|
|
||
|
|
if (!wavHeaderParser.readDataHeader((unsigned char *)buffer, 0, data_header)) {
|
||
|
|
Serial.println("Not able to read header! Aborting...");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
_header_offset = (44 + infoTagsSize) / 2;
|
||
|
|
_loop_finish = ((data_header.data_bytes) / 2) + _header_offset;
|
||
|
|
} else
|
||
|
|
_loop_finish = _file_size / 2;
|
||
|
|
|
||
|
|
__disable_irq();
|
||
|
|
file.close();
|
||
|
|
__enable_irq();
|
||
|
|
|
||
|
|
_sourceBuffer = new newdigate::IndexableFile<128, 2>(_filename);
|
||
|
|
_loop_start = _header_offset;
|
||
|
|
if (_file_size <= _header_offset * newdigate::IndexableFile<128, 2>::element_size) {
|
||
|
|
_playing = false;
|
||
|
|
if (_filename) delete [] _filename;
|
||
|
|
_filename = nullptr;
|
||
|
|
Serial.printf("Wave file contains no samples: %s\n", filename);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
reset();
|
||
|
|
_playing = true;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ResamplingSdReader::play()
|
||
|
|
{
|
||
|
|
stop();
|
||
|
|
reset();
|
||
|
|
_playing = true;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
void ResamplingSdReader::reset(){
|
||
|
|
initializeInterpolationPoints();
|
||
|
|
if (_playbackRate > 0.0) {
|
||
|
|
// forward playabck - set _file_offset to first audio block in file
|
||
|
|
_bufferPosition = _header_offset;
|
||
|
|
} else {
|
||
|
|
// reverse playback - forward _file_offset to last audio block in file
|
||
|
|
_bufferPosition = _loop_finish / _numChannels - _numChannels;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void ResamplingSdReader::stop()
|
||
|
|
{
|
||
|
|
if (_playing) {
|
||
|
|
__disable_irq();
|
||
|
|
_playing = false;
|
||
|
|
//_file.close();
|
||
|
|
__enable_irq();
|
||
|
|
//StopUsingSPI();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int ResamplingSdReader::available(void) {
|
||
|
|
return _playing;
|
||
|
|
}
|
||
|
|
|
||
|
|
void ResamplingSdReader::close(void) {
|
||
|
|
if (_playing)
|
||
|
|
stop();
|
||
|
|
if (_sourceBuffer) {
|
||
|
|
_sourceBuffer->close();
|
||
|
|
delete _sourceBuffer;
|
||
|
|
}
|
||
|
|
StopUsingSPI();
|
||
|
|
_sourceBuffer = nullptr;
|
||
|
|
//TODO: dispose _sourceBuffer properly
|
||
|
|
deleteInterpolationPoints();
|
||
|
|
}
|