update tmc

tmc.py

@@ -62,7 +62,7 @@ class TransitionMatrixCalculator:
                 # handle the fast lane
                 if k == 2 and s > 0:
                     p /= 2
-                    k_prime = 10 + (s - 1)
+                    k_prime = 10 + (s - 1) # rebalance the step before with s > 0
                     if layout[k_prime] in [0, 3]:  # normal or prison square
                         self.matrix_normal[k,k_prime] += p
                     elif layout[k_prime] == 1:  # handle type 1 trap
@@ -80,7 +80,7 @@ class TransitionMatrixCalculator:
                         else:
                             k_prime = max(0, k_prime - 3)
                         self.matrix_normal[k,k_prime] += p / 2
-                    k_prime = 3 + (s - 1)
+                    k_prime = 3 + (s - 1) # rebalance the step before with s > 0
                     if layout[k_prime] in [0, 3]:  # normal or prison square
                         self.matrix_normal[k,k_prime] += p
                     elif layout[k_prime] == 1:  # handle type 1 trap
@@ -94,7 +94,7 @@ class TransitionMatrixCalculator:
                     continue
 
                 k_prime = k + s
-                k_prime = k_prime % 15 if circle else min(14, k_prime)
+                k_prime = k_prime % 15 if circle else min(14, k_prime) # modulo
                 if layout[k_prime] in [1, 2]:
                     p /= 2
                     if layout[k_prime] == 1:
@@ -113,7 +113,6 @@ class TransitionMatrixCalculator:
                         self.matrix_normal[k,k_prime] += p
                         continue
                 self.matrix_normal[k,k_prime] += p
-
         return self.matrix_normal
 
     def _compute_risky_matrix(self, layout, circle):