Deleting duplicate UnityEnvironment

Author: eshvk

Assets/ECSEnvironment.py

@@ -2,9 +2,9 @@
 import struct
 import numpy as np
 
+from python_communication import UnityCommunication
 from .brain import BrainInfo, BrainParameters
 
-
 class ECSEnvironment(object):
     VEC_SIZE = 8
     ACT_SIZE = 3
@@ -90,76 +90,6 @@ def external_brain_names(self):
 
 
 
-
-
-class UnityCommunication:
-    FILE_CAPACITY = 200000
-    NUMBER_AGENTS_POSITION = 0
-    SENSOR_SIZE_POSITION = 4
-    ACTUATOR_SIZE_POSITION = 8
-    UNITY_READY_POSITION = 12
-    SENSOR_DATA_POSITION = 13
-
-    PYTHON_READY_POSITION = 100000
-    ACTUATOR_DATA_POSITION = 100001
-
-    # FILE_NAME = "../../../ml-agents-ecs/Assets/shared_communication_file.txt"
-    FILE_NAME = "shared_communication_file.txt" # This is relative to where the script is called
-
-    def __init__(self):
-        with open(self.FILE_NAME, "r+b") as f:
-            # memory-map the file, size 0 means whole file
-            self.accessor = mmap.mmap(f.fileno(), 0)
-
-    def get_int(self, position : int) -> int:
-        return struct.unpack("i", self.accessor[position:position + 4])[0]
-
-    def read_sensor(self) -> np.ndarray:
-        sensor_size = self.get_int(self.SENSOR_SIZE_POSITION)
-        number_agents = self.get_int(self.NUMBER_AGENTS_POSITION)
-
-        sensor = np.frombuffer(
-            buffer=self.accessor[self.SENSOR_DATA_POSITION: self.SENSOR_DATA_POSITION + 4*sensor_size*number_agents],
-            dtype=np.float32,
-            count=sensor_size * number_agents,
-            offset=0
-        )
-        return np.reshape(sensor, (number_agents, sensor_size))
-
-    def get_parameters(self) -> (int, int, int):
-        return self.get_int(self.NUMBER_AGENTS_POSITION), \
-               self.get_int(self.SENSOR_SIZE_POSITION), \
-               self.get_int(self.ACTUATOR_SIZE_POSITION)
-
-    def write_actuator(self, actuator: np.ndarray):
-        actuator_size = self.get_int(self.ACTUATOR_SIZE_POSITION)
-        number_agents = self.get_int(self.NUMBER_AGENTS_POSITION)
-
-        # TODO : Support more types ?
-        if actuator.dtype != np.float32:
-            actuator = actuator.astype(np.float32)
-
-        try:
-            assert(actuator.shape == (number_agents, actuator_size))
-        except:
-            print("_________")
-            print(actuator.shape)
-            print((number_agents, actuator_size))
-
-        self.accessor[self.ACTUATOR_DATA_POSITION: self.ACTUATOR_DATA_POSITION + 4*actuator_size*number_agents] = \
-            actuator.tobytes()
-
-    def set_ready(self):
-        self.accessor[self.UNITY_READY_POSITION: self.UNITY_READY_POSITION+1] = bytearray(struct.pack("b", False))
-        self.accessor[self.PYTHON_READY_POSITION: self.PYTHON_READY_POSITION+1] = bytearray(struct.pack("b", True))
-
-    def unity_ready(self) -> bool:
-        return self.accessor[self.UNITY_READY_POSITION]
-
-    def close(self):
-        self.accessor.close()
-
-
 # if __name__ == "__main__":
 #     comm = UnityCommunication()
 #

Assets/python_communication.py

@@ -4,6 +4,7 @@
 import time
 
 
+
 class UnityCommunication:
     FILE_CAPACITY = 200000
     NUMBER_AGENTS_POSITION = 0
@@ -14,7 +15,7 @@ class UnityCommunication:
 
     PYTHON_READY_POSITION = 100000
     ACTUATOR_DATA_POSITION = 100001
-
+    # FILE_NAME = "../../../ml-agents-ecs/Assets/shared_communication_file.txt"
     FILE_NAME = "shared_communication_file.txt"
 
     def __init__(self):