Added 'shared' dirt piles option for eval + Fixed usage of renderer + Added recorder option

marl_factory_grid/utils/renderer.py

@@ -156,7 +156,7 @@ class Renderer:
                     rects.append(dict(source=shape_surf, dest=visibility_rect))
         return rects
 
-    def render(self, entities):
+    def render(self, entities, recorder):
         """
         Renders the entities on the screen.
 
@@ -190,6 +190,11 @@ class Renderer:
         for blit in blits:
             self.screen.blit(**blit)
 
+        if recorder:
+            frame = pygame.surfarray.array3d(self.screen)
+            frame = np.transpose(frame, (1, 0, 2))  # Transpose to (height, width, channels)
+            recorder.append_data(frame)
+
         pygame.display.flip()
         self.clock.tick(self.fps)
         rgb_obs = pygame.surfarray.array3d(self.screen)