waveutils.h

/* Copyright 2017 ID R&D Inc. All Rights Reserved. */
 
#pragma once
 
#include <vector>
#include <string>
#include <fstream>
#include <voicesdk/core/config.h>
 
namespace voicesdk {
 
    struct AudioInfo {
 
        size_t samplesNum;
 
        size_t sampleRate;
 
        size_t channelsNum;
 
        AudioInfo() = default;
 
        AudioInfo(size_t samplesNum, size_t sampleRate, size_t channelsNum) :
            samplesNum(samplesNum),
            sampleRate(sampleRate),
            channelsNum(channelsNum)
        {}
 
        friend std::ostream &operator<<(std::ostream &os, const AudioInfo &obj) {
            os << "AudioInfo["
               << "samplesNum: " << obj.samplesNum << ", "
               << "sampleRate: " << obj.sampleRate << ", "
               << "channelsNum: " << obj.channelsNum << "]";
            return os;
        }
 
        bool operator==(const AudioInfo& other) const {
            return samplesNum  == other.samplesNum
                && sampleRate  == other.sampleRate
                && channelsNum == other.channelsNum;
        }
    };
 
    struct VOICE_SDK_API WavUtils {
 
        static std::vector<uint8_t> readAsPCM16Bytes(
            const std::string& wavFile,
            size_t& sampleRate);
 
        static std::vector<int16_t> readAsPCM16Samples(
            const std::string& wavFile,
            size_t& sampleRate);
 
        static std::vector<float> readAsFloatSamples(
            const std::string& wavFile,
            size_t& sampleRate);
 
        static std::vector<float> readAsFloatSamples16Bit(
            const std::string& wavFile,
            size_t& sampleRate);
 
        static std::vector<uint8_t> readAsPCM16Bytes(
            const uint8_t* data,
            size_t dataSize,
            size_t& sampleRate);
 
        static std::vector<int16_t> readAsPCM16Samples(
            const uint8_t* data,
            size_t dataSize,
            size_t& sampleRate);
 
        static std::vector<float> readAsFloatSamples(
            const uint8_t* data,
            size_t dataSize,
            size_t& sampleRate);
 
        static std::vector<float> readAsFloatSamples16Bit(
            const uint8_t* data,
            size_t dataSize,
            size_t& sampleRate);
 
        static AudioInfo getAudioInfo(const std::string& wavFile);
 
        static AudioInfo getAudioInfo(const uint8_t* data, size_t dataSize);
 
        WavUtils() = delete;
    };
}