96 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			96 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| from .Items import MinecraftItem, item_table, item_frequencies
 | |
| from .Locations import MinecraftAdvancement, advancement_table, exclusion_table, events_table
 | |
| from .Regions import mc_regions, link_minecraft_structures
 | |
| from .Rules import set_rules
 | |
| 
 | |
| from BaseClasses import Region, Entrance
 | |
| from .Options import minecraft_options
 | |
| from ..AutoWorld import World
 | |
| 
 | |
| client_version = (0, 4)
 | |
| 
 | |
| class MinecraftWorld(World):
 | |
|     game: str = "Minecraft"
 | |
|     options = minecraft_options
 | |
| 
 | |
| 
 | |
|     def _get_mc_data(self):
 | |
|         exits = ["Overworld Structure 1", "Overworld Structure 2", "Nether Structure 1", "Nether Structure 2",
 | |
|                  "The End Structure"]
 | |
|         return {
 | |
|             'world_seed': self.world.slot_seeds[self.player].getrandbits(32),
 | |
|             # consistent and doesn't interfere with other generation
 | |
|             'seed_name': self.world.seed_name,
 | |
|             'player_name': self.world.get_player_names(self.player),
 | |
|             'player_id': self.player,
 | |
|             'client_version': client_version,
 | |
|             'structures': {exit: self.world.get_entrance(exit, self.player).connected_region.name for exit in exits}
 | |
|         }
 | |
| 
 | |
| 
 | |
|     def generate_basic(self):
 | |
|         link_minecraft_structures(self.world, self.player)
 | |
| 
 | |
|         pool = []
 | |
|         pool_counts = item_frequencies.copy()
 | |
|         if getattr(self.world, "bee_traps")[self.player]: 
 | |
|             pool_counts.update({"Rotten Flesh": 0, "Bee Trap (Minecraft)": 4})
 | |
|         for item_name, item_data in item_table.items():
 | |
|             for count in range(pool_counts.get(item_name, 1)):
 | |
|                 pool.append(MinecraftItem(item_name, item_data.progression, item_data.code, self.player))
 | |
| 
 | |
|         prefill_pool = {}
 | |
|         prefill_pool.update(events_table)
 | |
|         exclusion_pools = ['hard', 'insane', 'postgame']
 | |
|         for key in exclusion_pools:
 | |
|             if not getattr(self.world, f"include_{key}_advancements")[self.player]:
 | |
|                 prefill_pool.update(exclusion_table[key])
 | |
| 
 | |
|         for loc_name, item_name in prefill_pool.items():
 | |
|             item_data = item_table[item_name]
 | |
|             location = self.world.get_location(loc_name, self.player)
 | |
|             item = MinecraftItem(item_name, item_data.progression, item_data.code, self.player)
 | |
|             self.world.push_item(location, item, collect=False)
 | |
|             pool.remove(item)
 | |
|             location.event = item_data.progression
 | |
|             location.locked = True
 | |
| 
 | |
|         self.world.itempool += pool
 | |
| 
 | |
| 
 | |
|     def set_rules(self):
 | |
|         set_rules(self.world, self.player)
 | |
| 
 | |
| 
 | |
|     def create_regions(self):
 | |
|         def MCRegion(region_name: str, exits=[]):
 | |
|             ret = Region(region_name, None, region_name, self.player)
 | |
|             ret.world = self.world
 | |
|             ret.locations = [ MinecraftAdvancement(self.player, loc_name, loc_data.id, ret) 
 | |
|                 for loc_name, loc_data in advancement_table.items() 
 | |
|                 if loc_data.region == region_name ]
 | |
|             for exit in exits: 
 | |
|                 ret.exits.append(Entrance(self.player, exit, ret))
 | |
|             return ret
 | |
| 
 | |
|         self.world.regions += [MCRegion(*r) for r in mc_regions]
 | |
| 
 | |
| 
 | |
|     def generate_output(self):
 | |
|         import json
 | |
|         from base64 import b64encode
 | |
|         from Utils import output_path
 | |
| 
 | |
|         data = self._get_mc_data()
 | |
|         filename = f"AP_{self.world.seed_name}_P{self.player}_{self.world.get_player_names(self.player)}.apmc"
 | |
|         with open(output_path(filename), 'wb') as f:
 | |
|             f.write(b64encode(bytes(json.dumps(data), 'utf-8')))
 | |
| 
 | |
| 
 | |
|     def fill_slot_data(self): 
 | |
|         slot_data = self._get_mc_data()
 | |
|         for option_name in minecraft_options:
 | |
|             option = getattr(self.world, option_name)[self.player]
 | |
|             slot_data[option_name] = int(option.value)
 | |
|         return slot_data
 | 
