119 lines
3.5 KiB
C++
119 lines
3.5 KiB
C++
#include "wav.h"
|
|
|
|
#include <vector>
|
|
#include <cstring>
|
|
#include <fstream>
|
|
#include <stdexcept>
|
|
|
|
#include "../audio/audio.h"
|
|
|
|
bool is_big_endian() {
|
|
uint32_t ui32_v = 0x01020304;
|
|
char bytes[sizeof(uint32_t)];
|
|
std::memcpy(bytes, &ui32_v, sizeof(uint32_t));
|
|
return bytes[0] == 1;
|
|
}
|
|
|
|
std::int32_t convert_to_int(char* buffer, std::size_t len){
|
|
std::int32_t a = 0;
|
|
if (!is_big_endian()) {
|
|
std::memcpy(&a, buffer, len);
|
|
}
|
|
else {
|
|
for (std::size_t i = 0; i < len; ++i) {
|
|
reinterpret_cast<char*>(&a)[3 - i] = buffer[i];
|
|
}
|
|
}
|
|
return a;
|
|
}
|
|
|
|
audio::PCM* wav::load_pcm(const std::filesystem::path& file, bool headerOnly) {
|
|
std::ifstream in(file, std::ios::binary);
|
|
if(!in.is_open()){
|
|
throw std::runtime_error("could not to open file '"+file.u8string()+"'");
|
|
}
|
|
|
|
char buffer[4];
|
|
// the RIFF
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not to read RIFF");
|
|
}
|
|
if(std::strncmp(buffer, "RIFF", 4) != 0){
|
|
throw std::runtime_error("file is not a valid WAVE file (header doesn't begin with RIFF)");
|
|
}
|
|
// the size of the file
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not read size of file");
|
|
}
|
|
// the WAVE
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not to read WAVE");
|
|
}
|
|
if(std::strncmp(buffer, "WAVE", 4) != 0){
|
|
throw std::runtime_error("file is not a valid WAVE file (header doesn't contain WAVE)");
|
|
}
|
|
// "fmt/0"
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not read fmt/0");
|
|
}
|
|
// this is always 16, the size of the fmt data chunk
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not read the 16");
|
|
}
|
|
// PCM should be 1?
|
|
if(!in.read(buffer, 2)){
|
|
throw std::runtime_error("could not read PCM");
|
|
}
|
|
// the number of channels
|
|
if(!in.read(buffer, 2)){
|
|
throw std::runtime_error("could not read number of channels");
|
|
}
|
|
int channels = convert_to_int(buffer, 2);
|
|
// sample rate
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not read sample rate");
|
|
}
|
|
int sampleRate = convert_to_int(buffer, 4);
|
|
if (!in.read(buffer, 6)) {
|
|
throw std::runtime_error("could not to read WAV header");
|
|
}
|
|
|
|
// bitsPerSample
|
|
if(!in.read(buffer, 2)){
|
|
throw std::runtime_error("could not read bits per sample");
|
|
}
|
|
int bitsPerSample = convert_to_int(buffer, 2);
|
|
|
|
// data chunk header "data"
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not read data chunk header");
|
|
}
|
|
if(std::strncmp(buffer, "data", 4) != 0){
|
|
throw std::runtime_error("file is not a valid WAVE file (doesn't have 'data' tag)");
|
|
}
|
|
|
|
// size of data
|
|
if(!in.read(buffer, 4)){
|
|
throw std::runtime_error("could not read data size");
|
|
}
|
|
size_t size = convert_to_int(buffer, 4);
|
|
|
|
/* cannot be at the end of file */
|
|
if(in.eof()){
|
|
throw std::runtime_error("reached EOF on the file");
|
|
}
|
|
if(in.fail()){
|
|
throw std::runtime_error("fail state set on the file");
|
|
}
|
|
|
|
std::vector<char> data;
|
|
size_t totalSamples = size / (bitsPerSample / 8);
|
|
if (!headerOnly) {
|
|
data.resize(size);
|
|
if (!in.read(data.data(), size)) {
|
|
throw std::runtime_error("could not load wav data of '"+file.u8string()+"'");
|
|
}
|
|
}
|
|
return new audio::PCM(std::move(data), totalSamples, channels, bitsPerSample, sampleRate);
|
|
}
|