Files
kitty/kitty/window_logo.c
Kovid Goyal 30023d9d9c DRYer
2024-07-16 10:57:00 +05:30

117 lines
3.4 KiB
C

/*
* window_logo.c
* Copyright (C) 2021 Kovid Goyal <kovid at kovidgoyal.net>
*
* Distributed under terms of the GPL3 license.
*/
#include "state.h"
#include "window_logo.h"
typedef struct WindowLogoItem {
WindowLogo wl;
unsigned int refcnt;
char *path;
window_logo_id_t id;
} WindowLogoItem;
#define NAME hash_by_id
#define KEY_TY window_logo_id_t
#define VAL_TY WindowLogoItem*
#include "kitty-verstable.h"
#define id_for_loop(table) vt_create_for_loop(hash_by_id_itr, itr, &(table)->by_id)
#define NAME hash_by_path
#define KEY_TY const char*
#define VAL_TY WindowLogoItem*
#include "kitty-verstable.h"
struct WindowLogoTable {
hash_by_id by_id;
hash_by_path by_path;
};
static void
free_window_logo(WindowLogoItem **itemref) {
WindowLogoItem *item = *itemref;
free(item->path);
free(item->wl.bitmap);
if (item->wl.texture_id) free_texture(&item->wl.texture_id);
free(item); itemref = NULL;
}
static void
send_logo_to_gpu(WindowLogo *s) {
send_image_to_gpu(&s->texture_id, s->bitmap, s->width, s->height, false, true, true, REPEAT_CLAMP);
free(s->bitmap); s->bitmap = NULL;
}
void
set_on_gpu_state(WindowLogo *s, bool on_gpu) {
if (s->load_from_disk_ok) {
if (on_gpu) { if (!s->texture_id) send_logo_to_gpu(s); }
else if (s->texture_id) free_texture(&s->texture_id);
}
}
window_logo_id_t
find_or_create_window_logo(WindowLogoTable *head, const char *path, void *png_data, size_t png_data_size) {
hash_by_path_itr n = vt_get(&head->by_path, path);
if (!vt_is_end(n)) { n.data->val->refcnt++; return n.data->val->id; }
WindowLogoItem *s = calloc(1, sizeof *s);
if (!s) { PyErr_NoMemory(); return 0; }
s->path = strdup(path);
if (!s->path) { free(s); PyErr_NoMemory(); return 0; }
size_t size;
bool ok = false;
if (png_data == NULL || !png_data_size) {
ok = png_path_to_bitmap(path, &s->wl.bitmap, &s->wl.width, &s->wl.height, &size);
} else {
ok = png_from_data(png_data, png_data_size, path, &s->wl.bitmap, &s->wl.width, &s->wl.height, &size);
}
if (ok) s->wl.load_from_disk_ok = true;
s->refcnt++;
static window_logo_id_t idc = 0;
s->id = ++idc;
if (vt_is_end(vt_insert(&head->by_path, s->path, s))) { free_window_logo(&s); PyErr_NoMemory(); return 0; }
if (vt_is_end(vt_insert(&head->by_id, s->id, s))) { vt_erase(&head->by_path, s->path); free_window_logo(&s); PyErr_NoMemory(); return 0; }
return s->id;
}
WindowLogo*
find_window_logo(WindowLogoTable *table, window_logo_id_t id) {
hash_by_id_itr n = vt_get(&table->by_id, id);
if (vt_is_end(n)) return NULL;
return &n.data->val->wl;
}
void
decref_window_logo(WindowLogoTable *table, window_logo_id_t id) {
hash_by_id_itr n = vt_get(&table->by_id, id);
if (!vt_is_end(n)) {
WindowLogoItem *s = n.data->val;
if (s->refcnt < 2) {
vt_erase(&table->by_id, s->id); vt_erase(&table->by_path, s->path);
free_window_logo(&s);
}
else s->refcnt--;
}
}
WindowLogoTable*
alloc_window_logo_table(void) {
WindowLogoTable *ans = calloc(1, sizeof(WindowLogoTable));
if (ans) { vt_init(&ans->by_path); vt_init(&ans->by_id); }
return ans;
}
void
free_window_logo_table(WindowLogoTable **table) {
id_for_loop(*table) free_window_logo(&itr.data->val);
vt_cleanup(&(*table)->by_id); vt_cleanup(&(*table)->by_path);
free(*table); *table = NULL;
}