native_crs_factory.hpp

Go to the documentation of this file.

#pragma once
#include "barretenberg/ecc/curves/bn254/bn254.hpp"
#include "barretenberg/srs/factories/crs_factory.hpp"
#include "barretenberg/srs/factories/mem_bn254_crs_factory.hpp"
#include "barretenberg/srs/factories/mem_grumpkin_crs_factory.hpp"
#include <filesystem>
#include <memory>
#ifndef NO_MULTITHREADING
#include <mutex>
#endif
 
namespace bb::srs::factories {
 
MemBn254CrsFactory init_bn254_crs(const std::filesystem::path& path,
                                  size_t dyadic_circuit_size,
                                  bool allow_download = true);
MemGrumpkinCrsFactory init_grumpkin_crs(const std::filesystem::path& path,
                                        size_t eccvm_dyadic_circuit_size,
                                        bool allow_download = true);
 
class NativeBn254CrsFactory : public CrsFactory<curve::BN254> {
  public:
    NativeBn254CrsFactory(const std::filesystem::path& path, bool allow_download = true)
        : path_(path)
        , allow_download_(allow_download)
    {}
    std::shared_ptr<Crs<curve::BN254>> get_crs(size_t degree) override
    {
#ifndef NO_MULTITHREADING
        std::lock_guard<std::mutex> lock(mutex_);
#endif
        if (degree > last_degree_ || mem_crs_ == nullptr) {
            mem_crs_ = std::make_shared<MemBn254CrsFactory>(init_bn254_crs(path_, degree, allow_download_));
            last_degree_ = degree;
        }
        return mem_crs_->get_crs(degree);
    }
 
  private:
    std::filesystem::path path_;
    bool allow_download_ = true;
    size_t last_degree_ = 0;
    std::shared_ptr<MemBn254CrsFactory> mem_crs_;
#ifndef NO_MULTITHREADING
    std::mutex mutex_;
#endif
};
 
class NativeGrumpkinCrsFactory : public CrsFactory<curve::Grumpkin> {
  public:
    NativeGrumpkinCrsFactory(const std::filesystem::path& path, bool allow_download = true)
        : path_(path)
        , allow_download_(allow_download)
    {}
 
    std::shared_ptr<Crs<curve::Grumpkin>> get_crs(size_t degree) override
    {
#ifndef NO_MULTITHREADING
        std::lock_guard<std::mutex> lock(mutex_);
#endif
        if (degree > last_degree_ || mem_crs_ == nullptr) {
            mem_crs_ = std::make_unique<MemGrumpkinCrsFactory>(init_grumpkin_crs(path_, degree, allow_download_));
            last_degree_ = degree;
        }
        return mem_crs_->get_crs(degree);
    }
 
  private:
    std::filesystem::path path_;
    bool allow_download_ = true;
    size_t last_degree_ = 0;
    std::unique_ptr<MemGrumpkinCrsFactory> mem_crs_;
#ifndef NO_MULTITHREADING
    std::mutex mutex_;
#endif
};
 
} // namespace bb::srs::factories