X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=plomrogue%2Fmapping.py;fp=plomrogue%2Fmapping.py;h=99924bc599bdf396c68115463fafbe4fba23eb99;hb=d9c9b5b7d5cac2469ac075010c4d729e1adf0cc4;hp=00b8b1d7a6771bd1ed6fcef34b835dd061ee38da;hpb=f0f0c01c001fef0bdc6bf8d60da329e4c7b55223;p=plomrogue2 diff --git a/plomrogue/mapping.py b/plomrogue/mapping.py index 00b8b1d..99924bc 100644 --- a/plomrogue/mapping.py +++ b/plomrogue/mapping.py @@ -24,22 +24,29 @@ class MapGeometry(): def get_directions(self): directions = [] + prefix = 'move__' for name in dir(self): - if name[:5] == 'move_': - directions += [name[5:]] + if name[:len(prefix)] == prefix: + directions += [name[len(prefix):]] return directions - def get_neighbors(self, pos): + def get_neighbors_yxyx(self, yxyx): neighbors = {} for direction in self.get_directions(): - neighbors[direction] = self.move(pos, direction) + neighbors[direction] = self.move_yxyx(yxyx, direction) + return neighbors + + def get_neighbors_yx(self, pos): + neighbors = {} + for direction in self.get_directions(): + neighbors[direction] = self.move_yx(pos, direction) return neighbors def get_neighbors_i(self, i): if i in self.neighbors_i: return self.neighbors_i[i] pos = YX(i // self.size.x, i % self.size.x) - neighbors_pos = self.get_neighbors(pos) + neighbors_pos = self.get_neighbors_yx(pos) neighbors_i = {} for direction in neighbors_pos: pos = neighbors_pos[direction] @@ -50,22 +57,41 @@ class MapGeometry(): self.neighbors_i[i] = neighbors_i return self.neighbors_i[i] - def move(self, start_pos, direction): - mover = getattr(self, 'move_' + direction) - target = mover(start_pos) + def move_yx(self, start_yx, direction, check=True): + mover = getattr(self, 'move__' + direction) + target = mover(start_yx) + # TODO refactor with SourcedMap.inside? if target.y < 0 or target.x < 0 or \ - target.y >= self.size.y or target.x >= self.size.x: + target.y >= self.size.y or target.x >= self.size.x: return None return target + def move_yxyx(self, start_yxyx, direction): + mover = getattr(self, 'move__' + direction) + start_yx = self.undouble_yxyx(*start_yxyx) + target_yx = mover(start_yx) + return self.double_yx(target_yx) + + def double_yx(self, absolute_yx): + big_y = absolute_yx.y // self.size.y + little_y = absolute_yx.y % self.size.y + big_x = absolute_yx.x // self.size.x + little_x = absolute_yx.x % self.size.x + return YX(big_y, big_x), YX(little_y, little_x) + + def undouble_yxyx(self, big_yx, little_yx): + y = big_yx.y * self.size.y + little_yx.y + x = big_yx.x * self.size.x + little_yx.x + return YX(y, x) + class MapGeometryWithLeftRightMoves(MapGeometry): - def move_LEFT(self, start_pos): + def move__LEFT(self, start_pos): return YX(start_pos.y, start_pos.x - 1) - def move_RIGHT(self, start_pos): + def move__RIGHT(self, start_pos): return YX(start_pos.y, start_pos.x + 1) @@ -75,18 +101,18 @@ class MapGeometrySquare(MapGeometryWithLeftRightMoves): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.fov_map_class = FovMapSquare - self.dijkstra_map_class = DijkstraMapSquare def define_segment(self, source_center, radius): + source_center = self.undouble_yxyx(*source_center) size = YX(2 * radius + 1, 2 * radius + 1) offset = YX(source_center.y - radius, source_center.x - radius) center = YX(radius, radius) return size, offset, center - def move_UP(self, start_pos): + def move__UP(self, start_pos): return YX(start_pos.y - 1, start_pos.x) - def move_DOWN(self, start_pos): + def move__DOWN(self, start_pos): return YX(start_pos.y + 1, start_pos.x) @@ -95,37 +121,37 @@ class MapGeometryHex(MapGeometryWithLeftRightMoves): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.fov_map_class = FovMapHex - self.dijkstra_map_class = DijkstraMapHex def define_segment(self, source_center, radius): + source_center = self.undouble_yxyx(*source_center) indent = 1 if (source_center.y % 2) else 0 size = YX(2 * radius + 1 + indent, 2 * radius + 1) offset = YX(source_center.y - radius - indent, source_center.x - radius) center = YX(radius + indent, radius) return size, offset, center - def move_UPLEFT(self, start_pos): + def move__UPLEFT(self, start_pos): start_indented = start_pos.y % 2 if start_indented: return YX(start_pos.y - 1, start_pos.x) else: return YX(start_pos.y - 1, start_pos.x - 1) - def move_UPRIGHT(self, start_pos): + def move__UPRIGHT(self, start_pos): start_indented = start_pos.y % 2 if start_indented: return YX(start_pos.y - 1, start_pos.x + 1) else: return YX(start_pos.y - 1, start_pos.x) - def move_DOWNLEFT(self, start_pos): + def move__DOWNLEFT(self, start_pos): start_indented = start_pos.y % 2 if start_indented: return YX(start_pos.y + 1, start_pos.x) else: return YX(start_pos.y + 1, start_pos.x - 1) - def move_DOWNRIGHT(self, start_pos): + def move__DOWNRIGHT(self, start_pos): start_indented = start_pos.y % 2 if start_indented: return YX(start_pos.y + 1, start_pos.x + 1) @@ -136,8 +162,8 @@ class MapGeometryHex(MapGeometryWithLeftRightMoves): class Map(): - def __init__(self, map_size): - self.size = map_size + def __init__(self, map_geometry): + self.geometry = map_geometry self.terrain = '.' * self.size_i def __getitem__(self, yx): @@ -152,23 +178,17 @@ class Map(): def __iter__(self): """Iterate over YX position coordinates.""" - for y in range(self.size.y): - for x in range(self.size.x): + for y in range(self.geometry.size.y): + for x in range(self.geometry.size.x): yield YX(y, x) - # TODO: use this for more refactoring - def inside(self, yx): - if yx.y < 0 or yx.x < 0 or yx.y >= self.size.y or yx.x >= self.size.x: - return False - return True - @property def size_i(self): - return self.size.y * self.size.x + return self.geometry.size.y * self.geometry.size.x def set_line(self, y, line): - height_map = self.size.y - width_map = self.size.x + height_map = self.geometry.size.y + width_map = self.geometry.size.x if y >= height_map: raise ArgError('too large row number %s' % y) width_line = len(line) @@ -178,36 +198,45 @@ class Map(): self.terrain[(y + 1) * width_map:] def get_position_index(self, yx): - return yx.y * self.size.x + yx.x + return yx.y * self.geometry.size.x + yx.x def lines(self): - width = self.size.x - for y in range(self.size.y): + width = self.geometry.size.x + for y in range(self.geometry.size.y): yield (y, self.terrain[y * width:(y + 1) * width]) - class SourcedMap(Map): - def __init__(self, source_map, source_center, radius): - self.source_map = source_map + def __init__(self, source_maps, source_center, radius, get_map): + self.source_maps = source_maps self.radius = radius - self.size, self.offset, self.center = \ - self.geometry_class.define_segment(None, source_center, radius) - self.geometry = self.geometry_class(self.size) + example_map = get_map(YX(0,0)) + self.source_geometry = example_map.geometry + size, self.offset, self.center = \ + self.source_geometry.define_segment(source_center, radius) + self.geometry = self.source_geometry.__class__(size) + for yx in self: + big_yx, _ = self.source_yxyx(yx) + get_map(big_yx) - def source_yx(self, yx, check=False): - source_yx = yx + self.offset - if check and not self.source_map.inside(source_yx): - return False - return source_yx + def source_yxyx(self, yx): + absolute_yx = yx + self.offset + big_yx, little_yx = self.source_geometry.double_yx(absolute_yx) + return big_yx, little_yx - def target_yx(self, yx, check=False): - target_yx = yx - self.offset + def target_yx(self, big_yx, little_yx, check=False): + target_yx = self.source_geometry.undouble_yxyx(big_yx, little_yx) - self.offset if check and not self.inside(target_yx): return False return target_yx + def inside(self, yx): + if yx.y < 0 or yx.x < 0 or \ + yx.y >= self.geometry.size.y or yx.x >= self.geometry.size.x: + return False + return True + class DijkstraMap(SourcedMap): @@ -219,11 +248,8 @@ class DijkstraMap(SourcedMap): shrunk = True source_map_segment = '' for yx in self: - yx_in_source = self.source_yx(yx, True) - if yx_in_source: - source_map_segment += self.source_map[yx_in_source] - else: - source_map_segment += 'X' + big_yx, little_yx = self.source_yxyx(yx) + source_map_segment += self.source_maps[big_yx][little_yx] while shrunk: shrunk = False for i in range(self.size_i): @@ -248,31 +274,21 @@ class DijkstraMap(SourcedMap): -class DijkstraMapHex(DijkstraMap): - geometry_class = MapGeometryHex - - - -class DijkstraMapSquare(DijkstraMap): - geometry_class = MapGeometrySquare - - - class FovMap(SourcedMap): # TODO: player visibility asymmetrical (A can see B when B can't see A): # does this make sense, or not? def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) - self.terrain = '?' * self.size.y * self.size.x + self.terrain = '?' * self.size_i #self.size.y * self.size.x self[self.center] = '.' self.shadow_cones = [] self.circle_out(self.center, self.shadow_process) - def throws_shadow(self, source_yx): - return self.source_map[source_yx] == 'X' + def throws_shadow(self, big_yx, little_yx): + return self.source_maps[big_yx][little_yx] == 'X' - def shadow_process(self, yx, source_yx, distance_to_center, dir_i, dir_progress): + def shadow_process(self, yx, source_yxyx, distance_to_center, dir_i, dir_progress): # Possible optimization: If no shadow_cones yet and self[yx] == '.', # skip all. CIRCLE = 360 # Since we'll float anyways, number is actually arbitrary. @@ -312,7 +328,7 @@ class FovMap(SourcedMap): if in_shadow_cone(cone): return self[yx] = '.' - if self.throws_shadow(source_yx): + if self.throws_shadow(*source_yxyx): unmerged = True while merge_cone(cone): unmerged = False @@ -333,8 +349,7 @@ class FovMap(SourcedMap): eval_cone([left_arm, right_arm]) def basic_circle_out_move(self, pos, direction): - #"""Move position pos into direction. Return whether still in map.""" - mover = getattr(self.geometry, 'move_' + direction) + mover = getattr(self.geometry, 'move__' + direction) return mover(pos) def circle_out(self, yx, f): @@ -353,9 +368,8 @@ class FovMap(SourcedMap): for dir_progress in range(distance): direction = self.circle_out_directions[dir_i] yx = self.circle_out_move(yx, direction) - source_yx = self.source_yx(yx, True) - if source_yx: - f(yx, source_yx, distance, dir_i, dir_progress) + source_yxyx = self.source_yxyx(yx) + f(yx, source_yxyx, distance, dir_i, dir_progress) distance += 1 @@ -363,7 +377,6 @@ class FovMap(SourcedMap): class FovMapHex(FovMap): circle_out_directions = ('DOWNLEFT', 'LEFT', 'UPLEFT', 'UPRIGHT', 'RIGHT', 'DOWNRIGHT') - geometry_class = MapGeometryHex def circle_out_move(self, yx, direction): return self.basic_circle_out_move(yx, direction) @@ -373,7 +386,6 @@ class FovMapHex(FovMap): class FovMapSquare(FovMap): circle_out_directions = (('DOWN', 'LEFT'), ('LEFT', 'UP'), ('UP', 'RIGHT'), ('RIGHT', 'DOWN')) - geometry_class = MapGeometrySquare def circle_out_move(self, yx, direction): yx = self.basic_circle_out_move(yx, direction[0])