/*
 * ProtocolSetup.h
 *
 */

#ifndef PROTOCOLS_PROTOCOLSETUP_H_
#define PROTOCOLS_PROTOCOLSETUP_H_

#include "Networking/Player.h"

/**
 * Global setup for an arithmetic share type
 */
template<class T>
class ProtocolSetup
{
    typename T::mac_key_type mac_key;

public:
    /**
     * @param P communication instance (used for MAC generation if needed)
     * @param prime_length length of prime if computing modulo a prime
     * @param directory location to read MAC if needed
     */
    ProtocolSetup(Player& P, int prime_length = 0, string directory = "")
    {
        // initialize fields
        if (prime_length == 0)
            prime_length = T::clear::MAX_N_BITS;

        T::clear::init_default(prime_length);
        T::clear::next::init_default(prime_length, false);

        // must initialize MAC key for security of some protocols
        T::read_or_generate_mac_key(directory, P, mac_key);

        T::MAC_Check::setup(P);
    }

    /**
     * @param prime modulus for computation
     * @param P communication instance (used for MAC generation if needed)
     * @param directory location to read MAC if needed
     */
    ProtocolSetup(bigint prime, Player& P, string directory = "")
    {
        static_assert(T::clear::prime_field, "must use computation modulo a prime");

        T::clear::init_field(prime);
        T::clear::next::init_field(prime, false);

        // must initialize MAC key for security of some protocols
        T::read_or_generate_mac_key(directory, P, mac_key);

        T::MAC_Check::setup(P);
    }

    ~ProtocolSetup()
    {
        T::LivePrep::teardown();
        T::MAC_Check::teardown();
    }

    typename T::mac_key_type get_mac_key() const
    {
        return mac_key;
    }

    /**
     * Set how much preprocessing is produced at once.
     */
    static void set_batch_size(size_t batch_size)
    {
        OnlineOptions::singleton.batch_size = batch_size;
    }
};

/**
 * Global setup for a binary share type
 */
template<class T>
class BinaryProtocolSetup
{
    typename T::mac_key_type mac_key;

public:
    /**
     * @param P communication instance (used for MAC generation if needed)
     * @param directory location to read MAC if needed
     */
    BinaryProtocolSetup(Player& P, string directory = "")
    {
        T::part_type::open_type::init_field();
        T::mac_key_type::init_field();
        T::part_type::read_or_generate_mac_key(directory, P, mac_key);

        T::MAC_Check::setup(P);
    }

    ~BinaryProtocolSetup()
    {
        T::MAC_Check::teardown();
    }

    typename T::mac_key_type get_mac_key() const
    {
        return mac_key;
    }
};

/**
 * Global setup for an arithmetic share type and the corresponding binary one
 */
template<class T>
class MixedProtocolSetup : public ProtocolSetup<T>
{
public:
    BinaryProtocolSetup<typename T::bit_type> binary;

    /**
     * @param P communication instance (used for MAC generation if needed)
     * @param prime_length length of prime if computing modulo a prime
     * @param directory location to read MAC if needed
     */
    MixedProtocolSetup(Player& P, int prime_length = 0, string directory = "") :
            ProtocolSetup<T>(P, prime_length, directory), binary(P, directory)
    {
    }
};

#endif /* PROTOCOLS_PROTOCOLSETUP_H_ */