ActivityTableGenerator.py

import gzip


class ATGenerator:

    """
    Reads saved activity binary files (table.bin)
    """

    def __init__(self, repre):

        self.representation = repre

    def read_table(self, file):
        activity_table = []
        with gzip.open(file, 'rb') as f:
            for line in f:
                for index, re in enumerate(self.representation):
                    if re in str(line):
                        activity_table.append(index)
                        break
        return bytearray(activity_table)