OcclusionTest.cpp

#include "common.h"
#include <bitset>
#include "Tile.h"
#include "WorldSegment.h"
#include "GUI.h"

using namespace std;
using namespace DFHack;
using namespace df::enums;

#define SCALE_FACTOR 4

#define S_PLATE_HEIGHT 4
#define S_FLOOR_HEIGHT 1
#define S_WALL_HEIGHT 4
#define S_TILE_HEIGHT (S_FLOOR_HEIGHT+S_WALL_HEIGHT)
#define S_SPRITE_HEIGHT (S_TILE_HEIGHT+S_PLATE_HEIGHT)

bitset<2*S_SPRITE_HEIGHT> base_mask_left;
bitset<2*S_SPRITE_HEIGHT> base_mask_right;
bitset<2*S_SPRITE_HEIGHT> wall_mask_left;
bitset<2*S_SPRITE_HEIGHT> wall_mask_right;
bitset<2*S_SPRITE_HEIGHT> floor_mask_left;
bitset<2*S_SPRITE_HEIGHT> floor_mask_right;

inline bool hasOpaqueSides(Tile * b){
    return IDhasOpaqueSides(b->tileType)
        || ( b->designation.bits.hidden && (ssConfig.shade_hidden_tiles && !ssConfig.show_hidden_tiles) );
}

inline bool hasOpaqueFloor(Tile * b){
    return IDhasOpaqueFloor(b->tileType)
        || ( b->designation.bits.hidden && (ssConfig.shade_hidden_tiles && !ssConfig.show_hidden_tiles) );
}

bool is_tile_solid(Tile * b)
{
    if(b->material.type == 3 ||
            b->material.type == 4 ||
            b->material.type == 5 ||
            b->material.type == 6) {
        return false;
    }
    return hasOpaqueSides(b) || hasOpaqueFloor(b);
}

void mask_center(Tile * b, int offset)
{
    if(!b) {
        return;
    }
    if(!is_tile_solid(b)) {
        return;
    }
    if(hasOpaqueSides(b)) {
        if(offset >= 0) {
            base_mask_left &= ~(wall_mask_left << offset*2);
            base_mask_right &= ~(wall_mask_right << offset*2);
        } else {
            base_mask_left &= ~(wall_mask_left >> -offset*2);
            base_mask_right &= ~(wall_mask_right >> -offset*2);
        }
    } else if(hasOpaqueFloor(b)) {
        if(offset >= 0) {
            base_mask_left &= ~(floor_mask_left << offset*2);
            base_mask_right &= ~(floor_mask_right << offset*2);
        } else {
            base_mask_left &= ~(floor_mask_left >> -offset*2);
            base_mask_right &= ~(floor_mask_right >> -offset*2);
        }
    }
}

void mask_left(Tile * b, int offset)
{
    if(!b) {
        return;
    }
    if(!is_tile_solid(b)) {
        return;
    }
    if(hasOpaqueSides(b)) {
        if(offset >= 0) {
            base_mask_left &= ~(wall_mask_right << offset*2);
        } else {
            base_mask_left &= ~(wall_mask_right >> -offset*2);
        }
    } else if(hasOpaqueFloor(b)) {
        if(offset >= 0) {
            base_mask_left &= ~(floor_mask_right << offset*2);
        } else {
            base_mask_left &= ~(floor_mask_right >> -offset*2);
        }
    }
}

void mask_right(Tile * b, int offset)
{
    if(!b) {
        return;
    }
    if(!is_tile_solid(b)) {
        return;
    }
    if(hasOpaqueSides(b)) {
        if(offset >= 0) {
            base_mask_right &= ~(wall_mask_left << offset*2);
        } else {
            base_mask_right &= ~(wall_mask_left >> -offset*2);
        }
    } else if(hasOpaqueFloor(b)) {
        if(offset >= 0) {
            base_mask_right &= ~(floor_mask_left << offset*2);
        } else {
            base_mask_right &= ~(floor_mask_left >> -offset*2);
        }
    }
}

void init_masks()
{
    //load up some mask files
    int flags = al_get_new_bitmap_flags();
    al_set_new_bitmap_flags(ALLEGRO_MEMORY_BITMAP);
    ALLEGRO_BITMAP * wall = load_bitmap_withWarning("stonesense/mask_wall.png");
    ALLEGRO_BITMAP * floor = load_bitmap_withWarning("stonesense/mask_floor.png");
    al_set_new_bitmap_flags(flags);

    //copy the mask to the bitsets. Each half is seperate to simplify things.
    for(int y = 0; y < S_SPRITE_HEIGHT; y++) {
        wall_mask_left.set(0 + (y*2), al_get_pixel(wall, 0, y).r > 0.5);
        wall_mask_left.set(1 + (y*2), al_get_pixel(wall, 1, y).r > 0.5);
        wall_mask_right.set(0 + (y*2), al_get_pixel(wall, 2, y).r > 0.5);
        wall_mask_right.set(1 + (y*2), al_get_pixel(wall, 3, y).r > 0.5);
    }
    for(int y = 0; y < S_SPRITE_HEIGHT; y++) {
        floor_mask_left.set(0 + (y*2), al_get_pixel(floor, 0, y).r > 0.5);
        floor_mask_left.set(1 + (y*2), al_get_pixel(floor, 1, y).r > 0.5);
        floor_mask_right.set(0 + (y*2), al_get_pixel(floor, 2, y).r > 0.5);
        floor_mask_right.set(1 + (y*2), al_get_pixel(floor, 3, y).r > 0.5);
    }

    al_destroy_bitmap(wall);
    al_destroy_bitmap(floor);

    base_mask_left.set();
    base_mask_right.set();
    int offset = 4;
    if(offset >= 0) {
        base_mask_left &= ~(wall_mask_left << offset*2);
        base_mask_right &= ~(wall_mask_right << offset*2);
    } else {
        base_mask_left &= ~(wall_mask_left >> -offset*2);
        base_mask_right &= ~(wall_mask_right >> -offset*2);
    }
}

void occlude_tile(Tile * b)
{
    WorldSegment * segment = b->ownerSegment;
    base_mask_left.set();
    base_mask_right.set();
    int baseX = b->x;
    int baseY = b->y;
    int baseZ = b->z;

    int stepX = 0, stepY = 0;

    switch(b->ownerSegment->segState.Rotation) {
    case 0:
        stepX = 1;
        stepY = 1;
        break;
    case 1:
        stepX = 1;
        stepY = -1;
        break;
    case 2:
        stepX = -1;
        stepY = -1;
        break;
    case 3:
        stepX = -1;
        stepY = 1;
        break;
    };

    bool done = 0;
    for(int relZ = 0; !done; relZ++) {
        int stepZ = relZ * S_TILE_HEIGHT / S_SPRITE_HEIGHT;
        if(relZ > 0) {
            stepZ--;
        }

        for(int relXY = 0; (relXY <= (((TILETOPHEIGHT + TILEHEIGHT) / (TILETOPHEIGHT))+1)) && !done ; relXY++) {
            int tempX = baseX + ((relXY + stepZ) * stepX);
            int tempY = baseY + ((relXY + stepZ) * stepY);
            int tempZ = baseZ + relZ;

            bool centerin = segment->CoordinateInteriorSegment(tempX + stepX, tempY + stepY, tempZ, 2);
            if(centerin) {
                Tile* center = segment->getTile(tempX + stepX, tempY + stepY, tempZ);
                mask_center(center, ((relXY+stepZ) * S_PLATE_HEIGHT + S_PLATE_HEIGHT) - (relZ * S_TILE_HEIGHT));
                if(base_mask_left.none() && base_mask_right.none()) {
                    done = true;
                    break;
                }
            }
            bool leftin = segment->CoordinateInteriorSegment(tempX, tempY + stepY, tempZ, 2);
            if(leftin) {
                Tile* left = segment->getTile(tempX, tempY + stepY, tempZ);
                mask_left(left, ((relXY+stepZ) * S_PLATE_HEIGHT + S_PLATE_HEIGHT/2) - (relZ * S_TILE_HEIGHT));
                if(base_mask_left.none() && base_mask_right.none()) {
                    done = true;
                    break;
                }
            }
            bool rightin = segment->CoordinateInteriorSegment(tempX + stepX, tempY, tempZ, 2);
            if(rightin) {
                Tile* right = segment->getTile(tempX + stepX, tempY, tempZ);
                mask_right(right, ((relXY+stepZ) * S_PLATE_HEIGHT + S_PLATE_HEIGHT/2) - (relZ * S_TILE_HEIGHT));
                if(base_mask_left.none() && base_mask_right.none()) {
                    done = true;
                    break;
                }
            }
            if(!leftin || !rightin || !centerin) {
                done = true;
                break;
            }
        }
    }

    if(base_mask_left.none() && base_mask_right.none()) {
        b->visible = 0;
    }
}