Snake/config.py at master · Wouterr0/Snake · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
import os
import sys
import colorsys

from PIL import Image, ImageEnhance
import numpy as np
import pygame

pygame.mixer.init()


def surface_to_PIL(surface):
	raw_str = pygame.image.tostring(surface, "RGBA", False)
	surface = Image.frombytes("RGBA", surface.get_size(), raw_str)
	return surface

def PIL_to_surface(surface):
	raw_str = surface.tobytes("raw", "RGBA")
	surface = pygame.image.fromstring(raw_str, surface.size, "RGBA")
	return surface

def changeBrightness(image, brightness):
	return ImageEnhance.Brightness(image).enhance(brightness)

def combineSufacesVertical(img1, img2):
	img1 = surface_to_PIL(img1)
	img2 = surface_to_PIL(img2)
	dst = Image.new("RGBA", (max(img1.width, img2.width), img1.height+img2.height))
	dst.paste(img1, ((dst.width-img1.width)//2, 0))					# (dst.width-img1.width)//2 to make sure the image is centered
	dst.paste(img2, ((dst.width-img2.width)//2, img1.height))
	return PIL_to_surface(dst)

def repeatTileImage(img, size):
	result = Image.new(img.mode, size)
	for i in range(0, size[0], img.width):
		for j in range(0, size[1], img.height):
			result.paste(img, (i, j))
	return result


def roundToNearestMultiple(number, multiple):
	return number - (number%multiple)

def unique(num_list):
	seen = []
	for i in num_list:
		if i not in seen:
			seen.append(i)
		else:
			return False
	return True


def mapArrayToRainBow(arr, length):
	new = np.zeros((*arr.shape, 3))
	new[:,0] = arr															# Make a new array in the RGB shape (..., 3)
	for i in range(len(arr)):
		new[i] = np.array(colorsys.hsv_to_rgb(new[i][0], 1-(1/length), 1-(1/length)))*255			# Convert hue to RGB
	new = np.array(new, dtype=np.uint8)
	return new.tolist()


# -- Run configurations --
width, height = 800, 800 # Initial window dimensions


# -- User Interface configurations --
gameFont = "assets/FSEX300.ttf"
startBgImage = Image.open("assets/brick.png")


# -- Gameplay configurations --
gamespeed = 10
maxFPS = 60

appleImage = Image.open("assets/apple.png")
berryFruitImage = Image.open("assets/berryFruit.png")

applePickupSound = pygame.mixer.Sound("assets/applePickup.wav")
berryPickupSound = pygame.mixer.Sound("assets/berryPickup.wav")

initSnakeLength = 4
snakeHeadImage = pygame.image.load("assets/snakeHead.png")
berryHeadImage = pygame.image.load("assets/berryHead.png")


# Color configurations
WHITE = (255, 255, 255)
GREY = (127, 127, 127)
BLACK = (0, 0, 0)
FUCHSIA = (255, 0, 255)
RED = (255, 0, 0)
BROWN = (150, 75, 0)
ORANGE = (255, 127, 0)
YELLOW = (255, 255, 0)
GREEN = (0, 255, 0)
AQUA = (0, 255, 255)
BLUE = (0, 0, 255)
PURPLE = (255, 0, 255)
DARK_RED = (127, 0, 0)
DARK_GREEN = (0, 127, 0)
DARK_BLUE = (0, 0, 127)
LIGHT_RED = (255, 127, 127)
LIGHT_GREEN = (127, 255, 127)
LIGHT_BLUE = (127, 127, 255)


# Constants
GOLDENRATIO = (1 + 5 ** 0.5) / 2