WIP Dither magic tool

For https://sourceforge.net/p/tuxpaint/feature-requests/256/
This commit is contained in:
Bill Kendrick 2024-02-29 01:13:21 -08:00
parent f21e5f136d
commit 53f6064aa0
2 changed files with 376 additions and 1 deletions

367
magic/src/dither.c Normal file
View file

@ -0,0 +1,367 @@
/* dither.c
A Magic tool for Tux Paint that turns (parts of) an image into
a black-and-white dithered representation.
(Using Atkinso dithering: https://en.wikipedia.org/wiki/Atkinson_dithering)
by Bill Kendrick <bill@newbreedsoftware.com>
Last updated: February 29, 2024
*/
/* Inclusion of header files */
/* ---------------------------------------------------------------------- */
#include <stdio.h>
#include <string.h>
#include <libintl.h>
#include "tp_magic_api.h"
#include "SDL_image.h"
#include "SDL_mixer.h"
/* Our global variables: */
/* ---------------------------------------------------------------------- */
/* Sound effects: */
Mix_Chunk *snd_effect;
/* The size the user has selected in Tux Paint: */
Uint8 dither_size = 16;
Uint32 dither_white, dither_color;
Uint8 * dither_touched;
float * dither_vals;
void dither_drag(magic_api * api, int which, SDL_Surface * canvas,
SDL_Surface * snapshot, int old_x, int old_y, int x, int y,
SDL_Rect * update_rect);
void dither_line_callback(void *pointer, int which, SDL_Surface * canvas,
SDL_Surface * snapshot, int x, int y);
Uint32 dither_api_version(void)
{
return (TP_MAGIC_API_VERSION);
}
int dither_init(magic_api * api, Uint8 disabled_features, Uint8 complexity_level)
{
int i;
char filename[1024];
snprintf(filename, sizeof(filename), "%ssounds/magic/%s", api->data_directory,
"reflection.ogg"); // FIXME
snd_effect = Mix_LoadWAV(filename);
return (1);
}
int dither_get_tool_count(magic_api * api)
{
return (1);
}
SDL_Surface *dither_get_icon(magic_api * api, int which)
{
char filename[1024];
snprintf(filename, sizeof(filename), "%simages/magic/%s",
api->data_directory, "reflection.png"); // FIXME
return (IMG_Load(filename));
}
char *dither_get_name(magic_api * api, int which)
{
return strdup(gettext("Dither"));
}
int dither_get_group(magic_api * api, int which)
{
return MAGIC_TYPE_COLOR_FILTERS;
}
int dither_get_order(int which)
{
return 610;
}
char *dither_get_description(magic_api * api, int which, int mode)
{
if (mode == MODE_PAINT) {
return strdup(gettext("Click and drag to replace parts of your image with dithered dots."));
} else if (mode == MODE_FULLSCREEN) {
return strdup(gettext("Click to replace your entire image with dithered dots."));
}
}
int dither_requires_colors(magic_api * api, int which)
{
return 1;
}
int dither_modes(magic_api * api, int which)
{
return MODE_PAINT | MODE_FULLSCREEN;
}
Uint8 dither_accepted_sizes(magic_api * api, int which, int mode)
{
return 4;
}
Uint8 dither_default_size(magic_api * api, int which, int mode)
{
return 2;
}
void dither_shutdown(magic_api * api)
{
if (snd_effect != NULL)
{
Mix_FreeChunk(snd_effect);
}
if (dither_touched == NULL)
{
free(dither_touched);
}
if (dither_vals == NULL)
{
free(dither_vals);
}
}
void
dither_click(magic_api * api, int which, int mode,
SDL_Surface * canvas, SDL_Surface * snapshot, int x, int y,
SDL_Rect * update_rect)
{
int xx, yy;
Uint8 r, g, b;
for (yy = 0; yy < canvas->h; yy++)
{
for (xx = 0; xx < canvas->w; xx++)
{
if (mode == MODE_PAINT)
{
dither_touched[yy * canvas->w + xx] = 0;
}
else
{
dither_touched[yy * canvas->w + xx] = 1;
SDL_GetRGB(api->getpixel(snapshot, xx, yy), snapshot->format, &r, &g, &b);
dither_vals[yy * canvas->w + xx] = (api->sRGB_to_linear(r) + api->sRGB_to_linear(g) + api->sRGB_to_linear(b)) / 3.0;
if (xx == 0)
{
api->update_progress_bar();
}
}
}
}
if (mode == MODE_PAINT)
{
dither_drag(api, which, canvas, snapshot, x, y, x, y, update_rect);
}
else
{
dither_release(api, which, canvas, snapshot, x, y, update_rect);
}
}
void
dither_drag(magic_api * api, int which,
SDL_Surface * canvas, SDL_Surface * snapshot,
int old_x, int old_y, int x, int y,
SDL_Rect * update_rect)
{
SDL_LockSurface(snapshot);
SDL_LockSurface(canvas);
api->line((void *) api, which, canvas, snapshot,
old_x, old_y, x, y, 1,
dither_line_callback);
SDL_UnlockSurface(canvas);
SDL_UnlockSurface(snapshot);
if (old_x > x)
{
int temp = old_x;
old_x = x;
x = temp;
}
if (old_y > y)
{
int temp = old_y;
old_y = y;
y = temp;
}
update_rect->x = old_x - dither_size;
update_rect->y = old_y - dither_size;
update_rect->w = (x + dither_size) - update_rect->x + 1;
update_rect->h = (y + dither_size) - update_rect->y + 1;
api->playsound(snd_effect, (x * 255) / canvas->w, 255);
}
/*
* [ . P 0 1 ]
* [ 2 3 4 . ]
* [ . 5 . . ]
*/
int dither_x_pos[6] = { 1, 2, -1, 0, 1, 0 };
int dither_y_pos[6] = { 0, 0, 1, 1, 1, 2 };
void
dither_release(magic_api * api, int which,
SDL_Surface * canvas, SDL_Surface * snapshot, int x, int y,
SDL_Rect * update_rect)
{
Uint8 r, g, b;
float val, err;
int i, nx, ny;
for (y = 0; y < canvas->h; y++)
{
for (x = 0; x < canvas->w; x++)
{
if (dither_touched[y * canvas->w + x])
{
val = dither_vals[y * canvas->w + x];
if (val >= 0.5) {
api->putpixel(canvas, x, y, dither_white);
err = val - 1.0;
}
else
{
api->putpixel(canvas, x, y, dither_color);
err = val;
}
/* Diffuse */
for (i = 0; i < 6; i++)
{
nx = x + dither_x_pos[i];
ny = y + dither_y_pos[i];
if (nx >= 0 && nx < canvas->w && ny >= 0 && ny < canvas->h)
{
if (dither_touched[ny * canvas->w + nx])
{
dither_vals[ny * canvas->w + nx] += (err / 8.0);
}
}
}
}
}
}
update_rect->x = 0;
update_rect->y = 0;
update_rect->w = canvas->w;
update_rect->h = canvas->h;
}
void dither_set_color(magic_api * api, int which, SDL_Surface * canvas, SDL_Surface * snapshot, Uint8 r, Uint8 g, Uint8 b, SDL_Rect * update_rect)
{
if (r <= 240 || g <= 240 || b <= 240)
{
dither_color = SDL_MapRGB(canvas->format, r, g, b);
}
else
{
/* If the chosen color is very bright or white, fall back to black */
dither_color = SDL_MapRGB(canvas->format, 0, 0, 0);
}
}
void dither_set_size(magic_api * api, int which, int mode, SDL_Surface * canvas, SDL_Surface * snapshot, Uint8 size, SDL_Rect * update_rect)
{
dither_size = size * 8;
}
void dither_line_callback(void *pointer, int which, SDL_Surface * canvas,
SDL_Surface * snapshot, int x, int y)
{
magic_api *api = (magic_api *) pointer;
int xx, yy;
Uint8 r, g, b;
float val;
if (dither_touched == NULL)
return;
/* Just do a simple threshold effect while interacting */
for (yy = -dither_size; yy < dither_size; yy++)
{
if (y + yy >= 0 && y + yy < canvas->h)
{
for (xx = -dither_size; xx < dither_size; xx++)
{
if (x + xx >= 0 && x + xx < canvas->w)
{
if (!dither_touched[(y + yy) * canvas->w + (x + xx)])
{
dither_touched[(y + yy) * canvas->w + (x + xx)] = 1;
SDL_GetRGB(api->getpixel(snapshot, x + xx, y + yy), snapshot->format, &r, &g, &b);
val = (api->sRGB_to_linear(r) + api->sRGB_to_linear(g) + api->sRGB_to_linear(b)) / 3.0;
dither_vals[(y + yy) * canvas->w + (x + xx)] = val;
/*
if (val >= 0.5)
api->putpixel(canvas, x + xx, y + yy, dither_white);
else
api->putpixel(canvas, x + xx, y + yy, dither_color);
*/
api->putpixel(canvas, x + xx, y + yy, SDL_MapRGB(canvas->format, val * 255, val * 255, val * 255));
}
}
}
}
}
}
void dither_switchin(magic_api * api, int which, int mode,
SDL_Surface * canvas)
{
if (dither_touched == NULL)
{
dither_touched = (Uint8 *) malloc(sizeof(Uint8) * canvas->h * canvas->w);
}
if (dither_vals == NULL)
{
dither_vals = (float *) malloc(sizeof(float) * canvas->h * canvas->w);
}
dither_white = SDL_MapRGB(canvas->format, 255, 255, 255);
}
void dither_switchout(magic_api * api, int which, int mode,
SDL_Surface * canvas)
{
}