File size: 2,023 Bytes
b87af40 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
#include <string>
#include <vector>
#include "Wav.hpp"
struct SliceResult
{
std::vector<unsigned long long> SliceOffset;
std::vector<bool> SliceTag;
cutResult(std::vector<unsigned long long>&& O, std::vector<bool>&& T) :SliceOffset(O), SliceTag(T) {}
};
double getAvg(const short* start, const short* end)
{
const auto size = end - start + 1;
auto avg = (double)(*start);
for (auto i = 1; i < size; i++)
{
avg = avg + (abs((double)start[i]) - avg) / (double)(i + 1ull);
}
return avg;
}
inline SliceResult SliceWav(Wav& input, double threshold, unsigned long minLen, unsigned short frame_len, unsigned short frame_shift)
{
const auto header = input.getHeader();
if (header.Subchunk2Size < minLen * header.bytesPerSec)
return { {0,header.Subchunk2Size},{true} };
auto ptr = input.getData();
std::vector<unsigned long long> output;
std::vector<bool> tag;
auto n = (header.Subchunk2Size / frame_shift) - 2 * (frame_len / frame_shift);
unsigned long nn = 0;
bool cutTag = true;
output.emplace_back(0);
while (n--)
{
//if (nn > minLen * header.bytesPerSec)
if (cutTag)
{
const auto vol = abs(getAvg((short*)ptr, (short*)ptr + frame_len));
if (vol < threshold)
{
cutTag = false;
if (nn > minLen * header.bytesPerSec)
{
nn = 0;
output.emplace_back((ptr - input.getData()) + (frame_len / 2));
}
}
else
{
cutTag = true;
}
}
else
{
const auto vol = abs(getAvg((short*)ptr, (short*)ptr + frame_len));
if (vol < threshold)
{
cutTag = false;
}
else
{
cutTag = true;
if (nn > minLen * header.bytesPerSec)
{
nn = 0;
output.emplace_back((ptr - input.getData()) + (frame_len / 2));
}
}
}
nn += frame_shift;
ptr += frame_shift;
}
output.push_back(header.Subchunk2Size);
for (size_t i = 1; i < output.size(); i++)
{
tag.push_back(abs(getAvg((short*)(input.getData() + output[i - 1]), (short*)(input.getData() + output[i]))) > threshold);
}
return { std::move(output),std::move(tag) };
}
|