aoc2020/day11/day11.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <panic.h>
#include <attrs.h>

const char* const input[] = {
#ifdef TEST
#include "input-test.h"
#else
#include "input.h"
#endif
};

// input_sz //
#ifdef TEST
#include "input-test-sz.h"
#else
#include "input-sz.h"
#endif

#define INPUT_WIDTH (input_sz-1)
#define INPUT_HEIGHT (sizeof(input)/sizeof(char*)) 

#define box(t) aligned_alloc(_Alignof(t), sizeof(t))

enum gol_state {
	STATE_INVALID=0,
	STATE_FLOOR='.',
	STATE_EMPTY='L',
	STATE_OCCU='#',
};

#define GOL_INDEX(w,h) ((h*INPUT_WIDTH)+w)

typedef struct gol_arena {
	enum gol_state arena[INPUT_HEIGHT*INPUT_WIDTH];
} gol_t;

gol_t* gol_clone(const gol_t* from)
{
	gol_t* out = box(gol_t);
	*out = *from;
	return out;
}

pure gol_t generate_arena()
{
	gol_t gol;
	for(size_t h=0;h<INPUT_HEIGHT;h++)
		for(size_t w=0;w<INPUT_WIDTH;w++)
			if( !(gol.arena[ (h*INPUT_WIDTH)+w ] = (enum gol_state) input[h][w]) ) panic("Invalid `gol_state` char at (%lu x %lu = %lu) '%c' (%d)", w, h, GOL_INDEX(w,h), input[h][w], (int)input[h][w]);

	return gol;
}

noglobal static inline int gol_in_bounds(int x, int y)
{
	return !(x<0 || x >= (int)INPUT_WIDTH || y<0 || y >= (int)INPUT_HEIGHT);
}

static inline pure enum gol_state gol_index(const gol_t* gol, int x, int y)
{
	if (!gol_in_bounds(x,y)) return STATE_FLOOR;
	return gol->arena[GOL_INDEX(x,y)];
}

static inline void gol_set(gol_t* restrict gol, int x,int y, enum gol_state st)
{
	if (!gol_in_bounds(x,y)) return;

	gol->arena[GOL_INDEX(x,y)] = st;
}

static inline int count_states(const enum gol_state* state, enum gol_state s, size_t n)
{
	register int j=0;
	for(register size_t i=0;i<n;i++)
		j+= (state[i] == s);
	return j;
}

static inline int gol_count(const gol_t* gol, enum gol_state s)
{
	return count_states(gol->arena, s, INPUT_WIDTH*INPUT_HEIGHT);
}

static inline int nneighbours(const gol_t* gol, int n, int x, int y, enum gol_state out[restrict 8])
{
	out[0] = gol_index(gol, x-n, y-n);
	out[1] = gol_index(gol, x, y-n);
	out[2] = gol_index(gol, x+n, y-n);

	out[3] = gol_index(gol, x-n, y);
	//out[4] = gol_index(gol, x, y);
	out[4] = gol_index(gol, x+n, y);

	out[5] = gol_index(gol, x-n, y+n);
	out[6] = gol_index(gol, x, y+n);
	out[7] = gol_index(gol, x+n, y+n);

	return gol_in_bounds(x-n, y-n) ||
		gol_in_bounds(x+n, y-n) ||
		gol_in_bounds(x-n, y+n) ||
		gol_in_bounds(x+n, y+n);
}
inline static int neighbours(const gol_t* gol, int x, int y, enum gol_state out[restrict 8])
{
	return nneighbours(gol, 1, x, y, out);
}

#define CEAR_WITH(clr, num, with) do { for(size_t __i=0;__i<num;__i++) clr[__i] = with; } while(0)

static inline void neighbours_far(const gol_t* gol, int x, int y, enum gol_state out[restrict 8])
{
	struct {
		int set;
		int x, y;
	} nco[8] = {0};

	int i=1;
	for(;;i++) {
#define X(mx, my, j) if (gol_index(gol, x + (mx), y + (my)) != STATE_FLOOR && !nco[j].set) { nco[j].set = 1; nco[j].x = x + (mx); nco[j].y = y + (my); } (void)0

		X( -i, -i, 0 );
		X(  0, -i, 1 );
		X( +i, -i, 2 );

		X( -i, 0,  3);
		X( +i, 0,  4);

		X( -i, +i, 5);
		X(  0, +i, 6);
		X( +i, +i, 7);

		if(! (gol_in_bounds(x-i, y-i) ||
			gol_in_bounds(x+i, y-i) ||
			gol_in_bounds(x-i, y+i) ||
			gol_in_bounds(x+i, y+i)) ) break;


		int red=1;
		for(int i=0;i<8;i++)
			red &= nco[i].set;
		if(red) break;
#undef X
	}

#define X(n) out[n] = nco[n].set ? gol_index(gol, nco[n].x, nco[n].y) : out[n]
	X(0);
	X(1);
	X(2);
	X(3);
	X(4);
	X(5);
	X(6);
	X(7);
#undef X
/*	enum gol_state tmp[8];
	neighbours(gol, x, y, tmp);
	int ndone[8] = {0};
	int rdone=0;
	for(register size_t i=1;rdone<8;i++) {
		if(!nneighbours(gol, i, x, y, tmp)) break;
		for(size_t j=0;j<8;j++) {
			if(!ndone[j] && tmp[j] != STATE_FLOOR) {
				out[j] = tmp[j];
				ndone[j] = 1;
				rdone+=1;
				break;
			}
		}
	}*/
}

static void simulate_ip(const gol_t* source, gol_t* restrict dest)
{
	for(size_t h=0;h<INPUT_HEIGHT;h++)
		for(size_t w=0;w<INPUT_WIDTH;w++) {
			enum gol_state next = gol_index(source, w, h);
			enum gol_state n[8]={0};
#ifdef PART2
			neighbours_far(source, w, h, n);
#else
			neighbours(source, w, h, n);
#endif
			switch(next) {
			case STATE_EMPTY:
				if(count_states(n, STATE_OCCU, 8)==0) next = STATE_OCCU;
				break;
			case STATE_OCCU:
#ifdef PART2
				if(count_states(n, STATE_OCCU, 8)>=5)
#else
				if(count_states(n, STATE_OCCU, 8)>=4)
#endif
					 next = STATE_EMPTY;
			default: break;
			}
			gol_set(dest, w, h, next);
		}
}

static inline int gol_eq(const gol_t* g1, const gol_t* g2)
{
	return memcmp(g1->arena, g2->arena, sizeof(g1->arena))==0;
}

static inline gol_t simulate_once(const gol_t* from)
{
	gol_t out;
	simulate_ip(from, &out);
	return out;
}

#ifdef DEBUG
static void print_arena(const gol_t* gol)
{
	for(size_t h=0;h<INPUT_HEIGHT;h++) {
		for(size_t w=0;w<INPUT_WIDTH;w++) fputc((char)gol_index(gol, w, h), stderr);
		fputc('\n', stderr);
	}
}
#endif

int main()
{
	gol_t* arena = box(gol_t);
	*arena = generate_arena();

#ifdef DEBUG
	printf("First: \n");
	print_arena(arena);
#endif
	gol_t tmp;
	while(1)
	{
#if !defined(_COPY_ALLOC_ARENA)
		simulate_ip(arena, &tmp);
#else
		tmp = simulate_once(arena);
#endif
#ifdef DEBUG
		printf("Next: \n");
		print_arena(&tmp);
#endif
		if(gol_eq(&tmp, arena))
		{
			printf("%d\n", gol_count(&tmp, STATE_OCCU));
			break;
		}
		*arena = tmp;
	}
	free(arena);
	return 0;
}