nullifier_memory_tree.hpp

#pragma once
#include "../hash.hpp"
#include "../memory_tree.hpp"
#include "nullifier_leaf.hpp"
 
namespace proof_system::plonk {
namespace stdlib {
namespace merkle_tree {
 
using namespace barretenberg;
 
class NullifierMemoryTree : public MemoryTree {
 
  public:
    NullifierMemoryTree(size_t depth);
 
    using MemoryTree::get_hash_path;
    using MemoryTree::root;
    using MemoryTree::update_element;
 
    fr update_element(fr const& value);
 
    const std::vector<barretenberg::fr>& get_hashes() { return hashes_; }
    const WrappedNullifierLeaf get_leaf(size_t index)
    {
        return (index < leaves_.size()) ? leaves_[index] : WrappedNullifierLeaf::zero();
    }
    const std::vector<WrappedNullifierLeaf>& get_leaves() { return leaves_; }
 
  protected:
    using MemoryTree::depth_;
    using MemoryTree::hashes_;
    using MemoryTree::root_;
    using MemoryTree::total_size_;
    std::vector<WrappedNullifierLeaf> leaves_;
};
 
} // namespace merkle_tree
} // namespace stdlib
} // namespace proof_system::plonk