pairwise_counts.cuh

Go to the documentation of this file.

#pragma once
 
#include <cuda/std/cstdint>
 
namespace cuddl {
 
struct pairwise_counts {
    uint32_t lower{};
    uint32_t equal{};
    uint32_t higher{};
    uint32_t both_empty{};
 
    __host__ __device__ pairwise_counts& operator+=(pairwise_counts const other) noexcept {
        lower += other.lower;
        equal += other.equal;
        higher += other.higher;
        both_empty += other.both_empty;
        return *this;
    }
 
    friend __host__ __device__ pairwise_counts
    operator+(pairwise_counts left, pairwise_counts const right) noexcept {
        return left += right;
    }
 
    [[nodiscard]] __host__ __device__ constexpr pairwise_counts mirrored() const noexcept {
        return {.lower = higher, .equal = equal, .higher = lower, .both_empty = both_empty};
    }
 
    friend bool operator==(pairwise_counts const&, pairwise_counts const&) = default;
};
 
struct packed_pairwise_counts {
    uint64_t bits{};
 
    [[nodiscard]] static __host__ __device__ constexpr packed_pairwise_counts
    pack(uint32_t lower, uint32_t equal, uint32_t higher) noexcept {
        return {
            static_cast<uint64_t>(lower) | (static_cast<uint64_t>(equal) << 21U) |
            (static_cast<uint64_t>(higher) << 42U)
        };
    }
 
    [[nodiscard]] __host__ __device__ constexpr pairwise_counts unpack(
        uint32_t bucket_count
    ) const noexcept {
        constexpr uint64_t field = (uint64_t{1} << 21U) - 1U;
        auto const lower = static_cast<uint32_t>(bits & field);
        auto const equal = static_cast<uint32_t>((bits >> 21U) & field);
        auto const higher = static_cast<uint32_t>(bits >> 42U);
        return {
            .lower = lower,
            .equal = equal,
            .higher = higher,
            .both_empty = bucket_count - lower - equal - higher,
        };
    }
 
    friend bool operator==(packed_pairwise_counts const&, packed_pairwise_counts const&) = default;
};
 
[[nodiscard]] __host__ __device__ constexpr float weighted_kmer_identity(
    pairwise_counts const& counts
) noexcept {
    auto const divisor =
        counts.equal + (counts.lower < counts.higher ? counts.lower : counts.higher);
    if (divisor == 0U) return 0.0F;
    return static_cast<float>(counts.equal) / static_cast<float>(divisor < 6U ? 6U : divisor);
}
 
struct pairwise_summary {
    pairwise_counts counts{};
    double cardinality{};
 
    friend bool operator==(pairwise_summary const&, pairwise_summary const&) = default;
};
 
}  // namespace cuddl