aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/moe/nea/firmament/features/diana/AncestralSpadeSolver.kt
blob: 459fa3c1af648085bac26ecf0348bc76a351d6c3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*
 * SPDX-FileCopyrightText: 2024 Linnea Gräf <nea@nea.moe>
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

package moe.nea.firmament.features.diana

import org.joml.Vector3f
import kotlin.time.Duration.Companion.seconds
import net.minecraft.particle.ParticleTypes
import net.minecraft.sound.SoundEvents
import net.minecraft.util.math.Vec3d
import moe.nea.firmament.events.ParticleSpawnEvent
import moe.nea.firmament.events.SoundReceiveEvent
import moe.nea.firmament.events.WorldReadyEvent
import moe.nea.firmament.events.WorldRenderLastEvent
import moe.nea.firmament.util.TimeMark
import moe.nea.firmament.util.render.RenderInWorldContext

object AncestralSpadeSolver {
    var lastDing = TimeMark.farPast()
        private set
    private val pitches = mutableListOf<Float>()
    val particlePositions = mutableListOf<Vec3d>()
    var nextGuess: Vec3d? = null
        private set

    fun onParticleSpawn(event: ParticleSpawnEvent) {
        if (!DianaWaypoints.TConfig.ancestralSpadeSolver) return
        if (event.particleEffect != ParticleTypes.DRIPPING_LAVA) return
        if (event.offset.x != 0.0F || event.offset.y != 0F || event.offset.z != 0F)
            return
        particlePositions.add(event.position)
        if (particlePositions.size > 20) {
            particlePositions.removeFirst()
        }
    }

    fun onPlaySound(event: SoundReceiveEvent) {
        if (!DianaWaypoints.TConfig.ancestralSpadeSolver) return
        if (!SoundEvents.BLOCK_NOTE_BLOCK_HARP.matchesId(event.sound.value().id)) return

        if (lastDing.passedTime() > 1.seconds) {
            particlePositions.clear()
            pitches.clear()
        }
        lastDing = TimeMark.now()

        pitches.add(event.pitch)
        if (pitches.size > 20) {
            pitches.removeFirst()
        }

        if (particlePositions.size < 3) {
            return
        }

        val averagePitchDelta =
            if (pitches.isEmpty()) 0.0
            else pitches
                .zipWithNext { a, b -> b - a }
                .average()

        val soundDistanceEstimate = (Math.E / averagePitchDelta) - particlePositions.first().distanceTo(event.position)

        if (soundDistanceEstimate > 1000) {
            return
        }

        val lastParticleDirection = particlePositions
            .takeLast(3)
            .let { (a, _, b) -> b.subtract(a) }
            .normalize()

        nextGuess = event.position.add(lastParticleDirection.multiply(soundDistanceEstimate))
    }

    fun onWorldRender(event: WorldRenderLastEvent) {
        if (!DianaWaypoints.TConfig.ancestralSpadeSolver) return
        RenderInWorldContext.renderInWorld(event) {
            nextGuess?.let {
                color(1f, 1f, 0f, 0.5f)
                tinyBlock(it, 1f)
                color(1f, 1f, 0f, 1f)
                val cameraForward = Vector3f(0f, 0f, 1f).rotate(event.camera.rotation)
                line(event.camera.pos.add(Vec3d(cameraForward)), it, lineWidth = 3f)
            }
            if (particlePositions.size > 2 && lastDing.passedTime() < 10.seconds) {
                color(0f, 1f, 0f, 0.7f)
                line(*particlePositions.toTypedArray())
            }
        }
    }

    fun onSwapWorld(event: WorldReadyEvent) {
        nextGuess = null
        particlePositions.clear()
        pitches.clear()
        lastDing = TimeMark.farPast()
    }

}