aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/at/hannibal2/skyhanni/utils/shader/ShaderManager.kt
blob: 3dbec3c6e76a379e39ae5e89df16f7aad9ec55f4 (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
package at.hannibal2.skyhanni.utils.shader

import at.hannibal2.skyhanni.features.chroma.ChromaShader
import at.hannibal2.skyhanni.utils.LorenzUtils
import net.minecraft.client.Minecraft
import net.minecraft.util.ResourceLocation
import org.apache.commons.lang3.StringUtils
import java.io.BufferedReader
import java.io.InputStreamReader

/**
 * Object to handle shaders for SkyHanni
 */
object ShaderManager {

    /**
     * For any future shaders add the object instance in this enum and
     * in the when expression
     */
    enum class Shaders(val shader: Shader) {
        CHROMA(ChromaShader.INSTANCE);

        companion object {
            fun getShaderInstance(shaderName: String): Shader? = when (shaderName) {
                "chroma" -> CHROMA.shader
                else -> {
                    null
                }
            }
        }
    }

    private val shaders: MutableMap<String, Shader> = mutableMapOf()
    private var activeShader: Shader? = null

    fun enableShader(shaderName: String) {
        var shader = shaders[shaderName]

        if (shader == null) {
            shader = Shaders.getShaderInstance(shaderName)
            if (shader == null) return
            shaders[shaderName] = shader
        }

        activeShader = shader
        shader.enable()
        shader.updateUniforms()
    }

    fun attachShader(shaderProgram: Int, shaderID: Int) {
        ShaderHelper.glAttachShader(shaderProgram, shaderID)
    }

    fun disableShader() {
        if (activeShader == null) return

        activeShader?.disable()
        activeShader = null
    }

    fun loadShader(type: ShaderType, fileName: String): Int {
        val resourceLocation = ResourceLocation("skyhanni:shaders/$fileName${type.extension}")

        val source = StringBuilder()

        val inputStream = Minecraft.getMinecraft().resourceManager.getResource(resourceLocation).inputStream
        BufferedReader(InputStreamReader(inputStream)).forEachLine {
            source.append(it).append("\n")
        }

        val shaderID = ShaderHelper.glCreateShader(type.shaderType)
        ShaderHelper.glShaderSource(shaderID, source.toString())
        ShaderHelper.glCompileShader(shaderID)

        if (ShaderHelper.glGetShaderi(shaderID, ShaderHelper.GL_COMPILE_STATUS) == 0) {
            LorenzUtils.consoleLog(
                "Error occurred when compiling shader $fileName${type.extension} : " +
                    StringUtils.trim(ShaderHelper.glGetShaderInfoLog(shaderID, 1024))
            )
        }

        return shaderID
    }
}

enum class ShaderType(val extension: String, val shaderType: Int) {
    VERTEX(".vsh", ShaderHelper.GL_VERTEX_SHADER),
    FRAGMENT(".fsh", ShaderHelper.GL_FRAGMENT_SHADER)
}