/
Snake.py
159 lines (148 loc) · 5.1 KB
/
Snake.py
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
import random
import sys
from get_list import get_list
import Base_Input as bi
from GameFlow import Game
import GameObject as go
from get_size import get_size
import Engine_Utils as utils
import time
class Snake(go.Object):
def __init__(self,name):
go.Object.__init__(self,name)
self.direction = [-1,0]
self.rows,self.cols = get_size()
self.position = [self.cols/2,self.rows/2]
self.add_pixel([go.pixel(self.position),go.pixel([self.position[0]+1,self.position[1]]),go.pixel([self.position[0]+2,self.position[1]]),go.pixel([self.position[0]+3,self.position[1]]),go.pixel([self.position[0]+4,self.position[1]])])
def judge(self,position):
if position in [snake.pixels[i].position for i in range(len(snake.pixels))]:
return False
else:
return True
def add_p(self,position):
self.pixels.insert(0,go.pixel(position))
return 0
class SnakeGame(Game):
def __init__(self):
snake = Snake("snake")
global snake
food = go.Object("food")
global food
Game.__init__(self)
#self.gameover = False
self.direction = [-1,0]
self.score = 0
self.add_object(snake)
self.add_object(food)
self.isPause = False
self.frame = 0.1
self.foodpos = []
self.instance_food()
def instance_food(self):
food.pixels = []
random.seed(int(time.time()))
temppos = [random.randrange(self.cols-1),random.randrange(self.rows-1)]
#if self.foodpos == temppos:
# self.foodpos = instance_food()
#else:
# self.foodpos = temppos
while self.foodpos == temppos:
random.seed(int(time.time()))
temppos = [random.randrange(self.cols-1),random.randrange(self.rows-1)]
self.foodpos = temppos
food.add_pixel([go.pixel(self.foodpos)])
self.iseaten = False
return 0
def get_command(self,s):
if s == ":q":
self.isrunning = False
if s == ":r":
self.__init__()
cs = s.split(' ')
if len(cs) == 2:
if cs[0] == ":set":
cs2 = cs[1].split('=')
if len(cs2) == 2:
if cs2[0] == "frame":
self.frame = float(cs2[1])
def Move(self,c):
if c == "w" and self.direction<>[0,-1]:
self.direction = [0,1]
if c == "s" and self.direction<>[0,1]:
self.direction = [0,-1]
if c == "d" and self.direction<>[-1,0]:
self.direction = [1,0]
if c == "a" and self.direction<>[1,0]:
self.direction = [-1,0]
if c == "\x1b":
self.set_pause()
return 0
snake.position = utils.vector_add(snake.position,self.direction)
temp,snake.position = self.transform(snake.position)
if snake.judge(snake.position):
if snake.position == self.foodpos:
snake.add_p(snake.position)
self.iseaten = True
else:
snake.add_p(snake.position)
snake.pixels.pop(-1)
else:
self.gameover()
return 0
def gameover(self):
self.objects = []
content = open('gameover','r').read()
pos = get_list(content)
gameover_obj = go.Object('gameover')
gameover_obj.add_pixel([go.pixel(i) for i in pos])
self.add_object(gameover_obj)
self.fill_panel()
self.render_once()
self.set_pause()
return 0
def fill_panel(self):
self.canva = []
if len(self.objects)<>0:
for obj in self.objects:
for p in obj.pixels:
temp, p.position = self.transform(p.position)
if temp not in self.canva and (temp[0]<=self.rows-1):
self.canva.append(temp)
return self.canva
def transform(self,pos):
trans=[self.rows - pos[1] - 1,pos[0]+1]
if pos[1]+1 <= 0:
trans[0] = 1
pos[1] = self.rows - 2
if pos[0] < 0:
trans[1] = self.cols
pos[0] = self.cols - 1
if pos[1] + 1 >= self.rows:
trans[0] = self.rows - 1
pos[1] = 0
if pos[0] > self.cols:
trans[1] = 1
pos[0] = 0
return trans, pos
def update(self):
s = ''
self.isrunning = True
self.iseaten = False
while(self.isrunning):
if self.iseaten:
self.score += 1
self.instance_food()
c = bi.get_input()
s = c
self.canva = self.fill_panel()
#self.isrunning = True
#self.current_log = str([snake.pixels[i].position for i in range(len(snake.pixels))]) + str(self.canva)
self.current_log = str(self.score)
self.update_log()
if not self.isPause:
self.Move(s)
self.render_once()
print "\x1b[1;1H\x1b[2J\x1b[0m",
if __name__== "__main__":
test = SnakeGame()
test.update()