MovementDetector.cpp

#define GRT_DLL_EXPORTS
#include "MovementDetector.h"

GRT_BEGIN_NAMESPACE

MovementDetector::MovementDetector( const UINT numDimensions, const Float upperThreshold, const Float lowerThreshold, const Float gamma, const UINT searchTimeout ) : MLBase("MovementDetector")
{    
    this->numInputDimensions = numDimensions;
    this->numOutputDimensions = 1;
    this->upperThreshold = upperThreshold;
    this->lowerThreshold = lowerThreshold;
    this->gamma = gamma;
    this->searchTimeout = searchTimeout;
    this->trained = true;
    
    reset();
}

MovementDetector::~MovementDetector(){
    
}

bool MovementDetector::predict_( VectorFloat &input ){
    
    movementDetected = false;
    noMovementDetected = false;
    
    if( !trained ){
        errorLog << "predict_(VectorFloat &input) - AdaBoost Model Not Trained!" << std::endl;
        return false;
    }
    
    if( input.size() != numInputDimensions ){
        errorLog << "predict_(VectorFloat &input) - The size of the input vector (" << input.size() << ") does not match the num features in the model (" << numInputDimensions << std::endl;
        return false;
    }
    
    //Compute the movement index, unless we are in the first sample
    Float x = 0;
    if( !firstSample ){
        for(UINT n=0; n<numInputDimensions; n++){
            x += SQR( input[n] - lastSample[n] );
        }
        movementIndex = (movementIndex*gamma) + sqrt( x );
    }
    
    //Flag that this is not the first sample and store the input for the next prediction
    firstSample = false;
    lastSample = input;
    
    switch( state ){
        case SEARCHING_FOR_MOVEMENT:
        if( movementIndex >= upperThreshold ){
            movementDetected = true;
            state = SEARCHING_FOR_NO_MOVEMENT;
        }
        break;
        case SEARCHING_FOR_NO_MOVEMENT:
        if( movementIndex < lowerThreshold ){
            noMovementDetected = true;
            state = SEARCH_TIMEOUT;
            searchTimer.start();
        }
        break;
        case SEARCH_TIMEOUT:
        // searchTimeout is cast because of a C4018 warning on visual (signed/unsigned incompatibility)
        if( searchTimer.getMilliSeconds() >= (signed long)searchTimeout ){
            state = SEARCH_TIMEOUT;
            searchTimer.stop();
        }
        break;
    }
    
    
    return true;
}

bool MovementDetector::clear(){
    
    trained = false;
    lastSample.clear();
    
    reset();
    
    return true;
}

bool MovementDetector::reset(){
    
    state = SEARCH_TIMEOUT;
    searchTimer.start();
    firstSample = true;
    movementDetected = false;
    noMovementDetected = false;
    movementIndex = 0;
    
    return true;
}

bool MovementDetector::save( std::fstream &file ) const{
    
    //Write the header info
    file << "GRT_MOVEMENT_DETECTOR_MODEL_FILE_V1.0\n";
    
    //Write the base settings to the file
    if( !MLBase::saveBaseSettingsToFile(file) ){
        errorLog <<"save(fstream &file) - Failed to save ML base settings to file!" << std::endl;
        return false;
    }
    
    //Write the detector variables
    file << "SearchTimeout: " << searchTimeout << std::endl;
    file << "UpperThreshold: " << upperThreshold << std::endl;
    file << "LowerThreshold: " << lowerThreshold << std::endl;
    file << "Gamma: " << gamma << std::endl;
    
    return true;
}

bool MovementDetector::load( std::fstream &file ){
    
    clear();
    
    if(!file.is_open())
    {
        errorLog << "load(string filename) - Could not open file to load model!" << std::endl;
        return false;
    }
    
    std::string word;
    file >> word;
    
    //Write the header info
    if( word != "GRT_MOVEMENT_DETECTOR_MODEL_FILE_V1.0" ){
        errorLog <<"load(fstream &file) - Failed to read file header!" << std::endl;
        return false;
    }
    
    //Load the base settings from the file
    if( !MLBase::loadBaseSettingsFromFile(file) ){
        errorLog << "load(string filename) - Failed to load base settings from file!" << std::endl;
        return false;
    }
    
    file >> word;
    if( word != "SearchTimeout:" ){
        errorLog <<"load(fstream &file) - Failed to read SearchTimeout header!" << std::endl;
        return false;
    }
    file >> searchTimeout;
    
    file >> word;
    if( word != "UpperThreshold:" ){
        errorLog <<"load(fstream &file) - Failed to read UpperThreshold header!" << std::endl;
        return false;
    }
    file >> upperThreshold;
    
    file >> word;
    if( word != "LowerThreshold:" ){
        errorLog <<"load(fstream &file) - Failed to read LowerThreshold header!" << std::endl;
        return false;
    }
    file >> lowerThreshold;
    
    file >> word;
    if( word != "Gamma:" ){
        errorLog <<"load(fstream &file) - Failed to read Gamma header!" << std::endl;
        return false;
    }
    file >> gamma;
    
    return true;
}

Float MovementDetector::getUpperThreshold() const {
    return upperThreshold;
}

Float MovementDetector::getLowerThreshold() const {
    return lowerThreshold;
}

Float MovementDetector::getMovementIndex() const {
    return movementIndex;
}

Float MovementDetector::getGamma() const {
    return gamma;
}

bool MovementDetector::getMovementDetected() const {
    return movementDetected;
}

bool MovementDetector::getNoMovementDetect() const {
    return noMovementDetected;
}

UINT MovementDetector::getState() const {
    return state;
}

UINT MovementDetector::getSearchTimeout() const {
    return searchTimeout;
}

bool MovementDetector::setUpperThreshold(const Float upperThreshold) {
    this->upperThreshold = upperThreshold;
    return true;
}

bool MovementDetector::setLowerThreshold(const Float lowerThreshold) {
    this->lowerThreshold = lowerThreshold;
    return true;
}

bool MovementDetector::setGamma(const Float gamma) {
    this->gamma = gamma;
    return true;
}

bool MovementDetector::setSearchTimeout(const UINT searchTimeout){
    this->searchTimeout = searchTimeout;
    return true;
}

GRT_END_NAMESPACE