source: snake.py @ 70:e8eabd9530a1

"""Guts of snakes."""

importˆàengine

defˆàpreprocess(line):
ˆà ˆà """Remove comments and junk spaces from line of snake definition file."""
ˆà ˆà ifˆà'//'ˆàinˆàline:
ˆà ˆà ˆà ˆà line =ˆàline[:line.index('//')]
ˆà ˆà line =ˆàline.rstrip()
ˆà ˆà returnˆàline

classˆàSnake(object):
ˆà ˆà """Snakes.

ˆà ˆà Attributes:

ˆà ˆà - `cells` -- list of cells belonging to the snake The first of these cells
ˆà ˆà ˆà becomes head, the last one becomes tail, the rest ar body. If snake has
ˆà ˆà ˆà only one cell, it is tail.
ˆà ˆà - `color` -- color of snake
ˆà ˆà - `rules` -- a list of Rule objects
ˆà ˆà """

ˆà ˆà defˆà__init__ˆà(self,ˆàcells,ˆàcolor):
ˆà ˆà ˆà ˆà self.cells =ˆàcells
ˆà ˆà ˆà ˆà self.color =ˆàcolor
ˆà ˆà ˆà ˆà self.rules =ˆà[]

ˆà ˆà defˆàloadˆà(self,ˆàfile):
ˆà ˆà ˆà ˆà """Load snake description from file.
ˆà ˆà ˆà ˆà 
ˆà ˆà ˆà ˆà See program design docs for file syntax.
ˆà ˆà ˆà ˆà """
ˆà ˆà ˆà ˆà forˆàline inˆàfile:
ˆà ˆà ˆà ˆà ˆà ˆà magic,ˆàname =ˆàpreprocess(line).split(' ',ˆà1)
ˆà ˆà ˆà ˆà ˆà ˆà break
ˆà ˆà ˆà ˆà assertˆàmagic ==ˆà"snake",ˆà"This is not snake file"
ˆà ˆà ˆà ˆà forˆàline inˆàfile:
ˆà ˆà ˆà ˆà ˆà ˆà line =ˆàpreprocess(line)
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàline ==ˆà'end':
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà break
ˆà ˆà ˆà ˆà ˆà ˆà assertˆàline ==ˆà'',ˆà"Rules must be separated by empty lines"
ˆà ˆà ˆà ˆà ˆà ˆà self.rules.append(Rule(self).load(file))

ˆà ˆà defˆàfillˆà(self):
ˆà ˆà ˆà ˆà """Mark every cell in `self.cells` as belonging to self."""
ˆà ˆà ˆà ˆà forˆàcell inˆàself.cells:
ˆà ˆà ˆà ˆà ˆà ˆà cell.snake =ˆàself
ˆà ˆà ˆà ˆà snake.cells[0].type =ˆà'head'
ˆà ˆà ˆà ˆà forˆàcell inˆàself.cells[1:-1]:
ˆà ˆà ˆà ˆà ˆà ˆà cell.type =ˆà'body'
ˆà ˆà ˆà ˆà snake.cells[-1].type =ˆà'tail'
ˆà ˆà ˆà ˆà return

classˆàRule(object):
ˆà ˆà """Rule defining possible behaviour of snake."""

ˆà ˆà codes =ˆà{
ˆà ˆà ˆà ˆà 'h':ˆà'head',
ˆà ˆà ˆà ˆà 'b':ˆà'body',
ˆà ˆà ˆà ˆà 't':ˆà'tail',
ˆà ˆà ˆà ˆà '#':ˆà'wall',
ˆà ˆà ˆà ˆà ' ':ˆà'any',
ˆà ˆà ˆà ˆà '-':ˆà'empty',
ˆà ˆà }

ˆà ˆà defˆà__init__ˆà(self,ˆàsnake):
ˆà ˆà ˆà ˆà self.snake =ˆàsnake
ˆà ˆà ˆà ˆà self.direction =ˆà(1,ˆà0)
ˆà ˆà ˆà ˆà self.pattern =ˆà{}

ˆà ˆà defˆàloadˆà(self,ˆàfile):
ˆà ˆà ˆà ˆà """Load rule definition from file.
ˆà ˆà ˆà ˆà 
ˆà ˆà ˆà ˆà Ignore any leading empty lines.
ˆà ˆà ˆà ˆà Return self.
ˆà ˆà ˆà ˆà """
ˆà ˆà ˆà ˆà y =ˆà0
ˆà ˆà ˆà ˆà forˆàline inˆàfile:
ˆà ˆà ˆà ˆà ˆà ˆà line =ˆàpreprocess(line)
ˆà ˆà ˆà ˆà ˆà ˆà ifˆày ==ˆà0ˆàandˆàline ==ˆà'':
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà continue
ˆà ˆà ˆà ˆà ˆà ˆà assertˆàlen(line)ˆà==ˆà8,ˆà"Rule lines must be exactly 7 chars long"
ˆà ˆà ˆà ˆà ˆà ˆà assertˆàline[-1]ˆà==ˆà';',ˆà"Rule lines must end with semicolon"
ˆà ˆà ˆà ˆà ˆà ˆà forˆàx,ˆàchar inˆàenumerate(line[:7]):
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.parse_cell(x,ˆày,ˆàchar)
ˆà ˆà ˆà ˆà ˆà ˆà y +=ˆà1
ˆà ˆà ˆà ˆà ˆà ˆà ifˆày ==ˆà7:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà break
ˆà ˆà ˆà ˆà returnˆàself

ˆà ˆà defˆàparse_cell(self,ˆàx,ˆày,ˆàchar):
ˆà ˆà ˆà ˆà """Parse definition of cell in rule file.

ˆà ˆà ˆà ˆà Cell is defined by one character.
ˆà ˆà ˆà ˆà """
ˆà ˆà ˆà ˆà assertˆàchar.lower()ˆàinˆàself.codes,ˆà"Illegal symbol in rule: %s"ˆà%ˆàchar
ˆà ˆà ˆà ˆà cell =ˆàengine.Cell(x,ˆày,ˆàself.snake)
ˆà ˆà ˆà ˆà ifˆàchar inˆà'htb':
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàchar.islower():
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà cell.snake_type =ˆà'my'
ˆà ˆà ˆà ˆà ˆà ˆà else:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà cell.snake_type =ˆà'enemy'
ˆà ˆà ˆà ˆà ifˆàchar ==ˆà'h':
ˆà ˆà ˆà ˆà ˆà ˆà assertˆà(x,ˆày)ˆà==ˆà(3,ˆà3),ˆà"Own head must in the center of rule"
ˆà ˆà ˆà ˆà ifˆà(x,ˆày)ˆà==ˆà(3,ˆà3):
ˆà ˆà ˆà ˆà ˆà ˆà assertˆàchar ==ˆà'h',ˆà"In the center of rule must be own head"
ˆà ˆà ˆà ˆà cell.type =ˆàself.codes[char.lower()]
ˆà ˆà ˆà ˆà self.pattern[x,ˆày]ˆà=ˆàcell

ˆà ˆà defˆàappliesˆà(self,ˆàfield,ˆàx,ˆày):
ˆà ˆà ˆà ˆà """True if the rule applies in the field at position (x,y)."""
ˆà ˆà ˆà ˆà forˆàpx,ˆàfx inˆàzip(range(7),ˆàrange(x -ˆà3,ˆàx +ˆà4)):
ˆà ˆà ˆà ˆà ˆà ˆà forˆàpy,ˆàfy inˆàzip(range(7),ˆàrange(y -ˆà3,ˆày +ˆà4)):
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆà(fx,ˆàfy)ˆàinˆàfield:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàfield[fx,ˆàfy]ˆà!=ˆàself.pattern[px,ˆàpy]:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà else:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ifˆàself.pattern[px,ˆàpy].type !=ˆà'any':
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse
ˆà ˆà ˆà ˆà returnˆàTrue

ˆà ˆà defˆàrotateˆà(self,ˆàrot):
ˆà ˆà ˆà ˆà """Rotate rule pattern `rot` times counterclockwise."""
ˆà ˆà ˆà ˆà forˆài inˆàrange(((rot %ˆà4)ˆà+ˆà4)ˆà%ˆà4):
ˆà ˆà ˆà ˆà ˆà ˆà self.rotate_ccw()

ˆà ˆà defˆàrotate_ccw(self):
ˆà ˆà ˆà ˆà """Rotate rule pattern one time counterclockwise."""
ˆà ˆà ˆà ˆà pattern =ˆà{}
ˆà ˆà ˆà ˆà forˆàx inˆàrange(7):
ˆà ˆà ˆà ˆà ˆà ˆà forˆày inˆàrange(7):
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà pattern[y,ˆà6ˆà-ˆàx]ˆà=ˆàself.pattern[x,ˆày]
ˆà ˆà ˆà ˆà self.pattern =ˆàpattern
ˆà ˆà ˆà ˆà x,ˆày =ˆàself.direction
ˆà ˆà ˆà ˆà self.direction =ˆày,ˆà-x

# vim: set ts=4 sts=4 sw=4 et: