plasma.c

/*

MIT No Attribution

Copyright (c) 2020-2023 Mika Tuupola

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.

-cut-

SPDX-License-Identifier: MIT-0

*/

#include <stdint.h>
#include <stdlib.h>
#include <math.h>
#include <hagl.h>

#include "plasma.h"

hagl_color_t *palette;
uint8_t *plasma;

static const uint8_t SPEED = 4;
static const uint8_t PIXEL_SIZE = 2;

void
plasma_init(hagl_backend_t const *display)
{
    uint8_t *ptr = plasma = malloc(display->width * display->height * sizeof(uint8_t));
    palette = malloc(256 * sizeof(hagl_color_t));

    /* Generate nice continous palette. */
    for(uint16_t i = 0; i < 256; i++) {
        const uint8_t r = 128.0f + 128.0f * sin((M_PI * i / 128.0f) + 1);
        const uint8_t g = 128.0f + 128.0f * sin((M_PI * i / 64.0f) + 1);
        const uint8_t b = 64;
        palette[i] = hagl_color(display, r, g, b);
    }

    for (uint16_t y = 0; y < display->height; y += PIXEL_SIZE) {
        for (uint16_t x = 0; x < display->width; x += PIXEL_SIZE) {
            /* Generate three different sinusoids. */
            const float v1 = 128.0f + (128.0f * sin(x / 32.0f));
            const float v2 = 128.0f + (128.0f * sin(y / 24.0f));
            const float v3 = 128.0f + (128.0f * sin(sqrt(x * x + y * y) / 24.0f));
            /* Calculate average of the three sinusoids */
            /* and use it as color index. */
            const uint8_t color = (v1 + v2 + v3) / 3;
            *(ptr++) = color;
        }
    }
}

void
plasma_render(hagl_backend_t const *display)
{
    uint8_t *ptr = plasma;

    for (uint16_t y = 0; y < display->height; y += PIXEL_SIZE) {
        for (uint16_t x = 0; x < display->width; x += PIXEL_SIZE) {
            /* Get a color for pixel from the plasma buffer. */
            const uint8_t index = *(ptr++);
            const hagl_color_t color = palette[index];
            /* Put a pixel to the display. */
            if (1 == PIXEL_SIZE) {
                hagl_put_pixel(display, x, y, color);
            } else {
                hagl_fill_rectangle(display, x, y, x + PIXEL_SIZE - 1, y + PIXEL_SIZE - 1, color);
            }
        }
    }
}

void
plasma_animate(hagl_backend_t const *display)
{
    uint8_t *ptr = plasma;

    for (uint16_t y = 0; y < display->height; y = y + PIXEL_SIZE) {
        for (uint16_t x = 0; x < display->width; x = x + PIXEL_SIZE) {
            /* Get a color from plasma and choose the next color. */
            /* Unsigned integers wrap automatically. */
            const uint8_t index = *ptr + SPEED;
            /* Put the new color back to the plasma buffer. */
            *(ptr++) = index;
        }
    }
}

void
plasma_close()
{
    free(plasma);
    free(palette);
}