encode.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
// Copyright Contributors to the OpenQMC Project.
 
 
#pragma once
 
#include "gpu.h"
 
#include <cstdint>
 
namespace oqmc
{
 
struct EncodeKey
{
    int x; 
    int y; 
    int z; 
};
 
template <int XBits, int YBits, int ZBits>
OQMC_HOST_DEVICE inline std::uint16_t encodeBits16(EncodeKey key)
{
    constexpr auto sum = XBits + YBits + ZBits;
 
    static_assert(sum <= 16, "Precision sum must be equal or less than 16");
 
    constexpr auto maskX = (1 << XBits) - 1;
    constexpr auto maskY = (1 << YBits) - 1;
    constexpr auto maskZ = (1 << ZBits) - 1;
 
    constexpr auto offsetX = 0;
    constexpr auto offsetY = XBits;
    constexpr auto offsetZ = XBits + YBits;
 
    std::uint16_t value = 0;
    value |= (key.x & maskX) << offsetX;
    value |= (key.y & maskY) << offsetY;
    value |= (key.z & maskZ) << offsetZ;
 
    return value;
}
 
template <int XBits, int YBits, int ZBits>
OQMC_HOST_DEVICE inline EncodeKey decodeBits16(std::uint16_t value)
{
    constexpr auto sum = XBits + YBits + ZBits;
 
    static_assert(sum <= 16, "Precision sum must be equal or less than 16");
 
    constexpr auto maskX = (1 << XBits) - 1;
    constexpr auto maskY = (1 << YBits) - 1;
    constexpr auto maskZ = (1 << ZBits) - 1;
 
    constexpr auto offsetX = 0;
    constexpr auto offsetY = XBits;
    constexpr auto offsetZ = XBits + YBits;
 
    const auto x = (value >> offsetX) & maskX;
    const auto y = (value >> offsetY) & maskY;
    const auto z = (value >> offsetZ) & maskZ;
 
    return {x, y, z};
}
 
} // namespace oqmc