Mag gen & work towards AI

scripts/v2/worldgen/standard_generator.gd

@@ -20,6 +20,8 @@ var _rooms_blocking: Array[PackedScene] = []
 
 var _progress_tracker: ProgressTracker
 
+var _markers: Dictionary = {}
+
 func _generate(map: TileMap) -> void:
 	self._progress_tracker = ProgressTracker.new(1)
 
@@ -32,6 +34,9 @@ func _generate(map: TileMap) -> void:
 	# Create rooms
 	self._create_rooms(map, init_room, self.max_room_path_length, _progress_tracker.next_step("Creating Rooms"))
 	map.update_internals()
+	
+	self._spawn_entities(map, _progress_tracker.next_step("Spawning Entities"))
+	map.update_internals()
 
 func _sort_rooms(step_tracker: ProgressStepTracker) -> void:
 	step_tracker.complete.call_deferred()
@@ -163,5 +168,95 @@ func _get_rooms(exit: Vector2i) -> Array[PackedScene]:
 		push_error("Invalid exit: " + str(exit))
 		return []
 
+func _register_marker(group: String, marker: Node2D) -> void:
+	if not group in self._markers:
+		self._markers[group] = [marker]
+	else:
+		self._markers[group].append(marker)
+
+func _spawn_entities(map: TileMap, step_tracker: ProgressStepTracker) -> void:
+	var tree = map.get_tree()
+	
+	# Register entity markers
+	for entity_marker in tree.get_nodes_in_group("genv2:entity_marker"):
+		if entity_marker is StandardEntityMarker:
+			entity_marker._register(self)
+	
+	for marker_id in self._markers:
+		var sample_marker: StandardEntityMarker = self._markers[marker_id][0]
+		
+		match sample_marker.spawn_method:
+			StandardEntityMarker.SPAWN_METHOD.ALL:
+				for marker in self._markers[marker_id]:
+					map.erase_cell.call_deferred(0, map.local_to_map(marker.position))
+					
+					var packed_entity: PackedScene = _select_entity(marker.entity_selection_method, marker.entities)
+					if packed_entity == null:
+						continue
+					
+					var entity: Node2D = packed_entity.instantiate()
+					entity.position = marker.position
+					map.add_child(entity)
+			StandardEntityMarker.SPAWN_METHOD.FURTHEST:
+				var furthest_marker: StandardEntityMarker = self._markers[marker_id].pop_back()
+				var furthest_distance: float = furthest_marker.position.distance_to(Vector2.ZERO)
+				
+				# Calculate furthest marker and destroy others
+				for marker in self._markers[marker_id]:
+					var calc_distance: float = marker.position.distance_to(Vector2.ZERO)
+					if calc_distance > furthest_distance:
+						map.erase_cell(0, map.local_to_map(furthest_marker.position))
+						furthest_marker = marker
+						furthest_distance = calc_distance
+					else:
+						map.erase_cell(0, map.local_to_map(marker.position))
+
+				# Erase furthest marker
+				map.erase_cell.call_deferred(0, map.local_to_map(furthest_marker.position))
+
+				# Get entity
+				var packed_entity: PackedScene = _select_entity(furthest_marker.entity_selection_method, furthest_marker.entities)
+				if packed_entity == null:
+					continue
+
+				var entity: Node2D = packed_entity.instantiate()
+				entity.position = furthest_marker.position
+				map.add_child(entity)
+				
+				
+			StandardEntityMarker.SPAWN_METHOD.ONCE:
+				var i = randi() % len(self._markers[marker_id])
+				var marker = self._markers[marker_id][i]
+				
+				var packed_entity: PackedScene = _select_entity(marker.entity_selection_method, marker.entities)
+				if packed_entity != null:
+					var entity: Node2D = packed_entity.instantiate()
+					entity.position = marker.position
+					map.add_child(entity)
+				
+				for _marker in self._markers[marker_id]:
+					map.erase_cell(0, map.local_to_map(_marker.position))
+				
+			StandardEntityMarker.SPAWN_METHOD.NONE:
+				for marker in self._markers[marker_id]:
+					map.erase_cell(0, map.local_to_map(marker.position))
+			_:
+				push_error("Invalid spawning method!")
+				continue
+
+func _select_entity(mode: StandardEntityMarker.ENTITY_SELECTION_METHOD, entities: Array) -> PackedScene:
+	if len(entities) == 0:
+		return null
+	
+	match mode:
+		StandardEntityMarker.ENTITY_SELECTION_METHOD.RANDOM:
+			var i = randi() % len(entities)
+			return entities[i]
+		StandardEntityMarker.ENTITY_SELECTION_METHOD.POP:
+			var i = randi() % len(entities)
+			return entities.pop_at(i)
+
+	return null
+
 func _get_progress_tracker() -> ProgressTracker:
 	return self._progress_tracker