from ..generic.Rules import add_item_rule, add_rule, location_item_name from .items import item_groups def set_rules(world, player): item_rules = { "Pallet Town - Player's PC": (lambda i: i.player == player and "Badge" not in i.name and "Trap" not in i.name and i.name != "Pokedex" and "Coins" not in i.name) } if world.prizesanity[player]: def prize_rule(i): return i.player != player or i.name in item_groups["Unique"] item_rules["Celadon Prize Corner - Item Prize 1"] = prize_rule item_rules["Celadon Prize Corner - Item Prize 2"] = prize_rule item_rules["Celadon Prize Corner - Item Prize 3"] = prize_rule if world.accessibility[player] != "locations": world.get_location("Cerulean City - Bicycle Shop", player).always_allow = (lambda state, item: item.name == "Bike Voucher" and item.player == player) world.get_location("Fuchsia City - Safari Zone Warden", player).always_allow = (lambda state, item: item.name == "Gold Teeth" and item.player == player) access_rules = { "Pallet Town - Rival's Sister": lambda state: state.has("Oak's Parcel", player), "Pallet Town - Oak's Post-Route-22-Rival Gift": lambda state: state.has("Oak's Parcel", player), "Viridian City - Sleepy Guy": lambda state: state.pokemon_rb_can_cut(player) or state.pokemon_rb_can_surf(player), "Route 2 - Oak's Aide": lambda state: state.pokemon_rb_oaks_aide(state.multiworld.oaks_aide_rt_2[player].value + 5, player), "Pewter City - Museum": lambda state: state.pokemon_rb_can_cut(player), "Cerulean City - Bicycle Shop": lambda state: state.has("Bike Voucher", player) or location_item_name(state, "Cerulean City - Bicycle Shop", player) == ("Bike Voucher", player), "Lavender Town - Mr. Fuji": lambda state: state.has("Fuji Saved", player), "Vermilion Gym - Lt. Surge 1": lambda state: state.pokemon_rb_can_cut(player or state.pokemon_rb_can_surf(player)), "Vermilion Gym - Lt. Surge 2": lambda state: state.pokemon_rb_can_cut(player or state.pokemon_rb_can_surf(player)), "Route 11 - Oak's Aide": lambda state: state.pokemon_rb_oaks_aide(state.multiworld.oaks_aide_rt_11[player].value + 5, player), "Celadon City - Stranded Man": lambda state: state.pokemon_rb_can_surf(player), "Silph Co 11F - Silph Co President (Card Key)": lambda state: state.has("Card Key", player), "Fuchsia City - Safari Zone Warden": lambda state: state.has("Gold Teeth", player) or location_item_name(state, "Fuchsia City - Safari Zone Warden", player) == ("Gold Teeth", player), "Route 12 - Island Item": lambda state: state.pokemon_rb_can_surf(player), "Route 12 - Item Behind Cuttable Tree": lambda state: state.pokemon_rb_can_cut(player), "Route 15 - Oak's Aide": lambda state: state.pokemon_rb_oaks_aide(state.multiworld.oaks_aide_rt_15[player].value + 5, player), "Route 15 - Item": lambda state: state.pokemon_rb_can_cut(player), "Route 25 - Item": lambda state: state.pokemon_rb_can_cut(player), "Fuchsia City - Warden's House Item": lambda state: state.pokemon_rb_can_strength(player), "Rocket Hideout B4F - Southwest Item (Lift Key)": lambda state: state.has("Lift Key", player), "Rocket Hideout B4F - Giovanni Item (Lift Key)": lambda state: state.has("Lift Key", player), "Silph Co 3F - Item (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 4F - Left Item (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 4F - Middle Item (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 4F - Right Item (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 5F - Northwest Item (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 6F - West Item (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 6F - Southwest Item (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 7F - East Item (Card Key)": lambda state: state.has("Card Key", player), "Safari Zone Center - Island Item": lambda state: state.pokemon_rb_can_surf(player), "Celadon Prize Corner - Item Prize 1": lambda state: state.has("Coin Case", player), "Celadon Prize Corner - Item Prize 2": lambda state: state.has("Coin Case", player), "Celadon Prize Corner - Item Prize 3": lambda state: state.has("Coin Case", player), "Celadon Game Corner - West Gambler's Gift (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Center Gambler's Gift (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - East Gambler's Gift (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Northwest By Counter (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Southwest Corner (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Near Rumor Man (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Near Speculating Woman (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Near West Gifting Gambler (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Near Wonderful Time Woman (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Near Failing Gym Information Guy (Coin Case)": lambda state: state.has( "Coin Case", player), "Celadon Game Corner - Hidden Item Near East Gifting Gambler (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item Near Hooked Guy (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item at End of Horizontal Machine Row (Coin Case)": lambda state: state.has("Coin Case", player), "Celadon Game Corner - Hidden Item in Front of Horizontal Machine Row (Coin Case)": lambda state: state.has("Coin Case", player), "Silph Co 11F - Silph Co Liberated": lambda state: state.has("Card Key", player), "Pallet Town - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Pallet Town - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Route 22 - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Route 22 - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Route 24 - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Route 24 - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Route 24 - Super Rod Pokemon - 3": lambda state: state.has("Super Rod", player), "Route 6 - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Route 6 - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Route 10 - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Route 10 - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Safari Zone Center - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Safari Zone Center - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Safari Zone Center - Super Rod Pokemon - 3": lambda state: state.has("Super Rod", player), "Safari Zone Center - Super Rod Pokemon - 4": lambda state: state.has("Super Rod", player), "Route 12 - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Route 12 - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Route 12 - Super Rod Pokemon - 3": lambda state: state.has("Super Rod", player), "Route 12 - Super Rod Pokemon - 4": lambda state: state.has("Super Rod", player), "Route 19 - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Route 19 - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Route 19 - Super Rod Pokemon - 3": lambda state: state.has("Super Rod", player), "Route 19 - Super Rod Pokemon - 4": lambda state: state.has("Super Rod", player), "Route 23 - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Route 23 - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Route 23 - Super Rod Pokemon - 3": lambda state: state.has("Super Rod", player), "Route 23 - Super Rod Pokemon - 4": lambda state: state.has("Super Rod", player), "Fuchsia City - Super Rod Pokemon - 1": lambda state: state.has("Super Rod", player), "Fuchsia City - Super Rod Pokemon - 2": lambda state: state.has("Super Rod", player), "Fuchsia City - Super Rod Pokemon - 3": lambda state: state.has("Super Rod", player), "Fuchsia City - Super Rod Pokemon - 4": lambda state: state.has("Super Rod", player), "Anywhere - Good Rod Pokemon - 1": lambda state: state.has("Good Rod", player), "Anywhere - Good Rod Pokemon - 2": lambda state: state.has("Good Rod", player), "Anywhere - Old Rod Pokemon": lambda state: state.has("Old Rod", player), "Celadon Prize Corner - Pokemon Prize - 1": lambda state: state.has("Coin Case", player), "Celadon Prize Corner - Pokemon Prize - 2": lambda state: state.has("Coin Case", player), "Celadon Prize Corner - Pokemon Prize - 3": lambda state: state.has("Coin Case", player), "Celadon Prize Corner - Pokemon Prize - 4": lambda state: state.has("Coin Case", player), "Celadon Prize Corner - Pokemon Prize - 5": lambda state: state.has("Coin Case", player), "Celadon Prize Corner - Pokemon Prize - 6": lambda state: state.has("Coin Case", player), "Cinnabar Island - Old Amber Pokemon": lambda state: state.has("Old Amber", player), "Cinnabar Island - Helix Fossil Pokemon": lambda state: state.has("Helix Fossil", player), "Cinnabar Island - Dome Fossil Pokemon": lambda state: state.has("Dome Fossil", player), "Route 12 - Sleeping Pokemon": lambda state: state.has("Poke Flute", player), "Route 16 - Sleeping Pokemon": lambda state: state.has("Poke Flute", player), "Seafoam Islands B4F - Legendary Pokemon": lambda state: state.pokemon_rb_can_strength(player), "Vermilion City - Legendary Pokemon": lambda state: state.pokemon_rb_can_surf(player) and state.has("S.S. Ticket", player), **{f"Pokemon Tower {floor}F - Wild Pokemon - {slot}": lambda state: state.has("Silph Scope", player) for floor in range(3, 8) for slot in range(1, 11)}, "Route 2 - Marcel Trade": lambda state: state.can_reach("Route 24 - Wild Pokemon - 6", "Location", player), "Underground Tunnel West-East - Spot Trade": lambda state: state.can_reach("Route 24 - Wild Pokemon - 6", "Location", player), "Route 11 - Terry Trade": lambda state: state.can_reach("Safari Zone Center - Wild Pokemon - 5", "Location", player), "Route 18 - Marc Trade": lambda state: state.can_reach("Route 23 - Super Rod Pokemon - 1", "Location", player), "Cinnabar Island - Sailor Trade": lambda state: state.can_reach("Pokemon Mansion 1F - Wild Pokemon - 3", "Location", player), "Cinnabar Island - Crinkles Trade": lambda state: state.can_reach("Route 12 - Wild Pokemon - 4", "Location", player), "Cinnabar Island - Doris Trade": lambda state: state.can_reach("Cerulean Cave 1F - Wild Pokemon - 9", "Location", player), "Vermilion City - Dux Trade": lambda state: state.can_reach("Route 3 - Wild Pokemon - 2", "Location", player), "Cerulean City - Lola Trade": lambda state: state.can_reach("Route 10 - Super Rod Pokemon - 1", "Location", player), # Pokédex check "Pallet Town - Oak's Parcel Reward": lambda state: state.has("Oak's Parcel", player), # trainers "Route 4 - Cooltrainer F": lambda state: state.pokemon_rb_can_surf(player), "Route 15 - Jr. Trainer F 1": lambda state: state.pokemon_rb_can_cut(player), "Silph Co 11F - Rocket 2 (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 9F - Rocket 2 (Card Key)": lambda state: state.has("Card Key", player), "Silph Co 3F - Scientist (Card Key)": lambda state: state.has("Card Key", player), "Route 10 North - Pokemaniac": lambda state: state.pokemon_rb_can_surf(player), "Rocket Hideout B1F - Rocket 5 (Lift Key)": lambda state: state.has("Lift Key", player), "Rocket Hideout B4F - Rocket 2 (Lift Key)": lambda state: state.has("Lift Key", player), "Rocket Hideout B4F - Rocket 3 (Lift Key)": lambda state: state.has("Lift Key", player), # hidden items "Viridian Forest - Hidden Item Northwest by Trainer": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Viridian Forest - Hidden Item Entrance Tree": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Mt Moon B2F - Hidden Item Dead End Before Fossils": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Route 25 - Hidden Item Fence Outside Bill's House": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Route 9 - Hidden Item Bush By Grass": lambda state: state.pokemon_rb_can_get_hidden_items(player), "S.S. Anne 1F - Hidden Item Kitchen Trash": lambda state: state.pokemon_rb_can_get_hidden_items(player), "S.S. Anne B1F - Hidden Item Under Pillow": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 10 - Hidden Item Behind Rock Tunnel Entrance Cuttable Tree": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 10 - Hidden Item Bush": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Rocket Hideout B1F - Hidden Item Pot Plant": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Rocket Hideout B3F - Hidden Item Near East Item": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Rocket Hideout B4F - Hidden Item Behind Giovanni (Lift Key)": lambda state: state.pokemon_rb_can_get_hidden_items(player) and state.has("Lift Key", player), "Pokemon Tower 5F - Hidden Item Near West Staircase": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Route 13 - Hidden Item Dead End Bush": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 13 - Hidden Item Dead End By Water Corner": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Pokemon Mansion B1F - Hidden Item Secret Key Room Corner": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Safari Zone West - Hidden Item Secret House Statue": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Silph Co 5F - Hidden Item Pot Plant": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Silph Co 9F - Hidden Item Nurse Bed (Card Key)": lambda state: state.pokemon_rb_can_get_hidden_items( player) and state.has("Card Key", player), "Copycat's House - Hidden Item Desk": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Cerulean Cave 1F - Hidden Item Center Rocks": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Cerulean Cave B1F - Hidden Item Northeast Rocks": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Power Plant - Hidden Item Central Dead End": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Power Plant - Hidden Item Before Zapdos": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Seafoam Islands B2F - Hidden Item Rock": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Seafoam Islands B4F - Hidden Item Corner Island": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Pokemon Mansion 1F - Hidden Item Block Near Entrance Carpet": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Pokemon Mansion 3F - Hidden Item Behind Burglar": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 23 - Hidden Item Rocks Before Victory Road": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Route 23 - Hidden Item East Bush After Water": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Route 23 - Hidden Item On Island": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Victory Road 2F - Hidden Item Rock Before Moltres": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Victory Road 2F - Hidden Item Rock In Final Room": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Viridian City - Hidden Item Cuttable Tree": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 11 - Hidden Item Isolated Bush Near Gate": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 12 - Hidden Item Bush Near Gate": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 17 - Hidden Item In Grass": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 17 - Hidden Item Near Northernmost Sign": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 17 - Hidden Item East Center": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 17 - Hidden Item West Center": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Route 17 - Hidden Item Before Final Bridge": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Underground Tunnel North-South - Hidden Item Near Northern Stairs": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Underground Tunnel North-South - Hidden Item Near Southern Stairs": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Underground Tunnel West-East - Hidden Item West": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Underground Tunnel West-East - Hidden Item East": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Celadon City - Hidden Item Dead End Near Cuttable Tree": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Route 25 - Hidden Item Northeast Of Grass": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Mt Moon B2F - Hidden Item Lone Rock": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Seafoam Islands B3F - Hidden Item Rock": lambda state: state.pokemon_rb_can_get_hidden_items(player), "Vermilion City - Hidden Item In Water Near Fan Club": lambda state: state.pokemon_rb_can_get_hidden_items( player) and state.pokemon_rb_can_surf(player), "Cerulean City - Hidden Item Gym Badge Guy's Backyard": lambda state: state.pokemon_rb_can_get_hidden_items( player), "Route 4 - Hidden Item Plateau East Of Mt Moon": lambda state: state.pokemon_rb_can_get_hidden_items(player), } for loc in world.get_locations(player): if loc.name in access_rules: add_rule(loc, access_rules[loc.name]) if loc.name in item_rules: add_item_rule(loc, item_rules[loc.name]) if loc.name.startswith("Pokedex"): mon = loc.name.split(" - ")[1] add_rule(loc, lambda state, i=mon: (state.has("Pokedex", player) or not state.multiworld.require_pokedex[player]) and (state.has(i, player) or state.has(f"Static {i}", player)))