steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-05-23 07:16:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

adjusted loop fpr shadow-casting

Browse Source
This commit is contained in:
romue 2021-07-27 17:10:06 +02:00
parent 9e8d2ac1dc
commit 0fac3bbe8c
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
environments/factory/renderer.py
View File

@ -2,6 +2,7 @@ import sys
import numpy as np
from pathlib import Path
from collections import deque
from itertools import product
import pygame
from typing import NamedTuple, Any
import time
@ -76,8 +77,8 @@ class Renderer:
def visibility_rects(self, bp, view):
rects = []
for i in range(-self.view_radius, self.view_radius+1):
for j in range(-self.view_radius, self.view_radius+1):
for i, j in product(range(-self.view_radius, self.view_radius+1),
range(-self.view_radius, self.view_radius+1)):
if bool(view[self.view_radius+j, self.view_radius+i]):
visibility_rect = bp['dest'].copy()
visibility_rect.centerx += i*self.cell_size