✨ Initial commit

Sly3Pool.py

@@ -0,0 +1,68 @@
+import typing
+
+from BaseClasses import Item
+
+from .data.Constants import EPISODES
+from .data.Items import item_groups
+
+if typing.TYPE_CHECKING:
+    from . import Sly3World
+
+def gen_powerups(world: "Sly3World") -> list[Item]:
+  """Generate the power-ups for the item pool"""
+  powerups = []
+  for item_name in item_groups["Power-Up"]:
+    if item_name == "Mega Jump" and not world.options.include_mega_jump:
+      continue
+    elif item_name == "Progressive Shadow Power":
+      powerups.append(world.create_item(item_name))
+      powerups.append(world.create_item(item_name))
+    elif item_name[:11] == "Progressive":
+      powerups.append(world.create_item(item_name))
+      powerups.append(world.create_item(item_name))
+      powerups.append(world.create_item(item_name))
+    else:
+      powerups.append(world.create_item(item_name))
+
+  return powerups
+
+def gen_crew(world: "Sly3World") -> list[Item]:
+  """Generate the crew for the item pool"""
+  crew = []
+  for item_name in item_groups["Crew"]:
+    crew.append(world.create_item(item_name))
+
+  return crew
+
+def gen_episodes(world: "Sly3World") -> list[Item]:
+  """Generate the progressive episodes items for the item pool"""
+  all_episodes = [
+    item_name for item_name in item_groups["Episode"]
+    for _ in range(4)
+  ]
+
+  # Make sure the starting episode is precollected
+  starting_episode_n = world.options.starting_episode.value
+  starting_episode = f"Progressive {list(EPISODES.keys())[starting_episode_n]}"
+  all_episodes.remove(starting_episode)
+  world.multiworld.push_precollected(world.create_item(starting_episode))
+
+  return [world.create_item(e) for e in all_episodes]
+
+
+
+def gen_pool_sly3(world: "Sly3World") -> list[Item]:
+  """Generate the item pool for the world"""
+  item_pool = []
+  item_pool += gen_powerups(world)
+  item_pool += gen_episodes(world)
+  item_pool += gen_crew(world)
+
+  unfilled_locations = world.multiworld.get_unfilled_locations(world.player)
+  remaining = len(unfilled_locations)-len(item_pool)
+  if world.options.goal.value < 6:
+      remaining -= 1
+  assert remaining >= 0, f"There are more items than locations ({len(item_pool)} items; {len(unfilled_locations)} locations)"
+  item_pool += [world.create_item(world.get_filler_item_name()) for _ in range(remaining)]
+
+  return item_pool