refactoring II

LP/InterruptIndices.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+# -*- coding: UTF-8 -*-
+from IOReader import load_indices
+from LPath import FILES_ALL, LPath
+
+
+#########################################
+#  InterruptIndices  :  Read chapters and extract indices (cluster by runes)
+#########################################
+
+class InterruptIndices(object):
+    def __init__(self):
+        self.pos = InterruptIndices.load()
+
+    def consider(self, name, irp, limit):
+        nums = self.pos[name]['pos'][irp]
+        total = self.pos[name]['total'] if len(nums) <= limit else nums[limit]
+        return nums[:limit], total
+
+    def total(self, name):
+        return self.pos[name]['total']
+
+    def longest_no_interrupt(self, name, irp, irpmax=0):
+        irpmax += 1
+        nums = self.pos[name]['pos'][irp] + [self.pos[name]['total']] * irpmax
+        ret = [(y - x, x) for x, y in zip(nums, nums[irpmax:])]
+        return sorted(ret, reverse=True)
+
+    @staticmethod
+    def write(dbname='db_indices'):
+        with open(LPath.InterruptDB(dbname), 'w') as f:
+            f.write('# file | total runes in file | interrupt | indices\n')
+            for name in FILES_ALL:
+                data = load_indices(LPath.page(name), 0)
+                total = len(data)
+                nums = [[] for x in range(29)]
+                for idx, rune in enumerate(data):
+                    nums[rune].append(idx)
+                for irp, pos in enumerate(nums):
+                    f.write('{}|{}|{}|{}\n'.format(
+                        name, total, irp, ','.join(map(str, pos))))
+
+    @staticmethod
+    def load(dbname='db_indices'):
+        with open(LPath.InterruptDB(dbname), 'r') as f:
+            ret = {}
+            for line in f.readlines():
+                if line.startswith('#'):
+                    continue
+                line = line.strip()
+                name, total, irp, nums = line.split('|')
+                if name not in ret:
+                    ret[name] = {'total': int(total),
+                                 'pos': [[] for _ in range(29)]}
+                pos = ret[name]['pos']
+                pos[int(irp)] = list(map(int, nums.split(','))) if nums else []
+            return ret
+
+
+if __name__ == '__main__':
+    # InterruptIndices.write()
+    for name, val in InterruptIndices.load().items():
+        print(name, 'total:', val['total'])
+        print(' ', [len(x) for x in val['pos']])