misc.py

def pack(words, wordsize=32):
    data = 0
    for i, word in enumerate(words):
        data = data | ((word & (2**wordsize-1)) << i*wordsize)
    return data

def unpack(data, n, wordsize=32):
    words = []
    for i in range(n):
        words.append((data >> i*wordsize) & (2**wordsize-1))
    return words

def repack(data, n, wordsize=32):
    packed_data = []
    for i in range(0, len(data), n):
        packed_data.append(pack(data[i:min(i+n, len(data))], wordsize=wordsize))
    return packed_data

def get_simulators(module, name, *args, **kwargs):
    simulators = []
    if hasattr(module, name):
        simulators.append(getattr(module, name)(*args, **kwargs))
    for _, submodule in module._submodules:
            for simulator in get_simulators(submodule, name, *args, **kwargs):
                    simulators.append(simulator)
    return simulators