Make fungeon precondition configurable. Add growth point forking in dungeon generator.

Author: AlexeyBond

addons/wfc/examples/demo_wfc_2d_tilemap.tscn

@@ -32,6 +32,19 @@ backtracking_limit = -1
 
 [sub_resource type="Resource" id="Resource_26rrk"]
 script = ExtResource("2_71acl")
+wall_border_size = Vector2i(1, 2)
+free_gap = Vector2i(2, 2)
+room_half_size_base = Vector2i(2, 2)
+room_half_size_variation = Vector2i(2, 2)
+room_probability = 0.5
+road_len_base = 15
+road_len_variation = 5
+fork_probability = 0.5
+full_fork_probability = 0.5
+passable_area_ratio = 0.1
+iterations_limit = 1000
+road_class = "wfc_dungeon_road"
+wall_class = "wfc_dungeon_wall"
 
 [sub_resource type="Resource" id="Resource_uafrt"]
 script = ExtResource("3_iuw6v")

addons/wfc/problems/2d/preconditions/precondition_2d_base.gd

@@ -4,17 +4,17 @@ extends RefCounted
 func prepare():
 	"""
 	May calculate data necessary for this precondition.
-	
-	Must be called exactly once on one object.
+
+	Must be called at most once on one object.
 	"""
 	pass
 
 func read_domain(coords: Vector2i) -> WFCBitSet:
 	"""
 	Returns initial domain of a cell at given coordinates.
-	
+
 	Returns null if this precondition doesn't limit domain of cell at given coordinates.
-	
+
 	Must not be called before completion of prepare() call.
 	"""
 	return null

addons/wfc/problems/2d/preconditions/precondition_2d_dungeon.gd

@@ -29,7 +29,7 @@ const STATE_PASSAGE = 2
 
 func coord_to_id(c: Vector2i) -> int:
 	assert(rect.has_point(c))
-	
+
 	var rel_c: Vector2i = c - rect.position
 
 	return rel_c.x + rel_c.y * rect.size.x
@@ -46,7 +46,7 @@ func _meta_to_bool(meta: Array) -> bool:
 	for x in meta:
 		if x:
 			return true
-	
+
 	return false
 
 func learn_classes(
@@ -56,7 +56,7 @@ func learn_classes(
 ):
 	passable_domain = WFCBitSet.new(mapper.size())
 	walls_domain = WFCBitSet.new(mapper.size())
-	
+
 	for i in range(mapper.size()):
 		if _meta_to_bool(mapper.read_tile_meta(i, road_class)):
 			passable_domain.set_bit(i)
@@ -106,7 +106,7 @@ func _get_start_point() -> Vector2i:
 class _GrowthPoint:
 	var position: Vector2i
 	var direction: Vector2i
-	
+
 	func _init(p: Vector2i, d: Vector2i):
 		self.position = p
 		self.direction = d
@@ -120,6 +120,11 @@ class _GrowthPoint:
 		if reverse:
 			self.direction *= -1
 
+	func rotated(reverse: bool) -> _GrowthPoint:
+		var gp := _GrowthPoint.new(position, direction)
+		gp.rotate(reverse)
+		return gp
+
 func _free_space_around(r: Rect2i):
 	_replace_rect(
 		r.grow_individual(
@@ -156,8 +161,6 @@ func _generate(start_point: Vector2i, start_directions: Array[Vector2i]):
 
 	var passable_rect: Rect2i = _get_safe_passable_rect()
 
-	print('Total rect: ', rect, ' passable rect: ', passable_rect)
-
 	assert(passable_area_ratio > 0.0 and passable_area_ratio < 1.0)
 	var remaining_area: int = passable_area_ratio * passable_rect.get_area()
 	assert(remaining_area > 0)
@@ -192,19 +195,21 @@ func _generate(start_point: Vector2i, start_directions: Array[Vector2i]):
 
 		_free_space_around(road_rect)
 
-		gp.rotate(randf() > 0.5)
-
-		growth_points.push_back(gp)
+		if randf() < fork_probability:
+			growth_points.push_back(gp.rotated(true))
+			growth_points.push_back(gp.rotated(false))
+			if randf() < full_fork_probability:
+				growth_points.push_back(gp)
+		else:
+			gp.rotate(randf() > 0.5)
+			growth_points.push_back(gp)
 
 func prepare():
 	assert(rect.has_area())
 
 	state.resize(rect.get_area())
 
 	_generate(_get_start_point(), [Vector2i.DOWN, Vector2i.UP])
-	#var test_road_rect: Rect2i = Rect2i(10, 10, 100, 1)
-	#_replace_rect(test_road_rect, STATE_WALL, STATE_PASSAGE)
-	#_replace_rect(test_road_rect.grow(2), STATE_WALL, STATE_FREE)
 
 
 func read_domain(coords: Vector2i) -> WFCBitSet:

addons/wfc/problems/2d/preconditions/precondition_2d_dungeon_settings.gd

@@ -1,12 +1,80 @@
 extends WFC2DPrecondition2DNullSettings
 class_name WFC2DPreconditionDungeonSettings
 
+@export
+var wall_border_size: Vector2i = Vector2i(1, 2)
+
+@export
+var free_gap: Vector2i = Vector2i(2, 2)
+
+@export
+var room_half_size_base: Vector2i = Vector2i(2, 2)
+
+@export
+var room_half_size_variation: Vector2i = Vector2i(2, 2)
+
+@export_range(0.0001, 1.0)
+var room_probability: float = 0.5
+
+@export
+var road_len_base: int = 15
+
+@export
+var road_len_variation: int = 5
+
+@export_range(0.0001, 1.0)
+var fork_probability: float = 0.5
+
+@export_range(0.0001, 1.0)
+var full_fork_probability: float = 0.5
+
+@export_range(0.0001, 1.0)
+var passable_area_ratio: float = 0.1
+
+@export
+var iterations_limit: int = 1000
+
+@export
+var road_class: String = "wfc_dungeon_road"
+
+@export
+var wall_class: String = "wfc_dungeon_wall"
+
 func create_precondition(parameters: WFC2DPrecondition2DNullSettings.CreationParameters) -> WFC2DPrecondition:
 	var res: WFC2DPreconditionDungeon = WFC2DPreconditionDungeon.new()
 
 	res.rect = parameters.problem_settings.rect
 
-	res.learn_classes(parameters.problem_settings.rules.mapper)
+	res.learn_classes(parameters.problem_settings.rules.mapper, road_class, wall_class)
+
+	assert(wall_border_size >= Vector2i.ZERO)
+	res.wall_border_size = wall_border_size
+
+	assert(free_gap >= Vector2i.ZERO)
+	res.free_gap = free_gap
+
+	assert(room_half_size_base > Vector2i.ZERO)
+	res.room_half_size_base = room_half_size_base
+
+	assert(room_half_size_variation > Vector2i.ZERO)
+	assert(room_half_size_variation <= room_half_size_variation)
+	res.room_half_size_variation = room_half_size_variation
+
+	res.room_probability = room_probability
+
+	assert(road_len_base > 0)
+	res.road_len_base = road_len_base
+
+	assert(road_len_variation > 0)
+	assert(road_len_variation < road_len_base)
+	res.road_len_variation = road_len_variation
+
+	res.fork_probability = fork_probability
+	res.full_fork_probability = full_fork_probability
+	res.passable_area_ratio = passable_area_ratio
+
+	assert(iterations_limit > 0)
+	res.iterations_limit = iterations_limit
 
 	return res
 

project.godot

@@ -19,6 +19,7 @@ config/icon="res://icon.png"
 
 window/size/viewport_width=1920
 window/size/viewport_height=1080
+window/size/mode=3
 
 [editor]