FrameBuffers, PostProcessing, ClickGui animations

bladekt · bladekt · commit 5cbe0b17738e · 2024-06-22T20:02:34.000+03:00
diff --git a/common/src/main/kotlin/com/lambda/graphics/buffer/FrameBuffer.kt b/common/src/main/kotlin/com/lambda/graphics/buffer/FrameBuffer.kt
@@ -0,0 +1,119 @@
+package com.lambda.graphics.buffer
+import com.lambda.Lambda.mc
+import com.lambda.graphics.RenderMain
+import com.lambda.graphics.buffer.vao.VAO
+import com.lambda.graphics.buffer.vao.vertex.VertexAttrib
+import com.lambda.graphics.buffer.vao.vertex.VertexMode
+import com.lambda.graphics.gl.GlStateUtils.withBlendFunc
+import com.lambda.graphics.shader.Shader
+import com.lambda.graphics.texture.TextureUtils.bindTexture
+import com.lambda.graphics.texture.TextureUtils.setupTexture
+import com.lambda.util.math.Vec2d
+import org.lwjgl.opengl.GL30.*
+import java.nio.IntBuffer
+
+class FrameBuffer {
+    private val fbo = glGenFramebuffers()
+
+    private val colorAttachment = glGenTextures()
+    private val depthAttachment = glGenTextures()
+
+    private var width = 0
+    private var height = 0
+
+    fun write(block: () -> Unit): FrameBuffer {
+        val prev = lastFrameBuffer ?: mc.framebuffer.fbo
+
+        glBindFramebuffer(GL_FRAMEBUFFER, fbo)
+        lastFrameBuffer = fbo
+
+        update()
+        withBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA, GL_ONE, GL_ONE_MINUS_SRC_ALPHA, block)
+
+        lastFrameBuffer = prev
+        glBindFramebuffer(GL_FRAMEBUFFER, prev)
+        return this
+    }
+
+    fun read(
+        shader: Shader,
+        pos1: Vec2d = Vec2d.ZERO,
+        pos2: Vec2d = RenderMain.screenSize,
+        shaderBlock: (Shader) -> Unit = {}
+    ): FrameBuffer {
+        bindColorTexture()
+
+        shader.use()
+        shaderBlock(shader)
+
+        vao.use {
+            val uv1 = pos1 / RenderMain.screenSize
+            val uv2 = pos2 / RenderMain.screenSize
+
+            putQuad(
+                vec2(pos1.x, pos1.y).vec2(uv1.x, 1.0 - uv1.y).end(),
+                vec2(pos1.x, pos2.y).vec2(uv1.x, 1.0 - uv2.y).end(),
+                vec2(pos2.x, pos2.y).vec2(uv2.x, 1.0 - uv2.y).end(),
+                vec2(pos2.x, pos1.y).vec2(uv2.x, 1.0 - uv1.y).end()
+            )
+        }
+
+        withBlendFunc(GL_ONE, GL_ONE_MINUS_SRC_ALPHA) {
+            vao.upload()
+            vao.render()
+            vao.clear()
+        }
+
+        return this
+    }
+
+    fun bindColorTexture(slot: Int = 0): FrameBuffer {
+        bindTexture(colorAttachment, slot)
+        return this
+    }
+
+    fun bindDepthTexture(slot: Int = 0): FrameBuffer {
+        bindTexture(depthAttachment, slot)
+        return this
+    }
+
+    private fun update() {
+        val widthIn = mc.window.framebufferWidth
+        val heightIn = mc.window.framebufferHeight
+
+        if (width != widthIn || height != heightIn) {
+            width = widthIn
+            height = heightIn
+
+            setupBufferTexture(colorAttachment)
+            glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, width, height, 0, GL_RGBA, GL_UNSIGNED_BYTE, null as IntBuffer?)
+            glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, colorAttachment, 0)
+
+            setupBufferTexture(depthAttachment)
+            glTexImage2D(GL_TEXTURE_2D, 0, GL_DEPTH_COMPONENT32F, width, height, 0, GL_DEPTH_COMPONENT, GL_FLOAT, null as IntBuffer?)
+            glFramebufferTexture2D(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT, GL_TEXTURE_2D, depthAttachment, 0)
+        }
+
+        clear()
+    }
+
+    private fun setupBufferTexture(id: Int) {
+        bindTexture(id)
+        setupTexture(GL_NEAREST, GL_NEAREST)
+        glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST)
+        glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST)
+        glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE)
+        glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE)
+    }
+
+    private fun clear() {
+        glClearColor(0f, 0f, 0f, 0f)
+        glClearDepth(1.0)
+        glClear(GL_COLOR_BUFFER_BIT or GL_DEPTH_BUFFER_BIT)
+    }
+
+    companion object {
+        private val vao = VAO(VertexMode.TRIANGLES, VertexAttrib.Group.POS_UV)
+        private var lastFrameBuffer: Int? = null
+    }
+}
diff --git a/common/src/main/kotlin/com/lambda/graphics/buffer/vao/vertex/VertexAttrib.kt b/common/src/main/kotlin/com/lambda/graphics/buffer/vao/vertex/VertexAttrib.kt
@@ -13,11 +13,12 @@ enum class VertexAttrib(val componentCount: Int, componentSize: Int, val normali
     val size = componentCount * componentSize
 
     enum class Group(vararg val attributes: VertexAttrib) {
+        POS_UV(Vec2, Vec2),
+
         // GUI
         FONT(Vec2, Vec2, Color), // pos, uv, color
         RECT_FILLED(Vec2, Vec2, Vec2, Float, Float, Color), // pos, uv, size, roundRadius, shade, color
         RECT_OUTLINE(Vec2, Float, Float, Color), // pos, alpha, shade, color
-        BLUR(Vec2, Vec2), // pos, uv
 
         // WORLD
         DYNAMIC_RENDERER(Vec3, Vec3, Color), // prev pos, pos, color
diff --git a/common/src/main/kotlin/com/lambda/graphics/gl/GlStateUtils.kt b/common/src/main/kotlin/com/lambda/graphics/gl/GlStateUtils.kt
@@ -47,6 +47,16 @@ object GlStateUtils {
         glLineWidth(1f)
     }
 
+    fun withBlendFunc(
+        sfactorRGB: Int, dfactorRGB: Int,
+        sfactorAlpha: Int = sfactorRGB, dfactorAlpha: Int = dfactorRGB,
+        block: () -> Unit
+    ) {
+        glBlendFuncSeparate(sfactorRGB, dfactorRGB, sfactorAlpha, dfactorAlpha)
+        block()
+        glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
+    }
+
     @JvmStatic
     fun capSet(id: Int, flag: Boolean) {
         val field = when (id) {
diff --git a/common/src/main/kotlin/com/lambda/graphics/renderer/gui/BlurPostProcessor.kt b/common/src/main/kotlin/com/lambda/graphics/renderer/gui/BlurPostProcessor.kt
diff --git a/common/src/main/kotlin/com/lambda/graphics/renderer/gui/rect/AbstractRectRenderer.kt b/common/src/main/kotlin/com/lambda/graphics/renderer/gui/rect/AbstractRectRenderer.kt
@@ -15,13 +15,13 @@ abstract class AbstractRectRenderer(
 ) {
     protected val vao = VAO(VertexMode.TRIANGLES, attribGroup)
 
-    open fun render() {
+    fun render() {
         shader.use()
         shader["u_Time"] = glfwGetTime() * GuiSettings.colorSpeed * 5.0
         shader["u_Color1"] = GuiSettings.shadeColor1
         shader["u_Color2"] = GuiSettings.shadeColor2
 
-        shader["u_Size"] = RenderMain.screenSize /Vec2d(GuiSettings.colorWidth, GuiSettings.colorHeight)
+        shader["u_Size"] = RenderMain.screenSize / Vec2d(GuiSettings.colorWidth, GuiSettings.colorHeight)
 
         vao.upload()
         vao.render()
diff --git a/common/src/main/kotlin/com/lambda/graphics/renderer/gui/rect/FilledRectRenderer.kt b/common/src/main/kotlin/com/lambda/graphics/renderer/gui/rect/FilledRectRenderer.kt
@@ -57,11 +57,6 @@ class FilledRectRenderer : AbstractRectRenderer(
         )
     }
 
-    override fun render() {
-        shader.use()
-        super.render()
-    }
-
     companion object {
         private const val MIN_SIZE = 0.5
         private const val MIN_ALPHA = 3
diff --git a/common/src/main/kotlin/com/lambda/graphics/renderer/gui/rect/OutlineRectRenderer.kt b/common/src/main/kotlin/com/lambda/graphics/renderer/gui/rect/OutlineRectRenderer.kt
@@ -93,11 +93,6 @@ class OutlineRectRenderer : AbstractRectRenderer(
         if (drawOuter) drawStripWith(genVertices(outerGlow, true))
     }
 
-    override fun render() {
-        shader.use()
-        super.render()
-    }
-
     companion object {
         private val shader = Shader("renderer/rect_outline")
     }
diff --git a/common/src/main/kotlin/com/lambda/gui/api/component/WindowComponent.kt b/common/src/main/kotlin/com/lambda/gui/api/component/WindowComponent.kt
@@ -35,7 +35,7 @@ abstract class WindowComponent<T : ChildComponent>(
     private var dragOffset: Vec2d? = null
     private val padding get() = ClickGui.windowPadding
 
-    final override val rect get() = Rect.basedOn(position, width, renderHeight + titleBarHeight)
+    final override val rect get() = Rect.basedOn(position, width, renderHeightAnimation + titleBarHeight)
     private val contentRect get() = rect.shrink(padding).moveFirst(Vec2d(0.0, titleBarHeight - padding))
 
     private val titleBar get() = Rect.basedOn(rect.leftTop, rect.size.x, titleBarHeight)
@@ -51,7 +51,6 @@ abstract class WindowComponent<T : ChildComponent>(
 
     private val actualHeight get() = height + padding * 2 * isOpen.toInt()
     private var renderHeightAnimation by animation.exp({ 0.0 }, ::actualHeight, 0.6, ::isOpen)
-    private val renderHeight get() = lerp(0.0, renderHeightAnimation, childShowAnimation)
 
     open val contentComponents = ChildLayer.Drawable<T, WindowComponent<T>>(gui, this, contentRenderer, ::contentRect)
 
@@ -61,6 +60,7 @@ abstract class WindowComponent<T : ChildComponent>(
         when (e) {
             is GuiEvent.Show -> {
                 dragOffset = null
+                renderHeightAnimation = if (isOpen) actualHeight else 0.0
             }
 
             is GuiEvent.Render -> {
@@ -119,7 +119,7 @@ abstract class WindowComponent<T : ChildComponent>(
                         Mouse.Button.Right -> {
                             // Don't let user spam
                             val targetHeight = if (isOpen) actualHeight else 0.0
-                            if (abs(targetHeight - renderHeight) > 1) return
+                            if (abs(targetHeight - renderHeightAnimation) > 1) return
 
                             isOpen = !isOpen
 
diff --git a/common/src/main/kotlin/com/lambda/gui/impl/AbstractClickGui.kt b/common/src/main/kotlin/com/lambda/gui/impl/AbstractClickGui.kt
@@ -2,6 +2,8 @@ package com.lambda.gui.impl
 
 import com.lambda.Lambda.mc
 import com.lambda.graphics.animation.Animation.Companion.exp
+import com.lambda.graphics.buffer.FrameBuffer
+import com.lambda.graphics.shader.Shader
 import com.lambda.gui.AbstractGuiConfigurable
 import com.lambda.gui.api.GuiEvent
 import com.lambda.gui.api.LambdaGui
@@ -29,6 +31,9 @@ abstract class AbstractClickGui(name: String, owner: Module? = null) : LambdaGui
         child == hoveredWindow && !closing
     }
 
+    private val frameBuffer = FrameBuffer()
+    private val shader = Shader("post/cgui_animation", "renderer/pos_tex")
+
     abstract val moduleFilter: (Module) -> Boolean
     abstract val configurable: AbstractGuiConfigurable
 
@@ -41,6 +46,16 @@ abstract class AbstractClickGui(name: String, owner: Module? = null) : LambdaGui
         while (actionPool.isNotEmpty()) actionPool.removeLast().invoke()
 
         when (e) {
+            is GuiEvent.Render -> {
+                frameBuffer.write {
+                    windows.onEvent(e)
+                }.read(shader) {
+                    it["u_Progress"] = childShowAnimation
+                }
+
+                return
+            }
+
             is GuiEvent.Show -> {
                 hoveredWindow = null
                 closing = false
diff --git a/common/src/main/resources/assets/lambda/shaders/fragment/post/blur.frag b/common/src/main/resources/assets/lambda/shaders/fragment/post/blur.frag
diff --git a/common/src/main/resources/assets/lambda/shaders/fragment/post/cgui_animation.frag b/common/src/main/resources/assets/lambda/shaders/fragment/post/cgui_animation.frag
@@ -0,0 +1,22 @@
+#version 330 core
+
+uniform sampler2D u_Texture;
+uniform float u_Progress;
+
+in vec2 v_TexCoord;
+out vec4 color;
+
+void main() {
+    vec2 coord = v_TexCoord;
+
+    coord -= 0.5;
+    coord *= 2.0;
+
+    coord *= mix(0.9, 1.0, u_Progress);
+
+    coord *= 0.5;
+    coord += 0.5;
+
+    vec4 tex = texture(u_Texture, coord);
+    color = vec4(tex.rgb * mix(1.1, 1.0, u_Progress), tex.a * u_Progress);
+}
diff --git a/common/src/main/resources/assets/lambda/shaders/vertex/renderer/pos_tex.vert b/common/src/main/resources/assets/lambda/shaders/vertex/renderer/pos_tex.vert
@@ -3,12 +3,11 @@
 layout (location = 0) in vec4 pos;
 layout (location = 1) in vec2 uv;
 
-uniform mat4 u_Projection;
-uniform mat4 u_ModelView;
+uniform mat4 u_ProjModel;
 
 out vec2 v_TexCoord;
 
 void main() {
-    gl_Position = u_Projection * u_ModelView * pos;
+    gl_Position = u_ProjModel * pos;
     v_TexCoord = uv;
 }

Original file line number	Diff line number	Diff line change
`@@ -57,11 +57,6 @@ class FilledRectRenderer : AbstractRectRenderer(`
`57`	`57`	`)`
`58`	`58`	`}`
`59`	`59`
`60`		`- override fun render() {`
`61`		`- shader.use()`
`62`		`- super.render()`
`63`		`- }`
`64`		`-`
`65`	`60`	`companion object {`
`66`	`61`	`private const val MIN_SIZE = 0.5`
`67`	`62`	`private const val MIN_ALPHA = 3`
Original file line number	Diff line number	Diff line change
`@@ -93,11 +93,6 @@ class OutlineRectRenderer : AbstractRectRenderer(`
`93`	`93`	`if (drawOuter) drawStripWith(genVertices(outerGlow, true))`
`94`	`94`	`}`
`95`	`95`
`96`		`- override fun render() {`
`97`		`- shader.use()`
`98`		`- super.render()`
`99`		`- }`
`100`		`-`
`101`	`96`	`companion object {`
`102`	`97`	`private val shader = Shader("renderer/rect_outline")`
`103`	`98`	`}`