Bootstrapper.hpp

Go to the documentation of this file.

//                                                                            //
// Copyright (C) 2021-2023 Crypto Lab Inc.                                    //
//                                                                            //
// - This file is part of HEaaN homomorphic encryption library.               //
// - HEaaN cannot be copied and/or distributed without the express permission //
//  of Crypto Lab Inc.                                                        //
//                                                                            //
 
#pragma once
 
#include "Context.hpp"
#include "HEaaN/HEaaNExport.hpp"
 
namespace HEaaN {
 
class BootstrapperImpl;
class Ciphertext;
class Device;
class HomEvaluator;
 
class HEAAN_API Bootstrapper {
public:
    explicit Bootstrapper(const HomEvaluator &eval, const u64 log_slots);
    explicit Bootstrapper(const HomEvaluator &eval);
 
    explicit Bootstrapper(const HomEvaluator &eval,
                          const Context &context_sparse, const u64 log_slots);
    explicit Bootstrapper(const HomEvaluator &eval,
                          const Context &context_sparse);
 
    bool isBootstrapReady(const u64 log_slots) const;
 
    void makeBootConstants(const u64 log_slots);
 
    // loaded on CPU when performing makeBootConstants. To perform
    // bootstrap on Ciphertext on GPU memory, these constants should
    // be loaded on GPU. You may manually load the constants to reduce
    // latency of bootstrap. Otherwise, the first execution of bootstrap
    // on GPU will automatically load the constants on the device.
    void loadBootConstants(const u64 log_slots, const Device &device) const;
 
    u64 getLevelAfterFullSlotBootstrap() const;
 
    u64 getMinLevelForBootstrap() const;
 
    void bootstrap(const Ciphertext &ctxt, Ciphertext &ctxt_out,
                   bool is_complex = false) const;
 
    void bootstrap(const Ciphertext &ctxt, Ciphertext &ctxt_out_real,
                   Ciphertext &ctxt_out_imag) const;
 
    void bootstrapExtended(const Ciphertext &ctxt, Ciphertext &ctxt_out,
                           bool is_complex = false) const;
 
    void bootstrapExtended(const Ciphertext &ctxt, Ciphertext &ctxt_out_real,
                           Ciphertext &ctxt_out_imag) const;
 
private:
    std::shared_ptr<BootstrapperImpl> impl_;
};
 
} // namespace HEaaN