def enteredlevel(allowcancel=1):
# always allows quit and cancel
# deals with cancel and quit elegantly
options = []
choicedict = {}
def myfunc(x, dir_name, filenames):
for x in filenames:
i = x.rfind(".txt")
if i == -1:
options.append(x)
else:
options.append(x[:i])
choicedict[x[:i]] = x
os.path.walk(os.path.join("data", "rooms"), myfunc, 0)
# don't want to accidentally overwrite this
options.append("quit")
choicedict["quit"] = "quit"
choice = getChoiceUnbounded("Select a level:", options, allowcancel=allowcancel)
if choice in choicedict:
return choicedict[choice]
else:
return choice
def prepareRandomLevel(screen, cols=0, rows=0):
messages.colorBackground()
if not cols:
cols = int(getChoiceUnbounded("Room width?", [str(x) for x in range(30, 61, 5)]))
if not rows:
rows = int(getChoiceUnbounded("Room height?", [str(x) for x in range(20, 61, 5)]))
room.globalize_dimensions(rows, cols)
lines = [" " * cols for x in range(rows + 1)]
room.randomlymodify(lines)
tiles = []
for c in range(cols):
tiles.append([])
for r in range(rows):
tiles[c].append(tile.tile[lines[r][c]])
r = room.Room(tiles, lines)
graphics.extract_dimensions(lines)
world = pygame.Surface((graphics.world_w, graphics.world_h))
t = topbar.TopBar(screen)
test.mark(r.background)
r.topbar = t
return world, t, r
def main_loop():
print "\n" * 12 + "* * * * * NEW GAME * * * * *"
"""these things only need to happen once"""
screen = pygame.display.get_surface()
screen.fill(graphics.background)
load_settings()
messages.screen = screen
clock = pygame.time.Clock()
dt = 0
world, t, r = 0, 0, 0
loadedSavedGame = 0
if getChoiceUnbounded("Load saved game?", ["yes", "no"]) == "yes":
r = game_progress.load_game()
if r:
messages.say("Loading saved game", down=4)
pygame.display.flip()
world = pygame.Surface((graphics.world_w, graphics.world_h))
t = r.topbar
loadedSavedGame = 1
else:
"""these are level-specific"""
level = enteredlevel(allowcancel=0)
if level == "quit":
return
world, t, r = preparelevel(screen, level)
pygame.display.get_surface().fill((50, 50, 50))
p = player.Player(r)
# h = zoo.Beehive()
h = zoo.Beehive("the fact that any argument is passed here means that I'll load saved bees")
c = camera.Camera(p, world, screen, r)
c.jump_to_player()
# myfamilytree = familytree.FamilyTree()
myspeciesplot = species_visualization.SpeciesPlot()
# Generating new bees
# r.bees = h.preserved_bees(r, p)
if loadedSavedGame:
for b in r.bees:
b.player = p
else:
r.generate_bees(settings[MAXIMUM_BEES])
r.h = h
# for b in r.bees:
# b.randomize_position()
framecount = 0
previous_key_states = []
key_ups = []
global time_to_quit
info_panel = StackingPanel()
left = 0
top = 0
for family_name in game_settings.families:
family_panel = ShrinkingPanel(min_width=250)
top = 0
title = MessagePanel(family_name)
family_panel.children.append(title)
title.onclick.append(get_toggler(family_panel))
top += 30
for entry in game_settings.families[family_name]:
new_button = SettingButton(entry, y=top, typeface="DejaVu Sans", bold=True, italic=True, visible=False)
top += 20
family_panel.children.append(new_button)
left += family_panel.rect.w
info_panel.children.append(family_panel)
# we should never see reassignment of gameobjects!
gameobjects = {
"room": r,
"player": p,
"hive": h,
"camera": c,
"world": world,
"screen": screen,
"info_panel": info_panel,
}
info_panel.children.append(StatisticsPanel(gameobjects))
while not close(r): # Main game loop # Don't need cancel and quit works
if time_to_quit:
try:
print "Quitting indirectly at main loop. Saving room..."
game_progress.save_game(r)
break
except Exception:
print "Failed to save"
framecount += 1
test.begin_timing(test.segs, test.segdescriptors)
key_presses = pygame.event.get(pygame.KEYDOWN)
key_states = pygame.key.get_pressed()
if previous_key_states:
key_ups = [old and not new for new, old in zip(key_states, previous_key_states)]
else:
key_ups = [0 for x in key_states]
previous_key_states = key_states[:]
time_gap = handle_player_input(key_presses, key_ups, key_states, gameobjects)
r = gameobjects["room"]
c = gameobjects["camera"]
h = gameobjects["hive"]
world = gameobjects["world"]
screen = gameobjects["screen"]
p = gameobjects["player"]
if time_gap:
clock.tick(400)
dt = 0
mod = 20
# each bee updates once every mod frames
"""Update"""
for i, bee in enumerate(r.bees):
bee.slow = 1 # (i + framecount) % mod
test.add_sticky("main:playerupdate")
p.update(dt, key_states, key_presses)
test.remove_sticky("main:playerupdate")
for phys in r.food + r.bullets:
phys.update(dt, key_states, key_presses)
test.add_sticky("bee")
for b in r.bees:
b.update(dt, key_states, key_presses)
if b.request_family_tree_update:
updatefamilytree = 1
b.request_family_tree_update = 0
test.remove_sticky("bee")
screen.fill([0, 0, 0])
if settings[SHOW_FAMILY_TREE]:
test.add_sticky("tree")
# draw every frame, since the tree follows the player around
if settings[SPECIES_STYLE] == 3:
if not framecount % settings[TREE_UPDATE_TIME]:
myspeciesplot.update(r.bees + r.deadbees)
for b in r.deadbees:
if b.dead == 2:
r.deadbees.remove(b)
myspeciesplot.draw(
world, (int(p.xy[0, 0]) - myspeciesplot.w / 2, int(p.xy[0, 1]) - myspeciesplot.h / 2)
)
else:
# update once every few frames
if not framecount % settings[TREE_UPDATE_TIME]:
myspeciesplot.update(r.bees + r.deadbees)
myspeciesplot.draw(screen, (840, 0))
for b in r.deadbees:
if b.dead == 2:
r.deadbees.remove(b)
updatefamilytree = 0
if updatefamilytree:
myfamilytree.update(r.bees)
myfamilytree.draw(screen, (840, 0))
test.remove_sticky("tree")
test.add_sticky("room")
# Necessarily comes afterwards so that the bees can see the player
r.update()
test.remove_sticky("room")
"""
if r.signal_new_tree:
r.signal_new_tree = 0
myfamilytree.depth += 2*settings[TREE_V_SPACING]
test.remove_sticky('tree') ##########
"""
test.add_sticky("camera_update")
c.follow_player(dt) # Here so that room knows what to draw
c.updatesight()
test.remove_sticky("camera_update")
test.add_sticky("main:drawing")
test.add_sticky("main:drawing:test")
test.draw(world, r)
test.remove_sticky("main:drawing:test")
test.add_sticky("main:drawing:bee")
for x in r.bees:
x.draw(world)
test.remove_sticky("main:drawing:bee")
test.add_sticky("main:drawing:other")
for x in r.food + [p] + r.bullets:
x.draw(world)
test.remove_sticky("main:drawing:other")
# Time
dt = clock.tick(120) # used to be 120
# Make it seem like a slowed down version of 22fps if necessary
dt = min(dt, 45)
# print dt, "this is dt"
test.add_sticky("main:drawing:camera")
c.draw()
test.remove_sticky("main:drawing:camera")
info_panel.draw_to(screen)
"""
seconds = pygame.time.get_ticks() / 1000
t0 = ""
if settings[SPECIES_STYLE] == 1:
t0 += "Plotted on right: x: horizontal movement, y: generation"
elif settings[SPECIES_STYLE] == 2:
t0 += "Plotted on right: x: generation, y: vertical movement"
elif settings[SPECIES_STYLE] == 3:
t0 += "angle: direction, radius: generation"
t.permanent_text = [t0]
ta = "Time: " + str(seconds)
tb = " | current bees: " + str(len(r.bees))
tbb = " | total dead bees " + str(len(r.deadbees))
tc = " |" + str(int(clock.get_fps()))+"fps"
t.permanent_text += [
ta + tb + tbb + tc,]
if settings[SHOW_HELP]:
t.permanent_text += [
"click anywhere for options",
"plotted on right: Recent lifetimes, max is %s" % (max(t.data) if t.data else "not defined"),
"[a]: %s birth and deaths" % ("resume" if r.stasis else "pause"),
"[g]: generate bees",
"[e]: visualize paths of a random bee",
"[r]: %s eyes" % ("hide" if settings[SHOW_EYES] else "show"),
"[n]: %s names" % ("hide" if settings[SHOW_NAMES] else "show"),
"[m]: %s visual madness" % ("decrease" if r.madness == 2 else "increase"),
"[SPACE]: teleport to a random position",
"[t]: turn %s random map generation" % ("off" if settings[GENERATE_RANDOM_MAP] else "on"),
]
t.permanent_text.append("[h]: %s help" % ("hide" if settings[SHOW_HELP] else "show"))
t.draw(screen)"""
test.add_sticky("main:drawing:displayflip")
pygame.display.flip()
test.remove_sticky("main:drawing:displayflip")
test.remove_sticky("main:drawing")
for phys in r.bees + r.food + [p]:
phys.visible = c.can_see(phys)
test.summarizetimings(test.segs, test.segdescriptors)
# if framecount % 30 == 0:
# print int(clock.get_fps()), "fps"
pygame.event.pump()
def handle_player_input(key_presses, key_ups, key_states, gameobjects):
time_gap = False
try:
r = gameobjects["room"]
p = gameobjects["player"]
c = gameobjects["camera"]
world = gameobjects["world"]
h = gameobjects["hive"]
screen = gameobjects["screen"]
except:
raise Exception("Missing gameobjects. Have following keys:", [key for key in gameobjects])
clicks.get_more_clicks()
global time_to_quit
# Handle player input
if key_ups[pygame.K_e]:
if r.bees:
# for x in r.bees + r.food + [p] + r.bullets:
# x.draw(world)
# c.draw()
# pygame.display.flip()
"""this is for picking the closest bee"""
"""minbee = r.bees[0]
for b in r.bees:
if linalg.norm(b.xy - p.xy) < linalg.norm(minbee.xy - p.xy):
minbee = b"""
minbee = random.choice(r.bees)
messages.say("Loading", down=1)
pygame.display.flip()
minbee.visualize_intelligence(world, c)
p.displaycolor = [255, 255, 255]
pygame.draw.circle(world, [0, 0, 0], array(p.xy.astype(int))[0], int(p.forcefield))
pygame.draw.circle(world, [255, 255, 255], array(p.xy.astype(int))[0], int(p.forcefield), 1)
p.draw(world)
# c.xy = minbee.xy * 1
# c.updatesight()
c.draw()
pygame.display.flip()
minbee.sample_path(world)
c.draw()
pygame.display.flip()
waitForKey(pygame.K_w)
time_gap = True
if time_to_quit:
game_progress.save_game(r)
return
test.record()
world.blit(r.background, (0, 0))
test.record("blitting background")
togglers = {pygame.K_r: SHOW_EYES, pygame.K_h: SHOW_HELP, pygame.K_t: GENERATE_RANDOM_MAP, pygame.K_n: SHOW_NAMES}
for key, variable in togglers.iteritems():
if key_ups[key]:
settings[variable] = not settings[variable]
if key_ups[pygame.K_m]:
print framecount
r.madness += 1
r.madness %= 5
for b in r.bees:
b.madness = r.madness
if key_ups[pygame.K_a]:
if key_states[pygame.K_UP]:
p.shoot("up")
else:
p.shoot()
if key_ups[pygame.K_x]:
r.stasis = not r.stasis
if key_ups[pygame.K_z]:
print pygame.mouse.get_pos()
# gameobjects["info_panel"].handle_event(pygame.mouse.get_pos(), "mouseover")
"""All mouse click stuff"""
if clicks.mouseups[0]:
event_taken = False
if gameobjects["info_panel"].rect.collidepoint(pygame.mouse.get_pos()):
mx, my = pygame.mouse.get_pos()
x, y = mx - gameobjects["info_panel"].rect.x, my - gameobjects["info_panel"].rect.y
event_taken = gameobjects["info_panel"].handle_event((x, y), "click")
if not event_taken:
for b in r.bees:
b.skipupdate = True
options = [
"Move to another map",
"Random 60 x 25",
"Random 35 x 35",
"Random 60 x 60",
"Examine bee",
"Save bees",
"Extinction",
"Load bees",
"Delete bees",
"Tweak variables",
"Modify map",
"View Shortcuts",
"quit",
]
choice = getChoiceUnbounded("select option", options, allowcancel=1)
if choice != "cancel":
if choice == "quit":
time_to_quit = True
elif choice in ["Move to another map", "Random 60 x 25", "Random 35 x 35", "Random 60 x 60"]:
level = choice
if choice == "Move to another map":
level = enteredlevel()
if level == "quit":
time_to_quit = True
if level != "cancel":
print "MAGIC LEVEL MOVE WHOA"
gameobjects2 = {}
gameobjects2["room"] = r
gameobjects2["player"] = p
gameobjects2["hive"] = h
gameobjects2["camera"] = c
gameobjects2["world"] = world
gameobjects2["screen"] = screen
move_to_level(level, gameobjects2)
try:
r = gameobjects2["room"]
p = gameobjects2["player"]
c = gameobjects2["camera"]
world = gameobjects2["world"]
h = gameobjects2["hive"]
screen = gameobjects2["screen"]
except:
raise Exception("Missing gameobjects. Have following keys:", [key for key in gameobjects])
elif choice == "Extinction":
if getChoiceUnbounded("Kill all bees?", ["no", "yes"], allowcancel=1) == "yes":
for b in r.bees:
b.health = -20
for b in r.bees + r.deadbees:
b.dead = 2
t.data = []
elif choice == "Examine bee":
"""this thing lets you examine the closest bee to you"""
if r.bees:
minbee = r.bees[0]
for b in r.bees:
if linalg.norm(b.xy - p.xy) < linalg.norm(minbee.xy - p.xy):
minbee = b
for x in r.bees + r.food + [p] + r.bullets:
x.draw(world)
c.draw()
pygame.display.flip()
while not time_to_quit: # quit should work fine here
choice = getChoiceUnbounded(
"Examining nearest bee...what feature?",
["Neural Network", "Movement"],
allowcancel=True,
)
if choice == "Neural Network":
command = minbee.show_brain(world)
if command:
return
elif choice == "Movement":
c.draw()
messages.say("Loading", down=1)
pygame.display.flip()
minbee.visualize_intelligence(world, c)
minbee.sample_path(world)
c.draw()
pygame.display.flip()
elif choice == "cancel":
done = 1
break
elif choice == "quit":
time_to_quit = True
elif choice == "Save bees":
beemap = {}
beenames = []
for b in r.bees:
label = b.firstname + " " + b.name
beemap[label] = b
beenames.append(label)
choice = getChoiceUnbounded("Pick a bee", beenames, allowcancel=True)
if choice == "quit":
time_to_quit = True
elif choice != "cancel":
b = beemap[choice]
# messages.colorBackground()
# messages.say("Pick a new name for the bee", down = 4)
rename = b.name
"""loop is for error checks"""
while not time_to_quit:
rename = getString(string=b.name, message="Pick a new name for the bee")
# quit
if rename == -1:
break
# cancel
elif rename == 0:
break
elif any(specimen[0] == rename for specimen in h.specimens):
messages.colorBackground()
messages.say("Name taken. Pick a new name for the bee", down=4)
continue
else:
b.name = rename
break
h.save_bee(b)
h.savedata()
elif choice == "Load bees":
loadedbeenames = []
for b in h.specimens:
loadedbeenames.append(b[0])
loadedbeenames.append("done")
loadedbeenames = loadedbeenames[-1::-1]
i = 0
while not time_to_quit:
name = getChoiceUnbounded("Loaded " + str(i) + " bees", loadedbeenames, allowcancel=True)
if name in ["done", "quit", "cancel"]:
break
else:
i += 1
b = h.make_bee(name, r, p)
r.bees.append(b)
b.flash = 50
b.xy[0, 0] = p.xy[0, 0]
elif choice == "Delete bees":
i = 0
message = "Choose a bee to delete"
while not time_to_quit:
indexfromname = {}
loadedbeenames = []
for index, b in enumerate(h.specimens):
name = b[0]
loadedbeenames.append(name)
indexfromname[name] = index
loadedbeenames.append("done")
loadedbeenames = loadedbeenames[-1::-1]
if i:
message = "Deleted " + str(i) + " bees"
name = getChoiceUnbounded(message, loadedbeenames, allowcancel=True)
if name in ["quit", "done", "cancel"]:
break
else:
confirm = getChoiceUnbounded("Delete %s?" % name, ["no", "yes"], allowcancel=True)
if confirm != "yes":
continue
i += 1
indextoremove = indexfromname[name]
h.specimens = h.specimens[:indextoremove] + h.specimens[(indextoremove + 1) :]
elif choice == "Tweak variables":
families = game_settings.families
"""picking kinds of variables"""
while not time_to_quit:
family = getChoiceUnbounded("What kind of variable?", families.keys(), allowcancel=1)
if family == "cancel":
break
"""picking variable"""
while not time_to_quit:
"""takes "name: value" string to "name"."""
label_to_key = {key + ": " + str(settings[key]): key for key in families[family]}
"""this is a bunch of informative labels"""
options = [x for x in sorted(label_to_key.keys())]
toChange = getChoiceUnbounded("Pick a variable to modify", options, allowcancel=1)
if toChange == "cancel":
break
toChange = label_to_key[toChange]
if toChange in want_bools:
settings[toChange] = not settings[toChange]
continue
elif toChange in want_ints and toChange in max_val and toChange in min_val:
settings[toChange] += 1
if settings[toChange] > max_val[toChange]:
settings[toChange] = min_val[toChange]
continue
else:
"""we have the name of the variable we want to change now"""
usedindex = 0
for i, entry in enumerate(options):
if entry == toChange:
usedindex = i
# messages.colorBackground()
# messages.say("Pick a new value", down = 4)
allowed_keys = range(pygame.K_0, pygame.K_9 + 1)
allowed_keys += [pygame.K_PERIOD]
allowed_keys += [pygame.K_e]
allowed_keys += [pygame.K_MINUS]
"""loop is for error checking"""
while not time_to_quit:
# message = messages.smallfont.render("Set new value for '%s'" % toChange, 1, [255, 255, 255])
# pygame.display.get_surface().blit(message, (600, 15 * (usedindex + 3)))
m = "Set new value for '%s'" % toChange
position = pygame.Rect(0, 0, 290, 30)
# position.x = 600 #Leaning right
position.centerx = graphics.screen_w / 2
# position.y = 15 * (usedindex + 4.3)
# Leaning up
position.centery = graphics.screen_h / 2
out = getString(allowed_keys, str(settings[toChange]), textpos=position, message=m)
# out = getString(allowed_keys, str(settings[toChange]))
# quit
if out == -1:
break
# cancel
elif out == 0:
break
try:
val = 0
if "." in out:
val = float(out)
else:
val = int(out)
problems = problem_with_setting(toChange, val)
if problems:
messages.colorBackground()
messages.say(problems, down=4)
continue
settings[toChange] = val
break
except:
messages.colorBackground()
messages.say("Error. Pick a new value", down=4)
save_settings()
# for key, value in settings.iteritems():
# print key + ":", str(value)
if family == "quit" or time_to_quit:
time_to_quit = True
break
elif choice == "Modify map":
messages.colorBackground()
messages.say("This is where I would modify the map", 500)
elif choice == "View Shortcuts":
messages.colorBackground()
messagecount = 2
if settings[SHOW_EYES]:
messagecount = messages.say("[r]: Hide eyes", down=messagecount)
else:
messagecount = messages.say("[r]: Show eyes", down=messagecount)
if r.madness:
messagecount = messages.say("[m]: Don't draw trails", down=messagecount)
else:
messagecount = messages.say("[m]: Draw trails", down=messagecount)
if settings[GENERATE_RANDOM_MAP]:
messagecount = messages.say("[t]: Don't randomize maps when loading", down=messagecount)
else:
messagecount = messages.say("[t]: Randomize maps when loading", down=messagecount)
if r.stasis:
messagecount = messages.say("[a]: Resume deaths and births", down=messagecount)
else:
messagecount = messages.say("[a]: Pause deaths and births", down=messagecount)
messagecount = messages.say("[e]: Examine a random bee", down=messagecount)
messagecount = messages.say("Click anywhere to continue", down=messagecount)
waitForEnter(hidemessage=1)
if time_to_quit:
print "Saving game very rapidly, whee"
game_progress.save_game(r)
return
screen.fill(graphics.background)
# This is to trick the computer into thinking no time has passed
time_gap = True
gameobjects["player"] = p
gameobjects["room"] = r
gameobjects["world"] = world
gameobjects["camera"] = c
gameobjects["screen"] = screen
gameobjects["hive"] = h
return time_gap