raylib-cs/Examples/Shaders/JuliaSet.cs

/*******************************************************************************************
*
*   raylib [shaders] example - julia sets
*
*   NOTE: This example requires raylib OpenGL 3.3 or ES2 versions for shaders support,
*         OpenGL 1.1 does not support shaders, recompile raylib to OpenGL 3.3 version.
*
*   NOTE: Shaders used in this example are #version 330 (OpenGL 3.3).
*
*   This example has been created using raylib 2.5 (www.raylib.com)
*   raylib is licensed under an unmodified zlib/libpng license (View raylib.h for details)
*
*   Example contributed by eggmund (@eggmund) and reviewed by Ramon Santamaria (@raysan5)
*
*   Copyright (c) 2019 eggmund (@eggmund) and Ramon Santamaria (@raysan5)
*
********************************************************************************************/

using System.Numerics;
using static Raylib_cs.Raylib;

namespace Examples.Shaders;

public class JuliaSet
{
    const int GlslVersion = 330;

    // A few good julia sets
    static float[][] PointsOfInterest = new float[][] {
            new float[] { -0.348827f, 0.607167f },
            new float[] { -0.786268f, 0.169728f },
            new float[] { -0.8f, 0.156f },
            new float[] { 0.285f, 0.0f },
            new float[] { -0.835f, -0.2321f },
            new float[] { -0.70176f, -0.3842f },
        };

    public static int Main()
    {
        // Initialization
        //--------------------------------------------------------------------------------------
        const int screenWidth = 800;
        const int screenHeight = 450;
        const float zoomSpeed = 1.01f;
        const float offsetSpeedMul = 2.0f;

        const float startingZoom = 0.75f;

        InitWindow(screenWidth, screenHeight, "raylib [shaders] example - julia sets");

        // Load julia set shader
        // NOTE: Defining 0 (NULL) for vertex shader forces usage of internal default vertex shader
        Shader shader = LoadShader(null, $"resources/shaders/glsl{GlslVersion}/julia_set.fs");

        // Create a RenderTexture2D to be used for render to texture
        RenderTexture2D target = LoadRenderTexture(screenWidth, screenHeight);

        // c constant to use in z^2 + c
        float[] c = { PointsOfInterest[0][0], PointsOfInterest[0][1] };

        // Offset and zoom to draw the julia set at. (centered on screen and default size)
        float[] offset = { 0, 0 };
        float zoom = startingZoom;

        // Get variable (uniform) locations on the shader to connect with the program
        // NOTE: If uniform variable could not be found in the shader, function returns -1
        int cLoc = GetShaderLocation(shader, "c");
        int zoomLoc = GetShaderLocation(shader, "zoom");
        int offsetLoc = GetShaderLocation(shader, "offset");

        // Upload the shader uniform values!
        Raylib.SetShaderValue(shader, cLoc, c, ShaderUniformDataType.Vec2);
        Raylib.SetShaderValue(shader, zoomLoc, zoom, ShaderUniformDataType.Float);
        Raylib.SetShaderValue(shader, offsetLoc, offset, ShaderUniformDataType.Vec2);

        // Multiplier of speed to change c value
        int incrementSpeed = 0;
        // Show controls
        bool showControls = true;

        SetTargetFPS(60);
        //--------------------------------------------------------------------------------------

        // Main game loop
        while (!WindowShouldClose())
        {
            // Update
            //----------------------------------------------------------------------------------
            // Press [1 - 6] to reset c to a point of interest
            if (IsKeyPressed(KeyboardKey.One) ||
                IsKeyPressed(KeyboardKey.Two) ||
                IsKeyPressed(KeyboardKey.Three) ||
                IsKeyPressed(KeyboardKey.Four) ||
                IsKeyPressed(KeyboardKey.Five) ||
                IsKeyPressed(KeyboardKey.Six))
            {

                if (IsKeyPressed(KeyboardKey.One))
                {
                    c[0] = PointsOfInterest[0][0];
                    c[1] = PointsOfInterest[0][1];
                }
                else if (IsKeyPressed(KeyboardKey.Two))
                {
                    c[0] = PointsOfInterest[1][0];
                    c[1] = PointsOfInterest[1][1];
                }
                else if (IsKeyPressed(KeyboardKey.Three))
                {
                    c[0] = PointsOfInterest[2][0];
                    c[1] = PointsOfInterest[2][1];
                }
                else if (IsKeyPressed(KeyboardKey.Four))
                {
                    c[0] = PointsOfInterest[3][0];
                    c[1] = PointsOfInterest[3][1];
                }
                else if (IsKeyPressed(KeyboardKey.Five))
                {
                    c[0] = PointsOfInterest[4][0];
                    c[1] = PointsOfInterest[4][1];
                }
                else if (IsKeyPressed(KeyboardKey.Six))
                {
                    c[0] = PointsOfInterest[5][0];
                    c[1] = PointsOfInterest[5][1];
                }
                Raylib.SetShaderValue(shader, cLoc, c, ShaderUniformDataType.Vec2);
            }

            if (IsKeyPressed(KeyboardKey.R))
            {
                zoom = startingZoom;
                offset[0] = 1f;
                offset[1] = 1f;
                Raylib.SetShaderValue(shader, zoomLoc, zoom, ShaderUniformDataType.Float);
                Raylib.SetShaderValue(shader, offsetLoc, offset, ShaderUniformDataType.Vec2);
            }

            // Pause animation (c change)
            if (IsKeyPressed(KeyboardKey.Space))
            {
                incrementSpeed = 0;
            }

            // Toggle whether or not to show controls
            if (IsKeyPressed(KeyboardKey.F1))
            {
                showControls = !showControls;
            }

            if (IsKeyPressed(KeyboardKey.Right))
            {
                incrementSpeed++;
            }
            else if (IsKeyPressed(KeyboardKey.Left))
            {
                incrementSpeed--;
            }

            // If either left or right button is pressed, zoom in/out
            if (IsMouseButtonDown(MouseButton.Left) || IsMouseButtonDown(MouseButton.Right))
            {
                if (IsMouseButtonDown(MouseButton.Left))
                {
                    zoom *= zoomSpeed;
                }

                if (IsMouseButtonDown(MouseButton.Right))
                {
                    zoom *= 1.0f / zoomSpeed;
                }

                Vector2 mousePos = GetMousePosition();
                Vector2 offsetVelocity = Vector2.Zero;

                offsetVelocity.X = (mousePos.X / screenWidth - 0.5f) * offsetSpeedMul / zoom;
                offsetVelocity.Y = (mousePos.Y / screenHeight - 0.5f) * offsetSpeedMul / zoom;

                // Apply move velocity to camera
                offset[0] += GetFrameTime() * offsetVelocity.X;
                offset[1] += GetFrameTime() * offsetVelocity.Y;

                Raylib.SetShaderValue(shader, zoomLoc, zoom, ShaderUniformDataType.Float);
                Raylib.SetShaderValue(shader, offsetLoc, offset, ShaderUniformDataType.Vec2);
            }

            // Increment c value with time
            float amount = GetFrameTime() * incrementSpeed * 0.0005f;
            c[0] += amount;
            c[1] += amount;

            Raylib.SetShaderValue(shader, cLoc, c, ShaderUniformDataType.Vec2);

            // Using a render texture to draw Julia set
            // Enable drawing to texture
            BeginTextureMode(target);
            ClearBackground(Color.Black);

            // Draw a rectangle in shader mode to be used as shader canvas
            // NOTE: Rectangle uses font Color.white character texture coordinates,
            // so shader can not be applied here directly because input vertexTexCoord
            // do not represent full screen coordinates (space where want to apply shader)
            DrawRectangle(0, 0, GetScreenWidth(), GetScreenHeight(), Color.Black);
            EndTextureMode();

            // Draw
            //----------------------------------------------------------------------------------
            BeginDrawing();
            ClearBackground(Color.Black);

            // Draw the saved texture and rendered julia set with shader
            // NOTE: We do not invert texture on Y, already considered inside shader
            BeginShaderMode(shader);
            DrawTexture(target.Texture, 0, 0, Color.White);
            EndShaderMode();

            if (showControls)
            {
                DrawText("Press Mouse buttons right/left to zoom in/out and move", 10, 15, 10, Color.RayWhite);
                DrawText("Press KEY_F1 to toggle these controls", 10, 30, 10, Color.RayWhite);
                DrawText("Press KEYS [1 - 6] to change point of interest", 10, 45, 10, Color.RayWhite);
                DrawText("Press KEY_LEFT | KEY_RIGHT to change speed", 10, 60, 10, Color.RayWhite);
                DrawText("Press KEY_SPACE to pause movement animation", 10, 75, 10, Color.RayWhite);
                DrawText("Press KEY_R to recenter the camera", 10, 90, 10, Color.RayWhite);
            }

            EndDrawing();
            //----------------------------------------------------------------------------------
        }

        // De-Initialization
        //--------------------------------------------------------------------------------------
        UnloadShader(shader);
        UnloadRenderTexture(target);

        CloseWindow();
        //--------------------------------------------------------------------------------------

        return 0;
    }
}