def __init__(self, env): self.env = env pygame.init() pygame.display.set_caption("Highway-env") panel_size = (self.env.config["screen_width"], self.env.config["screen_height"]) self.screen = pygame.display.set_mode([ self.env.config["screen_width"], self.env.config["screen_height"] ]) self.sim_surface = WorldSurface(panel_size, 0, pygame.Surface(panel_size)) self.sim_surface.centering_position = env.config.get( "centering_position", self.sim_surface.INITIAL_CENTERING) # Big screen # panel_size = (5000, 150) # self.screen = pygame.display.set_mode([5000, 150]) # self.sim_surface = WorldSurface(panel_size, 0, pygame.Surface(panel_size)) # self.sim_surface.centering_position = (0.01,0.5) self.clock = pygame.time.Clock() self.enabled = True if "SDL_VIDEODRIVER" in os.environ and os.environ[ "SDL_VIDEODRIVER"] == "dummy": self.enabled = False self.agent_display = None self.agent_surface = None self.vehicle_trajectory = None self.frame = 0
def __init__(self, env: 'AbstractEnv') -> None: self.env = env self.offscreen = env.config["offscreen_rendering"] pygame.init() pygame.display.set_caption("Highway-env") panel_size = (self.env.config["screen_width"], self.env.config["screen_height"]) # A display is not mandatory to draw things. Ignoring the display.set_mode() # instruction allows the drawing to be done on surfaces without # handling a screen display, useful for e.g. cloud computing if not self.offscreen: self.screen = pygame.display.set_mode([self.env.config["screen_width"], self.env.config["screen_height"]]) self.sim_surface = WorldSurface(panel_size, 0, pygame.Surface(panel_size)) self.sim_surface.scaling = env.config.get("scaling", self.sim_surface.INITIAL_SCALING) self.sim_surface.centering_position = env.config.get("centering_position", self.sim_surface.INITIAL_CENTERING) self.clock = pygame.time.Clock() self.enabled = True if os.environ.get("SDL_VIDEODRIVER", None) == "dummy": self.enabled = False self.agent_display = None self.agent_surface = None self.vehicle_trajectory = None self.frame = 0 self.directory = None
def __init__(self, env): self.env = env pygame.init() pygame.display.set_caption("Highway-env") panel_size = (self.SCREEN_WIDTH, self.SCREEN_HEIGHT) self.screen = pygame.display.set_mode([self.SCREEN_WIDTH, self.SCREEN_HEIGHT]) self.sim_surface = WorldSurface(panel_size, 0, pygame.Surface(panel_size)) self.sim_surface.centering_position = env.config.get("centering_position", self.sim_surface.INITIAL_CENTERING) self.clock = pygame.time.Clock() self.enabled = True if "SDL_VIDEODRIVER" in os.environ and os.environ["SDL_VIDEODRIVER"] == "dummy": self.enabled = False self.agent_display = None self.agent_surface = None