54 lines
1.4 KiB
C++
54 lines
1.4 KiB
C++
|
#include <iostream>
|
||
|
|
#include <fstream>
|
||
|
|
#include <iomanip>
|
||
|
|
#include <filesystem>
|
||
|
|
|
||
|
|
#include "AudioFile.h"
|
||
|
|
#include "json.hpp"
|
||
|
|
#include "FFT.hpp"
|
||
|
|
|
||
|
|
void PrintUsage();
|
||
|
|
|
||
|
|
int main(int argc, char** argv)
|
||
|
|
{
|
||
|
|
if (argc < 2) {
|
||
|
|
PrintUsage();
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
int numFiles = argc - 1;
|
||
|
|
for (int i = 1; i < argc; i++) {
|
||
|
|
AudioFile<double> audioFile;
|
||
|
|
audioFile.load(argv[i]);
|
||
|
|
|
||
|
|
std::string filename = std::filesystem::path(argv[i]).filename().string();
|
||
|
|
|
||
|
|
int sampleRate = audioFile.getSampleRate();
|
||
|
|
int numChannels = audioFile.getNumChannels();
|
||
|
|
|
||
|
|
nlohmann::json output;
|
||
|
|
for (int c = 1; c <= numChannels; c++) {
|
||
|
|
std::cout << "\rAnalyzing " << filename << "... Channel " << c << "/" << numChannels << " ";
|
||
|
|
std::vector<std::pair<double, double>> spectrum = FFT(audioFile.samples[c-1], sampleRate);
|
||
|
|
|
||
|
|
std::string chName = "channel_" + std::to_string(c);
|
||
|
|
output[chName] = nlohmann::json::array();
|
||
|
|
for (const std::pair<double, double>& pair : spectrum) {
|
||
|
|
output[chName].push_back({ {"freq", pair.first}, {"mag", pair.second } });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
std::ofstream ofs(std::filesystem::path(argv[i]).replace_extension("json"));
|
||
|
|
ofs << std::setw(4) << output << std::endl;
|
||
|
|
ofs.close();
|
||
|
|
std::cout << "\rAnalyzing " << filename << "... Done! " << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void PrintUsage()
|
||
|
|
{
|
||
|
|
std::cerr << "Usage: spectralyze file1 [file2...]" << std::endl;
|
||
|
|
}
|