mirror of
https://github.com/Relintai/entity_spell_system.git
synced 2025-02-20 17:14:44 +01:00
365 lines
11 KiB
C++
365 lines
11 KiB
C++
/*
|
|
Copyright (c) 2019-2022 Péter Magyar
|
|
|
|
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, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
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.
|
|
*/
|
|
|
|
#include "ess_material_cache.h"
|
|
|
|
#if PROPS_PRESENT
|
|
#include "../../props/props/prop_data.h"
|
|
#include "../../props/props/prop_data_prop.h"
|
|
#include "../../props/props/prop_data_tiled_wall.h"
|
|
#include "../../props/tiled_wall/tiled_wall_data.h"
|
|
|
|
#if MESH_DATA_RESOURCE_PRESENT
|
|
#define PROPS_PRESENT 1
|
|
#include "../../mesh_data_resource/props/prop_data_mesh_data.h"
|
|
#undef PROPS_PRESENT
|
|
#endif
|
|
|
|
#endif
|
|
|
|
#include "../singletons/ess.h"
|
|
#include "../../texture_packer/texture_packer.h"
|
|
|
|
|
|
#define VARIANT_ARRAY_GET(arr) \
|
|
Vector<Variant> r; \
|
|
for (int i = 0; i < arr.size(); i++) { \
|
|
r.push_back(arr[i]); \
|
|
} \
|
|
return r;
|
|
|
|
|
|
|
|
bool ESSMaterialCache::get_initialized() {
|
|
return _initialized;
|
|
}
|
|
void ESSMaterialCache::set_initialized(const bool value) {
|
|
_initialized = value;
|
|
}
|
|
|
|
bool ESSMaterialCache::mutex_locked() {
|
|
return _locked;
|
|
}
|
|
void ESSMaterialCache::mutex_lock() {
|
|
_mutex.lock();
|
|
}
|
|
void ESSMaterialCache::mutex_unlock() {
|
|
_mutex.unlock();
|
|
}
|
|
|
|
int ESSMaterialCache::get_ref_count() {
|
|
return _ref_count;
|
|
}
|
|
void ESSMaterialCache::set_ref_count(const int value) {
|
|
_ref_count = value;
|
|
}
|
|
void ESSMaterialCache::inc_ref_count() {
|
|
_ref_count += 1;
|
|
}
|
|
void ESSMaterialCache::dec_ref_count() {
|
|
_ref_count -= 1;
|
|
}
|
|
|
|
//Materials
|
|
Ref<Material> ESSMaterialCache::material_get(const int index) {
|
|
ERR_FAIL_INDEX_V(index, _materials.size(), Ref<Material>(NULL));
|
|
|
|
return _materials[index];
|
|
}
|
|
|
|
Ref<Material> ESSMaterialCache::material_lod_get(const int index) {
|
|
ERR_FAIL_COND_V(_materials.size() == 0, Ref<Material>(NULL));
|
|
|
|
if (index < 0) {
|
|
return _materials[0];
|
|
}
|
|
|
|
if (index >= _materials.size()) {
|
|
return _materials[_materials.size() - 1];
|
|
}
|
|
|
|
return _materials[index];
|
|
}
|
|
|
|
void ESSMaterialCache::material_add(const Ref<Material> &value) {
|
|
ERR_FAIL_COND(!value.is_valid());
|
|
|
|
_materials.push_back(value);
|
|
}
|
|
|
|
void ESSMaterialCache::material_set(const int index, const Ref<Material> &value) {
|
|
ERR_FAIL_INDEX(index, _materials.size());
|
|
|
|
_materials.set(index, value);
|
|
}
|
|
|
|
void ESSMaterialCache::material_remove(const int index) {
|
|
_materials.remove_at(index);
|
|
}
|
|
|
|
int ESSMaterialCache::material_get_num() const {
|
|
return _materials.size();
|
|
}
|
|
|
|
void ESSMaterialCache::materials_clear() {
|
|
_materials.clear();
|
|
}
|
|
|
|
Vector<Variant> ESSMaterialCache::materials_get() {
|
|
VARIANT_ARRAY_GET(_materials);
|
|
}
|
|
|
|
void ESSMaterialCache::materials_set(const Vector<Variant> &materials) {
|
|
_materials.clear();
|
|
|
|
for (int i = 0; i < materials.size(); i++) {
|
|
Ref<Material> material = Ref<Material>(materials[i]);
|
|
|
|
_materials.push_back(material);
|
|
}
|
|
}
|
|
|
|
void ESSMaterialCache::texture_add(const Ref<Texture> &texture) {
|
|
_textures.push_back(texture);
|
|
}
|
|
void ESSMaterialCache::texture_remove(const Ref<Texture> &texture) {
|
|
for (int i = 0; i < _textures.size(); ++i) {
|
|
if (_textures[i] == texture) {
|
|
_textures.remove_at(i);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
void ESSMaterialCache::texture_remove_index(const int index) {
|
|
ERR_FAIL_INDEX(index, _textures.size());
|
|
|
|
_textures.remove_at(index);
|
|
}
|
|
void ESSMaterialCache::textures_clear() {
|
|
_textures.clear();
|
|
}
|
|
int ESSMaterialCache::texture_count() {
|
|
return _textures.size();
|
|
}
|
|
Ref<Texture> ESSMaterialCache::texture_get(const int index) {
|
|
ERR_FAIL_INDEX_V(index, _textures.size(), Ref<Texture>());
|
|
|
|
return _textures[index];
|
|
}
|
|
Ref<AtlasTexture> ESSMaterialCache::texture_get_atlas(const int index) {
|
|
ERR_FAIL_INDEX_V(index, _textures.size(), Ref<AtlasTexture>());
|
|
|
|
return texture_get_atlas_tex(_textures[index]);
|
|
}
|
|
Ref<AtlasTexture> ESSMaterialCache::texture_get_atlas_tex(const Ref<Texture> &texture) {
|
|
return Ref<AtlasTexture>();
|
|
}
|
|
Rect2 ESSMaterialCache::texture_get_uv_rect(const Ref<Texture> &texture) {
|
|
return Rect2(0, 0, 1, 1);
|
|
}
|
|
|
|
#if PROPS_PRESENT
|
|
void ESSMaterialCache::prop_add_textures(const Ref<PropData> &prop) {
|
|
if (!prop.is_valid()) {
|
|
return;
|
|
}
|
|
|
|
for (int i = 0; i < prop->get_prop_count(); ++i) {
|
|
#if MESH_DATA_RESOURCE_PRESENT
|
|
Ref<PropDataMeshData> pdm = prop->get_prop(i);
|
|
|
|
if (pdm.is_valid()) {
|
|
Ref<Texture> tex = pdm->get_texture();
|
|
|
|
if (!tex.is_valid())
|
|
continue;
|
|
|
|
texture_add(tex);
|
|
|
|
continue;
|
|
}
|
|
#endif
|
|
|
|
Ref<PropDataTiledWall> pdtw = prop->get_prop(i);
|
|
|
|
if (pdtw.is_valid()) {
|
|
Ref<TiledWallData> twd = pdtw->get_data();
|
|
|
|
if (!twd.is_valid())
|
|
continue;
|
|
|
|
twd->setup_cache(Ref<ESSMaterialCache>(this));
|
|
|
|
continue;
|
|
}
|
|
|
|
Ref<PropDataProp> pdp = prop->get_prop(i);
|
|
|
|
if (pdp.is_valid()) {
|
|
prop_add_textures(pdp->get_prop());
|
|
}
|
|
}
|
|
}
|
|
void ESSMaterialCache::prop_remove_textures(const Ref<PropData> &prop) {
|
|
if (!prop.is_valid()) {
|
|
return;
|
|
}
|
|
|
|
for (int i = 0; i < prop->get_prop_count(); ++i) {
|
|
#if MESH_DATA_RESOURCE_PRESENT
|
|
Ref<PropDataMeshData> pdm = prop->get_prop(i);
|
|
|
|
if (pdm.is_valid()) {
|
|
Ref<Texture> tex = pdm->get_texture();
|
|
|
|
if (!tex.is_valid())
|
|
continue;
|
|
|
|
texture_remove(tex);
|
|
}
|
|
#endif
|
|
|
|
Ref<PropDataTiledWall> pdtw = prop->get_prop(i);
|
|
|
|
if (pdtw.is_valid()) {
|
|
Ref<TiledWallData> twd = pdtw->get_data();
|
|
|
|
if (!twd.is_valid())
|
|
continue;
|
|
|
|
for (int j = 0; j < twd->get_texture_count(); ++j) {
|
|
const Ref<Texture> &tex = twd->get_texture(j);
|
|
|
|
if (tex.is_valid()) {
|
|
texture_remove(tex);
|
|
}
|
|
}
|
|
|
|
for (int j = 0; j < twd->get_flavour_texture_count(); ++j) {
|
|
const Ref<Texture> &tex = twd->get_flavour_texture(j);
|
|
|
|
if (tex.is_valid()) {
|
|
texture_remove(tex);
|
|
}
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
Ref<PropDataProp> pdp = prop->get_prop(i);
|
|
|
|
if (pdp.is_valid()) {
|
|
prop_remove_textures(pdp);
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
void ESSMaterialCache::refresh_rects() {
|
|
_initialized = true;
|
|
}
|
|
|
|
void ESSMaterialCache::initial_setup_default() {
|
|
//Note: call only on the main thread! Shader->duplicate() can crash if done from an another thread!
|
|
|
|
ESS *ess = ESS::get_singleton();
|
|
|
|
ess->ensure_materials_loaded();
|
|
|
|
int matc = ess->material_get_num();
|
|
for (int i = 0; i < matc; ++i) {
|
|
Ref<Material> m = ess->material_get(i);
|
|
|
|
ERR_CONTINUE(!m.is_valid());
|
|
|
|
Ref<Material> md = m->duplicate();
|
|
|
|
_materials.push_back(md);
|
|
}
|
|
}
|
|
|
|
void ESSMaterialCache::setup_material_albedo(Ref<Texture> texture) {
|
|
if (has_method("_setup_material_albedo"))
|
|
call("_setup_material_albedo", texture);
|
|
}
|
|
|
|
ESSMaterialCache::ESSMaterialCache() {
|
|
_ref_count = 0;
|
|
_initialized = false;
|
|
_locked = false;
|
|
}
|
|
|
|
ESSMaterialCache::~ESSMaterialCache() {
|
|
_materials.clear();
|
|
}
|
|
|
|
void ESSMaterialCache::_bind_methods() {
|
|
ClassDB::bind_method(D_METHOD("get_initialized"), &ESSMaterialCache::get_initialized);
|
|
ClassDB::bind_method(D_METHOD("set_initialized", "value"), &ESSMaterialCache::set_initialized);
|
|
ADD_PROPERTY(PropertyInfo(Variant::BOOL, "initialized"), "set_initialized", "get_initialized");
|
|
|
|
ClassDB::bind_method(D_METHOD("mutex_locked"), &ESSMaterialCache::mutex_locked);
|
|
ClassDB::bind_method(D_METHOD("mutex_lock"), &ESSMaterialCache::mutex_lock);
|
|
ClassDB::bind_method(D_METHOD("mutex_unlock"), &ESSMaterialCache::mutex_unlock);
|
|
|
|
ClassDB::bind_method(D_METHOD("get_ref_count"), &ESSMaterialCache::get_ref_count);
|
|
ClassDB::bind_method(D_METHOD("set_ref_count", "value"), &ESSMaterialCache::set_ref_count);
|
|
ADD_PROPERTY(PropertyInfo(Variant::INT, "mat_ref_count"), "set_ref_count", "get_ref_count");
|
|
ClassDB::bind_method(D_METHOD("inc_ref_count"), &ESSMaterialCache::inc_ref_count);
|
|
ClassDB::bind_method(D_METHOD("dec_ref_count"), &ESSMaterialCache::dec_ref_count);
|
|
|
|
//GDVIRTUAL_BIND("_setup_material_albedo", "texture", "Texture");
|
|
|
|
ClassDB::bind_method(D_METHOD("material_get", "index"), &ESSMaterialCache::material_get);
|
|
ClassDB::bind_method(D_METHOD("material_lod_get", "index"), &ESSMaterialCache::material_lod_get);
|
|
ClassDB::bind_method(D_METHOD("material_add", "value"), &ESSMaterialCache::material_add);
|
|
ClassDB::bind_method(D_METHOD("material_set", "index", "value"), &ESSMaterialCache::material_set);
|
|
ClassDB::bind_method(D_METHOD("material_remove", "index"), &ESSMaterialCache::material_remove);
|
|
ClassDB::bind_method(D_METHOD("material_get_num"), &ESSMaterialCache::material_get_num);
|
|
ClassDB::bind_method(D_METHOD("materials_clear"), &ESSMaterialCache::materials_clear);
|
|
|
|
ClassDB::bind_method(D_METHOD("materials_get"), &ESSMaterialCache::materials_get);
|
|
ClassDB::bind_method(D_METHOD("materials_set"), &ESSMaterialCache::materials_set);
|
|
ADD_PROPERTY(PropertyInfo(Variant::ARRAY, "materials", PROPERTY_HINT_NONE, "17/17:Material", PROPERTY_USAGE_DEFAULT, "Material"), "materials_set", "materials_get");
|
|
|
|
ClassDB::bind_method(D_METHOD("texture_add", "texture"), &ESSMaterialCache::texture_add);
|
|
ClassDB::bind_method(D_METHOD("texture_remove", "texture"), &ESSMaterialCache::texture_remove);
|
|
ClassDB::bind_method(D_METHOD("texture_remove_index", "index"), &ESSMaterialCache::texture_remove_index);
|
|
ClassDB::bind_method(D_METHOD("textures_clear"), &ESSMaterialCache::textures_clear);
|
|
ClassDB::bind_method(D_METHOD("texture_count"), &ESSMaterialCache::texture_count);
|
|
ClassDB::bind_method(D_METHOD("texture_get", "index"), &ESSMaterialCache::texture_get);
|
|
ClassDB::bind_method(D_METHOD("texture_get_atlas", "index"), &ESSMaterialCache::texture_get_atlas);
|
|
ClassDB::bind_method(D_METHOD("texture_get_atlas_tex", "index"), &ESSMaterialCache::texture_get_atlas_tex);
|
|
ClassDB::bind_method(D_METHOD("texture_get_uv_rect", "texture"), &ESSMaterialCache::texture_get_uv_rect);
|
|
|
|
#if PROPS_PRESENT
|
|
ClassDB::bind_method(D_METHOD("prop_add_textures", "prop"), &ESSMaterialCache::prop_add_textures);
|
|
ClassDB::bind_method(D_METHOD("prop_remove_textures", "prop"), &ESSMaterialCache::prop_remove_textures);
|
|
#endif
|
|
|
|
ClassDB::bind_method(D_METHOD("refresh_rects"), &ESSMaterialCache::refresh_rects);
|
|
|
|
ClassDB::bind_method(D_METHOD("setup_material_albedo", "texture"), &ESSMaterialCache::setup_material_albedo);
|
|
|
|
ClassDB::bind_method(D_METHOD("initial_setup_default"), &ESSMaterialCache::initial_setup_default);
|
|
}
|