#ifndef TAK_H
#define TAK_H

#include <stdint.h>

enum E_RESULT {ACT_OK, ACT_ILLEGAL, ACT_OVERFLOW,
	PTN_VALID, PTN_INVALID, GAME_END };

// -------------------------------------------------------------------
// State management

enum COLOUR         { C_WHITE, C_BLACK };
enum STONE_VARIANT  { STONE_FLAT, STONE_STANDING, STONE_CAPSTONE };
enum MOVE_DIRECTION { M_UP, M_DOWN, M_LEFT, M_RIGHT };
enum WIN_TYPE       { WIN_DRAW,
											WIN_FLAT_WHITE, WIN_FLAT_BLACK,
											WIN_ROAD_WHITE, WIN_ROAD_BLACK };

typedef uint8_t data_t;
typedef uint16_t colour_stack_t;

#define NUM_SHIFT           4
#define NUM_INC             (0x1<<NUM_SHIFT) // 0b00010000
#define STONE_MASK          0b00000011
#define STONE_AT(l)         (celldat[(l)] & STONE_MASK)
#define COUNT_AT(l)         (celldat[(l)] >> NUM_SHIFT)
#define THE_COORDS(col,row) ((col)+(row)*board_size)

extern enum WIN_TYPE won;
extern uint8_t board_size;
extern data_t celldat[36];
extern colour_stack_t colours[36];
extern enum COLOUR current_colour;
extern uint8_t white_count, black_count, ply;

void reset_state(const uint8_t new_board_size);
void next_ply(void);

enum E_RESULT
try_place(const int8_t location, const enum COLOUR colour,
					const enum STONE_VARIANT stone);

enum E_RESULT
try_move(const int8_t location, const enum MOVE_DIRECTION direction,
				 const uint8_t steps, const uint8_t drops[5]);

enum WIN_TYPE
check_win(void);

// -------------------------------------------------------------------
// PTN related

enum E_RESULT
parse_place(const uint8_t board_size, char *ptn,
						uint8_t *out_location, enum STONE_VARIANT *out_stone);

enum E_RESULT
parse_move(const uint8_t board_size, char *ptn,
					 uint8_t *out_location, enum MOVE_DIRECTION *out_direction,
					 uint8_t *out_steps, uint8_t out_drops[5]);

void
generate_place(const uint8_t in_location,
							 const enum STONE_VARIANT in_stone, char out_ptn[4]);

void
generate_move(const uint8_t in_location,
							const enum MOVE_DIRECTION in_direction,
							const uint8_t in_steps, const uint8_t in_drops[5],
							char out_ptn[10]);

// -------------------------------------------------------------------
// Game driver

enum E_RESULT
do_ptn(char *ptn);
#endif