#include <stdint.h>
#include <string.h>
#include "../../compat/pgmspace.h"
#include "invaders2.h"

uint16_t const peter[8] PROGMEM =
		{0x0104, 0x0088, 0x01FC, 0x0376, 0x07FF, 0x05FD, 0x0505, 0x00D8};

uint16_t const hans[7] PROGMEM =
		{0x0000, 0x0372, 0x0552, 0x0372, 0x0552, 0x0356, 0x0000};

void initGuards(unsigned char *guards)
{
	memset(guards, 0, NUM_COLS);

#if NUM_COLS == 16
	guards[2] = 3;
	guards[5] = 3;
	guards[10] = 3;
	guards[13] = 3;
#else
	unsigned const guard_min_distance = 3;
	unsigned char pos;
	for (pos = 0; pos <= ((NUM_COLS - (guard_min_distance - 1)) / 2); ++pos)
	{
		if (((pos % guard_min_distance) == 0) && (pos != 0))
		{
			guards[pos - 1] = 3;
			guards[NUM_COLS - pos] = 3;
		}
	}
#endif
}

void initInvaders(Invaders * iv, unsigned char lv)
{
	unsigned char x, y;

	// first zero out map!
	memset(iv->map, 0, sizeof(iv->map));

	iv->speedinc = 0;
	iv->isEdged = 0;

	switch (lv)
	{
	case 0:
	default:
		for (x = 0; x < 8; ++x)
		{
			iv->map[x][0] = 2;
			iv->map[x][1] = 2;
			iv->map[x][2] = 2;
			iv->map[x][3] = 1;
		}

		iv->pos.x = (NUM_COLS - 8) / 2;
		iv->pos.y = SPACESHIP_LINE + 1;
		iv->speed = MIN_SPEED - 3;
		iv->direction = 1;
		break;

	case 1:
		for (x = 0; x < 8; ++x)
		{
			iv->map[x][0] = 3;
			iv->map[x][1] = 3;
			iv->map[x][2] = 2;
			iv->map[x][3] = 2;
		}

		iv->pos.x = (NUM_COLS - 8) / 2;
		iv->pos.y = SPACESHIP_LINE + 1;
		iv->speed = MIN_SPEED - 2;

		iv->direction = 1;
		break;

	case 2:
		for (x = 0; x < 8; ++x)
		{
			iv->map[x][0] = 3;
			iv->map[x][1] = 3;
			iv->map[x][2] = 2;
			iv->map[x][3] = 2;
			iv->map[x][4] = 1;
		}

		iv->pos.x = (NUM_COLS - 8) / 2;
		iv->pos.y = SPACESHIP_LINE + 1;
		iv->speed = MIN_SPEED - 1;

		iv->direction = 1;
		break;

	case 3:
		for (y = 7; y--;)
		{
			uint16_t hansrow = pgm_read_word(&hans[y]);
			uint16_t mask = 0x0001;
			for (x = 11; x--;)
			{
				iv->map[x][y] = (hansrow & mask) ? 2 : 1;
				mask <<= 1;
			}
		}

		iv->pos.x = (NUM_COLS - 11) / 2;
		iv->pos.y = SPACESHIP_LINE + 1;

		iv->speed = MIN_SPEED + 40;
		iv->direction = 1;
		break;

	case 4:
		for (y = 8; y--;)
		{
			uint16_t peterrow = pgm_read_word(&peter[y]);
			uint16_t mask = 0x0001;
			for (x = 11; x--;)
			{
				if (peterrow & mask)
				{
					iv->map[x][y] = 2;
				}
				mask <<= 1;
			}
		}

		iv->pos.x = (NUM_COLS - 11) / 2;
		iv->pos.y = SPACESHIP_LINE + 1;

		iv->speed = MIN_SPEED + 3;
		iv->direction = 1;
		break;
	}
}
