def draw_orbit(self, screen, zoom_factor, offset_pos=(0, 0)):
draw_pos = (int(
to_screen(self.parent_planet.get_position())[0] -
offset_pos[0] * zoom_factor),
int(
to_screen(self.parent_planet.get_position())[1] -
offset_pos[1] * zoom_factor))
pygame.draw.circle(screen, self.color, draw_pos,
int(self.distance * zoom_factor), 1)
def draw_satellite_overlay(self, universe):
cursor_pos = pygame.mouse.get_pos()
satellite_list = universe.get_all_satellites()
for s in satellite_list:
s_pos = to_ints(to_screen(scale_pos(s.get_pos(), universe.get_zoom_factor())))
if get_distance_between(s_pos, cursor_pos) < 30*universe.get_zoom_factor():
HUD_pos = (s_pos[0]+20, s_pos[1]+5)
HUD_block = pygame.Surface((245, 70))
HUD_block.fill((100,100,100))
if s.get_nationality() == "Roscosmos1" or s.get_nationality() == "Roscosmos2" or s.get_nationality() == "Roscosmos3" or s.get_nationality() == "Roscosmos":
HUD_block.blit(self.RUS_logo, (50-self.RUS_logo.get_width()/2,35-self.RUS_logo.get_height()/2))
elif s.get_nationality() == "ESA1" or s.get_nationality() == "ESA2":
HUD_block.blit(self.ESA_logo, (50-self.ESA_logo.get_width()/2,35-self.ESA_logo.get_height()/2))
elif s.get_nationality() == "NASA1"or s.get_nationality() == "NASA2"or s.get_nationality() == "NASA":
HUD_block.blit(self.USA_logo, (40-self.USA_logo.get_width()/2,35-self.USA_logo.get_height()/2))
elif s.get_nationality() == "Alien" or s.get_nationality() == "ALIEN1" or s.get_nationality() == "ALIEN2"or s.get_nationality() == "ALIEN3" or s.get_nationality() == "ALIEN4":
HUD_block.blit(self.ALIEN_logo, (40-self.USA_logo.get_width()/2,35-self.USA_logo.get_height()/2))
elif s.get_nationality() == "Space Rock" or s.get_nationality() == "ROCK1" or s.get_nationality() == "ROCK2"or s.get_nationality() == "ROCK3" or s.get_nationality() == "ROCK4":
HUD_block.blit(self.QUESTION_logo, (40-self.USA_logo.get_width()/2,35-self.USA_logo.get_height()/2))
font = pygame.font.SysFont("Courier", 12)
HUD_block.blit(font.render("Name: %s"%(s.get_name()), True, (255,255,255)), (90,5))
HUD_block.blit(font.render("Agency: %s"%(s.get_nationality()), True, (255,255,255)), (90,20))
HUD_block.blit(font.render("Altitude: %s"%(s.get_altitude()), True, (255,255,255)), (90,35))
HUD_block.blit(font.render("Right-click to target", True, (255,255,255)), (90,50))
pygame.draw.rect(HUD_block, (255,255,255), HUD_block.get_rect(), 1)
HUD_block.set_alpha(150)
self.screen.blit(HUD_block, HUD_pos)
def draw_stars(self, screen, list):
resolution = (screen.get_rect().width, screen.get_rect().height)
i=0
for pos in list:
i+=1
if(i%3)==0:
draw_pos = to_screen(scale_pos(pos, self.zoom_factor))
if draw_pos[0] > -10 and draw_pos[0] < resolution[0]+10 and\
draw_pos[1] > -10 and draw_pos[1] < resolution[1]+10:
pygame.gfxdraw.filled_circle(screen, int(draw_pos[0]), int(draw_pos[1]), 1, (200,0,0))
if(i%3)==1:
draw_pos = to_screen(scale_pos(pos, self.zoom_factor))
if draw_pos[0] > -10 and draw_pos[0] < resolution[0]+10 and\
draw_pos[1] > -10 and draw_pos[1] < resolution[1]+10:
pygame.gfxdraw.filled_circle(screen, int(draw_pos[0]), int(draw_pos[1]), 1, (0,0,200))
if(i%3)==2:
draw_pos = to_screen(scale_pos(pos, self.zoom_factor))
if draw_pos[0] > -10 and draw_pos[0] < resolution[0]+10 and\
draw_pos[1] > -10 and draw_pos[1] < resolution[1]+10:
pygame.gfxdraw.filled_circle(screen, int(draw_pos[0]), int(draw_pos[1]), 1, (255,255,255))
def draw(self, screen, zoom_factor, offset_pos=(0, 0)):
draw_pos = (self.pos[0] - offset_pos[0], self.pos[1] - offset_pos[1])
self.launch_angle = get_angle_between(
pygame.mouse.get_pos(), to_screen(scale_pos(self.pos,
zoom_factor)))
if self.empty:
draw(screen, self.sprite_pad, draw_pos, self.angle - 90,
self.scale, zoom_factor)
else:
draw(screen, self.sprite_pad, draw_pos, self.angle - 90,
self.scale, zoom_factor)
draw(screen, self.sprite_rocket, draw_pos,
-1 * self.launch_angle + 90, self.scale, zoom_factor)
def draw(self, screen, zoom_factor, offset_pos=(0, 0)):
draw_pos = (self.pos[0] - offset_pos[0], self.pos[1] - offset_pos[1])
if not self.exploded:
if offset_pos == (0, 0):
self.angle = -1 * get_angle_between(
pygame.mouse.get_pos(),
to_screen(scale_pos(self.pos, zoom_factor))) + 90
if self.thrusting and self.fuel > 0:
draw(screen, self.sprite_thrusting, draw_pos, self.angle,
self.scale, zoom_factor)
else:
draw(screen, self.sprite, draw_pos, self.angle, self.scale,
zoom_factor)
def draw(self, screen):
if not self.init_view:
self.view_manager.set_target(0, self.planet.get_launcher(), screen)
self.init_view = True
if self.rocket_shot:
self.view_manager.set_target(0, self.rockets[-1], screen)
self.rocket_shot = False
if self.reset_view:
self.view_manager.set_target(0, self.get_launcher(), screen)
self.reset_view = False
if pygame.mouse.get_pressed()[2]:
for satellite in self.planet.get_satellites():
s_pos = to_ints(to_screen(scale_pos(satellite.get_pos(), self.zoom_factor)))
if get_distance_between(s_pos, pygame.mouse.get_pos()) < 30*self.zoom_factor:
self.view_manager.set_target(0, satellite, screen)
for rocket in self.planet.get_rockets():
rocket_pos = to_ints(to_screen(scale_pos(rocket.get_pos(), self.zoom_factor)))
if get_distance_between(rocket_pos, pygame.mouse.get_pos()) < 30*self.zoom_factor:
self.view_manager.set_target(0, rocket, screen)
launcher = self.get_launcher()
launcher_pos = to_ints(to_screen(scale_pos(launcher.get_pos(), self.zoom_factor)))
if get_distance_between(launcher_pos, pygame.mouse.get_pos()) < 30*self.zoom_factor:
self.view_manager.set_target(0, launcher, screen)
planet_pos = to_ints(to_screen(scale_pos(self.planet.get_pos(), self.zoom_factor)))
if get_distance_between(planet_pos, pygame.mouse.get_pos()) < 30*self.zoom_factor:
self.view_manager.set_target(0, self.planet, screen)
self.draw_stars(screen, self.stars)
self.planet.draw(screen, self.zoom_factor)
for rocket in self.rockets:
rocket.draw(screen, self.zoom_factor)
for explosion in self.explosions:
explosion.draw(screen, self.zoom_factor)
self.view_manager.draw(screen, self.planet, self.rockets, 1.25)
def draw_launcher_overlay(self, universe):
cursor_pos = pygame.mouse.get_pos()
launcher = universe.get_launcher()
launcher_pos = to_ints(to_screen(scale_pos(launcher.get_pos(), universe.get_zoom_factor())))
if get_distance_between(launcher_pos, cursor_pos) < 30*universe.get_zoom_factor():
HUD_pos = (launcher_pos[0]+20, launcher_pos[1]+5)
HUD_block = pygame.Surface((245, 70))
HUD_block.fill((100,100,100))
HUD_block.blit(self.launcher_sprite, (40-self.launcher_sprite.get_width()/2,35-self.launcher_sprite.get_height()/2))
font = pygame.font.SysFont("Courier", 12)
HUD_block.blit(font.render("Agency: %s"%(launcher.get_nationality()), True, (255,255,255)), (90,5))
HUD_block.blit(font.render("Rockets left: %s"%(launcher.get_rockets_left()), True, (255,255,255)), (90,20))
HUD_block.blit(font.render("Right-click to target", True, (255,255,255)), (90,50))
pygame.draw.rect(HUD_block, (255,255,255), HUD_block.get_rect(), 1)
HUD_block.set_alpha(150)
self.screen.blit(HUD_block, HUD_pos)
def draw_rocket_overlay(self, universe):
cursor_pos = pygame.mouse.get_pos()
for rocket in universe.get_rockets():
rocket_pos = to_ints(to_screen(scale_pos(rocket.get_pos(), universe.get_zoom_factor())))
if get_distance_between(rocket_pos, cursor_pos) < 30*universe.get_zoom_factor():
HUD_pos = (rocket_pos[0]+20, rocket_pos[1]+5)
HUD_block = pygame.Surface((245, 70))
HUD_block.fill((100,100,100))
HUD_block.blit(self.rocket_icon, (50-self.rocket_icon.get_width()/2,35-self.rocket_icon.get_height()/2))
font = pygame.font.SysFont("Courier", 12)
HUD_block.blit(font.render("Fuel left: %s"%(rocket.get_fuel()), True, (255,255,255)), (90,5))
HUD_block.blit(font.render("Space to self-destruct", True, (255,255,255)), (90,35))
HUD_block.blit(font.render("Right-click to target", True, (255,255,255)), (90,50))
pygame.draw.rect(HUD_block, (255,255,255), HUD_block.get_rect(), 1)
HUD_block.set_alpha(150)
self.screen.blit(HUD_block, HUD_pos)
def draw_planet_overlay(self, universe):
cursor_pos = pygame.mouse.get_pos()
planet = universe.get_planet()
planet_pos = to_ints(to_screen(scale_pos(planet.get_pos(), universe.get_zoom_factor())))
if get_distance_between(planet_pos, cursor_pos) < 30*universe.get_zoom_factor():
HUD_pos = (planet_pos[0]+20, planet_pos[1]+5)
HUD_block = pygame.Surface((245, 70))
HUD_block.fill((100,100,100))
if planet.get_name() == "Earth":
HUD_block.blit(self.Earth_icon, (40-self.Earth_icon.get_width()/2,35-self.Earth_icon.get_height()/2))
if planet.get_name() == "Moon":
HUD_block.blit(self.Moon_icon, (40-self.Moon_icon.get_width()/2,35-self.Moon_icon.get_height()/2))
if planet.get_name() == "Jupiter":
HUD_block.blit(self.Jupiter_icon, (40-self.Jupiter_icon.get_width()/2,35-self.Jupiter_icon.get_height()/2))
if planet.get_name() == "Unknown Meteor":
HUD_block.blit(self.Comet_icon, (40-self.Comet_icon.get_width()/2,35-self.Comet_icon.get_height()/2))
if planet.get_name() == "Neptune":
HUD_block.blit(self.Neptune_icon, (40-self.Neptune_icon.get_width()/2,35-self.Neptune_icon.get_height()/2))
if planet.get_name() == "Exoplanet I":
HUD_block.blit(self.Exoplanet1_icon, (40-self.Exoplanet1_icon.get_width()/2,35-self.Exoplanet1_icon.get_height()/2))
if planet.get_name() == "Exoplanet II":
HUD_block.blit(self.Exoplanet2_icon, (40-self.Exoplanet2_icon.get_width()/2,35-self.Exoplanet2_icon.get_height()/2))
if planet.get_name() == "Exoplanet III":
HUD_block.blit(self.Exoplanet3_icon, (40-self.Exoplanet3_icon.get_width()/2,35-self.Exoplanet3_icon.get_height()/2))
if planet.get_name() == "Unkown Exometeor":
HUD_block.blit(self.Comet2_icon, (40-self.Comet2_icon.get_width()/2,35-self.Comet2_icon.get_height()/2))
if planet.get_name() == "Exoplanet IV":
HUD_block.blit(self.Exoplanet4_icon, (40-self.Exoplanet4_icon.get_width()/2,35-self.Exoplanet4_icon.get_height()/2))
font = pygame.font.SysFont("Courier", 12)
HUD_block.blit(font.render("Name: %s"%(planet.get_name()), True, (255,255,255)), (90,5))
HUD_block.blit(font.render("Right-click to target", True, (255,255,255)), (90,50))
pygame.draw.rect(HUD_block, (255,255,255), HUD_block.get_rect(), 1)
HUD_block.set_alpha(150)
self.screen.blit(HUD_block, HUD_pos)