Moved most of the code from the gltf importer to a new base class, and made it more modular.

This commit is contained in:
Relintai 2020-07-26 12:41:35 +02:00
parent 85417ebee9
commit 515a3b250f
5 changed files with 459 additions and 294 deletions

2
SCsub
View File

@ -14,6 +14,8 @@ module_env.add_source_files(env.modules_sources,"register_types.cpp")
module_env.add_source_files(env.modules_sources,"mesh_data_resource.cpp") module_env.add_source_files(env.modules_sources,"mesh_data_resource.cpp")
module_env.add_source_files(env.modules_sources,"plugin/mdr_import_plugin_base.cpp")
module_env.add_source_files(env.modules_sources,"plugin_collada/editor_import_collada_mdr.cpp") module_env.add_source_files(env.modules_sources,"plugin_collada/editor_import_collada_mdr.cpp")
module_env.add_source_files(env.modules_sources,"plugin_collada/editor_plugin_collada_mdr.cpp") module_env.add_source_files(env.modules_sources,"plugin_collada/editor_plugin_collada_mdr.cpp")

View File

@ -0,0 +1,379 @@
/*
Copyright (c) 2019-2020 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 "mdr_import_plugin_base.h"
#include "scene/resources/box_shape.h"
#include "scene/resources/capsule_shape.h"
#include "scene/resources/concave_polygon_shape.h"
#include "scene/resources/convex_polygon_shape.h"
#include "scene/resources/cylinder_shape.h"
#include "scene/resources/shape.h"
#include "scene/resources/sphere_shape.h"
void MDRImportPluginBase::get_import_options(List<ImportOption> *r_options, int p_preset) const {
r_options->push_back(ImportOption(PropertyInfo(Variant::INT, "collider_type", PROPERTY_HINT_ENUM, MeshDataResource::BINDING_STRING_COLLIDER_TYPE), MeshDataResource::COLLIDER_TYPE_NONE));
r_options->push_back(ImportOption(PropertyInfo(Variant::VECTOR3, "offset"), Vector3(0, 0, 0)));
//Todo this should be changed to 0, 0, 0
r_options->push_back(ImportOption(PropertyInfo(Variant::VECTOR3, "rotation"), Vector3(0, 0, -(3.141592 / 2.0))));
//Todo this should be changed to 1, 1, 1
r_options->push_back(ImportOption(PropertyInfo(Variant::VECTOR3, "scale"), Vector3(0.011, 0.011, 0.011)));
}
bool MDRImportPluginBase::get_option_visibility(const String &p_option, const Map<StringName, Variant> &p_options) const {
return true;
}
int MDRImportPluginBase::get_mesh_count(Node *n) {
int count = 0;
for (int i = 0; i < n->get_child_count(); ++i) {
Node *c = n->get_child(i);
if (Object::cast_to<MeshInstance>(c)) {
++count;
}
count += get_mesh_count(c);
}
return count;
}
Error MDRImportPluginBase::process_node_single(Node *n, const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files, Variant *r_metadata) {
MeshDataResource::ColliderType collider_type = static_cast<MeshDataResource::ColliderType>(static_cast<int>(p_options["collider_type"]));
Vector3 scale = p_options["scale"];
ERR_FAIL_COND_V(n == NULL, Error::ERR_PARSE_ERROR);
for (int i = 0; i < n->get_child_count(); ++i) {
Node *c = n->get_child(i);
if (Object::cast_to<MeshInstance>(c)) {
MeshInstance *mi = Object::cast_to<MeshInstance>(c);
Ref<MeshDataResource> mdr = get_mesh(mi, p_options, collider_type, scale);
ERR_FAIL_COND_V(!mdr.is_valid(), Error::ERR_PARSE_ERROR);
return ResourceSaver::save(p_save_path + "." + get_save_extension(), mdr);
}
}
return Error::ERR_PARSE_ERROR;
}
Error MDRImportPluginBase::process_node_multi(Node *n, const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files, Variant *r_metadata) {
MeshDataResource::ColliderType collider_type = static_cast<MeshDataResource::ColliderType>(static_cast<int>(p_options["collider_type"]));
Vector3 scale = p_options["scale"];
ERR_FAIL_COND_V(n == NULL, Error::ERR_PARSE_ERROR);
for (int i = 0; i < n->get_child_count(); ++i) {
Node *c = n->get_child(i);
if (Object::cast_to<MeshInstance>(c)) {
MeshInstance *mi = Object::cast_to<MeshInstance>(c);
Ref<MeshDataResource> mdr = get_mesh(mi, p_options, collider_type, scale);
String node_name = n->get_name();
node_name = node_name.to_lower();
Error err = ResourceSaver::save(p_save_path + "_" + node_name + "." + get_save_extension(), mdr);
if (err != Error::OK) {
return err;
}
}
process_node_multi(c, p_source_file, p_save_path, p_options, r_platform_variants, r_gen_files, r_metadata);
}
return Error::OK;
}
Ref<MeshDataResource> MDRImportPluginBase::get_mesh(MeshInstance *mi, const Map<StringName, Variant> &p_options, MeshDataResource::ColliderType collider_type, Vector3 scale) {
Ref<ArrayMesh> mesh = mi->get_mesh();
if (mesh.is_valid()) {
Ref<MeshDataResource> mdr;
mdr.instance();
Array arrays = mesh->surface_get_arrays(0);
mdr->set_array(apply_transforms(arrays, p_options));
if (collider_type == MeshDataResource::COLLIDER_TYPE_TRIMESH_COLLISION_SHAPE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<Shape> shape = m->create_trimesh_shape();
if (!shape.is_null()) {
mdr->add_collision_shape(Transform(), scale_shape(shape, scale));
}
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_SINGLE_CONVEX_COLLISION_SHAPE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<Shape> shape = mesh->create_convex_shape();
if (!shape.is_null()) {
mdr->add_collision_shape(Transform(), scale_shape(shape, scale));
}
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_MULTIPLE_CONVEX_COLLISION_SHAPES) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Vector<Ref<Shape> > shapes = mesh->convex_decompose();
for (int j = 0; j < shapes.size(); ++j) {
scale_shape(shapes[j], scale);
}
for (int j = 0; j < shapes.size(); ++j) {
mdr->add_collision_shape(Transform(), shapes[j]);
}
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_BOX) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<BoxShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_extents(size * 0.5);
Vector3 pos = aabb.position;
pos += size / 2.0;
Transform t;
t.origin = pos;
mdr->add_collision_shape(t, shape);
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_CAPSULE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<CapsuleShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_height(size.y * 0.5);
shape->set_radius(MIN(size.x, size.z) * 0.5);
Vector3 pos = aabb.position;
pos += size / 2.0;
Transform t = Transform(Basis().rotated(Vector3(1, 0, 0), M_PI_2));
t.origin = pos;
mdr->add_collision_shape(t, shape);
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_CYLINDER) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<CylinderShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_height(size.y * 0.5);
shape->set_radius(MIN(size.x, size.z) * 0.5);
Vector3 pos = aabb.position;
pos += size / 2.0;
Transform t;
t.origin = pos;
mdr->add_collision_shape(t, shape);
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_SPHERE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<SphereShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_radius(MIN(size.x, MIN(size.y, size.z)) * 0.5);
Vector3 mid = aabb.get_size() / 2.0;
Transform t;
t.origin = aabb.position + mid;
mdr->add_collision_shape(t, shape);
}
return mdr;
}
return Ref<MeshDataResource>();
}
Array MDRImportPluginBase::apply_transforms(Array &array, const Map<StringName, Variant> &p_options) {
Vector3 offset = p_options["offset"];
Vector3 rotation = p_options["rotation"];
Vector3 scale = p_options["scale"];
Transform transform = Transform(Basis(rotation).scaled(scale), offset);
PoolVector3Array verts = array.get(Mesh::ARRAY_VERTEX);
for (int i = 0; i < verts.size(); ++i) {
Vector3 vert = verts[i];
vert = transform.xform(vert);
verts.set(i, (vert));
}
PoolVector3Array normals = array.get(Mesh::ARRAY_NORMAL);
for (int i = 0; i < normals.size(); ++i) {
Vector3 normal = normals[i];
normal = transform.basis.xform(normal);
normals.set(i, normal);
}
/*
Array tangents = array.get(Mesh::ARRAY_TANGENT);
if (tangents.size() == verts.size() * 4) {
for (int i = 0; i < verts.size(); ++i) {
Plane p(tangents[i * 4 + 0], tangents[i * 4 + 1], tangents[i * 4 + 2], tangents[i * 4 + 3]);
Vector3 tangent = p.normal;
tangent = transform.basis.xform(tangent);
tangents.set(i, tangent);
}
}
*/
array.set(Mesh::ARRAY_VERTEX, verts);
array.set(Mesh::ARRAY_NORMAL, normals);
//array.set(Mesh::ARRAY_TANGENT, tangents);
return array;
}
Ref<Shape> MDRImportPluginBase::scale_shape(Ref<Shape> shape, const Vector3 &scale) {
if (shape.is_null())
return shape;
if (Object::cast_to<SphereShape>(*shape)) {
Ref<SphereShape> ss = shape;
ss->set_radius(ss->get_radius() * MAX(scale.x, MAX(scale.y, scale.z)));
}
if (Object::cast_to<BoxShape>(*shape)) {
Ref<BoxShape> bs = shape;
bs->set_extents(bs->get_extents() * scale);
}
if (Object::cast_to<CapsuleShape>(*shape)) {
Ref<CapsuleShape> cs = shape;
float sc = MAX(scale.x, MAX(scale.y, scale.z));
cs->set_radius(cs->get_radius() * sc);
cs->set_height(cs->get_height() * sc);
}
if (Object::cast_to<CylinderShape>(*shape)) {
Ref<CylinderShape> cs = shape;
float sc = MAX(scale.x, MAX(scale.y, scale.z));
cs->set_radius(cs->get_radius() * sc);
cs->set_height(cs->get_height() * sc);
}
if (Object::cast_to<ConcavePolygonShape>(*shape)) {
Ref<ConcavePolygonShape> cps = shape;
PoolVector3Array arr = cps->get_faces();
Basis b = Basis().scaled(scale);
for (int i = 0; i < arr.size(); ++i) {
arr.set(i, b.xform(arr[i]));
}
cps->set_faces(arr);
}
if (Object::cast_to<ConvexPolygonShape>(*shape)) {
Ref<ConvexPolygonShape> cps = shape;
PoolVector3Array arr = cps->get_points();
Basis b = Basis().scaled(scale);
for (int i = 0; i < arr.size(); ++i) {
arr.set(i, b.xform(arr[i]));
}
cps->set_points(arr);
}
return shape;
}
MDRImportPluginBase::MDRImportPluginBase() {
}
MDRImportPluginBase::~MDRImportPluginBase() {
}

View File

@ -0,0 +1,70 @@
/*
Copyright (c) 2019-2020 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.
*/
#ifndef MDR_IMPORT_PLUGIN_BASE
#define MDR_IMPORT_PLUGIN_BASE
#include "core/array.h"
#include "core/io/resource_saver.h"
#include "core/math/basis.h"
#include "core/math/transform.h"
#include "core/ustring.h"
#include "editor/import/editor_import_plugin.h"
#include "scene/main/node.h"
#include "scene/resources/mesh.h"
#include "../mesh_data_resource.h"
#include "editor/import/editor_scene_importer_gltf.h"
#include "core/math/transform.h"
#include "core/version.h"
#if VERSION_MAJOR < 4
#include "scene/3d/mesh_instance.h"
#else
#include "scene/3d/mesh_instance_3d.h"
#define MeshInstance MeshInstance3D
#endif
class MDRImportPluginBase : public EditorImportPlugin {
GDCLASS(MDRImportPluginBase, EditorImportPlugin);
public:
virtual void get_import_options(List<ImportOption> *r_options, int p_preset = 0) const;
virtual bool get_option_visibility(const String &p_option, const Map<StringName, Variant> &p_options) const;
int get_mesh_count(Node *n);
Error process_node_single(Node *n, const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files = NULL, Variant *r_metadata = NULL);
Error process_node_multi(Node *n, const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files = NULL, Variant *r_metadata = NULL);
Ref<MeshDataResource> get_mesh(MeshInstance *mi, const Map<StringName, Variant> &p_options, MeshDataResource::ColliderType collider_type, Vector3 scale);
Array apply_transforms(Array &array, const Map<StringName, Variant> &p_options);
Ref<Shape> scale_shape(Ref<Shape> shape, const Vector3 &scale);
MDRImportPluginBase();
~MDRImportPluginBase();
};
#endif

View File

@ -63,298 +63,17 @@ String EditorImportGLTFMdr::get_preset_name(int p_idx) const {
return ""; return "";
} }
void EditorImportGLTFMdr::get_import_options(List<ImportOption> *r_options, int p_preset) const {
r_options->push_back(ImportOption(PropertyInfo(Variant::INT, "collider_type", PROPERTY_HINT_ENUM, MeshDataResource::BINDING_STRING_COLLIDER_TYPE), MeshDataResource::COLLIDER_TYPE_NONE));
r_options->push_back(ImportOption(PropertyInfo(Variant::VECTOR3, "offset"), Vector3(0, 0, 0)));
r_options->push_back(ImportOption(PropertyInfo(Variant::VECTOR3, "rotation"), Vector3(0, 0, -(3.141592 / 2.0))));
r_options->push_back(ImportOption(PropertyInfo(Variant::VECTOR3, "scale"), Vector3(0.011, 0.011, 0.011)));
}
bool EditorImportGLTFMdr::get_option_visibility(const String &p_option, const Map<StringName, Variant> &p_options) const {
return true;
}
Error EditorImportGLTFMdr::import(const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files, Variant *r_metadata) { Error EditorImportGLTFMdr::import(const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files, Variant *r_metadata) {
MeshDataResource::ColliderType collider_type = static_cast<MeshDataResource::ColliderType>(static_cast<int>(p_options["collider_type"])); //MeshDataResource::ColliderType collider_type = static_cast<MeshDataResource::ColliderType>(static_cast<int>(p_options["collider_type"]));
Node *n = _importer->import_scene(p_source_file, 0, 15); Node *n = _importer->import_scene(p_source_file, 0, 15);
Vector3 scale = p_options["scale"]; ERR_FAIL_COND_V(!n, Error::ERR_PARSE_ERROR);
if (n == NULL) { Error err = process_node_single(n, p_source_file, p_save_path, p_options, r_platform_variants, r_gen_files, r_metadata);
n->queue_delete();
return Error::ERR_PARSE_ERROR;
}
for (int i = 0; i < n->get_child_count(); ++i) {
Node *c = n->get_child(i);
if (c == NULL) {
continue;
}
if (Object::cast_to<MeshInstance>(c)) {
MeshInstance *mi = Object::cast_to<MeshInstance>(c);
Ref<ArrayMesh> mesh = mi->get_mesh();
if (mesh.is_valid()) {
Ref<MeshDataResource> mdr;
mdr.instance();
Array arrays = mesh->surface_get_arrays(0);
mdr->set_array(apply_transforms(arrays, p_options));
if (collider_type == MeshDataResource::COLLIDER_TYPE_TRIMESH_COLLISION_SHAPE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<Shape> shape = m->create_trimesh_shape();
if (!shape.is_null()) {
mdr->add_collision_shape(Transform(), scale_shape(shape, scale));
}
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_SINGLE_CONVEX_COLLISION_SHAPE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<Shape> shape = mesh->create_convex_shape();
if (!shape.is_null()) {
mdr->add_collision_shape(Transform(), scale_shape(shape, scale));
}
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_MULTIPLE_CONVEX_COLLISION_SHAPES) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Vector<Ref<Shape> > shapes = mesh->convex_decompose();
for (int j = 0; j < shapes.size(); ++j) {
scale_shape(shapes[j], scale);
}
for (int j = 0; j < shapes.size(); ++j) {
mdr->add_collision_shape(Transform(), shapes[j]);
}
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_BOX) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<BoxShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_extents(size * 0.5);
Vector3 pos = aabb.position;
pos += size / 2.0;
Transform t;
t.origin = pos;
mdr->add_collision_shape(t, shape);
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_CAPSULE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<CapsuleShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_height(size.y * 0.5);
shape->set_radius(MIN(size.x, size.z) * 0.5);
Vector3 pos = aabb.position;
pos += size / 2.0;
Transform t = Transform(Basis().rotated(Vector3(1, 0, 0), M_PI_2));
t.origin = pos;
mdr->add_collision_shape(t, shape);
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_CYLINDER) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<CylinderShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_height(size.y * 0.5);
shape->set_radius(MIN(size.x, size.z) * 0.5);
Vector3 pos = aabb.position;
pos += size / 2.0;
Transform t;
t.origin = pos;
mdr->add_collision_shape(t, shape);
} else if (collider_type == MeshDataResource::COLLIDER_TYPE_APPROXIMATED_SPHERE) {
Ref<ArrayMesh> m;
m.instance();
m->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES, mdr->get_array());
Ref<SphereShape> shape;
shape.instance();
AABB aabb = m->get_aabb();
Vector3 size = aabb.get_size();
shape->set_radius(MIN(size.x, MIN(size.y, size.z)) * 0.5);
Vector3 mid = aabb.get_size() / 2.0;
Transform t;
t.origin = aabb.position + mid;
mdr->add_collision_shape(t, shape);
}
n->queue_delete();
return ResourceSaver::save(p_save_path + "." + get_save_extension(), mdr);
}
}
}
n->queue_delete(); n->queue_delete();
return Error::ERR_PARSE_ERROR; return err;
}
Array EditorImportGLTFMdr::apply_transforms(Array &array, const Map<StringName, Variant> &p_options) {
Vector3 offset = p_options["offset"];
Vector3 rotation = p_options["rotation"];
Vector3 scale = p_options["scale"];
Transform transform = Transform(Basis(rotation).scaled(scale), offset);
PoolVector3Array verts = array.get(Mesh::ARRAY_VERTEX);
for (int i = 0; i < verts.size(); ++i) {
Vector3 vert = verts[i];
vert = transform.xform(vert);
verts.set(i, (vert));
}
PoolVector3Array normals = array.get(Mesh::ARRAY_NORMAL);
for (int i = 0; i < normals.size(); ++i) {
Vector3 normal = normals[i];
normal = transform.basis.xform(normal);
normals.set(i, normal);
}
/*
Array tangents = array.get(Mesh::ARRAY_TANGENT);
if (tangents.size() == verts.size() * 4) {
for (int i = 0; i < verts.size(); ++i) {
Plane p(tangents[i * 4 + 0], tangents[i * 4 + 1], tangents[i * 4 + 2], tangents[i * 4 + 3]);
Vector3 tangent = p.normal;
tangent = transform.basis.xform(tangent);
tangents.set(i, tangent);
}
}
*/
array.set(Mesh::ARRAY_VERTEX, verts);
array.set(Mesh::ARRAY_NORMAL, normals);
//array.set(Mesh::ARRAY_TANGENT, tangents);
return array;
}
Ref<Shape> EditorImportGLTFMdr::scale_shape(Ref<Shape> shape, const Vector3 &scale) {
if (shape.is_null())
return shape;
if (Object::cast_to<SphereShape>(*shape)) {
Ref<SphereShape> ss = shape;
ss->set_radius(ss->get_radius() * MAX(scale.x, MAX(scale.y, scale.z)));
}
if (Object::cast_to<BoxShape>(*shape)) {
Ref<BoxShape> bs = shape;
bs->set_extents(bs->get_extents() * scale);
}
if (Object::cast_to<CapsuleShape>(*shape)) {
Ref<CapsuleShape> cs = shape;
float sc = MAX(scale.x, MAX(scale.y, scale.z));
cs->set_radius(cs->get_radius() * sc);
cs->set_height(cs->get_height() * sc);
}
if (Object::cast_to<CylinderShape>(*shape)) {
Ref<CylinderShape> cs = shape;
float sc = MAX(scale.x, MAX(scale.y, scale.z));
cs->set_radius(cs->get_radius() * sc);
cs->set_height(cs->get_height() * sc);
}
if (Object::cast_to<ConcavePolygonShape>(*shape)) {
Ref<ConcavePolygonShape> cps = shape;
PoolVector3Array arr = cps->get_faces();
Basis b = Basis().scaled(scale);
for (int i = 0; i < arr.size(); ++i) {
arr.set(i, b.xform(arr[i]));
}
cps->set_faces(arr);
}
if (Object::cast_to<ConvexPolygonShape>(*shape)) {
Ref<ConvexPolygonShape> cps = shape;
PoolVector3Array arr = cps->get_points();
Basis b = Basis().scaled(scale);
for (int i = 0; i < arr.size(); ++i) {
arr.set(i, b.xform(arr[i]));
}
cps->set_points(arr);
}
return shape;
} }
EditorImportGLTFMdr::EditorImportGLTFMdr() { EditorImportGLTFMdr::EditorImportGLTFMdr() {

View File

@ -23,12 +23,13 @@ SOFTWARE.
#ifndef EDITOR_IMPORT_GLTF_MDR #ifndef EDITOR_IMPORT_GLTF_MDR
#define EDITOR_IMPORT_GLTF_MDR #define EDITOR_IMPORT_GLTF_MDR
#include "../plugin/mdr_import_plugin_base.h"
#include "core/array.h" #include "core/array.h"
#include "core/io/resource_saver.h" #include "core/io/resource_saver.h"
#include "core/math/basis.h" #include "core/math/basis.h"
#include "core/math/transform.h" #include "core/math/transform.h"
#include "core/ustring.h" #include "core/ustring.h"
#include "editor/import/editor_import_plugin.h"
#include "scene/main/node.h" #include "scene/main/node.h"
#include "scene/resources/mesh.h" #include "scene/resources/mesh.h"
@ -47,9 +48,9 @@ SOFTWARE.
#define MeshInstance MeshInstance3D #define MeshInstance MeshInstance3D
#endif #endif
class EditorImportGLTFMdr : public EditorImportPlugin { class EditorImportGLTFMdr : public MDRImportPluginBase {
GDCLASS(EditorImportGLTFMdr, EditorImportPlugin); GDCLASS(EditorImportGLTFMdr, MDRImportPluginBase);
public: public:
virtual String get_importer_name() const; virtual String get_importer_name() const;
@ -62,14 +63,8 @@ public:
virtual int get_preset_count() const; virtual int get_preset_count() const;
virtual String get_preset_name(int p_idx) const; virtual String get_preset_name(int p_idx) const;
virtual void get_import_options(List<ImportOption> *r_options, int p_preset = 0) const;
virtual bool get_option_visibility(const String &p_option, const Map<StringName, Variant> &p_options) const;
virtual Error import(const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files = NULL, Variant *r_metadata = NULL); virtual Error import(const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files = NULL, Variant *r_metadata = NULL);
Array apply_transforms(Array &array, const Map<StringName, Variant> &p_options);
Ref<Shape> scale_shape(Ref<Shape> shape, const Vector3 &scale);
EditorImportGLTFMdr(); EditorImportGLTFMdr();
~EditorImportGLTFMdr(); ~EditorImportGLTFMdr();