This commit merge the full development done in greasepencil-object branch and include mainly the following features. - New grease pencil object. - New drawing engine. - New grease pencil modes Draw/Sculpt/Edit and Weight Paint. - New brushes for grease pencil. - New modifiers for grease pencil. - New shaders FX. - New material system (replace old palettes and colors). - Split of annotations (old grease pencil) and new grease pencil object. - UI adapted to blender 2.8. You can get more info here: https://code.blender.org/2017/12/drawing-2d-animation-in-blender-2-8/ https://code.blender.org/2018/07/grease-pencil-status-update/ This is the result of nearly two years of development and I want thanks firstly the other members of the grease pencil team: Daniel M. Lara, Matias Mendiola and Joshua Leung for their support, ideas and to keep working in the project all the time, without them this project had been impossible. Also, I want thanks other Blender developers for their help, advices and to be there always to help me, and specially to Clément Foucault, Dalai Felinto, Pablo Vázquez and Campbell Barton.
865 lines
20 KiB
C
865 lines
20 KiB
C
/*
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* The Original Code is Copyright (C) 2006-2007 Blender Foundation.
|
|
* All rights reserved.
|
|
*
|
|
* The Original Code is: all of this file.
|
|
*
|
|
* Contributor(s): none yet.
|
|
*
|
|
* ***** END GPL LICENSE BLOCK *****
|
|
*
|
|
*/
|
|
|
|
/** \file blender/blenkernel/intern/icons.c
|
|
* \ingroup bke
|
|
*/
|
|
|
|
|
|
#include <math.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
#include "DNA_brush_types.h"
|
|
#include "DNA_gpencil_types.h"
|
|
#include "DNA_group_types.h"
|
|
#include "DNA_lamp_types.h"
|
|
#include "DNA_material_types.h"
|
|
#include "DNA_object_types.h"
|
|
#include "DNA_scene_types.h"
|
|
#include "DNA_screen_types.h"
|
|
#include "DNA_texture_types.h"
|
|
#include "DNA_world_types.h"
|
|
|
|
#include "BLI_utildefines.h"
|
|
#include "BLI_ghash.h"
|
|
#include "BLI_linklist_lockfree.h"
|
|
#include "BLI_string.h"
|
|
#include "BLI_fileops.h"
|
|
#include "BLI_threads.h"
|
|
|
|
#include "BKE_icons.h"
|
|
#include "BKE_global.h" /* only for G.background test */
|
|
#include "BKE_studiolight.h"
|
|
|
|
#include "BLI_sys_types.h" // for intptr_t support
|
|
|
|
#include "GPU_texture.h"
|
|
|
|
#include "IMB_imbuf.h"
|
|
#include "IMB_imbuf_types.h"
|
|
#include "IMB_thumbs.h"
|
|
|
|
/**
|
|
* Only allow non-managed icons to be removed (by Python for eg).
|
|
* Previews & ID's have their own functions to remove icons.
|
|
*/
|
|
enum {
|
|
ICON_FLAG_MANAGED = (1 << 0),
|
|
};
|
|
|
|
/* GLOBALS */
|
|
|
|
static GHash *gIcons = NULL;
|
|
|
|
static int gNextIconId = 1;
|
|
|
|
static int gFirstIconId = 1;
|
|
|
|
static GHash *gCachedPreviews = NULL;
|
|
|
|
/* Queue of icons for deferred deletion. */
|
|
typedef struct DeferredIconDeleteNode {
|
|
struct DeferredIconDeleteNode *next;
|
|
int icon_id;
|
|
} DeferredIconDeleteNode;
|
|
static LockfreeLinkList g_icon_delete_queue;
|
|
|
|
static void icon_free(void *val)
|
|
{
|
|
Icon *icon = val;
|
|
|
|
if (icon) {
|
|
if (icon->obj_type == ICON_DATA_GEOM) {
|
|
struct Icon_Geom *obj = icon->obj;
|
|
if (obj->mem) {
|
|
/* coords & colors are part of this memory. */
|
|
MEM_freeN((void *)obj->mem);
|
|
}
|
|
else {
|
|
MEM_freeN((void *)obj->coords);
|
|
MEM_freeN((void *)obj->colors);
|
|
}
|
|
MEM_freeN(icon->obj);
|
|
}
|
|
|
|
if (icon->drawinfo_free) {
|
|
icon->drawinfo_free(icon->drawinfo);
|
|
}
|
|
else if (icon->drawinfo) {
|
|
MEM_freeN(icon->drawinfo);
|
|
}
|
|
MEM_freeN(icon);
|
|
}
|
|
}
|
|
|
|
static void icon_free_data(int icon_id, Icon *icon)
|
|
{
|
|
if (icon->obj_type == ICON_DATA_ID) {
|
|
((ID *)(icon->obj))->icon_id = 0;
|
|
}
|
|
else if (icon->obj_type == ICON_DATA_PREVIEW) {
|
|
((PreviewImage *)(icon->obj))->icon_id = 0;
|
|
}
|
|
else if (icon->obj_type == ICON_DATA_GPLAYER) {
|
|
((bGPDlayer *)(icon->obj))->runtime.icon_id = 0;
|
|
}
|
|
else if (icon->obj_type == ICON_DATA_GEOM) {
|
|
((struct Icon_Geom *)(icon->obj))->icon_id = 0;
|
|
}
|
|
else if (icon->obj_type == ICON_DATA_STUDIOLIGHT) {
|
|
StudioLight *sl = icon->obj;
|
|
if (sl != NULL) {
|
|
BKE_studiolight_unset_icon_id(sl, icon_id);
|
|
}
|
|
}
|
|
else {
|
|
BLI_assert(0);
|
|
}
|
|
}
|
|
|
|
/* create an id for a new icon and make sure that ids from deleted icons get reused
|
|
* after the integer number range is used up */
|
|
static int get_next_free_id(void)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
int startId = gFirstIconId;
|
|
|
|
/* if we haven't used up the int number range, we just return the next int */
|
|
if (gNextIconId >= gFirstIconId)
|
|
return gNextIconId++;
|
|
|
|
/* now we try to find the smallest icon id not stored in the gIcons hash */
|
|
while (BLI_ghash_lookup(gIcons, SET_INT_IN_POINTER(startId)) && startId >= gFirstIconId)
|
|
startId++;
|
|
|
|
/* if we found a suitable one that isn't used yet, return it */
|
|
if (startId >= gFirstIconId)
|
|
return startId;
|
|
|
|
/* fail */
|
|
return 0;
|
|
}
|
|
|
|
void BKE_icons_init(int first_dyn_id)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
gNextIconId = first_dyn_id;
|
|
gFirstIconId = first_dyn_id;
|
|
|
|
if (!gIcons) {
|
|
gIcons = BLI_ghash_int_new(__func__);
|
|
BLI_linklist_lockfree_init(&g_icon_delete_queue);
|
|
}
|
|
|
|
if (!gCachedPreviews) {
|
|
gCachedPreviews = BLI_ghash_str_new(__func__);
|
|
}
|
|
}
|
|
|
|
void BKE_icons_free(void)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
if (gIcons) {
|
|
BLI_ghash_free(gIcons, NULL, icon_free);
|
|
gIcons = NULL;
|
|
}
|
|
|
|
if (gCachedPreviews) {
|
|
BLI_ghash_free(gCachedPreviews, MEM_freeN, BKE_previewimg_freefunc);
|
|
gCachedPreviews = NULL;
|
|
}
|
|
|
|
BLI_linklist_lockfree_free(&g_icon_delete_queue, MEM_freeN);
|
|
}
|
|
|
|
void BKE_icons_deferred_free(void)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
for (DeferredIconDeleteNode *node =
|
|
(DeferredIconDeleteNode *)BLI_linklist_lockfree_begin(&g_icon_delete_queue);
|
|
node != NULL;
|
|
node = node->next)
|
|
{
|
|
BLI_ghash_remove(gIcons, SET_INT_IN_POINTER(node->icon_id), NULL, icon_free);
|
|
}
|
|
BLI_linklist_lockfree_clear(&g_icon_delete_queue, MEM_freeN);
|
|
}
|
|
|
|
static PreviewImage *previewimg_create_ex(size_t deferred_data_size)
|
|
{
|
|
PreviewImage *prv_img = NULL;
|
|
int i;
|
|
|
|
prv_img = MEM_mallocN(sizeof(PreviewImage) + deferred_data_size, "img_prv");
|
|
memset(prv_img, 0, sizeof(*prv_img)); /* leave deferred data dirty */
|
|
|
|
if (deferred_data_size) {
|
|
prv_img->tag |= PRV_TAG_DEFFERED;
|
|
}
|
|
|
|
for (i = 0; i < NUM_ICON_SIZES; ++i) {
|
|
prv_img->flag[i] |= PRV_CHANGED;
|
|
prv_img->changed_timestamp[i] = 0;
|
|
}
|
|
return prv_img;
|
|
}
|
|
|
|
PreviewImage *BKE_previewimg_create(void)
|
|
{
|
|
return previewimg_create_ex(0);
|
|
}
|
|
|
|
void BKE_previewimg_freefunc(void *link)
|
|
{
|
|
PreviewImage *prv = (PreviewImage *)link;
|
|
if (prv) {
|
|
int i;
|
|
|
|
for (i = 0; i < NUM_ICON_SIZES; ++i) {
|
|
if (prv->rect[i]) {
|
|
MEM_freeN(prv->rect[i]);
|
|
}
|
|
if (prv->gputexture[i])
|
|
GPU_texture_free(prv->gputexture[i]);
|
|
}
|
|
|
|
MEM_freeN(prv);
|
|
}
|
|
}
|
|
|
|
void BKE_previewimg_free(PreviewImage **prv)
|
|
{
|
|
if (prv && (*prv)) {
|
|
BKE_previewimg_freefunc(*prv);
|
|
*prv = NULL;
|
|
}
|
|
}
|
|
|
|
void BKE_previewimg_clear_single(struct PreviewImage *prv, enum eIconSizes size)
|
|
{
|
|
MEM_SAFE_FREE(prv->rect[size]);
|
|
if (prv->gputexture[size]) {
|
|
GPU_texture_free(prv->gputexture[size]);
|
|
}
|
|
prv->h[size] = prv->w[size] = 0;
|
|
prv->flag[size] |= PRV_CHANGED;
|
|
prv->flag[size] &= ~PRV_USER_EDITED;
|
|
prv->changed_timestamp[size] = 0;
|
|
}
|
|
|
|
void BKE_previewimg_clear(struct PreviewImage *prv)
|
|
{
|
|
int i;
|
|
for (i = 0; i < NUM_ICON_SIZES; ++i) {
|
|
BKE_previewimg_clear_single(prv, i);
|
|
}
|
|
}
|
|
|
|
PreviewImage *BKE_previewimg_copy(const PreviewImage *prv)
|
|
{
|
|
PreviewImage *prv_img = NULL;
|
|
int i;
|
|
|
|
if (prv) {
|
|
prv_img = MEM_dupallocN(prv);
|
|
for (i = 0; i < NUM_ICON_SIZES; ++i) {
|
|
if (prv->rect[i]) {
|
|
prv_img->rect[i] = MEM_dupallocN(prv->rect[i]);
|
|
}
|
|
prv_img->gputexture[i] = NULL;
|
|
}
|
|
}
|
|
return prv_img;
|
|
}
|
|
|
|
/** Duplicate preview image from \a id and clear icon_id, to be used by datablock copy functions. */
|
|
void BKE_previewimg_id_copy(ID *new_id, const ID *old_id)
|
|
{
|
|
PreviewImage **old_prv_p = BKE_previewimg_id_get_p(old_id);
|
|
PreviewImage **new_prv_p = BKE_previewimg_id_get_p(new_id);
|
|
|
|
if (old_prv_p && *old_prv_p) {
|
|
BLI_assert(new_prv_p != NULL && ELEM(*new_prv_p, NULL, *old_prv_p));
|
|
// const int new_icon_id = get_next_free_id();
|
|
|
|
// if (new_icon_id == 0) {
|
|
// return; /* Failure. */
|
|
// }
|
|
*new_prv_p = BKE_previewimg_copy(*old_prv_p);
|
|
new_id->icon_id = (*new_prv_p)->icon_id = 0;
|
|
}
|
|
}
|
|
|
|
PreviewImage **BKE_previewimg_id_get_p(const ID *id)
|
|
{
|
|
switch (GS(id->name)) {
|
|
#define ID_PRV_CASE(id_code, id_struct) case id_code: { return &((id_struct *)id)->preview; } ((void)0)
|
|
ID_PRV_CASE(ID_MA, Material);
|
|
ID_PRV_CASE(ID_TE, Tex);
|
|
ID_PRV_CASE(ID_WO, World);
|
|
ID_PRV_CASE(ID_LA, Lamp);
|
|
ID_PRV_CASE(ID_IM, Image);
|
|
ID_PRV_CASE(ID_BR, Brush);
|
|
ID_PRV_CASE(ID_OB, Object);
|
|
ID_PRV_CASE(ID_GR, Collection);
|
|
ID_PRV_CASE(ID_SCE, Scene);
|
|
ID_PRV_CASE(ID_SCR, bScreen);
|
|
#undef ID_PRV_CASE
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void BKE_previewimg_id_free(ID *id)
|
|
{
|
|
PreviewImage **prv_p = BKE_previewimg_id_get_p(id);
|
|
if (prv_p) {
|
|
BKE_previewimg_free(prv_p);
|
|
}
|
|
}
|
|
|
|
PreviewImage *BKE_previewimg_id_ensure(ID *id)
|
|
{
|
|
PreviewImage **prv_p = BKE_previewimg_id_get_p(id);
|
|
|
|
if (prv_p) {
|
|
if (*prv_p == NULL) {
|
|
*prv_p = BKE_previewimg_create();
|
|
}
|
|
return *prv_p;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
PreviewImage *BKE_previewimg_cached_get(const char *name)
|
|
{
|
|
return BLI_ghash_lookup(gCachedPreviews, name);
|
|
}
|
|
|
|
/**
|
|
* Generate an empty PreviewImage, if not yet existing.
|
|
*/
|
|
PreviewImage *BKE_previewimg_cached_ensure(const char *name)
|
|
{
|
|
PreviewImage *prv = NULL;
|
|
void **key_p, **prv_p;
|
|
|
|
if (!BLI_ghash_ensure_p_ex(gCachedPreviews, name, &key_p, &prv_p)) {
|
|
*key_p = BLI_strdup(name);
|
|
*prv_p = BKE_previewimg_create();
|
|
}
|
|
prv = *prv_p;
|
|
BLI_assert(prv);
|
|
|
|
return prv;
|
|
}
|
|
|
|
/**
|
|
* Generate a PreviewImage from given file path, using thumbnails management, if not yet existing.
|
|
*/
|
|
PreviewImage *BKE_previewimg_cached_thumbnail_read(
|
|
const char *name, const char *path, const int source, bool force_update)
|
|
{
|
|
PreviewImage *prv = NULL;
|
|
void **prv_p;
|
|
|
|
prv_p = BLI_ghash_lookup_p(gCachedPreviews, name);
|
|
|
|
if (prv_p) {
|
|
prv = *prv_p;
|
|
BLI_assert(prv);
|
|
}
|
|
|
|
if (prv && force_update) {
|
|
const char *prv_deferred_data = PRV_DEFERRED_DATA(prv);
|
|
if (((int)prv_deferred_data[0] == source) && STREQ(&prv_deferred_data[1], path)) {
|
|
/* If same path, no need to re-allocate preview, just clear it up. */
|
|
BKE_previewimg_clear(prv);
|
|
}
|
|
else {
|
|
BKE_previewimg_free(&prv);
|
|
}
|
|
}
|
|
|
|
if (!prv) {
|
|
/* We pack needed data for lazy loading (source type, in a single char, and path). */
|
|
const size_t deferred_data_size = strlen(path) + 2;
|
|
char *deferred_data;
|
|
|
|
prv = previewimg_create_ex(deferred_data_size);
|
|
deferred_data = PRV_DEFERRED_DATA(prv);
|
|
deferred_data[0] = source;
|
|
memcpy(&deferred_data[1], path, deferred_data_size - 1);
|
|
|
|
force_update = true;
|
|
}
|
|
|
|
if (force_update) {
|
|
if (prv_p) {
|
|
*prv_p = prv;
|
|
}
|
|
else {
|
|
BLI_ghash_insert(gCachedPreviews, BLI_strdup(name), prv);
|
|
}
|
|
}
|
|
|
|
return prv;
|
|
}
|
|
|
|
void BKE_previewimg_cached_release_pointer(PreviewImage *prv)
|
|
{
|
|
if (prv) {
|
|
if (prv->tag & PRV_TAG_DEFFERED_RENDERING) {
|
|
/* We cannot delete the preview while it is being loaded in another thread... */
|
|
prv->tag |= PRV_TAG_DEFFERED_DELETE;
|
|
return;
|
|
}
|
|
if (prv->icon_id) {
|
|
BKE_icon_delete(prv->icon_id);
|
|
}
|
|
BKE_previewimg_freefunc(prv);
|
|
}
|
|
}
|
|
|
|
void BKE_previewimg_cached_release(const char *name)
|
|
{
|
|
PreviewImage *prv = BLI_ghash_popkey(gCachedPreviews, name, MEM_freeN);
|
|
|
|
BKE_previewimg_cached_release_pointer(prv);
|
|
}
|
|
|
|
/** Handle deferred (lazy) loading/generation of preview image, if needed.
|
|
* For now, only used with file thumbnails. */
|
|
void BKE_previewimg_ensure(PreviewImage *prv, const int size)
|
|
{
|
|
if ((prv->tag & PRV_TAG_DEFFERED) != 0) {
|
|
const bool do_icon = ((size == ICON_SIZE_ICON) && !prv->rect[ICON_SIZE_ICON]);
|
|
const bool do_preview = ((size == ICON_SIZE_PREVIEW) && !prv->rect[ICON_SIZE_PREVIEW]);
|
|
|
|
if (do_icon || do_preview) {
|
|
ImBuf *thumb;
|
|
char *prv_deferred_data = PRV_DEFERRED_DATA(prv);
|
|
int source = prv_deferred_data[0];
|
|
char *path = &prv_deferred_data[1];
|
|
int icon_w, icon_h;
|
|
|
|
thumb = IMB_thumb_manage(path, THB_LARGE, source);
|
|
|
|
if (thumb) {
|
|
/* PreviewImage assumes premultiplied alhpa... */
|
|
IMB_premultiply_alpha(thumb);
|
|
|
|
if (do_preview) {
|
|
prv->w[ICON_SIZE_PREVIEW] = thumb->x;
|
|
prv->h[ICON_SIZE_PREVIEW] = thumb->y;
|
|
prv->rect[ICON_SIZE_PREVIEW] = MEM_dupallocN(thumb->rect);
|
|
prv->flag[ICON_SIZE_PREVIEW] &= ~(PRV_CHANGED | PRV_USER_EDITED);
|
|
}
|
|
if (do_icon) {
|
|
if (thumb->x > thumb->y) {
|
|
icon_w = ICON_RENDER_DEFAULT_HEIGHT;
|
|
icon_h = (thumb->y * icon_w) / thumb->x + 1;
|
|
}
|
|
else if (thumb->x < thumb->y) {
|
|
icon_h = ICON_RENDER_DEFAULT_HEIGHT;
|
|
icon_w = (thumb->x * icon_h) / thumb->y + 1;
|
|
}
|
|
else {
|
|
icon_w = icon_h = ICON_RENDER_DEFAULT_HEIGHT;
|
|
}
|
|
|
|
IMB_scaleImBuf(thumb, icon_w, icon_h);
|
|
prv->w[ICON_SIZE_ICON] = icon_w;
|
|
prv->h[ICON_SIZE_ICON] = icon_h;
|
|
prv->rect[ICON_SIZE_ICON] = MEM_dupallocN(thumb->rect);
|
|
prv->flag[ICON_SIZE_ICON] &= ~(PRV_CHANGED | PRV_USER_EDITED);
|
|
}
|
|
IMB_freeImBuf(thumb);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void BKE_icon_changed(const int icon_id)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
Icon *icon = NULL;
|
|
|
|
if (!icon_id || G.background) return;
|
|
|
|
icon = BLI_ghash_lookup(gIcons, SET_INT_IN_POINTER(icon_id));
|
|
|
|
if (icon) {
|
|
/* We *only* expect ID-tied icons here, not non-ID icon/preview! */
|
|
BLI_assert(icon->id_type != 0);
|
|
BLI_assert(icon->obj_type == ICON_DATA_ID);
|
|
|
|
/* Do not enforce creation of previews for valid ID types using BKE_previewimg_id_ensure() here ,
|
|
* we only want to ensure *existing* preview images are properly tagged as changed/invalid, that's all. */
|
|
PreviewImage **p_prv = BKE_previewimg_id_get_p((ID *)icon->obj);
|
|
|
|
/* If we have previews, they all are now invalid changed. */
|
|
if (p_prv && *p_prv) {
|
|
int i;
|
|
for (i = 0; i < NUM_ICON_SIZES; ++i) {
|
|
(*p_prv)->flag[i] |= PRV_CHANGED;
|
|
(*p_prv)->changed_timestamp[i]++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static Icon *icon_create(int icon_id, int obj_type, void *obj)
|
|
{
|
|
Icon *new_icon = MEM_mallocN(sizeof(Icon), __func__);
|
|
|
|
new_icon->obj_type = obj_type;
|
|
new_icon->obj = obj;
|
|
new_icon->id_type = 0;
|
|
new_icon->flag = 0;
|
|
|
|
/* next two lines make sure image gets created */
|
|
new_icon->drawinfo = NULL;
|
|
new_icon->drawinfo_free = NULL;
|
|
|
|
BLI_ghash_insert(gIcons, SET_INT_IN_POINTER(icon_id), new_icon);
|
|
|
|
return new_icon;
|
|
}
|
|
|
|
static int icon_id_ensure_create_icon(struct ID *id)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
Icon *icon = icon_create(id->icon_id, ICON_DATA_ID, id);
|
|
icon->id_type = GS(id->name);
|
|
icon->flag = ICON_FLAG_MANAGED;
|
|
|
|
return id->icon_id;
|
|
}
|
|
|
|
int BKE_icon_id_ensure(struct ID *id)
|
|
{
|
|
/* Never handle icons in non-main thread! */
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
if (!id || G.background) {
|
|
return 0;
|
|
}
|
|
|
|
if (id->icon_id)
|
|
return id->icon_id;
|
|
|
|
id->icon_id = get_next_free_id();
|
|
|
|
if (!id->icon_id) {
|
|
printf("%s: Internal error - not enough IDs\n", __func__);
|
|
return 0;
|
|
}
|
|
|
|
/* Ensure we synchronize ID icon_id with its previewimage if it has one. */
|
|
PreviewImage **p_prv = BKE_previewimg_id_get_p(id);
|
|
if (p_prv && *p_prv) {
|
|
BLI_assert(ELEM((*p_prv)->icon_id, 0, id->icon_id));
|
|
(*p_prv)->icon_id = id->icon_id;
|
|
}
|
|
|
|
return icon_id_ensure_create_icon(id);
|
|
}
|
|
|
|
|
|
static int icon_gplayer_color_ensure_create_icon(bGPDlayer *gpl)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
/* NOTE: The color previews for GP Layers don't really need
|
|
* to be "rendered" to image per se (as it will just be a plain
|
|
* colored rectangle), we need to define icon data here so that
|
|
* we can store a pointer to the layer data in icon->obj.
|
|
*/
|
|
Icon *icon = icon_create(gpl->runtime.icon_id, ICON_DATA_GPLAYER, gpl);
|
|
icon->flag = ICON_FLAG_MANAGED;
|
|
|
|
return gpl->runtime.icon_id;
|
|
}
|
|
|
|
int BKE_icon_gplayer_color_ensure(bGPDlayer *gpl)
|
|
{
|
|
/* Never handle icons in non-main thread! */
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
if (!gpl || G.background) {
|
|
return 0;
|
|
}
|
|
|
|
if (gpl->runtime.icon_id)
|
|
return gpl->runtime.icon_id;
|
|
|
|
gpl->runtime.icon_id = get_next_free_id();
|
|
|
|
if (!gpl->runtime.icon_id) {
|
|
printf("%s: Internal error - not enough IDs\n", __func__);
|
|
return 0;
|
|
}
|
|
|
|
return icon_gplayer_color_ensure_create_icon(gpl);
|
|
}
|
|
|
|
/**
|
|
* Return icon id of given preview, or create new icon if not found.
|
|
*/
|
|
int BKE_icon_preview_ensure(ID *id, PreviewImage *preview)
|
|
{
|
|
if (!preview || G.background)
|
|
return 0;
|
|
|
|
if (id) {
|
|
BLI_assert(BKE_previewimg_id_ensure(id) == preview);
|
|
}
|
|
|
|
if (preview->icon_id) {
|
|
BLI_assert(!id || !id->icon_id || id->icon_id == preview->icon_id);
|
|
return preview->icon_id;
|
|
}
|
|
|
|
if (id && id->icon_id) {
|
|
preview->icon_id = id->icon_id;
|
|
return preview->icon_id;
|
|
}
|
|
|
|
preview->icon_id = get_next_free_id();
|
|
|
|
if (!preview->icon_id) {
|
|
printf("%s: Internal error - not enough IDs\n", __func__);
|
|
return 0;
|
|
}
|
|
|
|
/* Ensure we synchronize ID icon_id with its previewimage if available, and generate suitable 'ID' icon. */
|
|
if (id) {
|
|
id->icon_id = preview->icon_id;
|
|
return icon_id_ensure_create_icon(id);
|
|
}
|
|
|
|
Icon *icon = icon_create(preview->icon_id, ICON_DATA_PREVIEW, preview);
|
|
icon->flag = ICON_FLAG_MANAGED;
|
|
|
|
return preview->icon_id;
|
|
}
|
|
|
|
Icon *BKE_icon_get(const int icon_id)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
Icon *icon = NULL;
|
|
|
|
icon = BLI_ghash_lookup(gIcons, SET_INT_IN_POINTER(icon_id));
|
|
|
|
if (!icon) {
|
|
printf("%s: Internal error, no icon for icon ID: %d\n", __func__, icon_id);
|
|
return NULL;
|
|
}
|
|
|
|
return icon;
|
|
}
|
|
|
|
void BKE_icon_set(const int icon_id, struct Icon *icon)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
void **val_p;
|
|
|
|
if (BLI_ghash_ensure_p(gIcons, SET_INT_IN_POINTER(icon_id), &val_p)) {
|
|
printf("%s: Internal error, icon already set: %d\n", __func__, icon_id);
|
|
return;
|
|
}
|
|
|
|
*val_p = icon;
|
|
}
|
|
|
|
static void icon_add_to_deferred_delete_queue(int icon_id)
|
|
{
|
|
DeferredIconDeleteNode *node =
|
|
MEM_mallocN(sizeof(DeferredIconDeleteNode), __func__);
|
|
node->icon_id = icon_id;
|
|
BLI_linklist_lockfree_insert(&g_icon_delete_queue,
|
|
(LockfreeLinkNode *)node);
|
|
}
|
|
|
|
void BKE_icon_id_delete(struct ID *id)
|
|
{
|
|
const int icon_id = id->icon_id;
|
|
if (!icon_id) return; /* no icon defined for library object */
|
|
id->icon_id = 0;
|
|
|
|
if (!BLI_thread_is_main()) {
|
|
icon_add_to_deferred_delete_queue(icon_id);
|
|
return;
|
|
}
|
|
|
|
BKE_icons_deferred_free();
|
|
BLI_ghash_remove(gIcons, SET_INT_IN_POINTER(icon_id), NULL, icon_free);
|
|
}
|
|
|
|
/**
|
|
* Remove icon and free data.
|
|
*/
|
|
bool BKE_icon_delete(const int icon_id)
|
|
{
|
|
if (icon_id == 0) {
|
|
/* no icon defined for library object */
|
|
return false;
|
|
}
|
|
|
|
Icon *icon = BLI_ghash_popkey(gIcons, SET_INT_IN_POINTER(icon_id), NULL);
|
|
if (icon) {
|
|
icon_free_data(icon_id, icon);
|
|
icon_free(icon);
|
|
return true;
|
|
}
|
|
else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool BKE_icon_delete_unmanaged(const int icon_id)
|
|
{
|
|
if (icon_id == 0) {
|
|
/* no icon defined for library object */
|
|
return false;
|
|
}
|
|
|
|
Icon *icon = BLI_ghash_popkey(gIcons, SET_INT_IN_POINTER(icon_id), NULL);
|
|
if (icon) {
|
|
if (UNLIKELY(icon->flag & ICON_FLAG_MANAGED)) {
|
|
BLI_ghash_insert(gIcons, SET_INT_IN_POINTER(icon_id), icon);
|
|
return false;
|
|
}
|
|
else {
|
|
icon_free_data(icon_id, icon);
|
|
icon_free(icon);
|
|
return true;
|
|
}
|
|
}
|
|
else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Geometry Icon
|
|
* \{ */
|
|
|
|
int BKE_icon_geom_ensure(struct Icon_Geom *geom)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
|
|
if (geom->icon_id) {
|
|
return geom->icon_id;
|
|
}
|
|
|
|
geom->icon_id = get_next_free_id();
|
|
|
|
icon_create(geom->icon_id, ICON_DATA_GEOM, geom);
|
|
/* Not managed for now, we may want this to be configurable per icon). */
|
|
|
|
return geom->icon_id;
|
|
}
|
|
|
|
struct Icon_Geom *BKE_icon_geom_from_memory(const uchar *data, size_t data_len)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
if (data_len <= 8) {
|
|
goto fail;
|
|
}
|
|
/* Skip the header. */
|
|
data_len -= 8;
|
|
const int div = 3 * 2 * 3;
|
|
const int coords_len = data_len / div;
|
|
if (coords_len * div != data_len) {
|
|
goto fail;
|
|
}
|
|
|
|
const uchar header[4] = {'V', 'C', 'O', 0};
|
|
const uchar *p = data;
|
|
if (memcmp(p, header, ARRAY_SIZE(header)) != 0) {
|
|
goto fail;
|
|
}
|
|
p += 4;
|
|
|
|
struct Icon_Geom *geom = MEM_mallocN(sizeof(*geom), __func__);
|
|
geom->coords_range[0] = (int)*p++;
|
|
geom->coords_range[1] = (int)*p++;
|
|
/* x, y ignored for now */
|
|
p += 2;
|
|
|
|
geom->coords_len = coords_len;
|
|
geom->coords = (const void *)p;
|
|
geom->colors = (const void *)(p + (data_len / 3));
|
|
geom->icon_id = 0;
|
|
geom->mem = data;
|
|
return geom;
|
|
|
|
fail:
|
|
MEM_freeN((void *)data);
|
|
return NULL;
|
|
}
|
|
|
|
struct Icon_Geom *BKE_icon_geom_from_file(const char *filename)
|
|
{
|
|
BLI_assert(BLI_thread_is_main());
|
|
size_t data_len;
|
|
uchar *data = BLI_file_read_binary_as_mem(filename, 0, &data_len);
|
|
if (data == NULL) {
|
|
return NULL;
|
|
}
|
|
return BKE_icon_geom_from_memory(data, data_len);
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/** \name Studio Light Icon
|
|
* \{ */
|
|
int BKE_icon_ensure_studio_light(struct StudioLight *sl, int id_type)
|
|
{
|
|
int icon_id = get_next_free_id();
|
|
Icon *icon = icon_create(icon_id, ICON_DATA_STUDIOLIGHT, sl);
|
|
icon->id_type = id_type;
|
|
return icon_id;
|
|
}
|
|
/** \} */
|