dataMerger.cpp

Go to the documentation of this file.

#include <ISM/tools/DataMerger.hpp>

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <set>

#include <boost/program_options.hpp>

using namespace ISM;
using namespace std;
namespace po = boost::program_options;

int main (int argc, char** argv)
{
    bool mergeRecordings = false;
    bool mergeModels = false;

    po::options_description desc("Allowed options");
    desc.add_options()
        ("help,h", "produce help message")
        ("target-file,t", po::value<string>()->default_value("record.sqlite"), "target file")
    ;

    po::options_description required("Required");
    required.add_options()
        ("source-file,s", po::value<vector<string> >(), "source file (precede each path by -s)")
    ;

    po::options_description atLeastOne("At least one of these");
    atLeastOne.add_options()
        ("merge-recording,r", po::bool_switch(&mergeRecordings), "merge recordings")
        ("merge-models,m", po::bool_switch(&mergeModels), "merge models")
    ;

    desc.add(required);
    desc.add(atLeastOne);

    po::variables_map vm;
    po::store(po::parse_command_line(argc, argv, desc), vm);
    po::notify(vm);

    if (vm.count("help") || !vm.count("target-file") || !vm.count("source-file") || (!mergeRecordings && !mergeModels))
    {
        cout << desc << "\n";
        return 1;
    }

    bool first = true;
    vector<string> sourceFiles = vm["source-file"].as<vector<string> >();
    string targetFile = vm["target-file"].as<string>();

    cout<<"merging ";
    for (string& source : sourceFiles)
    {
        if (!first)
        {
            cout<<",";
        }
        else
        {
            first = false;
        }
        cout<<source;
    }
    cout<<" into "<<targetFile<<endl;
    cout<<"Type yes to continue"<<endl;
    string yes;
    cin>>yes;
    transform(yes.begin(), yes.end(), yes.begin(), ::tolower);
    if (yes != "yes")
    {
        cout<<"aborting"<<endl;
        return 1;
    }

    //actual merging process
    DataMerger data_merger;
    data_merger.merge(targetFile, sourceFiles,mergeRecordings, mergeModels);
}