source: snake.py @ 139:cd7658cb90eb

"""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ˆàFile(object):
ˆà ˆà """Wrapper around file that saves the current line number."""
ˆà ˆà defˆà__init__(self,ˆàfile):
ˆà ˆà ˆà ˆà self.file =ˆàfile
ˆà ˆà ˆà ˆà self.name =ˆàfile.name
ˆà ˆà ˆà ˆà self.line_no =ˆà0
ˆà ˆà ˆà ˆà self.iterator =ˆàself.enumerate_lines()
ˆà ˆà defˆà__iter__(self):
ˆà ˆà ˆà ˆà returnˆàself.iterator
ˆà ˆà defˆàenumerate_lines(self):
ˆà ˆà ˆà ˆà forˆàline_no,ˆàline inˆàenumerate(self.file,ˆàself.line_no):
ˆà ˆà ˆà ˆà ˆà ˆà self.line_no =ˆàline_no
ˆà ˆà ˆà ˆà ˆà ˆà yieldˆà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.
ˆà ˆà ˆà ˆà """
ˆà ˆà ˆà ˆà fileˆà=ˆàFile(file)
ˆà ˆà ˆà ˆà try:
ˆà ˆà ˆà ˆà ˆà ˆà self._load(file)
ˆà ˆà ˆà ˆà exceptˆàException,ˆàe:
ˆà ˆà ˆà ˆà ˆà ˆà raiseˆàException("%s:%s: %s"ˆà%ˆà(file.name,ˆàfile.line_no,ˆàe))

ˆà ˆà defˆà_loadˆà(self,ˆàfile):
ˆà ˆà ˆà ˆà """Actually do the loading."""
ˆà ˆà ˆà ˆà forˆàline inˆàfile:
ˆà ˆà ˆà ˆà ˆà ˆà magic,ˆàself.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))
ˆà ˆà ˆà ˆà forˆànumber,ˆàrule inˆàenumerate(self.rules):
ˆà ˆà ˆà ˆà ˆà ˆà rule.number =ˆànumber

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

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 =ˆà(0,ˆà-1)
ˆà ˆà ˆà ˆà 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.
ˆà ˆà ˆà ˆà """
ˆà ˆà ˆà ˆà is_my =ˆàchar.islower()
ˆà ˆà ˆà ˆà char =ˆàchar.lower()
ˆà ˆà ˆà ˆà assertˆàchar inˆàself.codes,ˆà"Illegal symbol in rule: %s"ˆà%ˆàchar
ˆà ˆà ˆà ˆà cell =ˆàengine.Cell(x,ˆày)
ˆà ˆà ˆà ˆà cell.snake =ˆàself.snake
ˆà ˆà ˆà ˆà cell.snake_type =ˆàNone
ˆà ˆà ˆà ˆà ifˆàchar inˆà'htb':
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàis_my:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà 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]
ˆà ˆà ˆà ˆà self.pattern[x,ˆày]ˆà=ˆàcell

ˆà ˆà defˆàappliesˆà(self,ˆàfield,ˆàx,ˆày):
ˆà ˆà ˆà ˆà """True if the rule applies in the field at position (x,y)."""
ˆà ˆà ˆà ˆà wall =ˆàengine.Cell(-1,ˆà-1)
ˆà ˆà ˆà ˆà wall.type =ˆà'void'

ˆà ˆà ˆà ˆà 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ˆàfield.get((fx,ˆàfy),ˆàwall)ˆà!=ˆàself.pattern[px,ˆàpy]:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà returnˆàFalse
ˆà ˆà ˆà ˆà returnˆàTrue

ˆà ˆà defˆàrotateˆà(self,ˆàdirection):
ˆà ˆà ˆà ˆà """Rotate rule pattern to head in `direction`."""
ˆà ˆà ˆà ˆà forˆài inˆàrange(4):
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàself.direction ==ˆàdirection:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà return
ˆà ˆà ˆà ˆà ˆà ˆà self.rotate_ccw()
ˆà ˆà ˆà ˆà raiseˆàAssertionError("Illegal direction: %s"ˆà%ˆàdirection)

ˆà ˆà 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: