#include <iostream>

#include "BGN.hpp"

int main(void)
{
    BGN system;

    // const PrivateKey sk = system.get_private_key();

    Scalar test0(0);
    Scalar test1(1);
    Scalar test3(3);
    Scalar test5(5);
    Scalar decrypted0, decrypted1, decrypted3, decrypted5, decrypted6, decrypted10, decrypted15, decrypted30;

    std::cout << "Scalar test value (0): " << test0 << std::endl;
    // std::cout << "Scalar test value (1): " << test1 << std::endl;
    std::cout << "Scalar test value (3): " << test3 << std::endl;
    std::cout << "Scalar test value (5): " << test5 << std::endl;

    CurveBipoint encrypted0, encrypted1, encrypted3, encrypted6;
    // CurveBipoint pi_1_encrypted3, pi_1_curvegen_times3;
    TwistBipoint encrypted5, encrypted10;
    Quadripoint encrypted15, encrypted30;

    std::cout << "Performing encryptions" << std::endl;

    system.encrypt(encrypted0, test0);
    // system.encrypt(encrypted1, test1);
    system.encrypt(encrypted3, test3);
    system.encrypt(encrypted5, test5);

    std::cout << "Performing additions" << std::endl;

    encrypted6 = system.homomorphic_addition(encrypted3, encrypted3);
    encrypted10 = system.homomorphic_addition(encrypted5, encrypted5);

    std::cout << "Performing multiplication" << std::endl;

    encrypted15 = system.homomorphic_multiplication(encrypted3, encrypted5);

    std::cout << "Performing L2 addition" << std::endl;

    encrypted30 = system.homomorphic_addition(encrypted15, encrypted15);

    std::cout << "Performing decryptions" << std::endl;

    decrypted0  = system.decrypt(encrypted0);
    std::cout << "Scalar decrypted value (0):   " << decrypted0  << std::endl;
    // decrypted1  = system.decrypt(encrypted1);
    // std::cout << "Scalar decrypted value (1):   " << decrypted1  << std::endl;
    // decrypted1  = system.decrypt(encrypted1);
    // std::cout << "Scalar decrypted value (1):   " << decrypted1  << std::endl;
    decrypted3  = system.decrypt(encrypted3);
    std::cout << "Scalar decrypted value (3):   " << decrypted3  << std::endl;

    decrypted5  = system.decrypt(encrypted5);
    std::cout << "Scalar decrypted value (5):   " << decrypted5  << std::endl;

    decrypted6  = system.decrypt(encrypted6);
    std::cout << "Scalar decrypted value (6):   " << decrypted6  << std::endl;

    decrypted10 = system.decrypt(encrypted10);
    std::cout << "Scalar decrypted value (10): "  << decrypted10 << std::endl;

    decrypted15 = system.decrypt(encrypted15);
    std::cout << "Scalar decrypted value (15): "  << decrypted15 << std::endl;

    decrypted30 = system.decrypt(encrypted30);
    std::cout << "Scalar decrypted value (30): "  << decrypted30 << std::endl;

    return 0;
}