Commit a862e105e62abb61d9a2aa88484bb008008fb4e2
1 parent
57a9ebb0
add function to tidy up main, minor formatting changes
Showing
2 changed files
with
6 additions
and
20 deletions
Show diff stats
main.py
@@ -46,22 +46,7 @@ def main(args): | @@ -46,22 +46,7 @@ def main(args): | ||
46 | print(d.toString()) | 46 | print(d.toString()) |
47 | print(d.getPositionInArray()) | 47 | print(d.getPositionInArray()) |
48 | """ | 48 | """ |
49 | - MM.moveToCellID(8) | ||
50 | - MM.moveToCellID(7) | ||
51 | - MM.moveToCellID(8) | ||
52 | - MM.moveToCellID(7) | ||
53 | - MM.moveToCellID(8) | ||
54 | - MM.moveToCellID(7) | ||
55 | - MM.moveToCellID(8) | ||
56 | - MM.moveToCellID(5) | ||
57 | - MM.moveToCellID(8) | ||
58 | - MM.moveToCellID(2) | ||
59 | - MM.moveToCellID(9) | ||
60 | - MM.moveToCellID(8) | ||
61 | - MM.moveToCellID(1) | ||
62 | - MM.moveToCell(Tf[2][2]) | ||
63 | - MM.moveToCellID(8) | ||
64 | - MM.moveToCell(Tf[2][2]) | 49 | + MM.path([8,7,8,7,8,7,8,5,8,2,9,8,1,9,8,9]) |
65 | 50 | ||
66 | MM.printValues() | 51 | MM.printValues() |
67 | print("") | 52 | print("") |
structure.py
@@ -10,13 +10,11 @@ class RSSVector(): | @@ -10,13 +10,11 @@ class RSSVector(): | ||
10 | self.n4 = n4 | 10 | self.n4 = n4 |
11 | 11 | ||
12 | class Location(): | 12 | class Location(): |
13 | - | ||
14 | def __init__(self, x, y, z=0): | 13 | def __init__(self, x, y, z=0): |
15 | self.x = x | 14 | self.x = x |
16 | self.y = y | 15 | self.y = y |
17 | self.z = z | 16 | self.z = z |
18 | 17 | ||
19 | - | ||
20 | def __mul__(self, multiplier): | 18 | def __mul__(self, multiplier): |
21 | returnValue = Location(self.x, self.y, self.z) | 19 | returnValue = Location(self.x, self.y, self.z) |
22 | returnValue.x *= multiplier | 20 | returnValue.x *= multiplier |
@@ -84,8 +82,7 @@ class MarkovModel(): | @@ -84,8 +82,7 @@ class MarkovModel(): | ||
84 | self.MarkovValues.append([]) | 82 | self.MarkovValues.append([]) |
85 | for k in range (0, 10): | 83 | for k in range (0, 10): |
86 | self.MarkovValues[i].append(MarkovValue()) | 84 | self.MarkovValues[i].append(MarkovValue()) |
87 | - | ||
88 | - | 85 | + |
89 | def moveToCellID(self, nextCell): | 86 | def moveToCellID(self, nextCell): |
90 | self.MarkovValues[self.previousCell][nextCell].nb += 1 | 87 | self.MarkovValues[self.previousCell][nextCell].nb += 1 |
91 | self.MarkovValues[10][nextCell].nb += 1 | 88 | self.MarkovValues[10][nextCell].nb += 1 |
@@ -135,6 +132,10 @@ class MarkovModel(): | @@ -135,6 +132,10 @@ class MarkovModel(): | ||
135 | max_id = k | 132 | max_id = k |
136 | return max_id | 133 | return max_id |
137 | 134 | ||
135 | + def path(self, locationIDs): | ||
136 | + for loc in locationIDs: | ||
137 | + self.moveToCellID(loc) | ||
138 | + | ||
138 | def toString(self): | 139 | def toString(self): |
139 | return "" | 140 | return "" |
140 | 141 |