1 | importˆàrandomˆàasˆàrnd |
---|
2 | importˆàTkinterˆàasˆàtk |
---|
3 | importˆàsnake |
---|
4 | |
---|
5 | directions =ˆà[(0,1),ˆà(1,0),ˆà(0,-1),ˆà(-1,0)] |
---|
6 | tm =ˆà[[0,ˆà-1],ˆà[1,ˆà0]] |
---|
7 | |
---|
8 | classˆàCell(object): |
---|
9 | ˆà ˆà defˆà__init__(self,ˆàx,ˆày,ˆàcanvas =ˆàNone): |
---|
10 | ˆà ˆà ˆà ˆà self.x =ˆàx |
---|
11 | ˆà ˆà ˆà ˆà self.y =ˆày |
---|
12 | ˆà ˆà ˆà ˆà self.canvas =ˆàcanvas |
---|
13 | ˆà ˆà ˆà ˆà self.snake =ˆàNone |
---|
14 | ˆà ˆà ˆà ˆà self.type =ˆà'empty' |
---|
15 | ˆà ˆà ˆà ˆà return |
---|
16 | ˆà ˆà defˆàredraw(self): |
---|
17 | ˆà ˆà ˆà ˆà field_size =ˆàmin(self.canvas.winfo_height(),ˆàself.canvas.winfo_width()) |
---|
18 | ˆà ˆà ˆà ˆà offset =ˆà((self.canvas.winfo_width()ˆà-ˆàfield_size)/2.0,ˆà(self.canvas.winfo_height()ˆà-ˆàfield_size)/2.0) |
---|
19 | ˆà ˆà ˆà ˆà x0=offset[0]ˆà+ˆàself.x*field_size/21.0 |
---|
20 | ˆà ˆà ˆà ˆà y0=offset[1]ˆà+ˆàself.y*field_size/21.0 |
---|
21 | ˆà ˆà ˆà ˆà x1=offset[0]ˆà+ˆà(self.x+1)*field_size/21.0 |
---|
22 | ˆà ˆà ˆà ˆà y1=offset[1]ˆà+ˆà(self.y+1)*field_size/21.0 |
---|
23 | ˆà ˆà ˆà ˆà x2=offset[0]ˆà+ˆà(self.x+1/2.0)*field_size/21.0 |
---|
24 | ˆà ˆà ˆà ˆà ifˆàself.type ==ˆà'wall': |
---|
25 | ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_rectangle(x0,ˆày0,ˆàx1,ˆày1,ˆàfill="grey") |
---|
26 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
27 | ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'empty': |
---|
28 | ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_rectangle(x0,ˆày0,ˆàx1,ˆày1,ˆàfill="black") |
---|
29 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
30 | ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'body': |
---|
31 | ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_rectangle(x0,ˆày0,ˆàx1,ˆày1,ˆàfill=self.snake.color) |
---|
32 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
33 | ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'head': |
---|
34 | ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_oval(x0,ˆày0,ˆàx1,ˆày1,ˆàfill=self.snake.color) |
---|
35 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
36 | ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'tail': |
---|
37 | ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_polygon(x0,ˆày0,ˆàx1,ˆày0,ˆàx2,ˆày1,ˆàfill=self.snake.color) |
---|
38 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
39 | ˆà ˆà ˆà ˆà return |
---|
40 | ˆà ˆà defˆà__eq__(self,ˆàpattern): |
---|
41 | ˆà ˆà ˆà ˆà ifˆàpattern.type ==ˆà'any': |
---|
42 | ˆà ˆà ˆà ˆà ˆà ˆà returnˆàTrue |
---|
43 | ˆà ˆà ˆà ˆà ifˆàpattern.type !=ˆàself.type: |
---|
44 | ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse |
---|
45 | ˆà ˆà ˆà ˆà ifˆàpattern.snake_type ==ˆà'my'ˆàandˆàpattern.snake !=ˆàself.snake: |
---|
46 | ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse |
---|
47 | ˆà ˆà ˆà ˆà elifˆàpattern.snake_type ==ˆà'enemy'ˆàandˆàpattern.snake ==ˆàself.snake: |
---|
48 | ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse |
---|
49 | ˆà ˆà ˆà ˆà returnˆàTrue |
---|
50 | ˆà ˆà defˆàclear(self): |
---|
51 | ˆà ˆà ˆà ˆà self.snake =ˆàNone |
---|
52 | ˆà ˆà ˆà ˆà self.type =ˆà'empty' |
---|
53 | ˆà ˆà ˆà ˆà return |
---|
54 | |
---|
55 | |
---|
56 | classˆàEngine(object): |
---|
57 | ˆà ˆà defˆà__init__(self,ˆàcanvas): |
---|
58 | ˆà ˆà ˆà ˆà self.canvas =ˆàcanvas |
---|
59 | ˆà ˆà ˆà ˆà self.w =ˆàmin(canvas.winfo_height(),ˆàcanvas.winfo_width()) |
---|
60 | ˆà ˆà ˆà ˆà self.h =ˆàmin(canvas.winfo_height(),ˆàcanvas.winfo_width()) |
---|
61 | ˆà ˆà ˆà ˆà self.snakes =ˆà[None,ˆàNone,ˆàNone,ˆàNone] |
---|
62 | ˆà ˆà ˆà ˆà self.init_field() |
---|
63 | ˆà ˆà ˆà ˆà return |
---|
64 | ˆà ˆà defˆàinit_fieldˆà(self): |
---|
65 | ˆà ˆà ˆà ˆà self.field =ˆà{} |
---|
66 | ˆà ˆà ˆà ˆà forˆàx inˆàrange(21): |
---|
67 | ˆà ˆà ˆà ˆà ˆà ˆà forˆày inˆàrange(21): |
---|
68 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày]ˆà=ˆàCell(x,ˆày,ˆàself.canvas) |
---|
69 | ˆà ˆà ˆà ˆà forˆày inˆàrange(21): |
---|
70 | ˆà ˆà ˆà ˆà ˆà ˆà self.field[0,ˆày].type =ˆà'wall' |
---|
71 | ˆà ˆà ˆà ˆà ˆà ˆà self.field[20,ˆày].type =ˆà'wall' |
---|
72 | ˆà ˆà ˆà ˆà forˆàx inˆàrange(1,20): |
---|
73 | ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆà0].type =ˆà'wall' |
---|
74 | ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆà20].type =ˆà'wall' |
---|
75 | ˆà ˆà ˆà ˆà self.refill() |
---|
76 | ˆà ˆà ˆà ˆà self.redraw() |
---|
77 | ˆà ˆà ˆà ˆà return |
---|
78 | ˆà ˆà defˆàstep(self): |
---|
79 | ˆà ˆà ˆà ˆà forˆài,ˆàsnake inˆàenumerate(self.snakes): |
---|
80 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆàsnake !=ˆàNone: |
---|
81 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàlen(snake.cells)ˆà==ˆà0: |
---|
82 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.snakes[i]ˆà=ˆàNone |
---|
83 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà continue |
---|
84 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.legal_moves(snake) |
---|
85 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.move_snake(snake) |
---|
86 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.refill() |
---|
87 | ˆà ˆà ˆà ˆà self.redraw()ˆà ˆà |
---|
88 | ˆà ˆà ˆà ˆà return |
---|
89 | ˆà ˆà defˆàmove_snake(self,ˆàsnake): |
---|
90 | ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0] |
---|
91 | ˆà ˆà ˆà ˆà forˆàrule inˆàsnake.rules: |
---|
92 | ˆà ˆà ˆà ˆà ˆà ˆà forˆàdirection inˆàsnake.legal_dir: |
---|
93 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà rule.rotate(direction) |
---|
94 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàrule.applies(self.field,ˆàhead.x,ˆàhead.y)ˆà==ˆàTrue: |
---|
95 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàdirection) |
---|
96 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà return |
---|
97 | ˆà ˆà ˆà ˆà ifˆàsnake.legal_dir !=ˆà[]: |
---|
98 | ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàsnake.legal_dir[0]) |
---|
99 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
100 | ˆà ˆà ˆà ˆà return |
---|
101 | ˆà ˆà defˆàmove_do(self,ˆàsnake,ˆàapplied_dir): |
---|
102 | ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0] |
---|
103 | ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[head.x +ˆàapplied_dir[0],ˆàhead.y +ˆàapplied_dir[1]] |
---|
104 | ˆà ˆà ˆà ˆà ifˆàdir_cell.type ==ˆà'empty': |
---|
105 | ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell) |
---|
106 | ˆà ˆà ˆà ˆà ˆà ˆà delˆàsnake.cells[-1] |
---|
107 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
108 | ˆà ˆà ˆà ˆà elifˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake): |
---|
109 | ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell) |
---|
110 | ˆà ˆà ˆà ˆà ˆà ˆà delˆàdir_cell.snake.cells[-1] |
---|
111 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
112 | |
---|
113 | ˆà ˆà defˆàcreate_snake(self,ˆàsnake_number): |
---|
114 | ˆà ˆà ˆà ˆà cells_id =ˆà[] |
---|
115 | ˆà ˆà ˆà ˆà forˆày inˆàrange(10): |
---|
116 | ˆà ˆà ˆà ˆà ˆà ˆà cells_id.append((10,ˆày+1)) |
---|
117 | ˆà ˆà ˆà ˆà forˆàrot_num inˆàrange(snake_number -ˆà1): |
---|
118 | ˆà ˆà ˆà ˆà ˆà ˆà forˆài,ˆàcell inˆàenumerate(cells_id): |
---|
119 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà cells_id[i]ˆà=ˆà((tm[0][0]*(cell[0]-10)ˆà+ˆàtm[0][1]*(cell[1]-10))+10,(tm[1][0]*(cell[0]-10)ˆà+ˆàtm[1][1]*(cell[1]-10))+10) |
---|
120 | ˆà ˆà ˆà ˆà cells =ˆà[] |
---|
121 | ˆà ˆà ˆà ˆà forˆàcell inˆàcells_id: |
---|
122 | ˆà ˆà ˆà ˆà ˆà ˆà cells.append(self.field[cell]) |
---|
123 | ˆà ˆà ˆà ˆà color_dic =ˆà{ |
---|
124 | ˆà ˆà ˆà ˆà ˆà ˆà 1:'blue', |
---|
125 | ˆà ˆà ˆà ˆà ˆà ˆà 2:'green', |
---|
126 | ˆà ˆà ˆà ˆà ˆà ˆà 3:'yellow', |
---|
127 | ˆà ˆà ˆà ˆà ˆà ˆà 4:'red',} |
---|
128 | ˆà ˆà ˆà ˆà self.snakes[snake_number-1]ˆà=ˆàsnake.Snake(cells,ˆàcolor_dic[snake_number]) |
---|
129 | ˆà ˆà ˆà ˆà returnˆàself.snakes[snake_number-1] |
---|
130 | ˆà ˆà defˆàrefill(self): |
---|
131 | ˆà ˆà ˆà ˆà forˆàx inˆàrange(1,20): |
---|
132 | ˆà ˆà ˆà ˆà ˆà ˆà forˆày inˆàrange(1,20): |
---|
133 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].type =ˆà'empty' |
---|
134 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].snake =ˆàNone |
---|
135 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
136 | ˆà ˆà ˆà ˆà forˆàsnake inˆàself.snakes: |
---|
137 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆàsnake ==ˆàNone: |
---|
138 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
139 | ˆà ˆà ˆà ˆà ˆà ˆà else: |
---|
140 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.fill() |
---|
141 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass |
---|
142 | ˆà ˆà ˆà ˆà return |
---|
143 | ˆà ˆà defˆàredraw(self): |
---|
144 | ˆà ˆà ˆà ˆà self.canvas.delete(all) |
---|
145 | ˆà ˆà ˆà ˆà forˆàcell_coord inˆàself.field: |
---|
146 | ˆà ˆà ˆà ˆà ˆà ˆà self.field[cell_coord].redraw() |
---|
147 | ˆà ˆà ˆà ˆà return |
---|
148 | ˆà ˆà defˆàlegal_moves(self,ˆàsnake): |
---|
149 | ˆà ˆà ˆà ˆà snake.legal_dir =ˆà[] |
---|
150 | ˆà ˆà ˆà ˆà forˆàdirection inˆàdirections: |
---|
151 | ˆà ˆà ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[snake.cells[0].y +ˆàdirection[0],ˆàsnake.cells[0].x +ˆàdirection[1]] |
---|
152 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆà(dir_cell.type ==ˆà'empty'ˆàorˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake)): |
---|
153 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.legal_dir.append(direction) |
---|
154 | ˆà ˆà ˆà ˆà rnd.shuffle(snake.legal_dir) |
---|
155 | ˆà ˆà ˆà ˆà return |
---|
156 | |
---|