TestSetGenerator.hpp

Go to the documentation of this file.

#pragma once

#include <utility>
#include <vector>
#include <boost/shared_ptr.hpp>
#include <boost/random.hpp>
#include <boost/random/variate_generator.hpp>
#include <boost/random/uniform_real.hpp>
#include <boost/random/mersenne_twister.hpp>
#include <boost/random/normal_distribution.hpp>

#include <Eigen/Geometry>

#include <mutex>

#include "common_type/ObjectSet.hpp"
#include "common_type/Tracks.hpp"

#include "ObjectSetValidator.hpp"
#include "ObjectRelation.hpp"
#include "ImplicitShapeModel.hpp"
#include "utility/LogHelper.hpp"

namespace ISM {

typedef std::map<std::string, std::vector<ISM::ObjectSetPtr> > TestSet;
typedef boost::mt19937                     ENG;    // Mersenne Twister
typedef boost::uniform_int<> DIST;   // Normal Distribution
typedef boost::variate_generator<ENG,DIST> GEN;    // Variate generator


class TestSetGenerator
{
        public:
                TestSetGenerator(const std::map<std::string, ISM::ObjectRelations> allObjectRelationsPerPattern,
                                ObjectSetValidatorPtr &objectSetValidator)
                        : mAllObjectRelationsPerPattern(allObjectRelationsPerPattern)
                        , mObjectSetValidator(objectSetValidator)
                {}

                std::pair<std::vector<ObjectSetPtr>, std::vector<ObjectSetPtr>> generateTestSets(const std::string &patternName,
                                const TracksPtr& tracks, const IsmPtr ism, unsigned int testSetCount);

        private:
                const std::map<std::string, ISM::ObjectRelations> mAllObjectRelationsPerPattern;
                ISM::ObjectSetValidatorPtr mObjectSetValidator;


                ObjectSetPtr generateRandomObjectSetFromTracks(const ISM::TracksPtr& allTracks,
                        const std::string& pattern, GEN &gen);

}; typedef boost::shared_ptr<TestSetGenerator> TestSetGeneratorPtr;

}