From 8fd6fa26656f1414ba9d10bb416771455bac5ddc Mon Sep 17 00:00:00 2001 From: Stefano Pigozzi Date: Thu, 4 Jun 2020 22:33:49 +0200 Subject: [PATCH] Use an enum for the difficulty --- lihzahrd/header/__init__.py | 4 ++- lihzahrd/header/difficulty.py | 11 ++++++++ lihzahrd/world.py | 47 +++++++++++++++-------------------- 3 files changed, 34 insertions(+), 28 deletions(-) create mode 100644 lihzahrd/header/difficulty.py diff --git a/lihzahrd/header/__init__.py b/lihzahrd/header/__init__.py index 33988d3..0cf7020 100644 --- a/lihzahrd/header/__init__.py +++ b/lihzahrd/header/__init__.py @@ -4,6 +4,7 @@ from .anglerquestfish import AnglerQuestFish from .backgrounds import Backgrounds from .bossesdefeated import BossesDefeated from .clouds import Clouds +from .difficulty import Difficulty from .events import Events from .fourpartsplit import FourPartSplit from .generatorinfo import GeneratorInfo @@ -36,6 +37,7 @@ __all__ = [ "Backgrounds", "BossesDefeated", "Clouds", + "Difficulty", "Events", "FourPartSplit", "GeneratorInfo", @@ -61,5 +63,5 @@ __all__ = [ "Time", "TreetopVariants", "Version", - "WorldEvilType" + "WorldEvilType", ] diff --git a/lihzahrd/header/difficulty.py b/lihzahrd/header/difficulty.py new file mode 100644 index 0000000..72ef612 --- /dev/null +++ b/lihzahrd/header/difficulty.py @@ -0,0 +1,11 @@ +import enum + + +class Difficulty(enum.IntEnum): + JOURNEY = 3 + CLASSIC = 0 + EXPERT = 1 + MASTER = 2 + + def __repr__(self): + return f"{self.__class__.__qualname__}.{self.NAME}" diff --git a/lihzahrd/world.py b/lihzahrd/world.py index cb2faa6..6097d5a 100644 --- a/lihzahrd/world.py +++ b/lihzahrd/world.py @@ -29,7 +29,7 @@ class World: id_: int, bounds: Rect, size: Coordinates, - game_mode: int, + difficulty: Difficulty, drunk_world: bool, get_good_world: bool, created_on, @@ -107,11 +107,8 @@ class World: self.size: Coordinates = size """The world size in tiles.""" - self.game_mode: int = game_mode - """Which mode the game is in. 0=classic, 1=expert, 2=master, 3=journey""" - - self.is_expert: bool = True if self.game_mode==1 else False - """If the world is in expert mode or not, no longer stored in worldfile.""" + self.difficulty: Difficulty = difficulty + """The `difficulty `_ the game is in.""" self.drunk_world: bool = drunk_world """Was this world created with the drunk worldgen seed.""" @@ -319,6 +316,13 @@ class World: tile = Tile(block=block, wall=wall, liquid=liquid, wiring=wiring) return [tile] * multiply_by + @property + def is_expert(self): + """If the world is in expert mode or not. + + Provided for compatibility purposes.""" + return self.difficulty == 1 + @classmethod def _create_tilematrix(cls, f, world_size: Coordinates, tileframeimportant: typing.List[bool]): """Create a TileMatrix object from a file.""" @@ -372,7 +376,7 @@ class World: id_ = f.int4() bounds = f.rect() world_size = Coordinates(y=f.int4(), x=f.int4()) - game_mode = f.int4() + difficulty = Difficulty(f.int4()) drunk_world = f.bool() get_good_world = f.bool() created_on = f.datetime() @@ -849,32 +853,21 @@ class World: journey_powers = JourneyPowers() power_id = f.int2() if power_id == 0: - freeze_time = f.bool() - journey_powers.freeze_time = freeze_time + journey_powers.freeze_time = f.bool() elif power_id == 5: - god_mode = f.bool() - journey_powers.god_mode = god_mode + journey_powers.god_mode = f.bool() elif power_id == 8: - time_rate = f.single() - journey_powers.time_rate = time_rate + journey_powers.time_rate = f.single() elif power_id == 9: - freeze_rain = f.bool() - journey_powers.freeze_rain = freeze_rain + journey_powers.freeze_rain = f.bool() elif power_id == 10: - freeze_wind = f.bool() - journey_powers.freeze_wind = freeze_wind + journey_powers.freeze_wind = f.bool() elif power_id == 11: - far_placement_range = f.bool() - journey_powers.far_placement_range = far_placement_range + journey_powers.far_placement_range = f.bool() elif power_id == 12: - difficulty = f.single() - journey_powers.difficulty = difficulty + journey_powers.difficulty = f.single() elif power_id == 13: - freeze_biome_spread = f.bool() - journey_powers.freeze_biome_spread = freeze_biome_spread - elif power_id == -1: - spawn_rate: f.single() - journey_powers.spawn_rate = spawn_rate + journey_powers.freeze_biome_spread = f.bool() else: print(f"Unknown id: {power_id}") @@ -883,7 +876,7 @@ class World: # Object creation world = cls(version=version, savefile_type=savefile_type, revision=revision, is_favorite=is_favorite, name=name, generator=generator, uuid_=uuid_, id_=id_, bounds=bounds, size=world_size, - game_mode=game_mode, drunk_world=drunk_world, get_good_world=get_good_world, + difficulty=difficulty, drunk_world=drunk_world, get_good_world=get_good_world, created_on=created_on, styles=world_styles, backgrounds=backgrounds, spawn_point=spawn_point, underground_level=underground_level, cavern_level=cavern_level, time=time, events=events, dungeon_point=dungeon_point, world_evil=world_evil,