def GetInputDirection(self, event):
if event.key == pygame.K_UP or joystick.JustWentUp():
return 0
if event.key == pygame.K_LEFT or joystick.JustWentLeft():
return 1
if event.key == pygame.K_DOWN or joystick.JustWentDown():
return 2
if event.key == pygame.K_RIGHT or joystick.JustWentRight():
return 3
def confirmExit(display, profile, num):
button_selected = 0
buttons = ["Save + Exit", "Exit", "Cancel"]
rect = pygame.Rect(0, 0, 150, 50)
points = [(150, 400), (400, 400), (650, 400)]
rect1, rect2, rect3 = rect.copy(), rect.copy(), rect.copy()
rect1.midleft = points[0]
rect2.center = points[1]
rect3.midright = points[2]
rects = [rect1, rect2, rect3]
old_display = display.copy()
surf2 = old_display.copy()
surf2.fill((30, 30, 30))
surf2.set_alpha(150)
old_display.blit(surf2, (0, 0))
while True:
click = False
for event in pygame.event.get():
mpos = pygame.mouse.get_pos()
joystick.Update(event)
if not hasattr(event, 'key'):
event.key = None
if event.type == pygame.KEYDOWN or joystick.WasEvent():
if event.key == pygame.K_F2 or joystick.JustPressedLB():
screenshot.capture(num, display)
if event.type == pygame.MOUSEMOTION:
for rect in rects:
if rect.collidepoint(mpos):
button_selected = rects.index(rect)
break
if event.type == pygame.MOUSEBUTTONDOWN:
for rect in rects:
if rect.collidepoint(mpos):
click = True
if event.type == pygame.KEYUP or joystick.WasEvent():
if event.key == pygame.K_ESCAPE or joystick.BackEvent():
return
if event.key == pygame.K_LEFT or joystick.JustWentLeft():
button_selected -= 1
if event.key == pygame.K_RIGHT or joystick.JustWentRight():
button_selected += 1
if button_selected < 0:
button_selected = 0
if button_selected > 2:
button_selected = 2
if event.key == pygame.K_RETURN or joystick.JustPressedA():
click = True
display.blit(old_display, (0, 0))
retro_text((400, 302), display, 30, "Exit?", anchor="midbottom", bold=True, color=(0, 0, 0))
retro_text((400, 300), display, 30, "Exit?", anchor="midbottom", bold=True)
for n, rect in enumerate(rects):
color = (75, 75, 75)
tcolor = (255, 255, 255)
if n == button_selected:
color = (50, 50, 50)
tcolor = (255, 255, 255)
pygame.draw.rect(display, (0, 0, 0), rect.move(2, 2))
pygame.draw.rect(display, color, rect)
retro_text(rect.move(0, 2).center, display, 12, buttons[n], color = (0, 0, 0), anchor="center")
retro_text(rect.center, display, 12, buttons[n], color = tcolor, anchor="center")
pygame.display.update()
clock.tick(10)
if click:
name = buttons[button_selected]
if name == "Cancel":
return
if name == "Save + Exit":
saves.save_data(num, profile)
pygame.quit()
sys.exit()
if name == "Exit":
transition(display, 5)
pygame.quit()
sys.exit()
def events(self):
for event in pygame.event.get():
joystick.Update(event)
click = False
if event.type == pygame.QUIT:
confirmExit(self.Display, self.profile, self.profile_number)
if event.type == pygame.MOUSEBUTTONDOWN and not self.focused in ("theSun", None):
planet = self.planets[self.focused]
rect = pygame.Rect((0, 0), (205, 50))
rect.midtop = planet.center[1][0], planet.center[1][1] + 35
if rect.collidepoint(pygame.mouse.get_pos()):
click = True
if event.type == pygame.MOUSEBUTTONDOWN:
self.start_click = pygame.math.Vector2(pygame.mouse.get_pos())
self.last_sample = copy.copy(self.start_click)
if pygame.Rect(0, 0, 32, 32).collidepoint(pygame.mouse.get_pos()):
self.done = True
rect = pygame.Rect(0, 0, 32, 32)
rect.midright = self.Display.get_rect().midright
if rect.collidepoint(pygame.mouse.get_pos()):
for e, p in enumerate(self.planets):
if p.name == "Earth":
self.focused = e
if event.type == pygame.MOUSEBUTTONUP:
self.start_click = None
if event.type == pygame.KEYDOWN or joystick.WasEvent():
if not hasattr(event, 'key'):
event.key = None
if event.key == pygame.K_ESCAPE or joystick.BackEvent():
self.done = True
if event.key == pygame.K_RETURN or event.key == pygame.K_SPACE or joystick.GoEvent():
click = True
if event.key == pygame.K_LEFT or joystick.JustWentLeft():
self.speed -= .1
if self.speed <= 0.2:
self.speed = 0.2
if event.key == pygame.K_RIGHT or joystick.JustWentRight():
self.speed += .1
if self.speed > 2:
self.speed = 2
if event.key == pygame.K_F2 or joystick.JustPressedLB():
screenshot.capture(self.profile_number, self.Display)
if event.key == pygame.K_COMMA or joystick.JustPressedLT():
if self.focused == None:
self.focused = 0
self.focused = (self.focused + 1) % len(self.planets)
self.target_zoom = 1.5
if event.key == pygame.K_PERIOD or joystick.JustPressedRT():
if self.focused == None:
self.focused = 0
self.focused = (self.focused - 1) % len(self.planets)
self.target_zoom = 1.5
if event.type == pygame.MOUSEMOTION:
self.mouse_on = True
if event.type == pygame.MOUSEBUTTONUP:
pos = pygame.mouse.get_pos()
nto = self.target_offset[:]
if event.button == 5:
self.target_zoom -= self.zoom_amount
nto[0] -= (pos[0] - self.Display.get_width() / 2) / 10
nto[1] -= (pos[1] - self.Display.get_height() / 2) / 10
if event.button == 4:
self.target_zoom += self.zoom_amount
nto[0] += (pos[0] - self.Display.get_width() / 2) / 10
nto[1] += (pos[1] - self.Display.get_height() / 2) / 10
if self.target_zoom < .5:
self.target_zoom = .5
elif self.target_zoom > 2:
self.target_zoom = 2
else:
self.target_offset = nto
if not self.focused in ("theSun", None) and self.zoom >= 1.25 and click and self.planets[self.focused] in self.unlocked_planets:
self.profile["planet"] = self.planets[self.focused]
self.done = True
def events(self):
for event in pygame.event.get():
joystick.Update(event)
if event.type == pygame.QUIT:
self.exit()
if event.type == pygame.MOUSEBUTTONDOWN:
if not self.finished and not self.options_lock:
self.finished = True
print(colorize(self.text, 'bold'))
self.text = ""
if event.type == pygame.KEYDOWN or joystick.WasEvent():
if not hasattr(event, "key"):
event.key = None
if event.key == pygame.K_F2 or joystick.JustPressedLB():
screenshot.capture("M", self.display)
if not self.finished and not self.options_lock:
self.finished = True
print(colorize(self.text, 'bold'))
self.text = ""
else:
item = self.item_selected
items = self.items
if self.options_mode:
item = self.option_selected
items = self.options
if self.play_mode:
item = self.profile_selected
items = self.profiles
if event.key == pygame.K_ESCAPE or joystick.BackEvent():
if self.play_mode:
self.play_mode = False
elif self.options_mode:
self.options_mode = False
self.options_dict = saves.load_options()
else:
self.exit()
if event.key == pygame.K_UP or joystick.JustWentUp():
item -= 1
if (event.key == pygame.K_DOWN) or joystick.JustWentDown():
item += 1
if item < 0:
item = 0
if item >= len(items):
item = len(items) - 1
if self.options_mode:
self.option_selected = item
if self.play_mode:
self.profile_selected = item
elif self.play_mode:
pass
else:
self.item_selected = item
if self.options_mode:
op = self.options[self.option_selected]
if op == "Volume":
vol = self.options_dict["volume"]
if event.key == pygame.K_LEFT or joystick.JustWentLeft():
vol -= .1
if event.key == pygame.K_RIGHT or joystick.JustWentRight():
vol += .1
if vol < 0:
vol = 0
if vol > 1:
vol = 1
self.options_dict["volume"] = vol
if op == "Cache Screenshots":
if event.key == pygame.K_RETURN or joystick.JustPressedA():
self.options_dict["cache_screen_shots"] = not self.options_dict["cache_screen_shots"]
if op == "Stretch to Fullscreen":
if event.key == pygame.K_RETURN or joystick.JustPressedA():
self.options_dict["fullscreen"] = not self.options_dict["fullscreen"]
if op == self.DEL:
if event.key == pygame.K_RETURN or joystick.JustPressedA():
try:
DID_SOMETHING = False
for e, x in enumerate(os.listdir(fix_path(get_file("data/screenshots")))):
os.remove(fix_path(get_file("data/screenshots/")) + x)
DID_SOMETHING = True
if DID_SOMETHING:
print(colorize(f"Deleted screenshots: {e+1}", "green"))
self.options[self.options.index(self.DEL)] = "Deleted screenshots"
self.DEL = "Deleted screenshots"
except FileNotFoundError:
print(colorize("Failed to delete screenshots!", "fail"))
print(colorize("File not found error.", "fail"))
if event.key == pygame.K_x or joystick.JustPressedX():
if self.play_mode:
if self.profile_selected < 5 and not self.profiles[self.profile_selected].startswith("New"):
self.confirm_delete()
if event.key == pygame.K_RETURN or joystick.JustPressedA() or joystick.JustPressedStart():
if self.options_mode:
sel = self.option_selected
if self.options[sel] == "Cancel":
self.option_selected = 2
self.options_mode = False
self.options_dict = saves.load_options()
if self.options[sel] == "Save":
self.options_mode = False
saves.save_data("options", self.options_dict)
elif self.play_mode:
if self.profiles[self.profile_selected] == "Back":
self.play_mode = False
else:
profile = saves.load_profile(self.profile_selected)
if profile["version"] != __version__:
if profile["new"]:
profile["version"] = __version__
else:
print(colorize("Warning: this profile is from a different version \
of Interplanetary Invaders; \nerrors may occur", "warning"))
profile["new"] = False
saves.save_data(self.profile_selected, profile)
self.done = True
else:
sel = self.item_selected
if self.items[sel] == "Play":
self.play_mode = True
if self.items[sel] == "Options":
self.options_mode = True
if self.items[sel] == "Quit":
self.exit()
if self.items[sel] == "Credits":
run_credits(self.display, self.images)
pygame.mixer.music.load(fix_path(get_file("audio/music/MainMenu.mp3")))
pygame.mixer.music.play(-1)
if not self.options_mode and self.options_lock:
self.done = True
def events(self):
for event in pygame.event.get():
click = False
joystick.Update(event)
if not hasattr(event, "key"):
event.key = None
if event.type == pygame.MOUSEBUTTONDOWN:
if self.button_rect.collidepoint(pygame.mouse.get_pos()):
click = True
if event.type == pygame.KEYUP or joystick.WasEvent():
if event.key == pygame.K_y or joystick.JustPressedY():
cat = self.catagories[self.selected]
sel = self.sel_num[self.selected]
if self.selected != 2:
item = list(cat.keys())[sel]
else:
item = cat[sel][1]
if item:
display_info(self.display, self.images, item)
if event.type == pygame.KEYDOWN or joystick.WasEvent():
if event.key in (pygame.K_UP,
pygame.K_w) or joystick.JustWentUp():
if not self.button_selected:
self.selected -= 1
else:
self.button_selected = False
if event.key in (pygame.K_DOWN,
pygame.K_s) or joystick.JustWentDown():
self.selected += 1
if self.selected < 0:
self.selected = 0
if self.selected == len(self.rects):
self.selected = len(self.rects) - 1
self.button_selected = True
if (event.key in (pygame.K_LEFT, pygame.K_a) or
joystick.JustWentLeft()) and not self.button_selected:
self.sel_num[self.selected] -= 1
if (event.key in (pygame.K_RIGHT, pygame.K_d) or
joystick.JustWentRight()) and not self.button_selected:
self.sel_num[self.selected] += 1
if self.sel_num[self.selected] < 0:
self.sel_num[self.selected] = 0
if self.sel_num[self.selected] == len(
self.catagories[self.selected]):
self.sel_num[self.selected] = len(
self.catagories[self.selected]) - 1
if (event.key == pygame.K_RETURN or
joystick.JustPressedA()) and not self.button_selected:
cat = self.catagories[self.selected]
lcat = list(cat)
sel = self.sel_num[self.selected]
didSomething = False
if self.selected == 1:
for x in self.catagories[2]:
if x[1] == None:
if not lcat:
return
didSomething = True
x[1] = lcat[sel]
if cat[lcat[sel]] == 1:
cat.pop(lcat[sel])
break
else:
cat[lcat[sel]] -= 1
break
if not didSomething:
Sound(fix_path("audio/donk.wav")).play()
if self.selected == 2 and cat[sel][1] != None:
done = False
for x in self.catagories[1]:
if x == cat[sel][1]:
self.catagories[1][x] += 1
done = True
if not done:
self.catagories[1][cat[sel][1]] = 1
cat[sel][1] = None
if (event.key == pygame.K_RETURN
or joystick.JustPressedA()) and self.button_selected:
click = True
if click and not self.cannotContinue():
self.done = True
def events(self):
for event in pygame.event.get():
joystick.Update(event)
cat = self.catagories[self.rect_sel]
click = False
if not hasattr(event, 'key'):
event.key = None
if event.type == pygame.QUIT:
# self.confirmExit()
self.done = True
if event.type == pygame.KEYUP or joystick.WasEvent():
if (event.key == pygame.K_y
or joystick.JustPressedY()) and cat:
selected = list(cat.keys())[self.sel_num[self.rect_sel]]
display_info(self.display, self.images, selected)
if event.type == pygame.KEYDOWN or joystick.WasEvent():
jwu = joystick.JustWentUp() or event.key in (pygame.K_w,
pygame.K_UP)
jwd = joystick.JustWentDown() or event.key in (pygame.K_s,
pygame.K_DOWN)
mod_sel = False
if jwu or jwd:
sel = self.sel_num[self.rect_sel]
new_sel = copy(sel)
if jwd and sel + MAX_STORE_ITEM_LENGTH < len(cat):
mod_sel = True
new_sel += MAX_STORE_ITEM_LENGTH
if jwu and sel - MAX_STORE_ITEM_LENGTH >= 0:
mod_sel = True
new_sel -= MAX_STORE_ITEM_LENGTH
if mod_sel:
self.sel_num[self.rect_sel] = new_sel
if event.key == pygame.K_ESCAPE or joystick.BackEvent():
self.done = True
if (jwu and not mod_sel) or joystick.JustPressedLT(
) or event.key == pygame.K_PAGEUP:
self.rect_sel -= 1
if (jwd and not mod_sel) or joystick.JustPressedRT(
) or event.key == pygame.K_PAGEDOWN:
self.rect_sel += 1
if self.rect_sel < 0:
self.rect_sel = 0
if self.rect_sel >= len(self.rects):
self.rect_sel = len(self.rects) - 1
cat = self.catagories[
self.rect_sel] # REMEMBER: this line must be here!!!
if event.key in (pygame.K_a,
pygame.K_LEFT) or joystick.JustWentLeft():
self.sel_num[self.rect_sel] -= 1
if event.key in (pygame.K_d,
pygame.K_RIGHT) or joystick.JustWentRight():
self.sel_num[self.rect_sel] += 1
if self.sel_num[self.rect_sel] < 0:
self.sel_num[self.rect_sel] = 0
if self.sel_num[self.rect_sel] >= len(cat):
self.sel_num[self.rect_sel] = len(cat) - 1
if event.key == pygame.K_RETURN or joystick.JustPressedA():
click = True
if event.type == pygame.MOUSEBUTTONDOWN:
if self.purchase_rect.collidepoint(pygame.mouse.get_pos()):
click = True
if self.exit_rect.collidepoint(pygame.mouse.get_pos()):
self.done = True
if click:
self.purchase()
