Äîêóìåíò âçÿò èç êýøà ïîèñêîâîé ìàøèíû. Àäðåñ îðèãèíàëüíîãî äîêóìåíòà : http://kodomo.cmm.msu.su/trac/snake/browser/engine.py?rev=155
Äàòà èçìåíåíèÿ: Unknown
Äàòà èíäåêñèðîâàíèÿ: Sun Apr 10 21:15:12 2016
Êîäèðîâêà: IBM-866
engine.py òÀÓ Python Battle

source: engine.py @ 155:382ab7b65331

Revision 155:382ab7b65331, 8.2 KB checked in by Alex Martynov, 5 years ago (diff)

added smth for snake restart, not working yet
  • Property exe set to *
Lineˆà
1importˆàrandomˆàasˆàrnd
2importˆàTkinterˆàasˆàtk
3importˆàsnake
4
5directions =ˆà[(0,1),ˆà(1,0),ˆà(0,-1),ˆà(-1,0)]
6
7classˆàDict(dict):
8ˆà ˆà """Create a dictionary."""
9ˆà ˆà pass
10
11classˆàCell(object):
12ˆà ˆà """Cells.
13
14ˆà ˆà Atributes:
15ˆà ˆà - 'x' - absciss of the cell in field
16ˆà ˆà - 'y' - ordinate of the cell in field
17ˆà ˆà - 'canvas' - Widget the cell belongs to
18ˆà ˆà - 'snake' - snake the cell belongs to, possible values:
19ˆà ˆà ˆà ˆà snake
20ˆà ˆà ˆà ˆà None
21ˆà ˆà ˆà ˆà 'my'ˆà ˆà ˆà ˆà \
22ˆà ˆà ˆà ˆà 'enemy'ˆà ˆà ˆà/ˆà for patterns only
23
24ˆà ˆà - 'type' - type of the cell, possible values:
25ˆà ˆà ˆà ˆà 'empty'
26ˆà ˆà ˆà ˆà 'wall'
27ˆà ˆà ˆà ˆà 'body'
28ˆà ˆà ˆà ˆà 'head'
29ˆà ˆà ˆà ˆà 'tail'
30ˆà ˆà ˆà ˆà 'any'ˆà } for patterns only
31ˆà ˆà ˆà ˆà 'void' } for cells out of the field
32ˆà ˆà """
33ˆà ˆà defˆà__init__(self,ˆàx,ˆày,ˆàcanvas =ˆàNone):
34ˆà ˆà ˆà ˆà """Initialyze the cell with default parameters:
35ˆà ˆà ˆà ˆà ˆà ˆà type = 'empty'
36ˆà ˆà ˆà ˆà ˆà ˆà snake = None"""
37ˆà ˆà ˆà ˆà self.x =ˆàx
38ˆà ˆà ˆà ˆà self.y =ˆày
39ˆà ˆà ˆà ˆà self.canvas =ˆàcanvas
40ˆà ˆà ˆà ˆà self.snake =ˆàNone
41ˆà ˆà ˆà ˆà self.type =ˆà'empty'
42ˆà ˆà ˆà ˆà return
43ˆà ˆà
44ˆà ˆà defˆàredraw(self,ˆàoffset,ˆàc_size):
45ˆà ˆà ˆà ˆà """Draw a cell based on it content"""
46ˆà ˆà ˆà ˆà x0=offset[0]ˆà+ˆàself.x*c_size
47ˆà ˆà ˆà ˆà y0=offset[1]ˆà+ˆàself.y*c_size
48ˆà ˆà ˆà ˆà x1=offset[0]ˆà+ˆà(self.x+1)*c_size
49ˆà ˆà ˆà ˆà y1=offset[1]ˆà+ˆà(self.y+1)*c_size
50ˆà ˆà ˆà ˆà x2=offset[0]ˆà+ˆà(self.x+1/2.0)*c_size
51ˆà ˆà ˆà ˆà ifˆàself.type ==ˆà'wall':
52ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_rectangle(x0,ˆày0,ˆàx1,ˆày1,ˆàfill="grey")
53ˆà ˆà ˆà ˆà ˆà ˆà pass
54ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'empty':
55ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_rectangle(x0,ˆày0,ˆàx1,ˆày1,ˆàfill="black")
56ˆà ˆà ˆà ˆà ˆà ˆà pass
57ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'body':
58ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_rectangle(x0,ˆày0,ˆàx1,ˆày1,ˆàfill=self.snake.color)
59ˆà ˆà ˆà ˆà ˆà ˆà pass
60ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'head':
61ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_oval(x0,ˆày0,ˆàx1,ˆày1,ˆàfill=self.snake.color)
62ˆà ˆà ˆà ˆà ˆà ˆà pass
63ˆà ˆà ˆà ˆà elifˆàself.type ==ˆà'tail':
64ˆà ˆà ˆà ˆà ˆà ˆà self.canvas.create_polygon(x0,ˆày0,ˆàx1,ˆày0,ˆàx2,ˆày1,ˆàfill=self.snake.color)
65ˆà ˆà ˆà ˆà ˆà ˆà pass
66ˆà ˆà ˆà ˆà return
67ˆà ˆà
68ˆà ˆà defˆà__eq__(self,ˆàpattern):
69ˆà ˆà ˆà ˆà """Check the equaliation of the cell to the pattern cell."""
70ˆà ˆà ˆà ˆà ifˆàpattern.type ==ˆà'any':
71ˆà ˆà ˆà ˆà ˆà ˆà returnˆàTrue
72ˆà ˆà ˆà ˆà ifˆàpattern.type !=ˆàself.type:
73ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse
74ˆà ˆà ˆà ˆà ifˆàpattern.snake_type ==ˆà'my'ˆàandˆàpattern.snake !=ˆàself.snake:
75ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse
76ˆà ˆà ˆà ˆà elifˆàpattern.snake_type ==ˆà'enemy'ˆàandˆàpattern.snake ==ˆàself.snake:
77ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse
78ˆà ˆà ˆà ˆà returnˆàTrue
79ˆà ˆà
80ˆà ˆà defˆà__ne__(self,ˆàpattern):
81ˆà ˆà ˆà ˆà """Check the discrepancy of the cell to the pattern cell."""
82ˆà ˆà ˆà ˆà returnˆànotˆàselfˆà==ˆàpattern
83ˆà ˆà
84ˆà ˆà defˆàclear(self):
85ˆà ˆà ˆà ˆà """Change the cell parameters back to default."""
86ˆà ˆà ˆà ˆà self.snake =ˆàNone
87ˆà ˆà ˆà ˆà self.type =ˆà'empty'
88ˆà ˆà ˆà ˆà return
89
90
91classˆàEngine(object):
92ˆà ˆà """Engine
93
94ˆà ˆà Atributes:
95
96ˆà ˆà - 'field' - game field:
97ˆà ˆà ˆà ˆà 'field.w' - width of the field count in cells
98ˆà ˆà ˆà ˆà 'field.h' - hight of the field count in cells
99ˆà ˆà - 'canvas' - Widget game field is showing on
100ˆà ˆà - 'snakes' - list of snakes loaded
101ˆà ˆà - 'psnakes' - list of snakes loaded in previous match, if other snakes are not loaded
102ˆà ˆà - 'start_snake_length' - starting length of the snake"""
103ˆà ˆà
104ˆà ˆà defˆà__init__(self,ˆàcanvas):
105ˆà ˆà ˆà ˆà """Initialyze the engine:
106ˆà ˆà ˆà ˆà start_snake_length = 10"""
107ˆà ˆà ˆà ˆà self.canvas =ˆàcanvas
108ˆà ˆà ˆà ˆà self.snakes =ˆà[None,ˆàNone,ˆàNone,ˆàNone]
109ˆà ˆà ˆà ˆà self.psnakes =ˆà[None,ˆàNone,ˆàNone,ˆàNone]
110ˆà ˆà ˆà ˆà self.init_field()
111ˆà ˆà ˆà ˆà self.start_snake_length =ˆà10
112ˆà ˆà ˆà ˆà return
113ˆà ˆà
114ˆà ˆà defˆàinit_fieldˆà(self):
115ˆà ˆà ˆà ˆà """Initialyze the field:
116ˆà ˆà ˆà ˆà width = 31
117ˆà ˆà ˆà ˆà hieght = 31
118ˆà ˆà ˆà ˆà perimeter is made by walls"""
119ˆà ˆà ˆà ˆà self.field =ˆàDict()
120ˆà ˆà ˆà ˆà self.field.w =ˆà31
121ˆà ˆà ˆà ˆà self.field.h =ˆà31
122ˆà ˆà ˆà ˆà f_w =ˆàself.field.w
123ˆà ˆà ˆà ˆà f_h =ˆàself.field.h
124ˆà ˆà ˆà ˆà forˆàx inˆàrange(f_w):
125ˆà ˆà ˆà ˆà ˆà ˆà forˆày inˆàrange(f_h):
126ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày]ˆà=ˆàCell(x,ˆày,ˆàself.canvas)
127ˆà ˆà ˆà ˆà forˆày inˆàrange(f_h):
128ˆà ˆà ˆà ˆà ˆà ˆà self.field[0,ˆày].type =ˆà'wall'
129ˆà ˆà ˆà ˆà ˆà ˆà self.field[f_w-1,ˆày].type =ˆà'wall'
130ˆà ˆà ˆà ˆà forˆàx inˆàrange(1,f_w-1):
131ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆà0].type =ˆà'wall'
132ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆàf_h-1].type =ˆà'wall'
133ˆà ˆà ˆà ˆà self.refill()
134ˆà ˆà ˆà ˆà self.redraw()
135ˆà ˆà ˆà ˆà return
136ˆà ˆà
137ˆà ˆà defˆàstep(self):
138ˆà ˆà ˆà ˆà """Do the step of the game."""
139ˆà ˆà ˆà ˆà forˆài,ˆàsnake inˆàenumerate(self.snakes):
140ˆà ˆà ˆà ˆà ˆà ˆà ifˆàsnake !=ˆàNone:
141ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàlen(snake.cells)ˆà==ˆà0:
142ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.snakes[i]ˆà=ˆàNone
143ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà continue
144ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.legal_moves(snake)
145ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.move_snake(snake)
146ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.refill()
147ˆà ˆà ˆà ˆà self.redraw()ˆà ˆà
148ˆà ˆà ˆà ˆà return
149ˆà ˆà
150ˆà ˆà defˆàmove_snake(self,ˆàsnake):
151ˆà ˆà ˆà ˆà """Check of movement direction based on the snake rule list and actual
152ˆà ˆà ˆà ˆà enviroment."""
153ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0]
154ˆà ˆà ˆà ˆà forˆàrule inˆàsnake.rules:
155ˆà ˆà ˆà ˆà ˆà ˆà forˆàdirection inˆàsnake.legal_dir:
156ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà rule.rotate(direction)
157ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàrule.applies(self.field,ˆàhead.x,ˆàhead.y)ˆà==ˆàTrue:
158ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàdirection)
159ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà return
160ˆà ˆà ˆà ˆà ifˆàsnake.legal_dir !=ˆà[]:
161ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàsnake.legal_dir[0])
162ˆà ˆà ˆà ˆà ˆà ˆà pass
163ˆà ˆà ˆà ˆà return
164ˆà ˆà
165ˆà ˆà defˆàmove_do(self,ˆàsnake,ˆàapplied_dir):
166ˆà ˆà ˆà ˆà """Do the move of the snake."""
167ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0]
168ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[head.x +ˆàapplied_dir[0],ˆàhead.y +ˆàapplied_dir[1]]
169ˆà ˆà ˆà ˆà ifˆàdir_cell.type ==ˆà'empty':
170ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell)
171ˆà ˆà ˆà ˆà ˆà ˆà delˆàsnake.cells[-1]
172ˆà ˆà ˆà ˆà ˆà ˆà pass
173ˆà ˆà ˆà ˆà elifˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake):
174ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell)
175ˆà ˆà ˆà ˆà ˆà ˆà delˆàdir_cell.snake.cells[-1]
176ˆà ˆà ˆà ˆà ˆà ˆà pass
177
178ˆà ˆà defˆàcreate_snake(self,ˆàsnake_number):
179ˆà ˆà ˆà ˆà """Create the snake:
180ˆà ˆà ˆà ˆà position choice is based on number or placement of 'Load' button
181ˆà ˆà ˆà ˆà ˆà ˆà snakes are placed with tails turned to the wall.
182ˆà ˆà ˆà ˆà color is chosen accorting to fen shui tradition"""
183ˆà ˆà ˆà ˆà cells_id =ˆà[]
184ˆà ˆà ˆà ˆà f_h =ˆàself.field.h
185ˆà ˆà ˆà ˆà f_w =ˆàself.field.w
186ˆà ˆà ˆà ˆà forˆày inˆàrange(self.start_snake_length):
187ˆà ˆà ˆà ˆà ˆà ˆà cells_id.insert(0,((f_w-1)/2,ˆày+1))
188ˆà ˆà ˆà ˆà forˆàrot_num inˆàrange(snake_number -ˆà1):
189ˆà ˆà ˆà ˆà ˆà ˆà forˆài,ˆàcell inˆàenumerate(cells_id):
190ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà cells_id[i]ˆà=ˆà(min(f_h,ˆàf_w)-1-cell[1],cell[0])
191ˆà ˆà ˆà ˆà cells =ˆà[]
192ˆà ˆà ˆà ˆà forˆàcell inˆàcells_id:
193ˆà ˆà ˆà ˆà ˆà ˆà cells.append(self.field[cell])
194ˆà ˆà ˆà ˆà color_dic =ˆà{
195ˆà ˆà ˆà ˆà ˆà ˆà 1:'blue',
196ˆà ˆà ˆà ˆà ˆà ˆà 2:'green',
197ˆà ˆà ˆà ˆà ˆà ˆà 3:'yellow',
198ˆà ˆà ˆà ˆà ˆà ˆà 4:'red',}
199ˆà ˆà ˆà ˆà self.snakes[snake_number-1]ˆà=ˆàsnake.Snake(cells,ˆàcolor_dic[snake_number])
200ˆà ˆà ˆà ˆà self.psnakes[snake_number-1]ˆà=ˆàself.snakes[snake_number-1]
201ˆà ˆà ˆà ˆà returnˆàself.snakes[snake_number-1]
202ˆà ˆà
203ˆà ˆà defˆàrefill(self):
204ˆà ˆà ˆà ˆà """Refill the field cells types and snakes according to the actual
205ˆà ˆà ˆà ˆà position"""
206ˆà ˆà ˆà ˆà f_w =ˆàself.field.w
207ˆà ˆà ˆà ˆà f_h =ˆàself.field.h
208ˆà ˆà ˆà ˆà forˆàx inˆàrange(1,f_w-1):
209ˆà ˆà ˆà ˆà ˆà ˆà forˆày inˆàrange(1,f_h-1):
210ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].type =ˆà'empty'
211ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].snake =ˆàNone
212ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass
213ˆà ˆà ˆà ˆà forˆàsnake inˆàself.snakes:
214ˆà ˆà ˆà ˆà ˆà ˆà ifˆàsnake ==ˆàNone:
215ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass
216ˆà ˆà ˆà ˆà ˆà ˆà else:
217ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.fill()
218ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass
219ˆà ˆà ˆà ˆà return
220ˆà ˆà
221ˆà ˆà defˆàredraw(self):
222ˆà ˆà ˆà ˆà """Clear the field Widget and redraw cells images on it"""
223ˆà ˆà ˆà ˆà self.canvas.delete("all")
224ˆà ˆà ˆà ˆà w =ˆàself.canvas.winfo_width()
225ˆà ˆà ˆà ˆà h =ˆàself.canvas.winfo_height()
226ˆà ˆà ˆà ˆà cw =ˆàw/float(self.field.w)
227ˆà ˆà ˆà ˆà ch =ˆàh/float(self.field.h)
228ˆà ˆà ˆà ˆà c =ˆàmin(cw,ˆàch)
229ˆà ˆà ˆà ˆà field_geometry =ˆà(self.field.w*c,self.field.h*c)
230ˆà ˆà ˆà ˆà offset =ˆà((w -ˆàfield_geometry[0])/2.0,ˆà(h -ˆàfield_geometry[1])/2.0)
231ˆà ˆà ˆà ˆà forˆàcell_coord inˆàself.field:
232ˆà ˆà ˆà ˆà ˆà ˆà self.field[cell_coord].redraw(offset,ˆàc)
233ˆà ˆà ˆà ˆà return
234ˆà ˆà
235ˆà ˆà defˆàlegal_moves(self,ˆàsnake):
236ˆà ˆà ˆà ˆà """Check for snake legal move directions according to the game rules."""
237ˆà ˆà ˆà ˆà snake.legal_dir =ˆà[]
238ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0]
239ˆà ˆà ˆà ˆà forˆàdirection inˆàdirections:
240ˆà ˆà ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[head.x +ˆàdirection[0],ˆàhead.y +ˆàdirection[1]]
241ˆà ˆà ˆà ˆà ˆà ˆà ifˆà(dir_cell.type ==ˆà'empty'ˆàorˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake)):
242ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.legal_dir.append(direction)
243ˆà ˆà ˆà ˆà rnd.shuffle(snake.legal_dir)
244ˆà ˆà ˆà ˆà return
245
Note: See TracBrowser for help on using the repository browser.