#ifndef HALIDE_CPU_FEATURES_H
#define HALIDE_CPU_FEATURES_H

#include "HalideRuntime.h"
#include "cpu_features.h"

namespace Halide { namespace Runtime { namespace Internal {

// Return two masks:
// One with all the CPU-specific features that might possible be available on this architecture ('known'),
// and one with the subset that are actually present ('available').
struct CpuFeatures {
    static const int kWordCount = (halide_target_feature_end + 63) / (sizeof(uint64_t) * 8);

    __attribute__((always_inline))
    void set_known(int i) {
        known[i >> 6] |= ((uint64_t) 1) << (i & 63);
    }

    __attribute__((always_inline))
    void set_available(int i) {
        available[i >> 6] |= ((uint64_t) 1) << (i & 63);
    }

    __attribute__((always_inline))
    bool test_known(int i) const {
        return (known[i >> 6] & ((uint64_t) 1) << (i & 63)) != 0;
    }

    __attribute__((always_inline))
    bool test_available(int i) const {
        return (available[i >> 6] & ((uint64_t) 1) << (i & 63)) != 0;
    }

    __attribute__((always_inline))
    CpuFeatures() {
        // Can't use in-class initing of these without C++11 enabled,
        // which isn't the case for all runtime builds
        for (int i = 0; i < kWordCount; ++i) {
            known[i] = 0;
            available[i] = 0;
        }
    }

    uint64_t known[kWordCount];     // mask of the CPU features we know how to detect
    uint64_t available[kWordCount]; // mask of the CPU features that are available
                                    // (always a subset of 'known')
};

extern WEAK CpuFeatures halide_get_cpu_features();

}}} // namespace Halide::Runtime::Internal

#endif  // HALIDE_CPU_FEATURES_H