prac/heap.hpp

#ifndef __HEAP_HPP__
#define __HEAP_HPP__

#include "types.hpp"
#include "mpcio.hpp"
#include "coroutine.hpp"
#include "options.hpp"
#include "mpcops.hpp"

class MinHeap {
    private: 
    Duoram < RegAS > oram;
    size_t MAX_SIZE;
    size_t num_items;

    public: 
    
    MinHeap(int player_num, size_t size) : oram(player_num, size){

    };

    void set_num_items (size_t n) {num_items = n;}

    // The extractmin protocol returns the minimum element (the root), removes it
    // and restores the heap property
    // and takes in a boolean parameter to decide if the basic or the optimized version needs to be run
    RegAS extract_min(MPCIO &mpcio, MPCTIO tio, yield_t & yield, int is_optimized);
    
    // Intializes the heap array with 0x7fffffffffffff
    void init(MPCTIO tio, yield_t & yield);
    
    // This function simply inits a heap with values 1,2,...,n
    // We use this function only to setup our heap 
    // to do timing experiments on insert and extractmins
    void init(MPCTIO tio, yield_t & yield, size_t which_init);
    
    // The Basic Insert Protocol
    // Takes in the additive share of the value to be inserted
    // And adds the the value into the heap while keeping the heap property intact
    int insert(MPCTIO tio, yield_t & yield, RegAS val);
    
    // The Optimized Insert Protocol
    // Takes in the additive share of the value to be inserted
    // And adds the the value into the heap while keeping the heap property intact
    void insert_optimized(MPCTIO tio, yield_t & yield, RegAS val);
    
    // Note: This function is intended for testing purposes only.
    // The purpose of this function is to verify that the heap property is satisfied.
    void verify_heap_property(MPCTIO tio, yield_t & yield);
    
    // Basic restore heap property at a secret shared index
    RegXS restore_heap_property(MPCIO &mpcio, MPCTIO tio, yield_t & yield, RegXS index);
    
    // Optimized restore heap property at a secret shared index
    std::pair<RegXS, RegBS> restore_heap_property_optimized(MPCTIO tio, yield_t & yield, RegXS index, size_t layer, typename Duoram<RegAS>::template OblivIndex<RegXS,3> oidx);
    
    // Restore heap property at an index in clear
    std::pair<RegXS, RegBS> restore_heap_property_at_explicit_index(MPCTIO tio, yield_t & yield,  size_t index);
    
    // Prints the current heap
    void print_heap(MPCTIO tio, yield_t & yield);
};

void Heap(MPCIO &mpcio,
   const PRACOptions &opts, char **args, int argc);
#endif