TouhouGD/addons/MetroidvaniaSystem/Scripts/MapView.gd

extends Control

@onready var map_overlay: Control = $MapOverlay
@onready var map: Control = %Map
@onready var current_layer_spinbox: SpinBox = %CurrentLayer
@onready var status_label: Label = %StatusLabel

const NULL_VECTOR2I = Vector2i(-9999999, -9999999)
var plugin: EditorPlugin

var drag_from: Vector2i = NULL_VECTOR2I
var view_drag: Vector4
var map_offset := Vector2i(10, 10)

var current_layer: int
var force_mapped: bool

func _enter_tree() -> void:
	if owner:
		plugin = owner.plugin

func _ready() -> void:
	map.draw.connect(_on_map_draw)
	map_overlay.mouse_exited.connect(status_label.hide)
	map_overlay.gui_input.connect(_on_overlay_input)
	map_overlay.draw.connect(_on_overlay_draw)
	
	current_layer_spinbox.value_changed.connect(on_layer_changed)
	%RecenterButton.pressed.connect(on_recenter_view)
	
	status_label.hide()
	await get_tree().process_frame
	update_map_position()
	map.size = MetSys.CELL_SIZE * 200
	
	var refresh := func():
		update_map_position()
		on_layer_changed(current_layer)
	
	MetSys.settings.theme_changed.connect(refresh)
	MetSys.theme_modified.connect(refresh.unbind(1))

func get_cursor_pos() -> Vector2i:
	var pos := (map_overlay.get_local_mouse_position() - MetSys.CELL_SIZE / 2).snapped(MetSys.CELL_SIZE) / MetSys.CELL_SIZE as Vector2i - map_offset
	return pos

func on_layer_changed(l: int):
	current_layer = l
	map.queue_redraw()
	map_overlay.queue_redraw()

func on_recenter_view() -> void:
	map_offset = Vector2i(10, 10)
	map_overlay.queue_redraw()
	update_map_position()

func _on_overlay_input(event: InputEvent) -> void:
	if event is InputEventMouseMotion:
		if view_drag != Vector4():
			if event.position.x >= map_overlay.size.x:
				map_overlay.warp_mouse(Vector2(event.position.x - map_overlay.size.x, event.position.y))
				view_drag.x -= map_overlay.size.x
			elif event.position.x < 0:
				map_overlay.warp_mouse(Vector2(map_overlay.size.x + event.position.x, event.position.y))
				view_drag.x += map_overlay.size.x
			
			if event.position.y >= map_overlay.size.y:
				map_overlay.warp_mouse(Vector2(event.position.x, event.position.y - map_overlay.size.y))
				view_drag.y -= map_overlay.size.y
			elif event.position.y < 0:
				map_overlay.warp_mouse(Vector2(event.position.x, map_overlay.size.y + event.position.y))
				view_drag.y += map_overlay.size.y
			
			map_offset = Vector2(view_drag.z, view_drag.w) + (map_overlay.get_local_mouse_position() - Vector2(view_drag.x, view_drag.y)) / MetSys.CELL_SIZE
			update_map_position()
			map_overlay.queue_redraw()
			_on_drag()
		else:
			map_overlay.queue_redraw()
		
		_update_status_label()
	
	if event is InputEventMouseButton:
		if event.button_index == MOUSE_BUTTON_MIDDLE:
			if event.pressed:
				view_drag.x = map_overlay.get_local_mouse_position().x
				view_drag.y = map_overlay.get_local_mouse_position().y
				view_drag.z = map_offset.x
				view_drag.w = map_offset.y
			else:
				view_drag = Vector4()

func _on_drag():
	pass

func _update_status_label():
	status_label.show()
	status_label.text = str(get_cursor_pos())

func _unhandled_key_input(event: InputEvent) -> void:
	if not visible:
		return
	
	if event is InputEventKey and event.pressed:
		if event.physical_keycode == KEY_Q:
			current_layer_spinbox.value -= 1
			accept_event()
		elif event.physical_keycode == KEY_E:
			current_layer_spinbox.value += 1
			accept_event()

func _notification(what: int) -> void:
	if what == NOTIFICATION_THEME_CHANGED:
		if map_overlay:
			map_overlay.queue_redraw()

func _on_map_draw() -> void:
	if not plugin:
		return
	
	MetroidvaniaSystem.RoomDrawer.force_mapped = force_mapped
	for x in range(-100, 100):
		for y in range(-100, 100):
			MetSys.draw_cell(map, Vector2i(x, y) + Vector2i(100, 100), Vector3i(x, y, current_layer), true, false)
	
	if MetSys.settings.theme.use_shared_borders:
		MetSys.draw_shared_borders()
	
	MetroidvaniaSystem.RoomDrawer.force_mapped = false

func _on_overlay_draw() -> void:
	MetSys.draw_custom_elements(map_overlay, Rect2i(-map_offset, map_overlay.size / MetSys.CELL_SIZE + Vector2.ONE), Vector2(), current_layer)

func update_map_position():
	map.position = Vector2(map_offset - Vector2i(100, 100)) * MetSys.CELL_SIZE