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

source: engine.py @ 179:d73d8cecc812

Revision 179:d73d8cecc812, 8.6 KB checked in by Alex Martynov, 5 years ago (diff)

engine.field_geometry_calc() added
  • 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)ˆà<=ˆà1:
142ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàlen(snake.cells)ˆà==ˆà0:
143ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.snakes[i]ˆà=ˆàNone
144ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà continue
145ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.legal_moves(snake)
146ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.move_snake(snake)
147ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.refill()
148ˆà ˆà ˆà ˆà self.redraw()ˆà ˆà
149ˆà ˆà ˆà ˆà return
150ˆà ˆà
151ˆà ˆà defˆàmove_snake(self,ˆàsnake):
152ˆà ˆà ˆà ˆà """Check of movement direction based on the snake rule list and actual
153ˆà ˆà ˆà ˆà enviroment."""
154ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0]
155ˆà ˆà ˆà ˆà forˆàrule inˆàsnake.rules:
156ˆà ˆà ˆà ˆà ˆà ˆà forˆàdirection inˆàsnake.legal_dir:
157ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà rule.rotate(direction)
158ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàrule.applies(self.field,ˆàhead.x,ˆàhead.y)ˆà==ˆàTrue:
159ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàdirection)
160ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà return
161ˆà ˆà ˆà ˆà ifˆàsnake.legal_dir !=ˆà[]:
162ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàsnake.legal_dir[0])
163ˆà ˆà ˆà ˆà ˆà ˆà pass
164ˆà ˆà ˆà ˆà return
165ˆà ˆà
166ˆà ˆà defˆàmove_do(self,ˆàsnake,ˆàapplied_dir):
167ˆà ˆà ˆà ˆà """Do the move of the snake."""
168ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0]
169ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[head.x +ˆàapplied_dir[0],ˆàhead.y +ˆàapplied_dir[1]]
170ˆà ˆà ˆà ˆà ifˆàdir_cell.type ==ˆà'empty':
171ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell)
172ˆà ˆà ˆà ˆà ˆà ˆà delˆàsnake.cells[-1]
173ˆà ˆà ˆà ˆà ˆà ˆà pass
174ˆà ˆà ˆà ˆà elifˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake):
175ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell)
176ˆà ˆà ˆà ˆà ˆà ˆà delˆàdir_cell.snake.cells[-1]
177ˆà ˆà ˆà ˆà ˆà ˆà pass
178
179ˆà ˆà defˆàcreate_snake(self,ˆàsnake_number,ˆàold_snake =ˆàNone):
180ˆà ˆà ˆà ˆà """Create the snake:
181ˆà ˆà ˆà ˆà position choice is based on number or placement of 'Load' button
182ˆà ˆà ˆà ˆà ˆà ˆà snakes are placed with tails turned to the wall.
183ˆà ˆà ˆà ˆà color is chosen accorting to fen shui tradition"""
184ˆà ˆà ˆà ˆà cells_id =ˆà[]
185ˆà ˆà ˆà ˆà f_h =ˆàself.field.h
186ˆà ˆà ˆà ˆà f_w =ˆàself.field.w
187ˆà ˆà ˆà ˆà forˆày inˆàrange(self.start_snake_length):
188ˆà ˆà ˆà ˆà ˆà ˆà cells_id.insert(0,((f_w-1)/2,ˆày+1))
189ˆà ˆà ˆà ˆà forˆàrot_num inˆàrange(snake_number):
190ˆà ˆà ˆà ˆà ˆà ˆà forˆài,ˆàcell inˆàenumerate(cells_id):
191ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà cells_id[i]ˆà=ˆà(min(f_h,ˆàf_w)-1-cell[1],cell[0])
192ˆà ˆà ˆà ˆà cells =ˆà[]
193ˆà ˆà ˆà ˆà forˆàcell inˆàcells_id:
194ˆà ˆà ˆà ˆà ˆà ˆà cells.append(self.field[cell])
195ˆà ˆà ˆà ˆà color_dic =ˆà{
196ˆà ˆà ˆà ˆà ˆà ˆà 0:'blue',
197ˆà ˆà ˆà ˆà ˆà ˆà 1:'green',
198ˆà ˆà ˆà ˆà ˆà ˆà 2:'yellow',
199ˆà ˆà ˆà ˆà ˆà ˆà 3:'red',}
200ˆà ˆà ˆà ˆà ifˆàold_snake ==ˆàNone:
201ˆà ˆà ˆà ˆà ˆà ˆà self.snakes[snake_number]ˆà=ˆàsnake.Snake(cells,ˆàcolor_dic[snake_number])
202ˆà ˆà ˆà ˆà else:
203ˆà ˆà ˆà ˆà ˆà ˆà old_snake.cells =ˆàcells
204ˆà ˆà ˆà ˆà returnˆàself.snakes[snake_number]
205ˆà ˆà
206ˆà ˆà defˆàrefill(self):
207ˆà ˆà ˆà ˆà """Refill the field cells types and snakes according to the actual
208ˆà ˆà ˆà ˆà position"""
209ˆà ˆà ˆà ˆà f_w =ˆàself.field.w
210ˆà ˆà ˆà ˆà f_h =ˆàself.field.h
211ˆà ˆà ˆà ˆà forˆàx inˆàrange(1,f_w-1):
212ˆà ˆà ˆà ˆà ˆà ˆà forˆày inˆàrange(1,f_h-1):
213ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].type =ˆà'empty'
214ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].snake =ˆàNone
215ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass
216ˆà ˆà ˆà ˆà forˆàsnake inˆàself.snakes:
217ˆà ˆà ˆà ˆà ˆà ˆà ifˆàsnake ==ˆàNone:
218ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass
219ˆà ˆà ˆà ˆà ˆà ˆà else:
220ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.fill()
221ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass
222ˆà ˆà ˆà ˆà return
223ˆà ˆà
224ˆà ˆà defˆàredraw(self):
225ˆà ˆà ˆà ˆà """Clear the field Widget and redraw cells images on it"""
226ˆà ˆà ˆà ˆà self.canvas.delete("all")
227ˆà ˆà ˆà ˆà offset,ˆàc =ˆàself.field_geometry_calc()[1:]
228ˆà ˆà ˆà ˆà forˆàcell_coord inˆàself.field:
229ˆà ˆà ˆà ˆà ˆà ˆà self.field[cell_coord].redraw(offset,ˆàc)
230ˆà ˆà ˆà ˆà return
231
232ˆà ˆà defˆàfield_geometry_calcˆà(self):
233ˆà ˆà ˆà ˆà """Calculate grafical parameters of the field.
234ˆà ˆà ˆà ˆà Return:
235ˆà ˆà ˆà ˆà field_geometry - (width, hight) of the field
236ˆà ˆà ˆà ˆà offset - (x, y) of the left top corner of the field
237ˆà ˆà ˆà ˆà c - cell size"""
238ˆà ˆà ˆà ˆà w =ˆàself.canvas.winfo_width()
239ˆà ˆà ˆà ˆà h =ˆàself.canvas.winfo_height()
240ˆà ˆà ˆà ˆà cw =ˆàw/float(self.field.w)
241ˆà ˆà ˆà ˆà ch =ˆàh/float(self.field.h)
242ˆà ˆà ˆà ˆà c =ˆàmin(cw,ˆàch)
243ˆà ˆà ˆà ˆà field_geometry =ˆà(self.field.w*c,self.field.h*c)
244ˆà ˆà ˆà ˆà offset =ˆà((w -ˆàfield_geometry[0])/2.0,ˆà(h -ˆàfield_geometry[1])/2.0)
245ˆà ˆà ˆà ˆà returnˆàfield_geometry,ˆàoffset,ˆàc
246ˆà ˆà
247ˆà ˆà defˆàlegal_moves(self,ˆàsnake):
248ˆà ˆà ˆà ˆà """Check for snake legal move directions according to the game rules."""
249ˆà ˆà ˆà ˆà snake.legal_dir =ˆà[]
250ˆà ˆà ˆà ˆà head =ˆàsnake.cells[0]
251ˆà ˆà ˆà ˆà forˆàdirection inˆàdirections:
252ˆà ˆà ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[head.x +ˆàdirection[0],ˆàhead.y +ˆàdirection[1]]
253ˆà ˆà ˆà ˆà ˆà ˆà ifˆà(dir_cell.type ==ˆà'empty'ˆàorˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake)):
254ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.legal_dir.append(direction)
255ˆà ˆà ˆà ˆà rnd.shuffle(snake.legal_dir)
256ˆà ˆà ˆà ˆà return
257
Note: See TracBrowser for help on using the repository browser.