Beispiel #1
0
    def test_readln_empty(self):
        read_pipe, lua_write_pipe = [None] * 2
        try:
            read_pipe = Pipe("env1", "read", 'r', "mame/pipes")
            lua_write_pipe = setup_pipe(read_pipe)

            with self.assertRaises(IOError) as context:
                read_pipe.readln(timeout=0.1)

            assert_that(
                str(context.exception),
                contains_string("Failed to read value from 'readPipe'"))
        finally:
            close_pipes(read_pipe, lua_write_pipe)
Beispiel #2
0
class DataPipe(object):

    def __init__(self, env_id, screen_dims, addresses, pipes_path):
        self.pipe = Pipe(env_id, "data", 'r', pipes_path)
        self.screenDims = screen_dims
        self.addresses = addresses

    def open(self, console):
        self.pipe.open(console)

    def close(self):
        self.pipe.close()

    # Generates the equivalent Lua code specifying what data the Lua engine should return every time step 
    def get_lua_string(self):
        return self.pipe.get_lua_string(args=[address.get_lua_string() for address in self.addresses.values()]+['s:bitmap_binary()'])

    def read_data(self, timeout=10):
        data = {}
        line = self.pipe.readln(timeout=timeout)
        cursor = 0
        for k in self.addresses.keys():
            cursor_end = cursor
            while line[cursor_end] != 43:
                cursor_end += 1
            part = line[cursor:cursor_end]
            data[k] = int(part.decode("utf-8"))
            cursor = cursor_end+1
        data["frame"] = np.frombuffer(line[cursor:], dtype='uint8').reshape(self.screenDims["height"], self.screenDims["width"], 3)
        return data
Beispiel #3
0
    def test_read_from_write_pipe(self):
        write_pipe, lua_read_pipe = [None] * 2
        try:
            write_pipe = Pipe("env1", "write", 'w', "mame/pipes")
            lua_read_pipe = setup_pipe(write_pipe)

            with self.assertRaises(IOError) as context:
                write_pipe.readln()

            assert_that(
                str(context.exception),
                contains_string(
                    "Attempted to read from '/home/michael/dev/MAMEToolkit/test/emulator/mame/pipes/write-env1.pipe' in 'w' mode"
                ))

        finally:
            close_pipes(write_pipe, lua_read_pipe)
Beispiel #4
0
def run_read(output_queue):
    read_pipe, lua_write_pipe = [None] * 2
    try:
        read_pipe = Pipe("env1", "read", 'r', "mame/pipes")
        lua_write_pipe = setup_pipe(read_pipe)

        lua_write_pipe.write("test\n")
        lua_write_pipe.flush()
        output_queue.put(read_pipe.readln(timeout=1))
    finally:
        close_pipes(read_pipe, lua_write_pipe)
Beispiel #5
0
    def test_read(self):
        read_pipe, lua_write_pipe = [None] * 2
        try:
            read_pipe = Pipe("env1", "read", 'r', "mame/pipes")
            lua_write_pipe = setup_pipe(read_pipe)

            lua_write_pipe.write("test\n")
            lua_write_pipe.flush()
            assert_that(read_pipe.readln(timeout=0.1), equal_to(b"test"))
        finally:
            close_pipes(read_pipe, lua_write_pipe)