ObjectSetValidator.cpp

Go to the documentation of this file.

#include "ObjectSetValidator.hpp"
#include "utility/LogHelper.hpp"
#include "boost/functional/hash.hpp"

#include <ctime>
#include <sys/time.h>
#include <chrono>

namespace  ISM {

        std::pair<bool, double> ObjectSetValidator::isSetValid(const ObjectSetPtr &testSet,
                        const std::string &patternName)
        {
                struct timeval start;
                struct timeval end;

                gettimeofday(&start, NULL);
                std::vector<RecognitionResultPtr> results = mRecognizer->recognizePattern(testSet);
                gettimeofday(&end, NULL);

                double recognitionRuntime, seconds, useconds;
                seconds = end.tv_sec - start.tv_sec;
                useconds = end.tv_usec - start.tv_usec;
                recognitionRuntime = seconds + useconds / 1000000;

                for (size_t i = 0; i < results.size(); ++i)
                {
                        if (results[i]->patternName == patternName)
                        {
                                bool isValid = results[i]->confidence >= mConfidenceThreshold;
                                return std::make_pair(isValid, recognitionRuntime);
                        }
                }

                return std::make_pair(false, recognitionRuntime);
        }

        void ObjectSetValidator::setISM(const IsmPtr ism)
        {
                this->mRecognizer->clearData();
                this->mRecognizer->setVoteSpecifiersPerObject(ism->voteSpecifiersPerObject);
                this->mRecognizer->setObjectTypes(ism->objectTypes);
                this->mRecognizer->setPatternDefinitions(ism->patternDefinitions);
                this->mRecognizer->arrangePatternsAccordingToTreeHeight();
        }
}