Function ex_bits_code

include/rabitqlib/quantization/rabitq_impl.hpp:407–433

template <typename T, typename TP>
inline T ex_bits_code(
    const T* residual, size_t dim, size_t ex_bits, TP* ex_code, double t_const = -1
) {
    ConstRowMajorArrayMap<T> res_arr(residual, 1, dim);

    // get normalized abs residual for plus code
    RowMajorArray<T> abs_res = res_arr.rowwise().normalized().abs();

    // quantize data
    T ipnorm_inv = 1;
    if (t_const > 0) {
        ipnorm_inv = faster_quantize_ex(abs_res.data(), ex_code, dim, ex_bits, t_const);
    } else {
        ipnorm_inv = quantize_ex(abs_res.data(), ex_code, dim, ex_bits);
    }

    // revert codes for negative dims
    int32_t mask = (1 << ex_bits) - 1;
    for (size_t j = 0; j < dim; ++j) {
        if (res_arr.data()[j] < 0) {
            TP tmp = ex_code[j];
            ex_code[j] = (~tmp) & mask;
        }
    }

    return ipnorm_inv;
}

template <typename T, typename TP>
inline void ex_bits_code_with_factor(
    const T* data,