DataMerger.cpp

Go to the documentation of this file.

#include "DataMerger.hpp"
#include "../utility/TableHelper.hpp"

#include <set>

namespace ISM
{
void DataMerger::merge(const std::string& targetFile, const std::vector<std::string>& sourceFiles, bool mergeRecordings, bool mergeModels)
{
    TableHelper target(targetFile);
    for (const std::string& sourceFile : sourceFiles)
    {
        TableHelper source(sourceFile);
        std::cout << sourceFile << ":" << std::endl;

        if (mergeRecordings)
        {
            std::cout << "merging recordings" << std::endl;
            std::vector<std::string> patternNames = source.getRecordedPatternNames();
            std::cout << "found " << patternNames.size() << " patterns" << std::endl;
            for (const std::string& patternName : patternNames)
            {
                std::cout << "merge pattern " << patternName << std::endl;
                RecordedPatternPtr pattern = source.getRecordedPattern(patternName);
                for (const ObjectSetPtr& set : pattern->objectSets)
                {
                    target.insertRecordedObjectSet(set, patternName);
                    std::cout << ".";
                    std::cout.flush();
                }
                std::cout << "done" << std::endl;
            }
        }

        if (mergeModels)
        {
            std::cout << "merging models" << std::endl;
            std::set<std::string> objectTypes = source.getObjectTypes();
            std::cout << "found " << objectTypes.size() << " object types" << std::endl;
            ObjectTypeToVoteMap typeToVoteMap = source.getVoteSpecifiersForObjectTypes(objectTypes);
            std::set<std::string> patternNames;
            for (const std::pair<std::string, std::vector<VoteSpecifierPtr>>& typeToVoteVecPair : typeToVoteMap)
            {
                std::string type = typeToVoteVecPair.first;
                std::vector<VoteSpecifierPtr> voteVec = typeToVoteVecPair.second;
                std::cout << "found " << voteVec.size() << " votes for object type " << type << std::endl;
                for (VoteSpecifierPtr& vote : voteVec) {
                    patternNames.insert(vote->patternName);
                    target.insertModelVoteSpecifier(vote);
                    std::cout << ".";
                    std::cout.flush();
                }
                std::cout << "done" << std::endl;
            }

            PatternNameToPatternMap patternMap = source.getPatternDefinitionsByName(patternNames);
            std::cout << "found " << patternMap.size() << " patterns" << std::endl;
            for (const std::pair<std::string, PatternPtr>& patternNameToPattern : patternMap)
            {
                PatternPtr pattern = patternNameToPattern.second;
                target.upsertModelPattern(pattern->name, pattern->expectedMaxWeight);
                std::cout << ".";
                std::cout.flush();
            }
            std::cout << "done" << std::endl;
        }
    }
}
}