# Animation control test: V15 male (Tripo mesh, Meshy rig renamed to Quaternius
# bones) driven by REAL UAL1 clips baked for this skeleton via
# ariki-game/tools/mixamo_retarget.py --map ual (world-rotation-delta retarget).
# Controls: same as anim_male.gd (see HUD).
extends Node3D

const DIR := "C:/Users/CAN/tinqs-ltd/tattoo-test/male/"
const BODY_GLB := DIR + "male-lena-v15-quatnamed.glb"
const CLIPS_GLB := DIR + "male-ual-clips.glb"
const PREFERRED_ORDER := ["Idle_Loop", "Walk_Loop", "Jog_Fwd_Loop", "Sprint_Loop", "Dance_Loop", "Punch_Jab"]

var model: Node3D
var skel: Skeleton3D
var cam: Camera3D
var player: AnimationPlayer
var hud: Label
var clips: Array[String] = []
var clip_idx := 0
var lights: Array[Light3D] = []
var env: Environment
var lights_on := true

var interactive := false
var dragging := false
var orbit_yaw := 0.0
var orbit_pitch := 0.08
var orbit_dist := 3.6
var orbit_target := Vector3(0, 0.95, 0)

func _ready() -> void:
	env = Environment.new()
	env.background_mode = Environment.BG_COLOR
	env.background_color = Color(0.12, 0.13, 0.16)
	env.ambient_light_source = Environment.AMBIENT_SOURCE_COLOR
	env.ambient_light_color = Color(0.75, 0.78, 0.85)
	env.ambient_light_energy = 0.7
	env.tonemap_mode = Environment.TONE_MAPPER_FILMIC
	var we := WorldEnvironment.new()
	we.environment = env
	add_child(we)

	var sun := DirectionalLight3D.new()
	sun.rotation_degrees = Vector3(-40, -35, 0)
	sun.light_energy = 1.5
	add_child(sun)
	var fill := DirectionalLight3D.new()
	fill.rotation_degrees = Vector3(-15, 140, 0)
	fill.light_energy = 0.6
	add_child(fill)
	var under := DirectionalLight3D.new()
	under.rotation_degrees = Vector3(35, 0, 0)
	under.light_energy = 0.35
	add_child(under)
	lights = [sun, fill, under]

	var ground := MeshInstance3D.new()
	var plane := PlaneMesh.new()
	plane.size = Vector2(12, 12)
	ground.mesh = plane
	var gmat := StandardMaterial3D.new()
	gmat.albedo_color = Color(0.16, 0.17, 0.21)
	gmat.roughness = 1.0
	ground.set_surface_override_material(0, gmat)
	add_child(ground)

	model = _load_glb(BODY_GLB)
	if model == null:
		return
	add_child(model)
	var stack: Array = [model]
	while not stack.is_empty():
		var n: Node = stack.pop_back()
		if n is Skeleton3D and skel == null:
			skel = n
		for c in n.get_children():
			stack.push_back(c)
	if skel == null:
		push_error("no skeleton in body GLB")
		return
	print("INFO bones=", skel.get_bone_count())

	# lift baked UAL clips onto a local player bound to this skeleton
	var clip_scene := _load_glb(CLIPS_GLB)
	if clip_scene == null:
		return
	var src := _find_player(clip_scene)
	if src == null:
		push_error("no AnimationPlayer in clips GLB")
		return
	player = AnimationPlayer.new()
	add_child(player)
	var skel_path := String(get_path_to(skel))
	# The Meshy body skeleton is authored in cm (armature node carries a 0.01
	# scale); the baked clips are in meters. Rotations are unit-free but
	# position keys must be rescaled into the body's bone space.
	var pos_scale := 1.0
	var cskel: Skeleton3D = null
	var cstack: Array = [clip_scene]
	while not cstack.is_empty():
		var cn: Node = cstack.pop_back()
		if cn is Skeleton3D:
			cskel = cn
			break
		for cc in cn.get_children():
			cstack.push_back(cc)
	if cskel != null:
		var body_y := skel.get_bone_global_rest(skel.find_bone("pelvis")).origin.y
		var clip_y := cskel.get_bone_global_rest(cskel.find_bone("pelvis")).origin.y
		if clip_y > 0.001:
			pos_scale = body_y / clip_y
	print("INFO pos_scale=", pos_scale)
	var lib := AnimationLibrary.new()
	for clip_name in src.get_animation_list():
		var anim := src.get_animation(clip_name).duplicate() as Animation
		anim.loop_mode = Animation.LOOP_LINEAR
		for i in anim.get_track_count():
			var path := String(anim.track_get_path(i))
			var bone := path.get_slice(":", 1)
			anim.track_set_path(i, skel_path + ":" + bone)
			if anim.track_get_type(i) == Animation.TYPE_POSITION_3D:
				for k in anim.track_get_key_count(i):
					anim.track_set_key_value(i, k, anim.track_get_key_value(i, k) * pos_scale)
		lib.add_animation(clip_name, anim)
	player.add_animation_library("", lib)
	clip_scene.queue_free()
	clips.assign(Array(lib.get_animation_list()))
	clips.sort_custom(func(a, b): return _clip_rank(a) < _clip_rank(b))
	print("clips: ", ",".join(clips))
	if clips.is_empty():
		push_error("no clips loaded")
		return

	cam = Camera3D.new()
	cam.fov = 50
	add_child(cam)
	cam.current = true

	hud = Label.new()
	hud.position = Vector2(16, 12)
	hud.add_theme_font_size_override("font_size", 22)
	hud.add_theme_color_override("font_color", Color(0.95, 0.95, 0.9))
	add_child(hud)

	_play_clip(0)
	_update_cam()
	# UAL_SHOTS=1: save mid-clip verification frames before going interactive
	if OS.get_environment("UAL_SHOTS") == "1":
		DirAccess.make_dir_recursive_absolute(DIR + "shots/")
		for shot_clip in ["Idle_Loop", "Walk_Loop", "Jog_Fwd_Loop", "Dance_Loop"]:
			var i := clips.find(shot_clip)
			if i < 0:
				continue
			_play_clip(i)
			await get_tree().create_timer(0.6).timeout
			await RenderingServer.frame_post_draw
			var img := get_viewport().get_texture().get_image()
			img.save_png(DIR + "shots/ual_" + shot_clip + ".png")
			print("shot saved: ", DIR + "shots/ual_" + shot_clip + ".png")
		print("SHOTS_DONE")
		_play_clip(0)
	interactive = true

func _clip_rank(n: String) -> int:
	var i := PREFERRED_ORDER.find(n)
	return i if i >= 0 else 99

func _load_glb(path: String) -> Node3D:
	var doc := GLTFDocument.new()
	var state := GLTFState.new()
	if doc.append_from_file(path, state) != OK:
		push_error("GLB load failed: " + path)
		return null
	return doc.generate_scene(state)

func _find_player(root: Node) -> AnimationPlayer:
	var stack: Array = [root]
	while not stack.is_empty():
		var n: Node = stack.pop_back()
		if n is AnimationPlayer:
			return n
		for c in n.get_children():
			stack.push_back(c)
	return null

func _play_clip(idx: int) -> void:
	if clips.is_empty():
		return
	clip_idx = wrapi(idx, 0, clips.size())
	player.play(clips[clip_idx])
	_update_hud()

func _set_lights(on: bool) -> void:
	lights_on = on
	for l in lights:
		l.visible = on
	env.ambient_light_energy = 0.7 if on else 1.6
	_update_hud()

func _update_hud() -> void:
	if hud == null or clips.is_empty():
		return
	var clip: String = clips[clip_idx]
	var anim := player.get_animation(clip)
	hud.text = "UAL on Quat-named Meshy rig   [%d/%d] %s   %.2fs / %.2fs   speed %.2fx   light: %s   %s\n←/→ clip   Space pause   ,/. step   +/- speed   R reset   U light   drag rotate   scroll zoom" \
		% [clip_idx + 1, clips.size(), clip, player.current_animation_position, anim.length,
			player.speed_scale, "SUN" if lights_on else "FLAT",
			"PLAYING" if player.is_playing() else "PAUSED"]

func _process(_dt: float) -> void:
	if interactive and player.is_playing():
		_update_hud()

func _step_frame(dir: int) -> void:
	if player.is_playing():
		player.pause()
	var anim := player.get_animation(clips[clip_idx])
	var t := fposmod(player.current_animation_position + dir / 30.0, anim.length)
	player.seek(t, true)
	_update_hud()

func _unhandled_input(event: InputEvent) -> void:
	if not interactive:
		return
	if event is InputEventKey and event.pressed and not event.echo:
		match event.keycode:
			KEY_RIGHT, KEY_D:
				_play_clip(clip_idx + 1)
			KEY_LEFT, KEY_A:
				_play_clip(clip_idx - 1)
			KEY_SPACE:
				if player.is_playing():
					player.pause()
				else:
					player.play()
				_update_hud()
			KEY_COMMA:
				_step_frame(-1)
			KEY_PERIOD:
				_step_frame(1)
			KEY_EQUAL, KEY_KP_ADD:
				player.speed_scale = clampf(player.speed_scale * 1.25, 0.1, 4.0)
				_update_hud()
			KEY_MINUS, KEY_KP_SUBTRACT:
				player.speed_scale = clampf(player.speed_scale / 1.25, 0.1, 4.0)
				_update_hud()
			KEY_R:
				player.speed_scale = 1.0
				_update_hud()
			KEY_U:
				_set_lights(not lights_on)
	elif event is InputEventMouseButton:
		if event.button_index == MOUSE_BUTTON_LEFT:
			dragging = event.pressed
		elif event.pressed and event.button_index == MOUSE_BUTTON_WHEEL_UP:
			orbit_dist = clampf(orbit_dist * 0.90, 0.35, 10.0)
			_update_cam()
		elif event.pressed and event.button_index == MOUSE_BUTTON_WHEEL_DOWN:
			orbit_dist = clampf(orbit_dist / 0.90, 0.35, 10.0)
			_update_cam()
	elif event is InputEventMouseMotion and dragging:
		orbit_yaw -= event.relative.x * 0.008
		orbit_pitch = clampf(orbit_pitch + event.relative.y * 0.008, -1.3, 1.3)
		_update_cam()

func _update_cam() -> void:
	var dir := Vector3(
		sin(orbit_yaw) * cos(orbit_pitch),
		sin(orbit_pitch),
		cos(orbit_yaw) * cos(orbit_pitch))
	cam.look_at_from_position(orbit_target + dir * orbit_dist, orbit_target)