#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include "seal/base64.h"
#include "seal/bigpoly.h"
#include "seal/bigpolyarray.h"
#include "seal/biguint.h"
#include "seal/chooser.h"
#include "seal/ciphertext.h"
#include "seal/decryptor.h"
#include "seal/encoder.h"
#include "seal/encryptor.h"
#include "seal/encryptionparams.h"
#include "seal/evaluator.h"
#include "seal/keygenerator.h"
#include "seal/memorypoolhandle.h"
#include "seal/plaintext.h"
#include "seal/defaultparams.h"
#include "seal/publickey.h"
#include "seal/secretkey.h"
#include "seal/polycrt.h"

namespace py = pybind11;

using namespace pybind11::literals;
using namespace seal;
using namespace std;

// http://pybind11.readthedocs.io/en/stable/classes.html

template<class T>
py::tuple serialize(T &c) {
    std::stringstream output(std::ios::binary | std::ios::out);
    c.save(output);
    std::string cipherstr = output.str();
    std::string base64_encoded_cipher = base64_encode(reinterpret_cast<const unsigned char*>(cipherstr.c_str()), cipherstr.length());
    return py::make_tuple(base64_encoded_cipher);
}

template<class T>
T deserialize(py::tuple t) {
    if (t.size() != 1)
        throw std::runtime_error("(Pickle) Invalid input tuple!");
    T c = T();
    std::string cipherstr_encoded = t[0].cast<std::string>();
    std::string cipherstr_decoded = base64_decode(cipherstr_encoded);
    std::stringstream input(std::ios::binary | std::ios::in);
    input.str(cipherstr_decoded);
    c.load(input);
    return c;
}

PYBIND11_MODULE(seal, m) {

  py::class_<BigPoly>(m, "BigPoly")
    .def(py::init<>())
    .def(py::init<int, int>())
    .def(py::init<const std::string &>())
    .def(py::init<int, int, const std::string &>())
    .def(py::init<int, int, std::uint64_t *>())
    .def(py::init<const BigPoly &>())
    .def(py::init<BigPoly &>())
    .def("coeff_count", &BigPoly::coeff_count,
          "Returns the coefficient count for the BigPoly.")
    .def("to_string", &BigPoly::to_string,
         "Returns a human-readable string description of the BigPoly");

  py::class_<BigPolyArray>(m, "BigPolyArray")
    .def(py::init<>());

  py::class_<BigUInt>(m, "BigUInt")
    .def(py::init<>())
    .def("to_double", &BigUInt::to_double,
          "Returns the BigUInt value as a double. Note that precision may be lost during the conversion.")
    .def("significant_bit_count", (int (BigUInt::*)()) &BigUInt::significant_bit_count, "Returns the value of the current SmallModulus");

  py::class_<ChooserEncoder>(m, "ChooserEncoder")
    .def(py::init<>())
    .def(py::init<std::uint64_t>())
    .def("encode", (ChooserPoly (ChooserEncoder::*)(std::uint64_t)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("encode", (void (ChooserEncoder::*)(std::uint64_t, ChooserPoly &)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("encode", (ChooserPoly (ChooserEncoder::*)(std::int64_t)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("encode", (void (ChooserEncoder::*)(std::int64_t, ChooserPoly &)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("encode", (ChooserPoly (ChooserEncoder::*)(BigUInt)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("encode", (void (ChooserEncoder::*)(BigUInt, ChooserPoly &)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("encode", (ChooserPoly (ChooserEncoder::*)(std::uint32_t)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("encode", (ChooserPoly (ChooserEncoder::*)(std::int32_t)) &ChooserEncoder::encode,
        "Encodes a number (represented by std::uint64_t) into a ChooserPoly object")
    .def("base", (std::uint64_t (ChooserEncoder::*)()) &ChooserEncoder::base,
        "Returns the base used for encoding");

  py::class_<ChooserEvaluator>(m, "ChooserEvaluator")
    .def(py::init<>())
    .def(py::init<const MemoryPoolHandle &>())
    .def("multiply_many", (ChooserPoly (ChooserEvaluator::*)(const std::vector<ChooserPoly> &,
        int)) &ChooserEvaluator::multiply_many,
        "Performs an operation modeling Evaluator::multiply_many() on ChooserPoly objects")
    .def("add", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        const ChooserPoly &)) &ChooserEvaluator::add,
        "Performs an operation modeling Evaluator::add() on ChooserPoly objects")
    .def("add_many", (ChooserPoly (ChooserEvaluator::*)(const std::vector<ChooserPoly> &
        )) &ChooserEvaluator::add_many,
        "Performs an operation modeling Evaluator::add() on ChooserPoly objects")
    .def("sub", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        const ChooserPoly &)) &ChooserEvaluator::sub,
        "Performs an operation modeling Evaluator::sub() on ChooserPoly objects")
    .def("multiply", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        const ChooserPoly &)) &ChooserEvaluator::multiply,
        "Performs an operation modeling Evaluator::multiply() on ChooserPoly objects")
    .def("square", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &
        )) &ChooserEvaluator::square,
        "Performs an operation modeling Evaluator::square() on ChooserPoly objects")
    .def("relinearize", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        int)) &ChooserEvaluator::relinearize,
        "Performs an operation modeling Evaluator::relinearize() on ChooserPoly objects")
    .def("multiply_plain", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        int, std::uint64_t)) &ChooserEvaluator::multiply_plain,
        "Performs an operation modeling Evaluator::multiply_plain() on ChooserPoly objects")
    .def("multiply_plain", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        const ChooserPoly &)) &ChooserEvaluator::multiply_plain,
        "Performs an operation modeling Evaluator::multiply_plain() on ChooserPoly objects")
    .def("add_plain", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        int, std::uint64_t)) &ChooserEvaluator::add_plain,
        "Performs an operation modeling Evaluator::add_plain() on ChooserPoly objects")
    .def("add_plain", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        const ChooserPoly &)) &ChooserEvaluator::add_plain,
        "Performs an operation modeling Evaluator::add_plain() on ChooserPoly objects")
    .def("sub_plain", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        int, std::uint64_t)) &ChooserEvaluator::sub_plain,
        "Performs an operation modeling Evaluator::sub_plain() on ChooserPoly objects")
    .def("sub_plain", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        const ChooserPoly &)) &ChooserEvaluator::sub_plain,
        "Performs an operation modeling Evaluator::sub_plain() on ChooserPoly objects")
    .def("exponentiate", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &,
        std::uint64_t, int)) &ChooserEvaluator::exponentiate,
        "Performs an operation modeling Evaluator::exponentiate() on ChooserPoly objects")
    .def("negate", (ChooserPoly (ChooserEvaluator::*)(const ChooserPoly &)) &ChooserEvaluator::negate,
        "Performs an operation modeling Evaluator::negate() on ChooserPoly objects")
    .def("select_parameters", (bool (ChooserEvaluator::*)(const std::vector<ChooserPoly> &,
        int, EncryptionParameters &)) &ChooserEvaluator::select_parameters,
        "Provides the user with optimized encryption parameters")
    .def("select_parameters", (bool (ChooserEvaluator::*)(const std::vector<ChooserPoly> &,
        int, double, const std::map<int, std::vector<SmallModulus> > &, EncryptionParameters &)) &ChooserEvaluator::select_parameters,
        "Provides the user with optimized encryption parameters");

  py::class_<ChooserPoly>(m, "ChooserPoly")
    .def(py::init<>())
    .def(py::init<int, std::uint64_t>())
    .def(py::init<const ChooserPoly &>())
    /*.def("max_coeff_count", (int (ChooserPoly::*)()) &ChooserPoly::max_coeff_count,
        "Returns the upper bound on the number of non-zero coefficients")*/
    .def("max_coeff_count", (int & (ChooserPoly::*)()) &ChooserPoly::max_coeff_count,
        "Returns a reference to the upper bound on the number of non-zero coefficients")
    .def("max_abs_value", (std::uint64_t & (ChooserPoly::*)()) &ChooserPoly::max_abs_value,
        "Returns a reference to the upper bound on the absolute value of coefficients")
    .def("test_parameters", (bool (ChooserPoly::*)(const EncryptionParameters &, int)) &ChooserPoly::test_parameters,
        "Determines whether given encryption parameters are large enough to support operations in the operation history of the current ChooserPoly")
    .def("simulate", (Simulation (ChooserPoly::*)(const EncryptionParameters &)) &ChooserPoly::simulate,
        "Simulates noise budget consumption")
    .def("reset", (void (ChooserPoly::*)()) &ChooserPoly::reset,
        "Sets the bounds on the degree and the absolute value of the coefficients of \
        the modeled plaintext polynomial to zero, and sets the operation history to null")
    .def("set_fresh", (void (ChooserPoly::*)()) &ChooserPoly::set_fresh,
        "Sets the operation history to that of a freshly encrypted ciphertext");

  py::class_<Ciphertext>(m, "Ciphertext")
    .def(py::init<>())
    .def(py::init<const Ciphertext &>())
    .def(py::init<const MemoryPoolHandle &>())
    .def(py::init<const EncryptionParameters &, const MemoryPoolHandle &>())
    .def(py::init<const EncryptionParameters &>())
    .def("reserve", (void (Ciphertext::*)(int)) &Ciphertext::reserve,
        "Allocates enough memory to accommodate the backing array of a ciphertext with given capacity")
    .def("reserve", (void (Ciphertext::*)(int, const MemoryPoolHandle &)) &Ciphertext::reserve,
        "Allocates enough memory to accommodate the backing array of a ciphertext with given capacity")
    .def("reserve", (void (Ciphertext::*)(const EncryptionParameters &, int)) &Ciphertext::reserve,
        "Allocates enough memory to accommodate the backing array of a ciphertext with given capacity")
    .def("reserve", (void (Ciphertext::*)(const EncryptionParameters &, int, const MemoryPoolHandle &)) &Ciphertext::reserve,
        "Allocates enough memory to accommodate the backing array of a ciphertext with given capacity")
    .def("size", &Ciphertext::size, "Returns the capacity of the allocation")
    .def(py::pickle(&serialize<Ciphertext>, &deserialize<Ciphertext>))
    .def("save", (void (Ciphertext::*)(std::string &)) &Ciphertext::python_save,
        "Saves Ciphertext object to file given filepath")
    .def("load", (void (Ciphertext::*)(std::string &)) &Ciphertext::python_load,
        "Loads Ciphertext object from file given filepath");

  py::class_<Decryptor>(m, "Decryptor")
    .def(py::init<const SEALContext &, const SecretKey &>())
    .def(py::init<const SEALContext &, const SecretKey &, const MemoryPoolHandle &>())
    .def("decrypt", (void (Decryptor::*)(const Ciphertext &, Plaintext &, const MemoryPoolHandle &)) &Decryptor::decrypt,
        "Decrypts a ciphertext and writes the result to a given destination.")
    .def("decrypt", (void (Decryptor::*)(const Ciphertext &, Plaintext &)) &Decryptor::decrypt,
        "Decrypts a ciphertext and writes the result to a given destination.")
    .def("invariant_noise_budget", (int (Decryptor::*)(const Ciphertext &))
        &Decryptor::invariant_noise_budget, "Returns noise budget")
    .def("invariant_noise_budget", (int (Decryptor::*)(const Ciphertext &, const MemoryPoolHandle &))
        &Decryptor::invariant_noise_budget, "Returns noise budget");

  py::class_<Encryptor>(m, "Encryptor")
    .def(py::init<const SEALContext &, const PublicKey &>())
    .def(py::init<const SEALContext &, const PublicKey &, const MemoryPoolHandle &>())
    .def(py::init<const Encryptor &>())
    .def(py::init<Encryptor &>())
    .def("encrypt", (void (Encryptor::*)(const Plaintext &, Ciphertext &,
        const MemoryPoolHandle &)) &Encryptor::encrypt,
        "Encrypts a plaintext and writes the result to a given destination")
    .def("encrypt", (void (Encryptor::*)(const Plaintext &, Ciphertext &)) &Encryptor::encrypt,
        "Encrypts a plaintext and writes the result to a given destination");

  py::class_<EncryptionParameters>(m, "EncryptionParameters")
    .def(py::init<>())
    .def(py::init<const EncryptionParameters &>())
    .def("plain_modulus", &EncryptionParameters::plain_modulus, "Returns the plaintext modulus")
    .def("poly_modulus", &EncryptionParameters::poly_modulus, "Returns the polynomial modulus")
    .def("set_coeff_modulus",
        (void (EncryptionParameters::*)(const std::vector<SmallModulus> &)) &EncryptionParameters::set_coeff_modulus,
        "Set coefficient modulus parameter")
    .def("set_plain_modulus",
        (void (EncryptionParameters::*)(const SmallModulus &)) &EncryptionParameters::set_plain_modulus,
        "Set plaintext modulus parameter")
    .def("set_plain_modulus",
        (void (EncryptionParameters::*)(std::uint64_t)) &EncryptionParameters::set_plain_modulus,
        "Set plaintext modulus parameter")
    .def("set_poly_modulus",
        (void (EncryptionParameters::*)(const BigPoly &)) &EncryptionParameters::set_poly_modulus,
        "Set polynomial modulus parameter")
    .def("set_poly_modulus",
        (void (EncryptionParameters::*)(const std::string &)) &EncryptionParameters::set_poly_modulus,
        "Set polynomial modulus parameter")
    .def(py::pickle(&serialize<EncryptionParameters>, &deserialize<EncryptionParameters>));

  py::class_<EncryptionParameterQualifiers>(m, "EncryptionParameterQuailifers");

  py::class_<EvaluationKeys>(m, "EvaluationKeys")
    .def(py::init<>())
    .def("decomposition_bit_count", &EvaluationKeys::decomposition_bit_count, "Returns the decomposition bit count");

  py::class_<Evaluator>(m, "Evaluator")
    .def(py::init<const SEALContext &>())
    .def(py::init<const SEALContext &, const MemoryPoolHandle &>())
    .def("square", (void (Evaluator::*)(Ciphertext &)) &Evaluator::square,
        "Squares a ciphertext")
    .def("square", (void (Evaluator::*)(Ciphertext &, const MemoryPoolHandle &)) &Evaluator::square,
        "Squares a ciphertext")
    .def("add_many", (void (Evaluator::*)(const std::vector<Ciphertext> &, Ciphertext &)) &Evaluator::add_many,
        "Adds together a vector of ciphertexts and stores the result in the destination parameter.")
    .def("add_plain", (void (Evaluator::*)(Ciphertext &, const Plaintext &)) &Evaluator::add_plain,
        "Adds a ciphertext and a plaintext.")
    .def("add_plain", (void (Evaluator::*)(const Ciphertext &, const Plaintext &, Ciphertext &))
        &Evaluator::add_plain, "Adds a ciphertext and a plaintext.")
    .def("sub_plain", (void (Evaluator::*)(Ciphertext &, const Plaintext &))
        &Evaluator::sub_plain, "Subtracts a plaintext from a ciphertext.")
    .def("sub_plain", (void (Evaluator::*)(const Ciphertext &, const Plaintext &, Ciphertext &))
        &Evaluator::sub_plain, "Subtracts a plaintext from a ciphertext.")
    .def("exponentiate", (void (Evaluator::*)(Ciphertext &, std::uint64_t,
        const EvaluationKeys &, const MemoryPoolHandle &))
        &Evaluator::exponentiate, "Exponentiates a ciphertext.")
    .def("exponentiate", (void (Evaluator::*)(Ciphertext &, std::uint64_t,
        const EvaluationKeys &))
        &Evaluator::exponentiate, "Exponentiates a ciphertext.")
    .def("exponentiate", (void (Evaluator::*)(const Ciphertext &, std::uint64_t,
        const EvaluationKeys &, Ciphertext &, const MemoryPoolHandle &))
        &Evaluator::exponentiate, "Exponentiates a ciphertext.")
    .def("exponentiate", (void (Evaluator::*)(const Ciphertext &, std::uint64_t,
        const EvaluationKeys &, Ciphertext &))
        &Evaluator::exponentiate, "Exponentiates a ciphertext.")
    .def("negate", (void (Evaluator::*)(Ciphertext &)) &Evaluator::negate,
        "Negates a ciphertext")
    .def("negate", (void (Evaluator::*)(const Ciphertext &, Ciphertext &)) &Evaluator::negate,
        "Negates a ciphertext and writes to a given destination")
    .def("add", (void (Evaluator::*)(const Ciphertext &, const Ciphertext &,
        Ciphertext &)) &Evaluator::add,
        "Adds two ciphertexts and writes to a given destination")
    .def("add", (void (Evaluator::*)(Ciphertext &, const Ciphertext &)) &Evaluator::add,
        "Adds two ciphertexts and writes output over the first ciphertext")
    .def("sub", (void (Evaluator::*)(Ciphertext &, const Ciphertext &)) &Evaluator::sub,
        "Subtracts two ciphertexts and writes output over the first ciphertext")
    .def("sub", (void (Evaluator::*)(const Ciphertext &, const Ciphertext &,
        Ciphertext &)) &Evaluator::sub,
        "Subtracts two ciphertexts and writes to a given destination")
    .def("multiply", (void (Evaluator::*)(Ciphertext &, const Ciphertext &,
        const MemoryPoolHandle &)) &Evaluator::multiply,
        "Multiplies two ciphertexts and writes output over the first ciphertext")
    .def("multiply", (void (Evaluator::*)(Ciphertext &, const Ciphertext &)) &Evaluator::multiply,
        "Multiplies two ciphertexts and writes output over the first ciphertext")
    .def("multiply", (void (Evaluator::*)(const Ciphertext &, const Ciphertext &,
        Ciphertext &, const MemoryPoolHandle &)) &Evaluator::multiply,
        "Multiplies two ciphertexts and writes to a given destination")
    .def("multiply", (void (Evaluator::*)(const Ciphertext &, const Ciphertext &,
        Ciphertext &)) &Evaluator::multiply,
        "Multiplies two ciphertexts and writes to a given destination")
    .def("multiply_plain", (void (Evaluator::*)(Ciphertext &, const Plaintext &,
        const MemoryPoolHandle &)) &Evaluator::multiply_plain,
        "Multiplies a ciphertext with a plaintext")
    .def("multiply_plain", (void (Evaluator::*)(Ciphertext &, const Plaintext &))
        &Evaluator::multiply_plain, "Multiplies a ciphertext with a plaintext")
    .def("multiply_plain", (void (Evaluator::*)(const Ciphertext &, const Plaintext &,
        Ciphertext &, const MemoryPoolHandle &)) &Evaluator::multiply_plain,
        "Multiplies a ciphertext with a plaintext and writes to a given destination")
    .def("multiply_plain", (void (Evaluator::*)(const Ciphertext &, const Plaintext &,
        Ciphertext &)) &Evaluator::multiply_plain,
        "Multiplies a ciphertext with a plaintext and writes to a given destination")
    .def("relinearize", (void (Evaluator::*)(Ciphertext &, const EvaluationKeys &,
        const MemoryPoolHandle &)) &Evaluator::relinearize, "Relinearizes a ciphertext")
    .def("relinearize", (void (Evaluator::*)(Ciphertext &, const EvaluationKeys &))
        &Evaluator::relinearize, "Relinearizes a ciphertext")
    .def("relinearize", (void (Evaluator::*)(const Ciphertext &, const EvaluationKeys &,
        Ciphertext &, const MemoryPoolHandle &)) &Evaluator::relinearize,
        "Relinearizes a ciphertext and writes to a given destination")
    .def("relinearize", (void (Evaluator::*)(const Ciphertext &, const EvaluationKeys &,
        Ciphertext &)) &Evaluator::relinearize,
        "Relinearizes a ciphertext and writes to a given destination")
    .def("rotate_rows", (void (Evaluator::*)(const Ciphertext &, int,
        const GaloisKeys &, Ciphertext &)) &Evaluator::rotate_rows,
        "Rotates plaintext matrix rows cyclically")
    .def("rotate_rows", (void (Evaluator::*)(const Ciphertext &, int,
        const GaloisKeys &, Ciphertext &, const MemoryPoolHandle &)) &Evaluator::rotate_rows,
        "Rotates plaintext matrix rows cyclically")
    .def("rotate_rows", (void (Evaluator::*)(Ciphertext &, int,
        const GaloisKeys &, const MemoryPoolHandle &)) &Evaluator::rotate_rows,
        "Rotates plaintext matrix rows cyclically")
    .def("rotate_rows", (void (Evaluator::*)(Ciphertext &, int,
        const GaloisKeys &)) &Evaluator::rotate_rows,
        "Rotates plaintext matrix rows cyclically")
    .def("rotate_columns", (void (Evaluator::*)(Ciphertext &,
        const GaloisKeys &, const MemoryPoolHandle &)) &Evaluator::rotate_columns,
        "Rotates plaintext matrix rows cyclically")
    .def("rotate_columns", (void (Evaluator::*)(Ciphertext &,
        const GaloisKeys &)) &Evaluator::rotate_columns,
        "Rotates plaintext matrix rows cyclically")
    .def("rotate_columns", (void (Evaluator::*)(const Ciphertext &, const GaloisKeys &,
        Ciphertext &, const MemoryPoolHandle &)) &Evaluator::rotate_columns,
        "Rotates plaintext matrix rows cyclically")
    .def("rotate_columns", (void (Evaluator::*)(const Ciphertext &, const GaloisKeys &,
        Ciphertext &)) &Evaluator::rotate_columns,
        "Rotates plaintext matrix rows cyclically");

  py::class_<FractionalEncoder>(m, "FractionalEncoder")
    .def(py::init<const SmallModulus &, const BigPoly &, int, int,
        std::uint64_t, const MemoryPoolHandle &>())
    .def(py::init<const SmallModulus &, const BigPoly &, int, int,
        std::uint64_t>())
    .def(py::init<const SmallModulus &, const BigPoly &, int, int,
        const MemoryPoolHandle &>())
    .def(py::init<const SmallModulus &, const BigPoly &, int, int>())
    .def(py::init<const FractionalEncoder &>())
    .def("encode", (Plaintext (FractionalEncoder::*)(double)) &FractionalEncoder::encode,
        "Encodes a double precision floating point number into a plaintext polynomial")
    .def("decode", (double (FractionalEncoder::*)(const Plaintext &)) &FractionalEncoder::decode,
        "Decodes a plaintext polynomial and returns the result as a double-precision floating-point number");

  py::class_<GaloisKeys>(m, "GaloisKeys")
    .def(py::init<>())
    .def(py::init<const GaloisKeys &>());

  py::class_<IntegerEncoder>(m, "IntegerEncoder")
    .def(py::init<const SmallModulus &, std::uint64_t, const MemoryPoolHandle &>())
    .def(py::init<const SmallModulus &, std::uint64_t>())
    .def(py::init<const SmallModulus &, const MemoryPoolHandle &>())
    .def(py::init<const SmallModulus &>())
    .def("encode", (Plaintext (IntegerEncoder::*)(std::uint64_t)) &IntegerEncoder::encode, "Encode integer")
    .def("encode", (void (IntegerEncoder::*)(std::uint64_t, Plaintext &)) &IntegerEncoder::encode, "Encode integer and store in given destination")
    .def("encode", (Plaintext (IntegerEncoder::*)(std::int64_t)) &IntegerEncoder::encode, "Encode integer")
    .def("encode", (void (IntegerEncoder::*)(std::int64_t, Plaintext &)) &IntegerEncoder::encode, "Encode integer and store in given destination")
    .def("encode", (Plaintext (IntegerEncoder::*)(const BigUInt &)) &IntegerEncoder::encode, "Encode integer")
    .def("encode", (void (IntegerEncoder::*)(const BigUInt &, Plaintext &)) &IntegerEncoder::encode, "Encode integer and store in given destination")
    .def("encode", (Plaintext (IntegerEncoder::*)(std::int32_t)) &IntegerEncoder::encode, "Encode integer")
    .def("encode", (Plaintext (IntegerEncoder::*)(std::uint32_t)) &IntegerEncoder::encode, "Encode integer")
    .def("encode", (void (IntegerEncoder::*)(std::int32_t, Plaintext &)) &IntegerEncoder::encode, "Encode integer and store in given destination")
    .def("encode", (void (IntegerEncoder::*)(std::uint32_t, Plaintext &)) &IntegerEncoder::encode, "Encode integer and store in given destination")
    .def("decode_biguint", (void (IntegerEncoder::*)(const Plaintext &, BigUInt &)) &IntegerEncoder::decode_biguint, "Decode a plaintext polynomial and store in a given destination")
    .def("decode_biguint", (BigUInt (IntegerEncoder::*)(const Plaintext &)) &IntegerEncoder::decode_biguint, "Decode a plaintext polynomial")
    .def("decode_int64", (std::int64_t (IntegerEncoder::*)(Plaintext &)) &IntegerEncoder::decode_int64, "Decode a plaintext polynomial")
    .def("decode_int32", (std::int32_t (IntegerEncoder::*)(Plaintext &)) &IntegerEncoder::decode_int32, "Decode a plaintext polynomial")
    .def("decode_uint64", (std::uint64_t (IntegerEncoder::*)(Plaintext &)) &IntegerEncoder::decode_uint64, "Decode a plaintext polynomial")
    .def("decode_uint32", (std::uint32_t (IntegerEncoder::*)(Plaintext &)) &IntegerEncoder::decode_uint32, "Decode a plaintext polynomial");
    //.def("decode_int32", &IntegerEncoder::decode_int32,
    //    "Decodes a plaintext polynomial and returns the result as std::uint32_t");

  py::class_<KeyGenerator>(m, "KeyGenerator")
    .def(py::init<const SEALContext &>())
    .def(py::init<const SEALContext &, const MemoryPoolHandle &>())
    .def(py::init<const SEALContext &, const SecretKey &, const PublicKey &, const MemoryPoolHandle &>())
    .def("generate_evaluation_keys", (void (KeyGenerator::*)(int, int,
        EvaluationKeys &)) &KeyGenerator::generate_evaluation_keys,
        "Generates the specified number of evaluation keys")
    .def("generate_evaluation_keys", (void (KeyGenerator::*)(int,
        EvaluationKeys &)) &KeyGenerator::generate_evaluation_keys,
        "Generates the specified number of evaluation keys")
    .def("generate_galois_keys", (void (KeyGenerator::*)(int,
        GaloisKeys &)) &KeyGenerator::generate_galois_keys,
        "Generates Galois keys")
    .def("public_key", &KeyGenerator::public_key, "Returns public key")
    .def("secret_key", &KeyGenerator::secret_key, "Returns secret key");

  py::class_<MemoryPoolHandle>(m, "MemoryPoolHandle")
    .def(py::init<>())
    .def(py::init<const MemoryPoolHandle &>())
    .def_static("New", &MemoryPoolHandle::New,
               "Returns a MemoryPoolHandle pointing to a new memory pool")
    .def_static("acquire_global", &MemoryPoolHandle::Global,
               "Returns a MemoryPoolHandle pointing to the global memory pool");

  py::class_<Plaintext>(m, "Plaintext")
     .def(py::init<>())
     .def(py::init<const BigPoly &>())
     .def(py::init<const std::string &, const MemoryPoolHandle &>())
     .def(py::init<const std::string &>())
     .def(py::init<int, int>())
     .def(py::init<int, int, const MemoryPoolHandle &>())
     .def(py::init<int, int, std::uint64_t *>())
     .def("significant_coeff_count", &Plaintext::significant_coeff_count,
        "Returns the significant coefficient count of the current plaintext polynomial")
     .def("to_string", &Plaintext::to_string, "Returns the plaintext as a formatted string")
     .def("coeff_count", &Plaintext::coeff_count, "Returns the coefficient count of the current plaintext polynomial")
     .def("coeff_at", &Plaintext::coeff_at, "Returns coefficient at a given index")
     .def(py::pickle(&serialize<Plaintext>, &deserialize<Plaintext> ))
     .def("save", (void (Plaintext::*)(std::string &)) &Plaintext::python_save,
        "Saves Plaintext object to file given filepath")
     .def("load", (void (Plaintext::*)(std::string &)) &Plaintext::python_load,
        "Loads Plaintext object from file given filepath");

  py::class_<PolyCRTBuilder>(m, "PolyCRTBuilder")
    .def(py::init<const SEALContext &, const MemoryPoolHandle &>())
    .def(py::init<const SEALContext &>())
    .def(py::init<const PolyCRTBuilder &>())
    .def("slot_count", (int (PolyCRTBuilder::*)())
        &PolyCRTBuilder::slot_count, "Returns the number of slots")
    .def("compose", (void (PolyCRTBuilder::*)(const std::vector<std::uint64_t> &, Plaintext &))
        &PolyCRTBuilder::compose, "Creates a SEAL plaintext from a given matrix")
    .def("compose", (void (PolyCRTBuilder::*)(const std::vector<std::int64_t> &, Plaintext &))
        &PolyCRTBuilder::compose, "Creates a SEAL plaintext from a given matrix")
    .def("compose", (void (PolyCRTBuilder::*)(Plaintext &, const MemoryPoolHandle &))
        &PolyCRTBuilder::compose, "Creates a SEAL plaintext from a given matrix")
    .def("compose", (void (PolyCRTBuilder::*)(Plaintext &))
        &PolyCRTBuilder::compose, "Creates a SEAL plaintext from a given matrix")

    /*Error with the commented-out below functions due to what appears to be a scope issue.
    Specifically, when a Python list is passed as the second argument, it's not actually modified by decompose().
    We can work around this temporarily using one extra line with the new coeff_at helper function that's been added to the plaintext.h file (pybound in this file).
    See the construction of the pod_result variable in the Batching with PolyCRTBuilder example for an example of this work around.*/

    /*.def("decompose", (void (PolyCRTBuilder::*)(const Plaintext &, std::vector<std::uint64_t> &,
        const MemoryPoolHandle &)) &PolyCRTBuilder::decompose,
        "Inverse of compose. This function unbatches a given SEAL plaintext")
    .def("decompose", (void (PolyCRTBuilder::*)(const Plaintext &, std::vector<std::uint64_t> &))
        &PolyCRTBuilder::decompose, "Inverse of compose. This function unbatches a given SEAL plaintext")
    .def("decompose", (void (PolyCRTBuilder::*)(const Plaintext &, std::vector<std::int64_t> &,
        const MemoryPoolHandle &)) &PolyCRTBuilder::decompose,
        "Inverse of compose. This function unbatches a given SEAL plaintext")
    .def("decompose", (void (PolyCRTBuilder::*)(const Plaintext &, std::vector<std::int64_t> &))
        &PolyCRTBuilder::decompose, "Inverse of compose. This function unbatches a given SEAL plaintext")*/

    .def("decompose", (void (PolyCRTBuilder::*)(Plaintext &, const MemoryPoolHandle &))
        &PolyCRTBuilder::decompose, "Inverse of compose. This function unbatches a given SEAL plaintext")
    .def("decompose", (void (PolyCRTBuilder::*)(Plaintext &))
        &PolyCRTBuilder::decompose, "Inverse of compose. This function unbatches a given SEAL plaintext");

  py::class_<PublicKey>(m, "PublicKey")
     .def(py::init<>())
     .def("save", (void (PublicKey::*)(std::string &)) &PublicKey::python_save,
        "Saves PublicKey object to file given filepath")
     .def("load", (void (PublicKey::*)(std::string &)) &PublicKey::python_load,
        "Loads PublicKey object from file given filepath")
     .def(py::pickle(&serialize<PublicKey>, &deserialize<PublicKey>));

  py::class_<SecretKey>(m, "SecretKey")
     .def(py::init<>())
     .def("save", (void (SecretKey::*)(std::string &)) &SecretKey::python_save,
        "Saves SecretKey object to file given filepath")
     .def("load", (void (SecretKey::*)(std::string &)) &SecretKey::python_load,
        "Loads PublicKey object from file given filepath")
     .def(py::pickle(&serialize<SecretKey>, &deserialize<SecretKey>));

  py::class_<SEALContext>(m, "SEALContext")
     .def(py::init<const EncryptionParameters &>())
     .def(py::init<const EncryptionParameters &, const MemoryPoolHandle &>())
     .def("parms", (const EncryptionParameters & (SEALContext::*)()) &SEALContext::parms,
        "Returns a constant reference to the underlying encryption parameters")
     .def("noise_standard_deviation", (double (SEALContext::*)()) &SEALContext::noise_standard_deviation,
        "Returns the maximum deviation of the noise distribution that was given in the encryption parameters")
     .def("total_coeff_modulus", (const BigUInt & (SEALContext::*)()) &SEALContext::total_coeff_modulus,
        "Returns a constant reference to a pre-computed product of all primes in the coefficient modulus.")
     .def("poly_modulus", (const BigPoly & (SEALContext::*)()) &SEALContext::poly_modulus, "Returns a constant reference to the polynomial modulus")
     .def("plain_modulus", (const SmallModulus & (SEALContext::*)()) &SEALContext::plain_modulus, "Returns a constant reference to the plaintext modulus")
     .def("qualifiers", (EncryptionParameterQualifiers (SEALContext::*)()) &SEALContext::qualifiers,
        "Returns a copy of EncryptionParameterQualifiers corresponding to the current encryption parameters")
        .def(py::pickle(
        [](const SEALContext &context) {
            EncryptionParameters parms_ = context.parms();
            return serialize<EncryptionParameters>(parms_);
        },
        [](py::tuple t) {
            /* Create a new C++ instance */
            EncryptionParameters parms_ = deserialize<EncryptionParameters>(t);
            SEALContext context(parms_);
            return context;
        }
    ));

  py::class_<SmallModulus>(m, "SmallModulus")
      .def(py::init<>())
      .def(py::init<std::uint64_t>())
      .def("value", (std::uint64_t (SmallModulus::*)()) &SmallModulus::value, "Returns the value of the current SmallModulus");

  m.def("coeff_modulus_192", &coeff_modulus_192, "Returns the default coefficients modulus for a given polynomial modulus degree.");
  m.def("coeff_modulus_128", &coeff_modulus_128, "Returns the default coefficients modulus for a given polynomial modulus degree.");
  m.def("dbc_max", &dbc_max, "Return dbc max value.");

}