more formatting

.ruff.toml

@@ -0,0 +1,8 @@
+indent-width = 2
+line-length = 320
+
+[format]
+quote-style = "single"
+
+[lint]
+ignore = ["E741"]

other/decompose_fsm.py

@@ -10,11 +10,11 @@ import argparse
 # Stap 1: pip3 install python-sat
 # Stap 2: python3 decompose_fsm.py -h
 
-parser = argparse.ArgumentParser(description="Decomposes a FSM into smaller components by remapping its outputs. Uses a SAT solver.")
+parser = argparse.ArgumentParser(description='Decomposes a FSM into smaller components by remapping its outputs. Uses a SAT solver.')
 parser.add_argument('-c', '--components', type=int, default=2, help='number of components')
 parser.add_argument('-n', '--total-size', type=int, help='total number of states of the components')
-parser.add_argument('--add-state-trans', default=False, action="store_true", help='adds state transitivity constraints')
+parser.add_argument('--add-state-trans', default=False, action='store_true', help='adds state transitivity constraints')
-parser.add_argument('-v', '--verbose', default=False, action="store_true", help='prints more info')
+parser.add_argument('-v', '--verbose', default=False, action='store_true', help='prints more info')
 parser.add_argument('filename', help='path to .dot file')
 args = parser.parse_args()
 
@@ -46,6 +46,7 @@ class FSM:
   def output(self, s, a):
     return self.output_map[(s, a)]
 
+
 def parse_dot_file(lines):
   def parse_transition(line):
     (l, _, r) = line.partition('->')
@@ -83,6 +84,7 @@ def parse_dot_file(lines):
 
   return FSM(initial_state, states, inputs, outputs, transition_map, output_map)
 
+
 with open(args.filename) as file:
   machine = parse_dot_file(file)
   if args.verbose:
@@ -94,6 +96,7 @@ print(f'Initial size: {len(machine.states)}')
 ###################################
 # Utility functions
 
+
 def print_table(cell, rs, cs):
   first_col_size = max([len(str(r)) for r in rs])
   col_size = 1 + max([len(str(c)) for c in cs] + [len(cell(r, c)) for c in cs for r in rs])
@@ -109,6 +112,7 @@ def print_table(cell, rs, cs):
       print(cell(r, c).rjust(col_size), end='')
     print('')
 
+
 class Progress:
   def __init__(self, name, guess):
     self.reset(name, guess, show=False)
@@ -131,6 +135,7 @@ class Progress:
       print(f'{self.percentage}%', end='', flush=True)
       print('\r', end='')
 
+
 progress = Progress('', 1)
 
 ########################
@@ -141,13 +146,16 @@ rids = [i for i in range(c)] # components
 vpool = IDPool()
 cnf = CNF()
 
+
 # Een hulp variabele voor False en True, maakt de andere variabelen eenvoudiger
 def var_const(b):
-  return(vpool.id(('const', b)))
+  return vpool.id(('const', b))
+
 
 cnf.append([var_const(True)])
 cnf.append([-var_const(False)])
 
+
 # Voor elke relatie en elke twee elementen o1 en o2, is er een variabele die
 # aangeeft of o1 en o2 gerelateerd zijn. Er is 1 variabele voor xRy en yRx, dus
 # symmetrie is al ingebouwd. Reflexiviteit is ook ingebouwd.
@@ -156,7 +164,8 @@ def var_rel(rid, o1, o2):
     return var_const(True)
 
   [so1, so2] = sorted([o1, o2])
-  return(vpool.id(('rel', rid, so1, so2)))
+  return vpool.id(('rel', rid, so1, so2))
+
 
 # De relatie op outputs geeft een relaties op states. Deze relatie moet ook een
 # bisimulatie zijn.
@@ -165,12 +174,14 @@ def var_state_rel(rid, s1, s2):
     return var_const(True)
 
   [ss1, ss2] = sorted([s1, s2])
-  return(vpool.id(('state_rel', rid, ss1, ss2)))
+  return vpool.id(('state_rel', rid, ss1, ss2))
+
 
 # Voor elke relatie, en elke equivalentie-klasse, kiezen we precies 1 state
 # als representant. Deze variabele geeft aan welk element.
 def var_state_rep(rid, s):
-  return(vpool.id(('state_rep', rid, s)))
+  return vpool.id(('state_rep', rid, s))
+
 
 # Contraints zodat de relatie een equivalentie relatie is. We hoeven alleen
 # maar transitiviteit te encoderen, want refl en symm zijn ingebouwd in de var.
@@ -244,6 +255,7 @@ print('size constraints')
 cnf_optim = CardEnc.atmost([var_state_rep(rid, sx) for rid in rids for sx in machine.states], total_size, vpool=vpool)
 cnf.extend(cnf_optim)
 
+
 def print_eqrel(rel, xs):
   print_table(lambda r, c: 'Y' if rel(r, c) else '·', xs, xs)
 
@@ -261,15 +273,14 @@ with Solver(bootstrap_with=cnf) as solver:
     print('unsat :-(')
     exit()
 
-  print(f'sat :-)')
+  print('sat :-)')
 
   # Even omzetten in een makkelijkere data structuur
   print('- get model')
   m = solver.get_model()
   model = {}
   for l in m:
-    if l < 0: model[-l] = False
+    model[abs(l)] = l > 0
-    else: model[l] = True
 
   if args.verbose:
     for rid in rids:

other/decompose_observation_table.py

@@ -11,6 +11,7 @@ from pysat.formula import CNF
 ###################################
 # Wat dingetjes over Mealy machines
 
+
 # Voorbeeld: 2n states, input-alfabet 'a' en 'b', outputs [0...n-1]
 def rick_koenders_machine(N):
   transition_fun = {((n, False), 'a'): ((n + 1) % N, False) for n in range(N)}
@@ -23,6 +24,7 @@ def rick_koenders_machine(N):
   outputs = [n for n in range(N)]
   return {'transition_fun': transition_fun, 'output_fun': output_fun, 'initial_state': initial_state, 'inputs': inputs, 'outputs': outputs}
 
+
 def mealy_sem_q(machine, word, state):
   if len(word) == 0:
     return None
@@ -31,9 +33,11 @@ def mealy_sem_q(machine, word, state):
   else:
     return mealy_sem_q(machine, word[1:], machine['transition_fun'][(state, word[0])])
 
+
 def mealy_sem(machine, word):
   return mealy_sem_q(machine, word, machine['initial_state'])
 
+
 def print_table(cell, rs, cs):
   first_col_size = max([len(str(r)) for r in rs])
   col_size = 1 + max([len(str(c)) for c in cs])
@@ -76,13 +80,16 @@ rids = [i for i in range(c)] # components
 vpool = IDPool()
 cnf = CNF()
 
+
 # Een hulp variabele voor False en True, maakt de andere variabelen eenvoudiger
 def var_const(b):
-  return(vpool.id(('const', b)))
+  return vpool.id(('const', b))
+
 
 cnf.append([var_const(True)])
 cnf.append([-var_const(False)])
 
+
 # Voor elke relatie en elke twee elementen o1 en o2, is er een variabele die
 # aangeeft of o1 en o2 gerelateerd zijn. Er is 1 variabele voor xRy en yRx, dus
 # symmetrie is al ingebouwd. Reflexiviteit is ook ingebouwd.
@@ -91,7 +98,8 @@ def var_rel(rid, o1, o2):
     return var_const(True)
 
   [so1, so2] = sorted([o1, o2])
-  return(vpool.id(('rel', rid, so1, so2)))
+  return vpool.id(('rel', rid, so1, so2))
+
 
 # De relatie op outputs geeft een relaties op rijen. Deze zijn geindexeerd
 # met de woorden uit 'rows'.
@@ -100,12 +108,14 @@ def var_row_rel(rid, r1, r2):
     return var_const(True)
 
   [sr1, sr2] = sorted([r1, r2])
-  return(vpool.id(('row_rel', rid, sr1, sr2)))
+  return vpool.id(('row_rel', rid, sr1, sr2))
+
 
 # Voor elke relatie, en elke equivalentie-klasse, kiezen we precies 1 rij
 # als representant. Deze variabele geeft aan welk element.
 def var_row_rep(rid, r):
-  return(vpool.id(('row_rep', rid, r)))
+  return vpool.id(('row_rep', rid, r))
+
 
 # Contraints zodat de relatie een equivalentie relatie is. We hoeven alleen
 # maar transitiviteit te encoderen, want refl en symm zijn ingebouwd in de var.
@@ -147,7 +157,7 @@ for rid in rids:
       cnf.append([-var_rel(rid, ox, oy) for (ox, oy) in oss] + [var_row_rel(rid, rx, ry)])
 
       # rx ~ ry => oxi ~ oyi
-      for (ox, oy) in oss:
+      for ox, oy in oss:
         cnf.append([-var_row_rel(rid, rx, ry), var_rel(rid, ox, oy)])
 
 # De constraints die zorgen dat representanten ook echt representanten zijn.
@@ -169,6 +179,7 @@ print('- representatives at most k')
 cnf_optim = CardEnc.atmost([var_row_rep(rid, rx) for rid in rids for rx in rows], total_size, vpool=vpool)
 cnf.extend(cnf_optim)
 
+
 def print_eqrel(rel, xs):
   print_table(lambda r, c: 'Y' if rel(r, c) else '·', xs, xs)
 
@@ -185,15 +196,14 @@ with Solver(bootstrap_with=cnf) as solver:
     print('unsat :-(')
     exit()
 
-  print(f'sat :-)')
+  print('sat :-)')
 
   # Even omzetten in een makkelijkere data structuur
   print('- get model')
   m = solver.get_model()
   model = {}
   for l in m:
-    if l < 0: model[-l] = False
+    model[abs(l)] = l > 0
-    else: model[l] = True
 
   for rid in rids:
     print(f'Relation {rid}:')

other/decompose_set.py

@@ -37,13 +37,16 @@ print('Start encoding')
 vpool = IDPool()
 cnf = CNF()
 
+
 # Een hulp variabele voor False en True, maakt de andere variabelen eenvoudiger
 def var_const(b):
-  return(vpool.id(('bool', b)))
+  return vpool.id(('bool', b))
+
 
 cnf.append([var_const(True)])
 cnf.append([-var_const(False)])
 
+
 # Voor elke relatie en elke twee elementen o1 en o2, is er een variabele die
 # aangeeft of o1 en o2 gerelateerd zijn. Er is 1 variabele voor xRy en yRx, dus
 # symmetrie is al ingebouwd. Reflexiviteit is ook ingebouwd.
@@ -52,12 +55,14 @@ def var_rel(rid, o1, o2):
     return var_const(True)
 
   [so1, so2] = sorted([o1, o2])
-  return(vpool.id(('var_rel', rid, so1, so2)))
+  return vpool.id(('var_rel', rid, so1, so2))
+
 
 # Voor elke relatie, en elke equivalentie-klasse, kiezen we precies 1 element
 # als representant. Deze variabele geeft aan welk element.
 def var_rep(rid, o):
-  return(vpool.id(('var_rep', rid, o)))
+  return vpool.id(('var_rep', rid, o))
+
 
 # Contraints zodat de relatie een equivalentie relatie is. We hoeven alleen
 # maar transitiviteit te encoderen, want refl en symm zijn ingebouwd in de var.
@@ -109,15 +114,14 @@ with Solver(bootstrap_with=cnf) as solver:
     print('unsat :-(')
     exit()
 
-  print(f'sat :-)')
+  print('sat :-)')
 
   # Even omzetten in een makkelijkere data structuur
   print('- get model')
   m = solver.get_model()
   model = {}
   for l in m:
-    if l < 0: model[-l] = False
+    model[abs(l)] = l > 0
-    else: model[l] = True
 
   # print equivalence classes
   count = 0

other/fsm-examples.py

@@ -1,5 +1,6 @@
 import argparse
 
+
 class RKMachine:
   # Rick Koenders came up with this example in the case of N=3.
   # FSM will have 2*N states, but can be decomposed into N + 2 states.
@@ -27,6 +28,7 @@ class RKMachine:
   def output(self, s, a):
     return self.output_map[(s, a)]
 
+
 def fsm_to_dot(name, m):
   def transition_string(s, i, o, t):
     return f'{s} -> {t} [label="{i}/{o}"]'
@@ -44,6 +46,7 @@ def fsm_to_dot(name, m):
 
   return ret
 
+
 parser = argparse.ArgumentParser()
 parser.add_argument('machine', nargs='?', default='rk', choices=['rk'])
 parser.add_argument('-n', type=int, default=3, help='size parameter')

other/partitions.py

@@ -4,6 +4,7 @@ from math import prod
 N = 99999
 C = 16
 
+
 # precondition k <= n
 def all_partitions_(n, k):
   if n == 0:
@@ -16,9 +17,11 @@ def all_partitions_(n, k):
 
   return acc
 
+
 def all_partitions(n):
   return all_partitions_(n, n)
 
+
 def highest_product(n):
   ps = all_partitions(n)
 
@@ -37,6 +40,7 @@ def highest_product(n):
 
   return highest_prod
 
+
 def tabulate(upper):
   highest_n_per_c = {}
   table = {}
@@ -51,6 +55,7 @@ def tabulate(upper):
 
   return {(n, c): s for (n, c), s in table.items() if s <= n}
 
+
 def trim_tab(tab):
   best_s = {}
   best_c = {}
@@ -65,6 +70,7 @@ def trim_tab(tab):
 
   return {(n, c): s for (n, c), s in tab.items() if c <= best_c[n] or s <= best_s[n]}
 
+
 def print_table(tab0):
   tab = trim_tab(tab0)