source: snake.py @ 43:2a7b76d66e78

importˆàengine

defˆàpreprocess(line):
ˆà ˆà ifˆà'//'ˆàinˆàline:
ˆà ˆà ˆà ˆà line =ˆàline[:line.index('//')]
ˆà ˆà line =ˆàline.rstrip()
ˆà ˆà returnˆàline

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

ˆà ˆà defˆàloadˆà(self,ˆàfile):
ˆà ˆà ˆà ˆà magic,ˆàname =ˆàpreprocess(file.readline()).split(' ',ˆà1)
ˆà ˆà ˆà ˆà assertˆàmagic ==ˆà"snake",ˆà"This is not snake file"
ˆà ˆà ˆà ˆà whileˆàTrue:
ˆà ˆà ˆà ˆà ˆà ˆà line =ˆàpreprocess(file.readline())
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàline ==ˆà'end':
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà break
ˆà ˆà ˆà ˆà ˆà ˆà assertˆàline ==ˆà'',ˆà"Rules must be separated by empty lines"
ˆà ˆà ˆà ˆà ˆà ˆà self.rules.append(Rule().load(file))

ˆà ˆà defˆàfillˆà(self):
ˆà ˆà ˆà ˆà forˆàcell inˆàself.cells:
ˆà ˆà ˆà ˆà ˆà ˆà cell.snake =ˆàself
ˆà ˆà ˆà ˆà snake.cells[0].type =ˆà'head'
ˆà ˆà ˆà ˆà snake.cells[1:-1].type =ˆà'body'
ˆà ˆà ˆà ˆà snake.cells[-1].type =ˆà'tail'
ˆà ˆà ˆà ˆà return

classˆàRule(object):

ˆà ˆà 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):
ˆà ˆà ˆà ˆà y =ˆà0
ˆà ˆà ˆà ˆà forˆàline inˆàfile:
ˆà ˆà ˆà ˆà ˆà ˆà line =ˆàpreprocess(line)
ˆà ˆà ˆà ˆà ˆà ˆà ifˆày ==ˆà0ˆàandˆàline ==ˆà'':
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà continue
ˆà ˆà ˆà ˆà ˆà ˆà ifˆày ==ˆà7:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà break
ˆà ˆà ˆà ˆà ˆà ˆà 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[:8]):
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà self.parse_cell(x,ˆày,ˆàchar)
ˆà ˆà ˆà ˆà ˆà ˆà y +=ˆà1

ˆà ˆà defˆàparse_cell(self,ˆàx,ˆày,ˆàchar):
ˆà ˆà ˆà ˆà assertˆàchar.lower()ˆàinˆàself.codes,ˆà"Illegal symbol in rule: %s"ˆà%ˆàchar
ˆà ˆà ˆà ˆà cell =ˆàengine.Cell(x,ˆày,ˆàNone)
ˆà ˆà ˆà ˆà ifˆàchar inˆà'htb':
ˆà ˆà ˆà ˆà ˆà ˆà ifˆàchar.islower():
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà cell.snake =ˆà'my'
ˆà ˆà ˆà ˆà ˆà ˆà else:
ˆà ˆà ˆà ˆà ˆà ˆà ˆà ˆà cell.snake =ˆà'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):
ˆà ˆà ˆà ˆà 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):
ˆà ˆà ˆà ˆà forˆài inˆàrange(((rot %ˆà4)ˆà+ˆà4)ˆà%ˆà4):
ˆà ˆà ˆà ˆà ˆà ˆà self.rotate_ccw()

ˆà ˆà defˆàrotate_ccw(self):
ˆà ˆà ˆà ˆà 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: