steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-05-23 15:26:43 +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 import numpy as np
from pathlib import Path from pathlib import Path
from collections import deque from collections import deque
from itertools import product
import pygame import pygame
from typing import NamedTuple, Any from typing import NamedTuple, Any
import time import time
@ -76,8 +77,8 @@ class Renderer:
def visibility_rects(self, bp, view): def visibility_rects(self, bp, view):
rects = [] rects = []
for i in range(-self.view_radius, self.view_radius+1): for i, j in product(range(-self.view_radius, self.view_radius+1),
for j in 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]): if bool(view[self.view_radius+j, self.view_radius+i]):
visibility_rect = bp['dest'].copy() visibility_rect = bp['dest'].copy()
visibility_rect.centerx += i*self.cell_size visibility_rect.centerx += i*self.cell_size