root/library/tests/square_mat_prep.cpp @ 467

#include "base/bdmbase.h"
#include "base/user_info.h"
#include "generator.h"
#include "additive_generator.h"
#include "size_generator.h"
#include <iostream>
#include <iomanip>
#include <memory>
#include <stdlib.h>

using namespace std;
using namespace bdm;
using namespace itpp;

UIREGISTER(size_generator);
UIREGISTER(additive_generator);

const char *generator_file_name = "generator.cfg";
const char *agenda_file_name = "agenda.cfg";
int agenda_length = 10;

int main(int argc, char const *argv[]) {
    RNG_randomize();

    bool unknown = false;
    int update_next = 0; // 1 generator file, 2 agenda file, 3 agenda length
    const char **param = argv + 1;
    while (*param && !unknown) {
        if (update_next) {
            if (update_next == 1) {
                generator_file_name = *param;
            } else if (update_next == 2) {
                agenda_file_name = *param;
            } else {
                int length = atoi(*param);
                if (length > 0) {
                    agenda_length = length;
                } else {
                    cerr << "invalid agenda length value ignored" << endl;
                }
            }

            update_next = 0;
        } else {
            if (!strcmp(*param, "-a")) {
                update_next = 2;
            } else if (!strcmp(*param, "-g")) {
                update_next = 1;
            } else if (!strcmp(*param, "-l")) {
                update_next = 3;
            } else {
                unknown = true;
            }
        }

        ++param;
    }

    if (unknown || update_next) {
        cerr << "usage: " << argv[0] << " [ -g generator.cfg ] [ -a agenda_output.cfg ] [ -l agenda_length ]" << endl;
    } else {
        Array<mat> mag(agenda_length);

        UIFile gspec(generator_file_name);
        auto_ptr<generator> gen(UI::build<generator>(gspec, "generator"));
        for (int i = 0; i < agenda_length; ++i) {
            mag(i) = gen->next();
        }

        UIFile fag;
        UI::save(mag, fag, "agenda");
        fag.save(agenda_file_name);
    }
}