sparse_form.hpp

#pragma once
#include "barretenberg/common/throw_or_abort.hpp"
#include <cstddef>
#include <cstdint>
#include <iostream>
#include <vector>
 
#include "../uint256/uint256.hpp"
 
namespace numeric {
 
inline std::vector<uint64_t> slice_input(const uint256_t& input, const uint64_t base, const size_t num_slices)
{
    uint256_t target = input;
    std::vector<uint64_t> slices;
    if (num_slices > 0) {
        for (size_t i = 0; i < num_slices; ++i) {
            slices.push_back((target % base).data[0]);
            target /= base;
        }
    } else {
        while (target > 0) {
            slices.push_back((target % base).data[0]);
            target /= base;
        }
    }
    return slices;
}
 
inline std::vector<uint64_t> slice_input_using_variable_bases(const uint256_t& input,
                                                              const std::vector<uint64_t>& bases)
{
    uint256_t target = input;
    std::vector<uint64_t> slices;
    for (size_t i = 0; i < bases.size(); ++i) {
        if (target >= bases[i] && i == bases.size() - 1) {
            throw_or_abort(format("Last key slice greater than ", bases[i]));
        }
        slices.push_back((target % bases[i]).data[0]);
        target /= bases[i];
    }
    return slices;
}
 
template <uint64_t base, uint64_t num_slices> constexpr std::array<uint256_t, num_slices> get_base_powers()
{
    std::array<uint256_t, num_slices> output{};
    output[0] = 1;
    for (size_t i = 1; i < num_slices; ++i) {
        output[i] = output[i - 1] * base;
    }
    return output;
}
 
template <uint64_t base> constexpr uint256_t map_into_sparse_form(const uint64_t input)
{
    uint256_t out = 0UL;
    auto converted = input;
 
    constexpr auto base_powers = get_base_powers<base, 32>();
    for (size_t i = 0; i < 32; ++i) {
        uint64_t sparse_bit = ((converted >> i) & 1U);
        if (sparse_bit) {
            out += base_powers[i];
        }
    }
    return out;
}
 
template <uint64_t base> constexpr uint64_t map_from_sparse_form(const uint256_t& input)
{
    uint256_t target = input;
    uint64_t output = 0;
 
    constexpr auto bases = get_base_powers<base, 32>();
 
    for (uint64_t i = 0; i < 32; ++i) {
        const auto& base_power = bases[static_cast<size_t>(31 - i)];
        uint256_t prev_threshold = 0;
        for (uint64_t j = 1; j < base + 1; ++j) {
            const auto threshold = prev_threshold + base_power;
            if (target < threshold) {
                bool bit = ((j - 1) & 1);
                if (bit) {
                    output += (1ULL << (31ULL - i));
                }
                if (j > 1) {
                    target -= (prev_threshold);
                }
                break;
            }
            prev_threshold = threshold;
        }
    }
 
    return output;
}
 
template <uint64_t base, size_t num_bits> class sparse_int {
  public:
    sparse_int(const uint64_t input = 0)
        : value(input)
    {
        for (size_t i = 0; i < num_bits; ++i) {
            const uint64_t bit = (input >> i) & 1U;
            limbs[i] = bit;
        }
    }
    sparse_int(const sparse_int& other) noexcept = default;
    sparse_int(sparse_int&& other) noexcept = default;
    sparse_int& operator=(const sparse_int& other) noexcept = default;
    sparse_int& operator=(sparse_int&& other) noexcept = default;
    ~sparse_int() noexcept = default;
 
    sparse_int operator+(const sparse_int& other) const
    {
        sparse_int result(*this);
        for (size_t i = 0; i < num_bits - 1; ++i) {
            result.limbs[i] += other.limbs[i];
            if (result.limbs[i] >= base) {
                result.limbs[i] -= base;
                ++result.limbs[i + 1];
            }
        }
        result.limbs[num_bits - 1] += other.limbs[num_bits - 1];
        result.limbs[num_bits - 1] %= base;
        result.value += other.value;
        return result;
    };
 
    sparse_int operator+=(const sparse_int& other)
    {
        *this = *this + other;
        return *this;
    }
 
    [[nodiscard]] uint64_t get_value() const { return value; }
 
    [[nodiscard]] uint64_t get_sparse_value() const
    {
        uint64_t result = 0;
        for (size_t i = num_bits - 1; i < num_bits; --i) {
            result *= base;
            result += limbs[i];
        }
        return result;
    }
 
    const std::array<uint64_t, num_bits>& get_limbs() const { return limbs; }
 
  private:
    std::array<uint64_t, num_bits> limbs;
    uint64_t value;
    uint64_t sparse_value;
};
 
} // namespace numeric