mirror of
https://github.com/Relintai/pandemonium_engine.git
synced 2024-12-22 03:46:50 +01:00
461 lines
13 KiB
C++
461 lines
13 KiB
C++
|
|
|
|
#include "terrain_library_merger.h"
|
|
|
|
#include "scene/resources/packed_scene.h"
|
|
#include "scene/resources/texture.h"
|
|
#include "scene/resources/material/spatial_material.h"
|
|
#include "scene/resources/material/shader_material.h"
|
|
|
|
#include "modules/modules_enabled.gen.h"
|
|
|
|
#ifdef MODULE_PROPS_ENABLED
|
|
#include "../../props/props/prop_data.h"
|
|
|
|
#include "../../props/props/prop_data_prop.h"
|
|
|
|
#ifdef MODULE_MESH_DATA_RESOURCE_ENABLED
|
|
#include "../../mesh_data_resource/props/prop_data_mesh_data.h"
|
|
#endif
|
|
#endif
|
|
|
|
#include "../defines.h"
|
|
|
|
int TerrainLibraryMerger::get_texture_flags() const {
|
|
return _packer->get_texture_flags();
|
|
}
|
|
void TerrainLibraryMerger::set_texture_flags(const int flags) {
|
|
_packer->set_texture_flags(flags);
|
|
_prop_packer->set_texture_flags(flags);
|
|
}
|
|
|
|
int TerrainLibraryMerger::get_max_atlas_size() const {
|
|
return _packer->get_max_atlas_size();
|
|
}
|
|
void TerrainLibraryMerger::set_max_atlas_size(const int size) {
|
|
_packer->set_max_atlas_size(size);
|
|
_prop_packer->set_max_atlas_size(size);
|
|
}
|
|
|
|
bool TerrainLibraryMerger::get_keep_original_atlases() const {
|
|
return _packer->get_keep_original_atlases();
|
|
}
|
|
void TerrainLibraryMerger::set_keep_original_atlases(const bool value) {
|
|
_packer->set_keep_original_atlases(value);
|
|
_prop_packer->set_keep_original_atlases(value);
|
|
}
|
|
|
|
Color TerrainLibraryMerger::get_background_color() const {
|
|
return _packer->get_background_color();
|
|
}
|
|
void TerrainLibraryMerger::set_background_color(const Color &color) {
|
|
_packer->set_background_color(color);
|
|
_prop_packer->set_background_color(color);
|
|
}
|
|
|
|
int TerrainLibraryMerger::get_margin() const {
|
|
return _packer->get_margin();
|
|
}
|
|
void TerrainLibraryMerger::set_margin(const int margin) {
|
|
_packer->set_margin(margin);
|
|
_prop_packer->set_margin(margin);
|
|
}
|
|
|
|
//Surfaces
|
|
Ref<TerrainSurface> TerrainLibraryMerger::terra_surface_get(const int index) {
|
|
ERR_FAIL_INDEX_V(index, _terra_surfaces.size(), Ref<TerrainSurface>(NULL));
|
|
|
|
return _terra_surfaces[index];
|
|
}
|
|
|
|
void TerrainLibraryMerger::terra_surface_add(Ref<TerrainSurface> value) {
|
|
ERR_FAIL_COND(!value.is_valid());
|
|
|
|
value->set_library(Ref<TerrainLibraryMerger>(this));
|
|
value->set_id(_terra_surfaces.size());
|
|
|
|
_terra_surfaces.push_back(value);
|
|
}
|
|
|
|
void TerrainLibraryMerger::terra_surface_set(const int index, Ref<TerrainSurface> value) {
|
|
ERR_FAIL_COND(index < 0);
|
|
|
|
if (_terra_surfaces.size() < index) {
|
|
_terra_surfaces.resize(index + 1);
|
|
}
|
|
|
|
if (_terra_surfaces[index].is_valid()) {
|
|
_terra_surfaces.get(index)->set_library(Ref<TerrainLibraryMerger>(NULL));
|
|
}
|
|
|
|
if (value.is_valid()) {
|
|
value->set_library(Ref<TerrainLibraryMerger>(this));
|
|
|
|
_terra_surfaces.set(index, value);
|
|
}
|
|
}
|
|
|
|
void TerrainLibraryMerger::terra_surface_remove(const int index) {
|
|
_terra_surfaces.remove(index);
|
|
}
|
|
|
|
int TerrainLibraryMerger::terra_surface_get_num() const {
|
|
return _terra_surfaces.size();
|
|
}
|
|
|
|
void TerrainLibraryMerger::terra_surfaces_clear() {
|
|
_packer->clear();
|
|
|
|
for (int i = 0; i < _terra_surfaces.size(); i++) {
|
|
Ref<TerrainSurfaceMerger> surface = _terra_surfaces[i];
|
|
|
|
if (surface.is_valid()) {
|
|
surface->set_library(NULL);
|
|
}
|
|
}
|
|
|
|
_terra_surfaces.clear();
|
|
}
|
|
|
|
Vector<Variant> TerrainLibraryMerger::get_terra_surfaces() {
|
|
VARIANT_ARRAY_GET(_terra_surfaces);
|
|
}
|
|
|
|
void TerrainLibraryMerger::set_terra_surfaces(const Vector<Variant> &surfaces) {
|
|
_terra_surfaces.clear();
|
|
|
|
for (int i = 0; i < surfaces.size(); i++) {
|
|
Ref<TerrainSurfaceMerger> surface = Ref<TerrainSurfaceMerger>(surfaces[i]);
|
|
|
|
if (surface.is_valid()) {
|
|
surface->set_library(Ref<TerrainLibraryMerger>(this));
|
|
}
|
|
|
|
_terra_surfaces.push_back(surface);
|
|
}
|
|
}
|
|
|
|
#ifdef MODULE_PROPS_ENABLED
|
|
Ref<PropData> TerrainLibraryMerger::get_prop(const int index) {
|
|
ERR_FAIL_INDEX_V(index, _props.size(), Ref<PropData>());
|
|
|
|
return _props[index];
|
|
}
|
|
void TerrainLibraryMerger::add_prop(Ref<PropData> value) {
|
|
_props.push_back(value);
|
|
}
|
|
bool TerrainLibraryMerger::has_prop(const Ref<PropData> &value) const {
|
|
return _props.find(value) != -1;
|
|
}
|
|
void TerrainLibraryMerger::set_prop(const int index, const Ref<PropData> &value) {
|
|
ERR_FAIL_INDEX(index, _props.size());
|
|
|
|
_props.write[index] = value;
|
|
}
|
|
void TerrainLibraryMerger::remove_prop(const int index) {
|
|
ERR_FAIL_INDEX(index, _props.size());
|
|
|
|
_props.remove(index);
|
|
}
|
|
int TerrainLibraryMerger::get_num_props() const {
|
|
return _props.size();
|
|
}
|
|
void TerrainLibraryMerger::clear_props() {
|
|
_props.clear();
|
|
}
|
|
|
|
Vector<Variant> TerrainLibraryMerger::get_props() {
|
|
VARIANT_ARRAY_GET(_props);
|
|
}
|
|
|
|
void TerrainLibraryMerger::set_props(const Vector<Variant> &props) {
|
|
VARIANT_ARRAY_SET(props, _props, PropData);
|
|
}
|
|
|
|
Rect2 TerrainLibraryMerger::get_prop_uv_rect(const Ref<Texture> &texture) {
|
|
if (!texture.is_valid()) {
|
|
return Rect2(0, 0, 1, 1);
|
|
}
|
|
|
|
Ref<AtlasTexture> at = _prop_packer->get_texture(texture);
|
|
|
|
if (!at.is_valid()) {
|
|
return Rect2(0, 0, 1, 1);
|
|
}
|
|
|
|
Rect2 region = at->get_region();
|
|
|
|
Ref<Texture> tex = at->get_atlas();
|
|
|
|
if (!tex.is_valid()) {
|
|
return Rect2(0, 0, 1, 1);
|
|
}
|
|
|
|
Ref<Image> image = tex->get_data();
|
|
|
|
if (!image.is_valid()) {
|
|
return Rect2(0, 0, 1, 1);
|
|
}
|
|
|
|
float w = tex->get_width();
|
|
float h = tex->get_height();
|
|
|
|
region.position = Size2(region.position.x / w, region.position.y / h);
|
|
region.size = Size2(region.size.x / w, region.size.y / h);
|
|
|
|
return region;
|
|
}
|
|
|
|
Ref<TexturePacker> TerrainLibraryMerger::get_prop_packer() {
|
|
return _prop_packer;
|
|
}
|
|
#endif
|
|
|
|
void TerrainLibraryMerger::refresh_rects() {
|
|
bool texture_added = false;
|
|
for (int i = 0; i < _terra_surfaces.size(); i++) {
|
|
Ref<TerrainSurfaceMerger> surface = Ref<TerrainSurfaceMerger>(_terra_surfaces[i]);
|
|
|
|
if (surface.is_valid()) {
|
|
for (int j = 0; j < TerrainSurface::TERRAIN_SIDES_COUNT; ++j) {
|
|
Ref<Texture> tex = surface->get_texture(static_cast<TerrainSurface::TerrainSurfaceSides>(j));
|
|
|
|
if (!tex.is_valid())
|
|
continue;
|
|
|
|
if (!_packer->contains_texture(tex)) {
|
|
texture_added = true;
|
|
surface->set_region(static_cast<TerrainSurface::TerrainSurfaceSides>(j), _packer->add_texture(tex));
|
|
} else {
|
|
surface->set_region(static_cast<TerrainSurface::TerrainSurfaceSides>(j), _packer->get_texture(tex));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (texture_added) {
|
|
_packer->merge();
|
|
|
|
ERR_FAIL_COND(_packer->get_texture_count() == 0);
|
|
|
|
Ref<Texture> tex = _packer->get_generated_texture(0);
|
|
|
|
setup_material_albedo(MATERIAL_INDEX_TERRAIN, tex);
|
|
setup_material_albedo(MATERIAL_INDEX_LIQUID, tex);
|
|
}
|
|
|
|
#ifdef MODULE_PROPS_ENABLED
|
|
//todo add this back
|
|
//texture_added = false;
|
|
for (int i = 0; i < _props.size(); i++) {
|
|
Ref<PropData> prop = _props.get(i);
|
|
|
|
if (prop.is_valid()) {
|
|
if (process_prop_textures(prop))
|
|
texture_added = true;
|
|
}
|
|
}
|
|
|
|
//if (texture_added) {
|
|
if (_prop_packer->get_texture_count() > 0) {
|
|
_prop_packer->merge();
|
|
|
|
//ERR_FAIL_COND(_prop_packer->get_texture_count() == 0);
|
|
|
|
Ref<Texture> tex = _prop_packer->get_generated_texture(0);
|
|
|
|
setup_material_albedo(MATERIAL_INDEX_PROP, tex);
|
|
}
|
|
#endif
|
|
|
|
for (int i = 0; i < _terra_surfaces.size(); i++) {
|
|
Ref<TerrainSurfaceMerger> surface = _terra_surfaces[i];
|
|
|
|
if (surface.is_valid()) {
|
|
surface->refresh_rects();
|
|
}
|
|
}
|
|
|
|
set_initialized(true);
|
|
}
|
|
|
|
void TerrainLibraryMerger::_setup_material_albedo(const int material_index, const Ref<Texture> &texture) {
|
|
Ref<SpatialMaterial> mat;
|
|
|
|
int count = 0;
|
|
|
|
switch (material_index) {
|
|
case MATERIAL_INDEX_TERRAIN:
|
|
count = material_get_num();
|
|
break;
|
|
case MATERIAL_INDEX_LIQUID:
|
|
count = liquid_material_get_num();
|
|
break;
|
|
case MATERIAL_INDEX_PROP:
|
|
count = prop_material_get_num();
|
|
break;
|
|
}
|
|
|
|
for (int i = 0; i < count; ++i) {
|
|
switch (material_index) {
|
|
case MATERIAL_INDEX_TERRAIN:
|
|
mat = material_get(i);
|
|
break;
|
|
case MATERIAL_INDEX_LIQUID:
|
|
mat = liquid_material_get(i);
|
|
break;
|
|
case MATERIAL_INDEX_PROP:
|
|
mat = prop_material_get(i);
|
|
break;
|
|
}
|
|
|
|
Ref<SpatialMaterial> spmat;
|
|
|
|
switch (material_index) {
|
|
case MATERIAL_INDEX_TERRAIN:
|
|
spmat = material_get(i);
|
|
break;
|
|
case MATERIAL_INDEX_LIQUID:
|
|
spmat = liquid_material_get(i);
|
|
break;
|
|
case MATERIAL_INDEX_PROP:
|
|
spmat = prop_material_get(i);
|
|
break;
|
|
}
|
|
|
|
if (spmat.is_valid()) {
|
|
spmat->set_texture(SpatialMaterial::TEXTURE_ALBEDO, texture);
|
|
return;
|
|
}
|
|
|
|
Ref<ShaderMaterial> shmat;
|
|
|
|
switch (material_index) {
|
|
case MATERIAL_INDEX_TERRAIN:
|
|
shmat = material_get(i);
|
|
break;
|
|
case MATERIAL_INDEX_LIQUID:
|
|
shmat = liquid_material_get(i);
|
|
break;
|
|
case MATERIAL_INDEX_PROP:
|
|
shmat = prop_material_get(i);
|
|
break;
|
|
}
|
|
|
|
if (shmat.is_valid()) {
|
|
shmat->set_shader_param("texture_albedo", texture);
|
|
}
|
|
}
|
|
}
|
|
|
|
TerrainLibraryMerger::TerrainLibraryMerger() {
|
|
_packer.instance();
|
|
|
|
_packer->set_texture_flags(Texture::FLAG_MIPMAPS | Texture::FLAG_FILTER);
|
|
|
|
_packer->set_max_atlas_size(1024);
|
|
_packer->set_keep_original_atlases(false);
|
|
_packer->set_margin(0);
|
|
|
|
_prop_packer.instance();
|
|
|
|
_prop_packer->set_texture_flags(Texture::FLAG_MIPMAPS | Texture::FLAG_FILTER);
|
|
|
|
_prop_packer->set_max_atlas_size(1024);
|
|
_prop_packer->set_keep_original_atlases(false);
|
|
_prop_packer->set_margin(0);
|
|
}
|
|
|
|
TerrainLibraryMerger::~TerrainLibraryMerger() {
|
|
for (int i = 0; i < _terra_surfaces.size(); ++i) {
|
|
Ref<TerrainSurface> surface = _terra_surfaces[i];
|
|
|
|
if (surface.is_valid()) {
|
|
surface->set_library(Ref<TerrainLibraryMerger>());
|
|
}
|
|
}
|
|
|
|
_terra_surfaces.clear();
|
|
|
|
_packer->clear();
|
|
_packer.unref();
|
|
|
|
_prop_packer->clear();
|
|
_prop_packer.unref();
|
|
}
|
|
|
|
#ifdef MODULE_PROPS_ENABLED
|
|
bool TerrainLibraryMerger::process_prop_textures(Ref<PropData> prop) {
|
|
if (!prop.is_valid()) {
|
|
return false;
|
|
}
|
|
|
|
bool texture_added = false;
|
|
|
|
for (int i = 0; i < prop->get_prop_count(); ++i) {
|
|
#ifdef MODULE_MESH_DATA_RESOURCE_ENABLED
|
|
Ref<PropDataMeshData> pdm = prop->get_prop(i);
|
|
|
|
if (pdm.is_valid()) {
|
|
Ref<Texture> tex = pdm->get_texture();
|
|
|
|
if (!tex.is_valid())
|
|
continue;
|
|
|
|
if (!_prop_packer->contains_texture(tex)) {
|
|
_prop_packer->add_texture(tex);
|
|
texture_added = true;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
Ref<PropDataProp> pdp = prop->get_prop(i);
|
|
|
|
if (pdp.is_valid()) {
|
|
if (process_prop_textures(pdp))
|
|
texture_added = true;
|
|
}
|
|
}
|
|
|
|
return texture_added;
|
|
}
|
|
#endif
|
|
|
|
void TerrainLibraryMerger::_bind_methods() {
|
|
ClassDB::bind_method(D_METHOD("get_texture_flags"), &TerrainLibraryMerger::get_texture_flags);
|
|
ClassDB::bind_method(D_METHOD("set_texture_flags", "flags"), &TerrainLibraryMerger::set_texture_flags);
|
|
ADD_PROPERTY(PropertyInfo(Variant::INT, "texture_flags", PROPERTY_HINT_FLAGS, "Mipmaps,Repeat,Filter,Anisotropic Linear,Convert to Linear,Mirrored Repeat,Video Surface"), "set_texture_flags", "get_texture_flags");
|
|
|
|
ClassDB::bind_method(D_METHOD("get_max_atlas_size"), &TerrainLibraryMerger::get_max_atlas_size);
|
|
ClassDB::bind_method(D_METHOD("set_max_atlas_size", "size"), &TerrainLibraryMerger::set_max_atlas_size);
|
|
ADD_PROPERTY(PropertyInfo(Variant::INT, "max_atlas_size"), "set_max_atlas_size", "get_max_atlas_size");
|
|
|
|
ClassDB::bind_method(D_METHOD("get_keep_original_atlases"), &TerrainLibraryMerger::get_keep_original_atlases);
|
|
ClassDB::bind_method(D_METHOD("set_keep_original_atlases", "value"), &TerrainLibraryMerger::set_keep_original_atlases);
|
|
ADD_PROPERTY(PropertyInfo(Variant::BOOL, "keep_original_atlases"), "set_keep_original_atlases", "get_keep_original_atlases");
|
|
|
|
ClassDB::bind_method(D_METHOD("get_background_color"), &TerrainLibraryMerger::get_background_color);
|
|
ClassDB::bind_method(D_METHOD("set_background_color", "color"), &TerrainLibraryMerger::set_background_color);
|
|
ADD_PROPERTY(PropertyInfo(Variant::COLOR, "background_color"), "set_background_color", "get_background_color");
|
|
|
|
ClassDB::bind_method(D_METHOD("get_margin"), &TerrainLibraryMerger::get_margin);
|
|
ClassDB::bind_method(D_METHOD("set_margin", "size"), &TerrainLibraryMerger::set_margin);
|
|
ADD_PROPERTY(PropertyInfo(Variant::INT, "margin"), "set_margin", "get_margin");
|
|
|
|
ClassDB::bind_method(D_METHOD("get_terra_surfaces"), &TerrainLibraryMerger::get_terra_surfaces);
|
|
ClassDB::bind_method(D_METHOD("set_terra_surfaces"), &TerrainLibraryMerger::set_terra_surfaces);
|
|
ADD_PROPERTY(PropertyInfo(Variant::ARRAY, "terra_surfaces", PROPERTY_HINT_NONE, "23/20:TerrainSurfaceMerger", PROPERTY_USAGE_DEFAULT, "TerrainSurfaceMerger"), "set_terra_surfaces", "get_terra_surfaces");
|
|
|
|
#ifdef MODULE_PROPS_ENABLED
|
|
ClassDB::bind_method(D_METHOD("get_props"), &TerrainLibraryMerger::get_props);
|
|
ClassDB::bind_method(D_METHOD("set_props"), &TerrainLibraryMerger::set_props);
|
|
ADD_PROPERTY(PropertyInfo(Variant::ARRAY, "props", PROPERTY_HINT_NONE, "23/20:PropData", PROPERTY_USAGE_DEFAULT, "PropData"), "set_props", "get_props");
|
|
|
|
ClassDB::bind_method(D_METHOD("get_prop_uv_rect", "texture"), &TerrainLibraryMerger::get_prop_uv_rect);
|
|
|
|
ClassDB::bind_method(D_METHOD("get_prop_packer"), &TerrainLibraryMerger::get_prop_packer);
|
|
#endif
|
|
|
|
ClassDB::bind_method(D_METHOD("_setup_material_albedo", "material_index", "texture"), &TerrainLibraryMerger::_setup_material_albedo);
|
|
}
|