Changeset 34:8f8af9ef99e6

Timestamp:

12/19/10 22:22:42 (5 years ago)

Author:

Daniil Alexeyevsky <me.dendik@òÀæ>

Branch:

default

Message:

snake: implemented Snake.load, Rule.load; untested yet

File:

• snake.py (modified) (4 diffs)

snake.py

@@ -ˆà +1 @@
+import engine
+
+def preprocess(line):
+    if '//' in line:
+        line = line[:line.index('//')]
+    line = line.rstrip()
+    return line
 
 class Snake(object):
@@ -5 +12 @@
         self.color = color
         self.rules = []
-        pass
-    def load (self, file_name):
-        snake_file = open(file_name, "r")
-        line = snake_file.readline()
-        if line.partition(' ')[0] == 'snake':
-            self.name = line.partition(' ')[2]
-            for line in snake_file:
-                if line.partition('\\')[0] == ('' or '\n'):
-                    continue
-                else:
-                    if line.partition(';')[0] != ''
-        
-        pass
+
+    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:
@@ -25 +30 @@
         snake.cells[1:-1].type = 'body'
         return
+
     def error (self):
         pass
@@ -30 +36 @@
 
 class Rule(object):
+
+    codes = {
+        'h': 'head',
+        'b': 'body',
+        't': 'tail',
+        '#': 'wall',
+        ' ': 'any',
+        '-': 'empty',
+    }
+
     def __init__ (self, snake):
         self.snake = snake
-        self.direction = (1, -1)
-        pass
-    def load (self, file, line):
-        
-        pass
+        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
+            assert line[-1] == ';', "Rule lines must end with semicolon"
+            assert len(line) == 8, "Rule lines must be exactly 7 chars long"
+            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'
+        cell.type = self.codes[char.lower()]
+        self.pattern[x, y] = cell
+
     def applies (self, field, x, y):
-        
+
         pass
     def rotate (self, rot):
         pass
+
+# vim: set ts=4 sts=4 sw=4 et: