| 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,ˆàself.canvas.winfo_height()ˆà-ˆàfield_size) |
|---|
| 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.redraw() |
|---|
| 76 | ˆà ˆà ˆà ˆà return |
|---|
| 77 | ˆà ˆà defˆàstep(self): |
|---|
| 78 | ˆà ˆà ˆà ˆà forˆài,ˆàsnake inˆàenumerate(self.snakes): |
|---|
| 79 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆàlen(snake.cells)ˆà==ˆà0: |
|---|
| 80 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.snakes[i]ˆà=ˆàNone |
|---|
| 81 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆàsnake ==ˆàNone: |
|---|
| 82 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass |
|---|
| 83 | ˆà ˆà ˆà ˆà ˆà ˆà else: |
|---|
| 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 | ˆà ˆà ˆà ˆà dir_to_numb_of_rot={ |
|---|
| 91 | ˆà ˆà ˆà ˆà ˆà ˆà (0,-1):0, |
|---|
| 92 | ˆà ˆà ˆà ˆà ˆà ˆà (1,0):1, |
|---|
| 93 | ˆà ˆà ˆà ˆà ˆà ˆà (0,1):2, |
|---|
| 94 | ˆà ˆà ˆà ˆà ˆà ˆà (-1,0):3} |
|---|
| 95 | ˆà ˆà ˆà ˆà applied_dir =ˆàNone |
|---|
| 96 | ˆà ˆà ˆà ˆà forˆàrule inˆàsnake.rules: |
|---|
| 97 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆàapplied_dir ==ˆàNone: |
|---|
| 98 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà forˆàdirection inˆàsnake.legal_dir: |
|---|
| 99 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà rule.rotate(dir_to_numb_of_rot[direction]) |
|---|
| 100 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàrule.applies()ˆà==ˆàTrue: |
|---|
| 101 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà applied_dir =ˆàdirection |
|---|
| 102 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà break |
|---|
| 103 | ˆà ˆà ˆà ˆà ˆà ˆà else: |
|---|
| 104 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàapplied_dir) |
|---|
| 105 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà break |
|---|
| 106 | ˆà ˆà ˆà ˆà ifˆàapplied_dir ==ˆàNone: |
|---|
| 107 | ˆà ˆà ˆà ˆà ˆà ˆà self.move_do(snake,ˆàsnake.legal_dir[0]) |
|---|
| 108 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
|---|
| 109 | ˆà ˆà ˆà ˆà return |
|---|
| 110 | ˆà ˆà defˆàmove_do(self,ˆàsnake,ˆàapplied_dir): |
|---|
| 111 | ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[snake.cells[0].y +ˆàapplied_dir[0],ˆàsnake.cells[0].x +ˆàapplied_dir[1]] |
|---|
| 112 | ˆà ˆà ˆà ˆà ifˆàdir_cell.type ==ˆà'empty': |
|---|
| 113 | ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell) |
|---|
| 114 | ˆà ˆà ˆà ˆà ˆà ˆà delˆàsnake.cells[-1] |
|---|
| 115 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
|---|
| 116 | ˆà ˆà ˆà ˆà elifˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake): |
|---|
| 117 | ˆà ˆà ˆà ˆà ˆà ˆà snake.cells.insert(0,dir_cell) |
|---|
| 118 | ˆà ˆà ˆà ˆà ˆà ˆà delˆàdir_cell.snake.cells[-1] |
|---|
| 119 | ˆà ˆà ˆà ˆà ˆà ˆà pass |
|---|
| 120 | |
|---|
| 121 | ˆà ˆà defˆàcreate_snake(self,ˆàsnake_number): |
|---|
| 122 | ˆà ˆà ˆà ˆà cells_id =ˆà[] |
|---|
| 123 | ˆà ˆà ˆà ˆà forˆày inˆàrange(10): |
|---|
| 124 | ˆà ˆà ˆà ˆà ˆà ˆà cells_id.append((10,ˆày+1)) |
|---|
| 125 | ˆà ˆà ˆà ˆà forˆàrot_num inˆàrange(snake_number -ˆà1): |
|---|
| 126 | ˆà ˆà ˆà ˆà ˆà ˆà forˆài,ˆàcell inˆàenumerate(cells_id): |
|---|
| 127 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà 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) |
|---|
| 128 | ˆà ˆà ˆà ˆà cells =ˆà[] |
|---|
| 129 | ˆà ˆà ˆà ˆà forˆàcell inˆàcells_id: |
|---|
| 130 | ˆà ˆà ˆà ˆà ˆà ˆà cells.append(self.field[cell]) |
|---|
| 131 | ˆà ˆà ˆà ˆà color_dic =ˆà{ |
|---|
| 132 | ˆà ˆà ˆà ˆà ˆà ˆà 1:'blue', |
|---|
| 133 | ˆà ˆà ˆà ˆà ˆà ˆà 2:'green', |
|---|
| 134 | ˆà ˆà ˆà ˆà ˆà ˆà 3:'yellow', |
|---|
| 135 | ˆà ˆà ˆà ˆà ˆà ˆà 4:'red',} |
|---|
| 136 | ˆà ˆà ˆà ˆà self.snakes[snake_number-1]ˆà=ˆàsnake.Snake(cells,ˆàcolor_dic[snake_number]) |
|---|
| 137 | ˆà ˆà ˆà ˆà returnˆàself.snakes[snake_number-1] |
|---|
| 138 | ˆà ˆà defˆàrefill(self): |
|---|
| 139 | ˆà ˆà ˆà ˆà forˆàx inˆàrange(1,20): |
|---|
| 140 | ˆà ˆà ˆà ˆà ˆà ˆà forˆày inˆàrange(1,20): |
|---|
| 141 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].type =ˆà'empty' |
|---|
| 142 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.field[x,ˆày].snake =ˆàNone |
|---|
| 143 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass |
|---|
| 144 | ˆà ˆà ˆà ˆà forˆàsnake inˆàself.snakes: |
|---|
| 145 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆàsnake ==ˆàNone: |
|---|
| 146 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass |
|---|
| 147 | ˆà ˆà ˆà ˆà ˆà ˆà else: |
|---|
| 148 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.fill() |
|---|
| 149 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pass |
|---|
| 150 | ˆà ˆà ˆà ˆà return |
|---|
| 151 | ˆà ˆà defˆàredraw(self): |
|---|
| 152 | ˆà ˆà ˆà ˆà self.canvas.delete(all) |
|---|
| 153 | ˆà ˆà ˆà ˆà forˆàcell_coord inˆàself.field: |
|---|
| 154 | ˆà ˆà ˆà ˆà ˆà ˆà self.field[cell_coord].redraw() |
|---|
| 155 | ˆà ˆà ˆà ˆà return |
|---|
| 156 | ˆà ˆà defˆàlegal_moves(self,ˆàsnake): |
|---|
| 157 | ˆà ˆà ˆà ˆà snake.legal_dir =ˆà[] |
|---|
| 158 | ˆà ˆà ˆà ˆà forˆàdirection inˆàdirections: |
|---|
| 159 | ˆà ˆà ˆà ˆà ˆà ˆà dir_cell =ˆàself.field[snake.cells[0].y +ˆàdirection[0],ˆàsnake.cells[0].x +ˆàdirection[1]] |
|---|
| 160 | ˆà ˆà ˆà ˆà ˆà ˆà ifˆà(dir_cell.type ==ˆà'empty'ˆàorˆà(dir_cell.type ==ˆà'tail'ˆàandˆàdir_cell.snake !=ˆàsnake)): |
|---|
| 161 | ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà snake.legal_dir.append(direction) |
|---|
| 162 | ˆà ˆà ˆà ˆà rnd.shuffle(snake.legal_dir) |
|---|
| 163 | ˆà ˆà ˆà ˆà return |
|---|
| 164 | |
|---|