// SPDX-License-Identifier: CC0-1.0
//
// SPDX-FileContributor: NightFox & Co., 2009-2011
//
// Example of rotating and scaling 3D sprites.
// http://www.nightfoxandco.com

#include <stdio.h>
#include <time.h>

#include <nds.h>
#include <filesystem.h>

#include <nf_lib.h>

#define MAXSPRITES 8

int main(int argc, char **argv)
{
    // Set random seed based on the current time
    srand(time(NULL));

    // Prepare a NitroFS initialization screen
    NF_Set2D(0, 0);
    NF_Set2D(1, 0);
    consoleDemoInit();
    printf("\n NitroFS init. Please wait.\n\n");
    printf(" Iniciando NitroFS,\n por favor, espere.\n\n");
    swiWaitForVBlank();

    // Initialize NitroFS and set it as the root folder of the filesystem
    nitroFSInit(NULL);
    NF_SetRootFolder("NITROFS");

    // Initialize 3D engine in the top screen in mode 0
    NF_Set3D(0, 0);

    // Initialize tiled backgrounds system
    NF_InitTiledBgBuffers();    // Initialize storage buffers
    NF_InitTiledBgSys(0);       // Top screen

    // Initialize 3D sprite system
    NF_InitSpriteBuffers();     // Initialize storage buffers
    NF_Init3dSpriteSys();

    // Load background files from NitroFS
    NF_LoadTiledBg("bg/bg3", "bg3", 256, 256);

    // Load sprite files from NitroFS
    NF_LoadSpriteGfx("sprite/blueball", 0, 64, 64);
    NF_LoadSpritePal("sprite/blueball", 0);
    NF_LoadSpriteGfx("sprite/redcar", 1, 128, 64);
    NF_LoadSpritePal("sprite/redcar", 1);

    // Transfer the required sprites to VRAM
    NF_Vram3dSpriteGfx(0, 0, true);
    NF_Vram3dSpritePal(0, 0);
    NF_Vram3dSpriteGfx(1, 1, true);
    NF_Vram3dSpritePal(1, 1);

    // Create background
    NF_CreateTiledBg(0, 3, "bg3");

    // Variables
    s16 x[MAXSPRITES];
    s16 y[MAXSPRITES];
    s16 ix[MAXSPRITES];
    s16 iy[MAXSPRITES];
    s16 rx[MAXSPRITES];
    s16 ry[MAXSPRITES];
    s16 rz[MAXSPRITES];
    s16 scale[MAXSPRITES];

    // Initialize sprite variables and create the sprites
    for (int n = 0; n < MAXSPRITES; n++)
    {
        int r;

        if ((n % 2) == 0)
            r = 1;
        else
            r = 0;

        x[n] = (rand() % 128) + 1;
        y[n] = (rand() % 112) + 1;
        if ((rand() % 100) > 50)
            ix[n] = 1;
        else
            ix[n] = -1;

        if ((rand() % 100) > 50)
            iy[n] = 1;
        else
            iy[n] = -1;

        rx[n] = 0;
        ry[n] = 0;
        rz[n] = 0;
        scale[n] = 64;

        NF_Create3dSprite(n, r, r, x[n], y[n]);
    }

    // Sort their priorites based on their IDs (lower IDs have higher priority)
    NF_Sort3dSprites();

    // Variable that contains the ID of the sprite selected by the user
    s16 id = 0;

    while (1)
    {
        // Read keys
        scanKeys();
        u16 press = keysDown();
        u16 held = keysHeld();

        // Move sprites
        for (int n = 0; n < MAXSPRITES; n++)
        {
            x[n] += ix[n];
            if ((x[n] < 1) || (x[n] > (255 - NF_3DSPRITE[n].width)))
                ix[n] = -ix[n];

            y[n] += iy[n];
            if ((y[n] < 1) || (y[n] > (191 - NF_3DSPRITE[n].height)))
                iy[n] = -iy[n];

            NF_Move3dSprite(n, x[n], y[n]);
        }

        // Change the selected sprite by pressing A or B
        if (press & KEY_A)
        {
            id++;
            if (id > (MAXSPRITES - 1))
                id = 0;
        }
        if (press & KEY_B)
        {
            id--;
            if (id < 0)
                id = MAXSPRITES - 1;
        }

        // Rotate Z axis of the selcted sprite
        if (held & KEY_RIGHT)
        {
            rz[id] += 2;
            if (rz[id] > 512)
                rz[id] -= 512;
        }
        if (held & KEY_LEFT)
        {
            rz[id] -= 2;
            if (rz[id] < 0)
                rz[id] += 512;
        }

        // Rotate Y axis of the selcted sprite
        if (held & KEY_DOWN)
        {
            ry[id] += 2;
            if (ry[id] > 512)
                ry[id] -= 512;
        }
        if (held & KEY_UP)
        {
            ry[id] -= 2;
            if (ry[id] < 0)
                ry[id] += 512;
        }

        // Rotate X axis of the selcted sprite
        if (held & KEY_X)
        {
            rx[id] += 2;
            if (rx[id] > 512)
                rx[id] -= 512;
        }
        if (held & KEY_Y)
        {
            rx[id] -= 2;
            if (rx[id] < 0)
                rx[id] += 512;
        }

        // Scale sprite
        if (held & KEY_R)
        {
            scale[id] += 2;
            if (scale[id] > 512)
                scale[id] = 512;
        }
        if (held & KEY_L)
        {
            scale[id] -= 2;
            if (scale[id] < 0)
                scale[id] = 0;
        }

        // Apply rotation and scale
        NF_Rotate3dSprite(id, rx[id], ry[id], rz[id]);
        NF_Scale3dSprite(id, scale[id], scale[id]);

        // Draw all 3D sprites
        NF_Draw3dSprites();

        // Tell the GPU to draw the scene and wait until it's done
        glFlush(0);

        // Print debug information
        consoleClear();
        printf("A / B - Sprite select %d\n", id);
        printf("Rotate Z (LEFT/RIGHT) %d\n", rz[id]);
        printf("Rotate Y (UP/DOWN) %d\n", ry[id]);
        printf("Rotate X (X/Y) %d\n", rx[id]);
        printf("Scale (R/L) %d\n", scale[id]);

        // Wait for the screen refresh
        swiWaitForVBlank();
    }

    return 0;
}