Relintai/mesh_data_resource
1
0
Fork 0
mirror of https://github.com/Relintai/mesh_data_resource.git synced 2025-02-20 17:14:31 +01:00
Code Issues Packages Projects Releases Wiki Activity

The collada importer is now also inherited from the new base.

Browse Source
This commit is contained in:
Relintai 2020-07-26 23:28:17 +02:00
parent 5a26317a42
commit 66db595720
2 changed files with 8 additions and 294 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

289
plugin_collada/editor_import_collada_mdr.cpp
View File

@ -62,298 +62,17 @@ String EditorImportColladaMdr::get_preset_name(int p_idx) const {
return ""; return "";
} }
void EditorImportColladaMdr::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 EditorImportColladaMdr::get_option_visibility(const String &p_option, const Map<StringName, Variant> &p_options) const {
return true;
}
Error EditorImportColladaMdr::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 EditorImportColladaMdr::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(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()) {
}
} 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(); n->queue_delete();
return err;
return ResourceSaver::save(p_save_path + "." + get_save_extension(), mdr);
}
}
}
n->queue_delete();
return Error::ERR_PARSE_ERROR;
}
Array EditorImportColladaMdr::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);
ERR_FAIL_COND_V(array.size() != Mesh::ARRAY_MAX, Array());
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> EditorImportColladaMdr::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;
} }
EditorImportColladaMdr::EditorImportColladaMdr() { EditorImportColladaMdr::EditorImportColladaMdr() {

13
plugin_collada/editor_import_collada_mdr.h
View File

@ -23,6 +23,8 @@ SOFTWARE.
#ifndef EDITOR_IMPORT_COLLADA_MDR #ifndef EDITOR_IMPORT_COLLADA_MDR
#define EDITOR_IMPORT_COLLADA_MDR #define EDITOR_IMPORT_COLLADA_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"
@ -45,9 +47,8 @@ SOFTWARE.
#define MeshInstance MeshInstance3D #define MeshInstance MeshInstance3D
#endif #endif
class EditorImportColladaMdr : public EditorImportPlugin { class EditorImportColladaMdr : public MDRImportPluginBase {
GDCLASS(EditorImportColladaMdr, MDRImportPluginBase);
GDCLASS(EditorImportColladaMdr, EditorImportPlugin);
public: public:
virtual String get_importer_name() const; virtual String get_importer_name() const;
@ -60,14 +61,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);
EditorImportColladaMdr(); EditorImportColladaMdr();
~EditorImportColladaMdr(); ~EditorImportColladaMdr();