block_class.c

/* 
 * This file is part of the Minecraft Overviewer.
 *
 * Minecraft Overviewer is free software: you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or (at
 * your option) any later version.
 *
 * Minecraft Overviewer is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
 * Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with the Overviewer.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "block_class.h"
#include "utils.h"

#if defined(__i386__) || defined(__x86_64__)
#include <immintrin.h>
#endif
#if defined(__aarch64__)
#include <arm_neon.h>
#endif

bool block_class_is_subset(
    mc_block_t block,
    const mc_block_t block_class[],
    size_t block_class_len) {
    size_t i = 0;

#if defined(__AVX512F__) && defined(__AVX512BW__)
    for (; i / 32 < block_class_len / 32; i += 32) {
        const __m512i block_class_vec = _mm512_loadu_si512(
            (__m512i*)&block_class[i]);
        const __m512i block_vec = _mm512_set1_epi16(block);
        const __mmask32 block_cmp = _mm512_cmpeq_epi16_mask(block_vec, block_class_vec);
        if (block_cmp) {
            return true;
        }
    }
#endif
#if defined(__AVX2__)
    for (; i / 16 < block_class_len / 16; i += 16) {
        const __m256i block_class_vec = _mm256_loadu_si256(
            (__m256i*)&block_class[i]);
        const __m256i block_vec = _mm256_set1_epi16(block);
        const __m256i block_cmp = _mm256_cmpeq_epi16(block_vec, block_class_vec);
        if (_mm256_movemask_epi8(block_cmp)) {
            return true;
        }
    }
#endif
#ifdef __SSE2__
    for (; i / 8 < block_class_len / 8; i += 8) {
        const __m128i block_class_vec = _mm_loadu_si128(
            (__m128i*)&block_class[i]);
        const __m128i block_vec = _mm_set1_epi16(block);
        const __m128i block_cmp = _mm_cmpeq_epi16(block_vec, block_class_vec);
        if (_mm_movemask_epi8(block_cmp)) {
            return true;
        }
    }
#endif
#if defined(__aarch64__)
    for (; i / 8 < block_class_len / 8; i += 8) {
        const uint16x8_t block_class_vec = vld1q_u16(
            (uint16_t*)&block_class[i]);
        const uint16x8_t block_vec = vmovq_n_u16(block);
        const uint16x8_t block_cmp = vceqq_u16(block_vec, (uint16x8_t) block_class_vec);
        if(vgetq_lane_s64((int64x2_t) block_cmp, 0) +
           vgetq_lane_s64((int64x2_t) block_cmp, 1)) {
            return true;
        }
    }
#endif
#ifdef __MMX__
    for (; i / 4 < block_class_len / 4; i += 4) {
        const __m64 block_class_vec = _mm_cvtsi64_m64(
            *(uint64_t*)&block_class[i]);
        const __m64 block_vec = _mm_set1_pi16(block);
        const __m64 block_cmp = _mm_cmpeq_pi16(block_vec, block_class_vec);
        if (_mm_cvtm64_si64(block_cmp)) {
            return true;
        }
    }
#endif
    for (; i < block_class_len; ++i) {
        if (block == block_class[i]) {
            return true;
        }
    }
    return false;
}

bool block_class_is_wall(mc_block_t block) {
    mc_block_t mask = 0b11111111;
    mc_block_t prefix = 0b111 << 8;     // 1792 is the starting offset
    // if the xor zeroes all bits, the prefix must've matched.
    return ((block & ~mask) ^ prefix) == 0;
}

const mc_block_t block_class_stair[] = {
    block_oak_stairs,
    block_brick_stairs,
    block_stone_brick_stairs,
    block_nether_brick_stairs,
    block_sandstone_stairs,
    block_spruce_stairs,
    block_birch_stairs,
    block_jungle_stairs,
    block_crimson_stairs,
    block_warped_stairs,
    block_quartz_stairs,
    block_acacia_stairs,
    block_dark_oak_stairs,
    block_red_sandstone_stairs,
    block_smooth_red_sandstone_stairs,
    block_purpur_stairs,
    block_prismarine_stairs,
    block_dark_prismarine_stairs,
    block_prismarine_brick_stairs,
    block_mossy_cobblestone_stairs,
    block_cobblestone_stairs,
    block_smooth_quartz_stairs,
    block_polished_granite_stairs,
    block_polished_diorite_stairs,
    block_polished_andesite_stairs,
    block_stone_stairs,
    block_granite_stairs,
    block_diorite_stairs,
    block_andesite_stairs,
    block_end_stone_brick_stairs,
    block_red_nether_brick_stairs,
    block_mossy_stone_brick_stairs,
    block_smooth_sandstone_stairs,
    block_blackstone_stairs,
    block_polished_blackstone_stairs,
    block_polished_blackstone_brick_stairs,
    block_cut_copper_stairs,
    block_exposed_cut_copper_stairs,
    block_weathered_cut_copper_stairs,
    block_oxidized_cut_copper_stairs,
    block_waxed_cut_copper_stairs,
    block_waxed_exposed_cut_copper_stairs,
    block_waxed_weathered_cut_copper_stairs,
    block_waxed_oxidized_cut_copper_stairs,
    block_cobbled_deepslate_stairs,
    block_polished_deepslate_stairs,
    block_deepslate_brick_stairs,
    block_deepslate_tile_stairs,
    block_mangrove_stairs,
    block_mud_brick_stairs};
const size_t block_class_stair_len = COUNT_OF(block_class_stair);

const mc_block_t block_class_door[] = {
    block_wooden_door,
    block_iron_door,
    block_spruce_door,
    block_birch_door,
    block_jungle_door,
    block_acacia_door,
    block_dark_oak_door,
    block_crimson_door,
    block_mangrove_door,
    block_warped_door};
const size_t block_class_door_len = COUNT_OF(block_class_door);

const mc_block_t block_class_ancil[] = {
    block_wooden_door,
    block_iron_door,
    block_spruce_door,
    block_birch_door,
    block_jungle_door,
    block_acacia_door,
    block_dark_oak_door,
    block_crimson_door,
    block_mangrove_door,
    block_oak_stairs,
    block_brick_stairs,
    block_stone_brick_stairs,
    block_nether_brick_stairs,
    block_sandstone_stairs,
    block_spruce_stairs,
    block_birch_stairs,
    block_jungle_stairs,
    block_quartz_stairs,
    block_acacia_stairs,
    block_dark_oak_stairs,
    block_crimson_stairs,
    block_warped_stairs,
    block_red_sandstone_stairs,
    block_smooth_red_sandstone_stairs,
    block_purpur_stairs,
    block_prismarine_stairs,
    block_dark_prismarine_stairs,
    block_prismarine_brick_stairs,
    block_cobblestone_stairs,
    block_mossy_cobblestone_stairs,
    block_mossy_stone_brick_stairs,
    block_smooth_quartz_stairs,
    block_polished_granite_stairs,
    block_polished_diorite_stairs,
    block_polished_andesite_stairs,
    block_stone_stairs,
    block_granite_stairs,
    block_diorite_stairs,
    block_andesite_stairs,
    block_end_stone_brick_stairs,
    block_red_nether_brick_stairs,
    block_smooth_sandstone_stairs,
    block_blackstone_stairs,
    block_polished_blackstone_stairs,
    block_polished_blackstone_brick_stairs,    
    block_cut_copper_stairs,
    block_exposed_cut_copper_stairs,
    block_weathered_cut_copper_stairs,
    block_oxidized_cut_copper_stairs,
    block_waxed_cut_copper_stairs,
    block_waxed_exposed_cut_copper_stairs,
    block_waxed_weathered_cut_copper_stairs,
    block_waxed_oxidized_cut_copper_stairs,
    block_cobbled_deepslate_stairs,
    block_polished_deepslate_stairs,
    block_deepslate_brick_stairs,
    block_deepslate_tile_stairs,
    block_mangrove_stairs,
    block_mud_brick_stairs,
    block_grass,
    block_flowing_water,
    block_water,
    block_glass,
    block_redstone_wire,
    block_ice,
    block_portal,
    block_waterlily,
    block_andesite_wall,
    block_brick_wall,
    block_cobblestone_wall,
    block_diorite_wall,
    block_end_stone_brick_wall,
    block_granite_wall,
    block_mossy_cobblestone_wall,
    block_mossy_stone_brick_wall,
    block_nether_brick_wall,
    block_prismarine_wall,
    block_red_nether_brick_wall,
    block_red_sandstone_wall,
    block_sandstone_wall,
    block_stone_brick_wall,
    block_blackstone_wall,
    block_polished_blackstone_wall,
    block_polished_blackstone_brick_wall,
    block_double_plant,
    block_stained_glass,
    block_cobbled_deepslate_wall,
    block_polished_deepslate_wall,
    block_deepslate_brick_wall,
    block_deepslate_tile_wall,
    block_mud_brick_wall};
const size_t block_class_ancil_len = COUNT_OF(block_class_ancil);

const mc_block_t block_class_alt_height[] = {
    block_stone_slab,
    block_oak_stairs,
    block_brick_stairs,
    block_stone_brick_stairs,
    block_nether_brick_stairs,
    block_sandstone_stairs,
    block_spruce_stairs,
    block_birch_stairs,
    block_jungle_stairs,
    block_quartz_stairs,
    block_acacia_stairs,
    block_dark_oak_stairs,
    block_crimson_stairs,
    block_warped_stairs,
    block_red_sandstone_stairs,
    block_smooth_red_sandstone_stairs,
    block_prismarine_stairs,
    block_dark_prismarine_stairs,
    block_prismarine_brick_stairs,
    block_cobblestone_stairs,
    block_mossy_cobblestone_stairs,
    block_mossy_stone_brick_stairs,
    block_smooth_quartz_stairs,
    block_polished_granite_stairs,
    block_polished_diorite_stairs,
    block_polished_andesite_stairs,
    block_stone_stairs,
    block_granite_stairs,
    block_diorite_stairs,
    block_andesite_stairs,
    block_end_stone_brick_stairs,
    block_red_nether_brick_stairs,
    block_smooth_sandstone_stairs,
    block_blackstone_stairs,
    block_polished_blackstone_stairs,
    block_polished_blackstone_brick_stairs,
    block_mud_brick_stairs,   
    block_prismarine_slab,
    block_dark_prismarine_slab,
    block_prismarine_brick_slab,
    block_andesite_slab,
    block_diorite_slab,
    block_granite_slab,
    block_polished_andesite_slab,
    block_polished_diorite_slab,
    block_polished_granite_slab,
    block_red_nether_brick_slab,
    block_smooth_sandstone_slab,
    block_cut_sandstone_slab,
    block_smooth_red_sandstone_slab,
    block_cut_red_sandstone_slab,
    block_end_stone_brick_slab,
    block_blackstone_slab,
    block_polished_blackstone_slab,
    block_polished_blackstone_brick_slab,
    block_mossy_cobblestone_slab,
    block_mossy_stone_brick_slab,
    block_smooth_quartz_slab,
    block_smooth_stone_slab,
    block_stone_slab2,
    block_purpur_stairs,
    block_purpur_slab,
    block_wooden_slab,
    block_cut_copper_stairs,
    block_exposed_cut_copper_stairs,
    block_weathered_cut_copper_stairs,
    block_oxidized_cut_copper_stairs,
    block_waxed_cut_copper_stairs,
    block_waxed_exposed_cut_copper_stairs,
    block_waxed_weathered_cut_copper_stairs,
    block_waxed_oxidized_cut_copper_stairs,
    block_cut_copper_slab,
    block_exposed_cut_copper_slab,
    block_weathered_cut_copper_slab,
    block_oxidized_cut_copper_slab,
    block_waxed_cut_copper_slab,
    block_waxed_exposed_cut_copper_slab,
    block_waxed_weathered_cut_copper_slab,
    block_waxed_oxidized_cut_copper_slab,
    block_cobbled_deepslate_stairs,
    block_polished_deepslate_stairs,
    block_deepslate_brick_stairs,
    block_deepslate_tile_stairs, 
    block_mangrove_stairs,
    block_cobbled_deepslate_slab,
    block_polished_deepslate_slab,
    block_deepslate_brick_slab,
    block_deepslate_tile_slab,
    block_mud_brick_slab,
    block_mangrove_slab};
const size_t block_class_alt_height_len = COUNT_OF(block_class_alt_height);

const mc_block_t block_class_nether_roof[] = {
    block_bedrock,
    block_netherrack,
    block_quartz_ore,
    block_lava,
    block_soul_sand,
    block_basalt,
    block_blackstone,
    block_soul_soil,
    block_nether_gold_ore,
    block_ancient_debris};
const size_t block_class_nether_roof_len = COUNT_OF(block_class_nether_roof);