TouhouGD/addons/MetroidvaniaSystem/Scripts/RoomDrawer.gd

extends RefCounted

static var force_mapped: bool

static func draw(canvas_item: CanvasItem, offset: Vector2, coords: Vector3i, skip_empty: bool, map_data: MetroidvaniaSystem.MapData, save_data: MetroidvaniaSystem.SaveData):
	if MetSys.settings.custom_elements and not map_data.custom_elements.is_empty():
		setup_custom_elements(canvas_item, offset, coords)
	
	var cell_data := map_data.get_cell_at(coords)
	if not cell_data:
		if not skip_empty:
			draw_empty(canvas_item, offset)
		return
	
	var discovered := 2
	if save_data:
		discovered = save_data.is_cell_discovered(coords)
	elif force_mapped:
		discovered = 1
	
	if discovered == 0:
		if not skip_empty:
			draw_empty(canvas_item, offset)
		return
	
	var theme: MapTheme = MetSys.settings.theme
	
	var ci := canvas_item.get_canvas_item()
	var display_flags: int = (int(discovered == 2) * 255) | theme.mapped_display
	
	# center
	if bool(display_flags & MetroidvaniaSystem.DISPLAY_CENTER):
		var room_color := cell_data.get_color()
		theme.center_texture.draw(ci, offset * MetSys.CELL_SIZE, room_color if discovered == 2 else theme.mapped_center_color)
	
	# corners
	if theme.use_shared_borders:
		if not MetSys.has_meta(&"shared_borders_to_draw"):
			MetSys._add_meta(&"shared_borders_to_draw", {})#[Vector3i, Vector2]
			MetSys._add_meta(&"shared_borders_data", {"canvas_item": canvas_item, "display_flags": display_flags})
		
		var shared_borders_to_draw: Dictionary = MetSys.get_meta(&"shared_borders_to_draw")
		shared_borders_to_draw[coords] = offset
	else:
		draw_regular_borders(canvas_item, offset, coords, map_data, cell_data, display_flags, discovered)
	
	canvas_item.draw_set_transform_matrix(Transform2D())
	
	if bool(display_flags & MetroidvaniaSystem.DISPLAY_SYMBOLS):
		var symbol: int = -1
		
		if save_data and coords in save_data.custom_markers:
			var custom: int = save_data.custom_markers[coords]
			for i in range(63, -1, -1):
				if custom & 1 << i:
					symbol = i
					break
		
		if symbol == -1:
			symbol = cell_data.get_symbol()
		
		if symbol > - 1:
			if symbol >= theme.symbols.size():
				push_error("Bad symbol '%s' at '%s'" % [symbol, coords])
			else:
				canvas_item.draw_texture(theme.symbols[symbol], offset * MetSys.CELL_SIZE + MetSys.CELL_SIZE * 0.5 - theme.symbols[symbol].get_size() / 2)

static func draw_regular_borders(canvas_item: CanvasItem, offset: Vector2, coords: Vector3i, map_data: MetroidvaniaSystem.MapData, cell_data: MetroidvaniaSystem.MapData.CellData, display_flags: int, discovered: int):
	var theme: MapTheme = MetSys.settings.theme
	var ci := canvas_item.get_canvas_item()
	
	# borders
	var borders: Array[int] = [-1, -1, -1, -1]
	
	var display_outlines := bool(display_flags & MetroidvaniaSystem.DISPLAY_OUTLINE)
	for i in 4:
		var border: int = cell_data.get_border(i)
		if not display_outlines and border == 0:
			borders[i] = -1
		elif not bool(display_flags & MetroidvaniaSystem.DISPLAY_BORDERS):
			borders[i] = mini(border, 0 if display_outlines else -1)
		else:
			borders[i] = border
	
	for i in 4:
		var texture: Texture2D
		var color: Color
		
		var rotation := PI * 0.5 * i
		if borders[i] == -1:
			if display_outlines:
				texture = get_border_texture(theme, -1, i)
				color = Color.WHITE
		else:
			var border: int = borders[i]
			
			if not bool(display_flags & MetroidvaniaSystem.DISPLAY_BORDERS):
				border = 0
			
			texture = get_border_texture(theme, border, i)
			if discovered == 2:
				color = cell_data.get_border_color(i)
			else:
				color = theme.mapped_border_color
		
		if not texture:
			continue
		
		canvas_item.draw_set_transform(offset * MetSys.CELL_SIZE + MetSys.CELL_SIZE * 0.5, rotation, Vector2.ONE)
		match i:
			MetroidvaniaSystem.R, MetroidvaniaSystem.L:
				texture.draw(ci, -texture.get_size() / 2 + Vector2.RIGHT * (MetSys.CELL_SIZE.x * 0.5 - texture.get_width() / 2), color)
			MetroidvaniaSystem.D, MetroidvaniaSystem.U:
				texture.draw(ci, -texture.get_size() / 2 + Vector2.RIGHT * (MetSys.CELL_SIZE.y * 0.5 - texture.get_width() / 2), color)
	
	if not display_outlines:
		return
	
	# outer corner
	for i in 4:
		var j := rotate(i)
		if borders[i] == -1 or borders[j] == -1:
			continue
		
		var texture: Texture2D = theme.outer_corner
		var corner_color: Color
		if discovered == 2:
			corner_color = get_shared_color(cell_data.get_border_color(i), cell_data.get_border_color(j), theme.default_border_color)
		else:
			corner_color = theme.mapped_border_color
		
		canvas_item.draw_set_transform(offset * MetSys.CELL_SIZE + MetSys.CELL_SIZE * 0.5, PI * 0.5 * i, Vector2.ONE)
		
		var corner_offset := -texture.get_size()
		if theme.use_shared_borders:
			corner_offset /= 2
		
		match i:
			MetroidvaniaSystem.R, MetroidvaniaSystem.L:
				texture.draw(ci, MetSys.CELL_SIZE * 0.5 + corner_offset, corner_color)
			MetroidvaniaSystem.D, MetroidvaniaSystem.U:
				texture.draw(ci, Vector2(MetSys.CELL_SIZE.y, MetSys.CELL_SIZE.x) * 0.5 + corner_offset, corner_color)
	
	# inner corner
	for i in 4:
		var j := rotate(i)
		if borders[i] != -1 or borders[j] != -1:
			continue
		
		var neighbor_room := get_neighbor(map_data, coords, map_data.FWD[i] + map_data.FWD[j])
		if neighbor_room:
			if neighbor_room.borders[opposite(i)] == -1 and neighbor_room.borders[opposite(j)] == -1:
				continue
		
		var texture: Texture2D = theme.inner_corner
		var color1 := get_neighbor(map_data, coords, map_data.FWD[i]).get_border_color(rotate(i))
		var color2 := get_neighbor(map_data, coords, map_data.FWD[j]).get_border_color(rotate(j, -1))
		
		var corner_color := get_shared_color(color1, color2, theme.default_border_color)
		
		canvas_item.draw_set_transform(offset * MetSys.CELL_SIZE + MetSys.CELL_SIZE * 0.5, PI * 0.5 * i, Vector2.ONE)
		
		var corner_offset := -texture.get_size()
		if theme.use_shared_borders:
			corner_offset /= 2
		
		match i:
			MetroidvaniaSystem.R, MetroidvaniaSystem.L:
				theme.inner_corner.draw(ci, MetSys.CELL_SIZE * 0.5 + corner_offset, corner_color)
			MetroidvaniaSystem.D, MetroidvaniaSystem.U:
				theme.inner_corner.draw(ci, Vector2(MetSys.CELL_SIZE.y, MetSys.CELL_SIZE.x) * 0.5 + corner_offset, corner_color)

static func draw_shared_borders():
	assert(MetSys.settings.theme.use_shared_borders, "This function requires shared borders to be enabled.")
	if not MetSys.has_meta(&"shared_borders_to_draw"):
		return
	
	var theme: MapTheme = MetSys.settings.theme
	
	var shared_borders_to_draw: Dictionary = MetSys.get_meta(&"shared_borders_to_draw")
	var shared_borders_data: Dictionary = MetSys.get_meta(&"shared_borders_data")
	
	var shared_borders_colors: Dictionary#[Vector3i, Color]
	var shared_corners_data: Dictionary#[Vector3i, int]
	var shared_corners_colors: Dictionary#[Vector3i, Color]
	
	var display_flags: int = shared_borders_data["display_flags"]
	var canvas_item: CanvasItem = shared_borders_data["canvas_item"]
	var ci := canvas_item.get_canvas_item()
	
	var additional_corners: Dictionary#[Vector3i, Vector2]
	for coords in shared_borders_to_draw:
		for x in range (-1, 1):
			for y in range(-1, 1):
				var coords2: Vector3i = coords + Vector3i(x, y, 0)
				if not coords2 in shared_borders_to_draw:
					additional_corners[coords2] = shared_borders_to_draw[coords] + Vector2(x, y)
	
	shared_borders_to_draw.merge(additional_corners)
	for coords in shared_borders_to_draw:
		var corner_data := 0
		corner_data |= (1 << 0) * (get_corner_bit(coords, MetroidvaniaSystem.R) | get_corner_bit(coords + Vector3i(1, 0, 0), MetroidvaniaSystem.L))
		corner_data |= (1 << 1) * (get_corner_bit(coords, MetroidvaniaSystem.D) | get_corner_bit(coords + Vector3i(0, 1, 0), MetroidvaniaSystem.U))
		corner_data |= (1 << 2) * (get_corner_bit(coords + Vector3i(0, 1, 0), MetroidvaniaSystem.R) | get_corner_bit(coords + Vector3i(1, 1, 0), MetroidvaniaSystem.L))
		corner_data |= (1 << 3) * (get_corner_bit(coords + Vector3i(1, 0, 0), MetroidvaniaSystem.D) | get_corner_bit(coords + Vector3i(1, 1, 0), MetroidvaniaSystem.U))
		
		if theme.is_unicorner():
			corner_data = signi(corner_data) * 15
		
		var use_default: bool
		var color_blend: Array[Color]
		for i in 4:
			var color: Color
			if i < 2:
				color = get_shared_border_color(coords, i)
			else:
				color = get_shared_border_color(coords + Vector3i(1, 1, 0), i)
			
			if color == theme.default_border_color:
				use_default = true
			elif color.a > 0:
				color_blend.append(color)
		
		shared_corners_data[coords] = corner_data
		if not color_blend.is_empty():
			var ratio: float = 1.0 / color_blend.size()
			shared_corners_colors[coords] = color_blend.reduce(func(final: Color, current: Color) -> Color: return final.lerp(current, ratio))
		elif use_default:
			shared_corners_colors[coords] = theme.default_border_color
		
		var border_data: Array[int] = [-1, -1]
		var border_colors: Array[Color] = [Color(), Color()]
		for i in 2:
			var fwd := Vector3i(MetroidvaniaSystem.MapData.FWD[i].x, MetroidvaniaSystem.MapData.FWD[i].y, 0)
			border_data[i] = maxi(get_border_at(coords, i), get_border_at(coords + fwd, opposite(i)))
			border_colors[i] = get_shared_border_color(coords, i)
		
		shared_borders_data[coords] = border_data
		shared_borders_colors[coords] = border_colors
	
	# borders
	for coords in shared_borders_to_draw:
		var borders: Array[int] = shared_borders_data[coords]
		for i in 2:
			var texture: Texture2D
			var color: Color
			
			var rotation := PI * 0.5 * i
			if borders[i] == -1:
				texture = get_border_texture(theme, -1, i)
				color = Color.WHITE
			else:
				var border: int = borders[i]
				
				if not bool(display_flags & MetroidvaniaSystem.DISPLAY_BORDERS):
					border = 0
				
				texture = get_border_texture(theme, border, i)
				color = shared_borders_colors[coords][i]
			
			if not texture:
				continue
			
			canvas_item.draw_set_transform(shared_borders_to_draw[coords] * MetSys.CELL_SIZE + MetSys.CELL_SIZE * 0.5, rotation, Vector2.ONE)
			match i:
				MetroidvaniaSystem.R, MetroidvaniaSystem.L:
					texture.draw(ci, -texture.get_size() / 2 + Vector2.RIGHT * (MetSys.CELL_SIZE.x * 0.5), color)
				MetroidvaniaSystem.D, MetroidvaniaSystem.U:
					texture.draw(ci, -texture.get_size() / 2 + Vector2.RIGHT * (MetSys.CELL_SIZE.y * 0.5), color)
	
	# corners
	for coords in shared_borders_to_draw:
		var offset: Vector2 = shared_borders_to_draw[coords]
		var corner_data: int = shared_corners_data[coords]
		var corner_type = signi(corner_data & 1) + signi(corner_data & 2) + signi(corner_data & 4) + signi(corner_data & 8)
		
		var corner_texture: Texture2D
		var corner_rotation := -1
		match corner_type:
			1:
				corner_texture = theme.u_corner
				match corner_data:
					1:
						corner_rotation = 0
					2:
						corner_rotation = 3
					4:
						corner_rotation = 2
					8:
						corner_rotation = 1
			2:
				corner_texture = theme.l_corner
				match corner_data:
					3:
						corner_rotation = 3
					6:
						corner_rotation = 2
					9:
						corner_rotation = 0
					12:
						corner_rotation = 1
			3:
				corner_texture = theme.t_corner
				match corner_data:
					7:
						corner_rotation = 1
					11:
						corner_rotation = 2
					13:
						corner_rotation = 3
					14:
						corner_rotation = 0
			4:
				corner_texture = theme.cross_corner
				corner_rotation = 0
		
		if corner_rotation == -1 or not corner_texture:
			continue
		
		canvas_item.draw_set_transform(offset * MetSys.CELL_SIZE + MetSys.CELL_SIZE, PI * 0.5 * corner_rotation, Vector2.ONE)
		corner_texture.draw(canvas_item.get_canvas_item(), -corner_texture.get_size() / 2, shared_corners_colors[coords])
	
	MetSys.remove_meta(&"shared_borders_to_draw")
	MetSys.remove_meta(&"shared_borders_data")
	canvas_item.draw_set_transform_matrix(Transform2D())

static func setup_custom_elements(canvas_item: CanvasItem, offset: Vector2, coords: Vector3i):
		var custom_element_data: Dictionary
		if not MetSys.has_meta(&"custom_elements_data"):
			custom_element_data["canvas_item"] = canvas_item
			MetSys._add_meta(&"custom_elements_data", custom_element_data)

static func draw_custom_elements(canvas_item: CanvasItem, elements: Dictionary, base_offset: Vector2, rect: Rect2i, layer: int):
	var element_manager: MetroidvaniaSystem.CustomElementManager = MetSys.settings.custom_elements
	var already_drawn: Array[Dictionary]
	
	for y in rect.size.y:
		for x in rect.size.y:
			var pos := rect.position + Vector2i(x, y)
			for coords in elements:
				if coords.z != layer:
					continue
				
				var element: Dictionary = elements[coords]
				if element in already_drawn:
					continue
				
				var elerect := Rect2i(coords.x, coords.y, element["size"].x, element["size"].y)
				if elerect.has_point(pos):
					element_manager.draw_element(canvas_item, coords, element.name, (base_offset + Vector2(coords.x, coords.y) - Vector2(pos) + Vector2(x, y)) * MetSys.CELL_SIZE, Vector2(element.size) * MetSys.CELL_SIZE, element.data)
					already_drawn.append(element)

static func get_border_at(coords: Vector3i, idx: int) -> int:
	var cell_data = get_discovered_cell_at(coords)
	if not cell_data:
		return -1
	return cell_data.get_border(idx)

static func get_corner_bit(coords: Vector3i, idx: int) -> int:
	return int(get_border_at(coords, idx) > -1)

static func get_shared_border_color(coords: Vector3i, idx: int) -> Color:
	var fwd := Vector3i(MetroidvaniaSystem.MapData.FWD[idx].x, MetroidvaniaSystem.MapData.FWD[idx].y, 0)
	var color := Color.TRANSPARENT
	
	var cell_data = get_discovered_cell_at(coords)
	if cell_data:
		color = cell_data.get_border_color(idx)
	
	cell_data = get_discovered_cell_at(coords + fwd)
	if cell_data:
		if color.a > 0:
			color = get_shared_color(color, cell_data.get_border_color(opposite(idx)), MetSys.settings.theme.default_border_color)
		else:
			color = cell_data.get_border_color(opposite(idx))
	
	return color

static func get_discovered_cell_at(coords: Vector3i) -> MetroidvaniaSystem.MapData.CellData:
	var cell_data = MetSys.map_data.get_cell_at(coords)
	if cell_data and MetSys.is_cell_discovered(coords):
		return cell_data
	return null

static func get_shared_color(color1: Color, color2: Color, default: Color) -> Color:
	if color1 == default:
		return color2
	elif color2 == default:
		return color1
	return color1.lerp(color2, 0.5)

static func get_border_texture(theme: MapTheme, idx: int, direction: int) -> Texture2D:
	var texture_name: StringName
	
	if theme.rectangle:
		if direction == MetroidvaniaSystem.R or direction == MetroidvaniaSystem.L:
			match idx:
				-1:
					texture_name = &"vertical_separator"
				0:
					texture_name = &"vertical_wall"
				1:
					texture_name = &"vertical_passage"
				2:
					texture_name = &"vertical_borders"
		else:
			match idx:
				-1:
					texture_name = &"horizontal_separator"
				0:
					texture_name = &"horizontal_wall"
				1:
					texture_name = &"horizontal_passage"
				_:
					texture_name = &"vertical_borders"
	else:
		match idx:
			-1:
				texture_name = &"separator"
			0:
				texture_name = &"wall"
			1:
				texture_name = &"passage"
			_:
				texture_name = &"borders"
	
	if idx >= 2:
		return theme.get(texture_name)[idx - 2]
	else:
		return theme.get(texture_name)

static func draw_empty(canvas_item: CanvasItem, offset: Vector2):
	var theme: MapTheme = MetSys.settings.theme
	if theme.empty_space_texture:
		var ci := canvas_item.get_canvas_item()
		theme.empty_space_texture.draw(ci, offset * MetSys.CELL_SIZE, Color.WHITE)

static func get_neighbor(map_data: MetroidvaniaSystem.MapData, coords: Vector3i, offset: Vector2i) -> MetroidvaniaSystem.MapData.CellData:
	var neighbor: Vector2i = Vector2i(coords.x, coords.y) + offset
	return map_data.get_cell_at(Vector3i(neighbor.x, neighbor.y, coords.z))

static func rotate(i: int, amount := 1) -> int:
	return (i + amount) % 4

static func opposite(i: int) -> int:
	return (i + 2) % 4