import struct
from block import Block

class Demo(object):
    def __init__(self, filename):
        self.file = open(filename, 'rb')
        self.cdtrack = self.file.readline()

    def __iter__(self):
        while 1:
            b = self.file.read(4)
            if not b:
                return
            blocksize = struct.unpack('<i', b)[0]
            yield Block(b + self.file.read(12 + blocksize))


def test():
    for block in Demo('original.dem'):
        blocktime = None
        for message in block:
            if message.name == "Time":
                blocktime = message.time
                if blocktime > 10:
                    return

            if message.name == "UpdateEntity":
                print blocktime, message.entity, tuple(message.origin)
                import code
                code.interact(local=locals())
                return

if __name__ == "__main__":
    if 0:
        import profile
        profile.run("test()")
    else:
        test()