voice_activity_detection.cpp

/**
 * Copyright 2017 Google Inc. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <iostream>
#include <fstream>
#include <string>

#include <apiai/vad/VADState.h>
#include <apiai/vad/VoiceActivityDetector.h>

#include <unistd.h>

#include <memory>



VADState analyseAudioFile() {
    char *cwd = getcwd(NULL, 0);
    std::string filePath(cwd);
    free(cwd);
    cwd = NULL;

    filePath.append("/../tests/vad/sounds/count-hammer.raw");

    std::cout << "Analysing: " << filePath << std::endl;

    std::ifstream fstream(filePath.c_str(), std::ifstream::binary);

    fstream.seekg(0, fstream.end);
    std::streamsize length = fstream.tellg();
    fstream.seekg(0, fstream.beg);

    VADState state = VADStateUnknown;
    ai::vad::VoiceActivityDetector detector;

    const size_t bufferSize = 160;

    std::shared_ptr<short> buffer(new short[bufferSize]);
    while (length > 0) {
        const size_t read = (bufferSize > length) ? length : bufferSize;

        fstream.read((char *)buffer.get(), read * sizeof(short));
        state = detector.analyseFrames((const short *)buffer.get(), read);
        if ((state != VADStateUnknown) && (state != VADStateInProgress)) {
            break;
        }

        length -= read;
    }

    std::cout << "Analysis has been finished with result: " << state << std::endl;

    return state;
}

int main(int argc, char *argv[]) {
    analyseAudioFile();

    return 0;
}