/*
 * RingOptions.cpp
 *
 */

#include "RingOptions.h"
#include "BaseMachine.h"

#include <iostream>
using namespace std;

RingOptions::RingOptions(ez::ezOptionParser& opt, int argc, const char** argv)
{
    opt.add(
        "64", // Default.
        0, // Required?
        1, // Number of args expected.
        0, // Delimiter if expecting multiple args.
        "Number of integer bits (default: 64)", // Help description.
        "-R", // Flag token.
        "--ring" // Flag token.
    );
    opt.parse(argc, argv);
    opt.get("-R")->getInt(R);
    R_is_set = opt.isSet("-R");
    opt.resetArgs();
    if (R_is_set)
        cerr << "Trying to run " << R << "-bit computation" << endl;
}

int RingOptions::ring_size_from_opts_or_schedule(string progname)
{
    if (BaseMachine::prime_from_schedule(progname)
            or BaseMachine::prime_length_from_schedule(progname))
    {
        cerr << "Program was compiled for a prime field, "
                << "not a ring modulo a power of two. "
                << "Use './compile.py -R <size>'." << endl;
        exit(1);
    }

    int r = BaseMachine::ring_size_from_schedule(progname);
    if (R_is_set)
    {
        if (r and r != R)
            cerr << "Different -R option in compilation and run-time: " << r
                    << " vs " << R << endl;
        return R;
    }
    if (r == 0)
        r = R;
    cerr << "Trying to run " << r << "-bit computation" << endl;
    return r;
}