class Snake:
"""
This class represents a snake itself.
It has, most importantly, a body and a direction.
This snake can exist in N dimensions, and as such can have N directions.
--------------------------------------------------------------------------
Properties
--------------------------------------------------------------------------
snakeSegments: List of Snake_Segment type object
Represents the snake's body. The 0th segment is the head and is treated
as such, the rest of the segments follow the head traveling where it u-
sed to be. The body is important only for the cases of collision, and
non-trivial food spawning.
direction: A single direction object
Represents the snake's direction of travel. Realistically only affects
the head
segmentsToGrow: An integer
Represents the number of segments the snake has to grow. It will grow
by one segment in length per turn, and the number of segments gained by
eating food could hypothetically be variable.
"""
def __init__(self, position, segmentsPerFood=1):
self.snakeSegments = [Snake_Segment(position)]
self.direction = Direction()
self.segmentsToGrow = 0
self.segmentsPerFood = 1
def ChangeDirection(self, direction):
"""
input:
direction - direction type object
Indicates the direction in which the snake's head will travel
output:
none
Description:
Changes the direction in which the snake's head travels. Must pass
a validity check first.
"""
if len(self.snakeSegments) == 1 or not (
self.snakeSegments[0].position + direction
== self.snakeSegments[1].position):
return self.direction.ChangeDirection(direction)
else:
return False
def Move(self):
"""
input:
none
output:
none
Description:
Moves the snake's head one unit in the direction of travel, as def-
ined by self.direction. The rest of the snake's body will move to
where the previous segment used to be.
"""
grow = self.segmentsToGrow > 0
if (grow):
self.segmentsToGrow -= 1
lastPosition = Position(
self.snakeSegments[-1].position.coordinates)
oldSnakeSegments = [
Snake_Segment(Position(snakeSeg.position.coordinates[:]))
for snakeSeg in self.snakeSegments
]
for index, segment in enumerate(self.snakeSegments[1:]):
self.snakeSegments[index +
1].position.coordinates = oldSnakeSegments[
index].position.coordinates[:]
if (grow):
self.snakeSegments.append(Snake_Segment(lastPosition))
self.snakeSegments[
0].position = self.snakeSegments[0].position + self.direction
def Eat(self):
"""
input:
none
output:
none
Description:
Called by the game once the snake eats some food. Causes the snake
to grow on subsequent turns. Increases the dimensionality of the s-
nake by one.
"""
self.segmentsToGrow += self.segmentsPerFood
for i, e in enumerate(self.snakeSegments):
self.snakeSegments[i].position.coordinates.append(0)
self.direction.directionalCoordinates.append(0)
