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 | 46 | print(d.toString()) |
47 | 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 | 51 | MM.printValues() |
67 | 52 | print("") | ... | ... |
structure.py
... | ... | @@ -10,13 +10,11 @@ class RSSVector(): |
10 | 10 | self.n4 = n4 |
11 | 11 | |
12 | 12 | class Location(): |
13 | - | |
14 | 13 | def __init__(self, x, y, z=0): |
15 | 14 | self.x = x |
16 | 15 | self.y = y |
17 | 16 | self.z = z |
18 | 17 | |
19 | - | |
20 | 18 | def __mul__(self, multiplier): |
21 | 19 | returnValue = Location(self.x, self.y, self.z) |
22 | 20 | returnValue.x *= multiplier |
... | ... | @@ -84,8 +82,7 @@ class MarkovModel(): |
84 | 82 | self.MarkovValues.append([]) |
85 | 83 | for k in range (0, 10): |
86 | 84 | self.MarkovValues[i].append(MarkovValue()) |
87 | - | |
88 | - | |
85 | + | |
89 | 86 | def moveToCellID(self, nextCell): |
90 | 87 | self.MarkovValues[self.previousCell][nextCell].nb += 1 |
91 | 88 | self.MarkovValues[10][nextCell].nb += 1 |
... | ... | @@ -135,6 +132,10 @@ class MarkovModel(): |
135 | 132 | max_id = k |
136 | 133 | return max_id |
137 | 134 | |
135 | + def path(self, locationIDs): | |
136 | + for loc in locationIDs: | |
137 | + self.moveToCellID(loc) | |
138 | + | |
138 | 139 | def toString(self): |
139 | 140 | return "" |
140 | 141 | ... | ... |