bntables.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
// Copyright Contributors to the OpenQMC Project.
 
 
#pragma once
 
#include "encode.h"
#include "gpu.h"
 
#include <cstdint>
 
namespace oqmc
{
 
namespace bntables
{
 
struct TableReturnValue
{
    std::uint32_t key;  
    std::uint32_t rank; 
};
 
template <int XBits, int YBits, int ZBits>
OQMC_HOST_DEVICE constexpr TableReturnValue
tableValue(std::uint16_t pixel, std::uint16_t shift,
           const std::uint32_t keyTable[], const std::uint32_t rankTable[])
{
    const auto pixelOffset = decodeBits16<XBits, YBits, ZBits>(pixel);
    const auto shiftOffset = decodeBits16<XBits, YBits, ZBits>(shift);
 
    const auto x = pixelOffset.x + shiftOffset.x;
    const auto y = pixelOffset.y + shiftOffset.y;
    const auto z = pixelOffset.z + shiftOffset.z;
    const auto index = encodeBits16<XBits, YBits, ZBits>({x, y, z});
 
    return {
        keyTable[index],
        rankTable[index],
    };
}
 
constexpr auto xBits = 8;                   
constexpr auto yBits = 8;                   
constexpr auto size = 1 << (xBits + yBits); 
 
static_assert(xBits == yBits,
              "Optimisation tables have equal resolution in x and y");
 
// Following optimised blue noise randomisation values were generated using the
// optimise cli tool found in the source file src/tools/lib/optimise.cpp.
 
namespace pmj
{
 
#if defined(OQMC_ENABLE_BINARY)
 
extern const std::uint32_t keyTable[size];
 
extern const std::uint32_t rankTable[size];
 
#else
 
constexpr std::uint32_t keyTable[] = {
#include "data/pmj/keys.txt"
};
 
constexpr std::uint32_t rankTable[] = {
#include "data/pmj/ranks.txt"
};
 
#endif
 
} // namespace pmj
 
namespace sobol
{
 
#if defined(OQMC_ENABLE_BINARY)
 
extern const std::uint32_t keyTable[size];
 
extern const std::uint32_t rankTable[size];
 
#else
 
constexpr std::uint32_t keyTable[] = {
#include "data/sobol/keys.txt"
};
 
constexpr std::uint32_t rankTable[] = {
#include "data/sobol/ranks.txt"
};
 
#endif
 
} // namespace sobol
 
namespace lattice
{
 
#if defined(OQMC_ENABLE_BINARY)
 
extern const std::uint32_t keyTable[size];
 
extern const std::uint32_t rankTable[size];
 
#else
 
constexpr std::uint32_t keyTable[] = {
#include "data/lattice/keys.txt"
};
 
constexpr std::uint32_t rankTable[] = {
#include "data/lattice/ranks.txt"
};
 
#endif
 
} // namespace lattice
 
} // namespace bntables
 
} // namespace oqmc