LatAnalyze/utils/sample-fake.cpp

/*
 * make_fake_sample.cpp, part of LatAnalyze 3
 *
 * Copyright (C) 2013 - 2020 Antonin Portelli
 *
 * LatAnalyze 3 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * LatAnalyze 3 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with LatAnalyze 3.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <LatAnalyze/Io/Io.hpp>
#include <LatAnalyze/Core/OptParser.hpp>


using namespace std;
using namespace Latan;

int main(int argc, char *argv[])
{
    OptParser            opt;
    Index  nSample;
    double val, err;
    string outFileName;

    opt.addOption("r", "seed"      , OptParser::OptType::value,   true,
                    "random generator seed (default: random)"); 
    opt.addOption("", "help"      , OptParser::OptType::trigger, true,
                  "show this help message and exit");

    bool parsed = opt.parse(argc, argv);
    if (!parsed or (opt.getArgs().size() != 4) or opt.gotOption("help"))
    {
        cerr << "usage: " << argv[0];
        cerr << " <central value> <error> <nSample> <output file>" << endl;
        cerr << endl << "Possible options:" << endl << opt << endl;

        return EXIT_FAILURE;
    }

    val         = strTo<double>(argv[1]);
    err         = strTo<double>(argv[2]);
    nSample     = strTo<Index>(argv[3]);
    outFileName = argv[4];

    random_device         rd;
    SeedType              seed = (opt.gotOption("r")) ? opt.optionValue<SeedType>("r") : rd();
    mt19937               gen(seed);
    normal_distribution<> dis(val, err);
    DSample               res(nSample);

    FOR_STAT_ARRAY(res, s)
    {
        if (s == central)
        {
            res[s] = val;
        }
        else
        {
            res[s] = dis(gen);
        }
    }
    Io::save<DSample>(res, outFileName);

    return EXIT_SUCCESS;
}