746dddc067
* Map is unnecessary and inefficient in almost every case. * Replaced by the new HashMap. * Renamed Map to RBMap and Set to RBSet for cases that still make sense (order matters) but use is discouraged. There were very few cases where replacing by HashMap was undesired because keeping the key order was intended. I tried to keep those (as RBMap) as much as possible, but might have missed some. Review appreciated!
577 lines
13 KiB
C++
577 lines
13 KiB
C++
/*************************************************************************/
|
|
/* collada.h */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* */
|
|
/* 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 COLLADA_H
|
|
#define COLLADA_H
|
|
|
|
#include "core/config/project_settings.h"
|
|
#include "core/io/xml_parser.h"
|
|
#include "core/templates/rb_map.h"
|
|
#include "scene/resources/material.h"
|
|
|
|
class Collada {
|
|
public:
|
|
enum ImportFlags {
|
|
IMPORT_FLAG_SCENE = 1,
|
|
IMPORT_FLAG_ANIMATION = 2
|
|
};
|
|
|
|
struct Image {
|
|
String path;
|
|
};
|
|
|
|
struct Material {
|
|
String name;
|
|
String instance_effect;
|
|
};
|
|
|
|
struct Effect {
|
|
String name;
|
|
HashMap<String, Variant> params;
|
|
|
|
struct Channel {
|
|
int uv_idx = 0;
|
|
String texture;
|
|
Color color;
|
|
Channel() {}
|
|
};
|
|
|
|
Channel diffuse, specular, emission, bump;
|
|
float shininess = 40;
|
|
bool found_double_sided = false;
|
|
bool double_sided = true;
|
|
bool unshaded = false;
|
|
|
|
String get_texture_path(const String &p_source, Collada &state) const;
|
|
|
|
Effect() {
|
|
diffuse.color = Color(1, 1, 1, 1);
|
|
}
|
|
};
|
|
|
|
struct CameraData {
|
|
enum Mode {
|
|
MODE_PERSPECTIVE,
|
|
MODE_ORTHOGONAL
|
|
};
|
|
|
|
Mode mode = MODE_PERSPECTIVE;
|
|
|
|
union {
|
|
struct {
|
|
float x_fov = 0;
|
|
float y_fov = 0;
|
|
} perspective;
|
|
struct {
|
|
float x_mag = 0;
|
|
float y_mag = 0;
|
|
} orthogonal;
|
|
};
|
|
|
|
float aspect = 1;
|
|
float z_near = 0.05;
|
|
float z_far = 4000;
|
|
|
|
CameraData() {}
|
|
};
|
|
|
|
struct LightData {
|
|
enum Mode {
|
|
MODE_AMBIENT,
|
|
MODE_DIRECTIONAL,
|
|
MODE_OMNI,
|
|
MODE_SPOT
|
|
};
|
|
|
|
Mode mode = MODE_AMBIENT;
|
|
|
|
Color color = Color(1, 1, 1, 1);
|
|
|
|
float constant_att = 0;
|
|
float linear_att = 0;
|
|
float quad_att = 0;
|
|
|
|
float spot_angle = 45;
|
|
float spot_exp = 1;
|
|
|
|
LightData() {}
|
|
};
|
|
|
|
struct MeshData {
|
|
String name;
|
|
struct Source {
|
|
Vector<float> array;
|
|
int stride = 0;
|
|
};
|
|
|
|
HashMap<String, Source> sources;
|
|
|
|
struct Vertices {
|
|
HashMap<String, String> sources;
|
|
};
|
|
|
|
HashMap<String, Vertices> vertices;
|
|
|
|
struct Primitives {
|
|
struct SourceRef {
|
|
String source;
|
|
int offset = 0;
|
|
};
|
|
|
|
String material;
|
|
HashMap<String, SourceRef> sources;
|
|
Vector<float> polygons;
|
|
Vector<float> indices;
|
|
int count = 0;
|
|
int vertex_size = 0;
|
|
};
|
|
|
|
Vector<Primitives> primitives;
|
|
|
|
bool found_double_sided = false;
|
|
bool double_sided = true;
|
|
|
|
MeshData() {}
|
|
};
|
|
|
|
struct CurveData {
|
|
String name;
|
|
bool closed = false;
|
|
|
|
struct Source {
|
|
Vector<String> sarray;
|
|
Vector<float> array;
|
|
int stride = 0;
|
|
};
|
|
|
|
HashMap<String, Source> sources;
|
|
|
|
HashMap<String, String> control_vertices;
|
|
|
|
CurveData() {}
|
|
};
|
|
|
|
struct SkinControllerData {
|
|
String base;
|
|
bool use_idrefs = false;
|
|
|
|
Transform3D bind_shape;
|
|
|
|
struct Source {
|
|
Vector<String> sarray; //maybe for names
|
|
Vector<float> array;
|
|
int stride = 1;
|
|
Source() {}
|
|
};
|
|
|
|
HashMap<String, Source> sources;
|
|
|
|
struct Joints {
|
|
HashMap<String, String> sources;
|
|
} joints;
|
|
|
|
struct Weights {
|
|
struct SourceRef {
|
|
String source;
|
|
int offset = 0;
|
|
};
|
|
|
|
String material;
|
|
HashMap<String, SourceRef> sources;
|
|
Vector<float> sets;
|
|
Vector<float> indices;
|
|
int count = 0;
|
|
} weights;
|
|
|
|
HashMap<String, Transform3D> bone_rest_map;
|
|
|
|
SkinControllerData() {}
|
|
};
|
|
|
|
struct MorphControllerData {
|
|
String mesh;
|
|
String mode;
|
|
|
|
struct Source {
|
|
int stride = 1;
|
|
Vector<String> sarray; //maybe for names
|
|
Vector<float> array;
|
|
Source() {}
|
|
};
|
|
|
|
HashMap<String, Source> sources;
|
|
|
|
HashMap<String, String> targets;
|
|
MorphControllerData() {}
|
|
};
|
|
|
|
struct Vertex {
|
|
int idx = 0;
|
|
Vector3 vertex;
|
|
Vector3 normal;
|
|
Vector3 uv;
|
|
Vector3 uv2;
|
|
Plane tangent;
|
|
Color color;
|
|
int uid = 0;
|
|
struct Weight {
|
|
int bone_idx = 0;
|
|
float weight = 0;
|
|
bool operator<(const Weight w) const { return weight > w.weight; } //heaviest first
|
|
};
|
|
|
|
Vector<Weight> weights;
|
|
|
|
void fix_weights() {
|
|
weights.sort();
|
|
if (weights.size() > 4) {
|
|
//cap to 4 and make weights add up 1
|
|
weights.resize(4);
|
|
float total = 0;
|
|
for (int i = 0; i < 4; i++) {
|
|
total += weights[i].weight;
|
|
}
|
|
if (total) {
|
|
for (int i = 0; i < 4; i++) {
|
|
weights.write[i].weight /= total;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void fix_unit_scale(const Collada &state);
|
|
|
|
bool operator<(const Vertex &p_vert) const {
|
|
if (uid == p_vert.uid) {
|
|
if (vertex == p_vert.vertex) {
|
|
if (normal == p_vert.normal) {
|
|
if (uv == p_vert.uv) {
|
|
if (uv2 == p_vert.uv2) {
|
|
if (!weights.is_empty() || !p_vert.weights.is_empty()) {
|
|
if (weights.size() == p_vert.weights.size()) {
|
|
for (int i = 0; i < weights.size(); i++) {
|
|
if (weights[i].bone_idx != p_vert.weights[i].bone_idx) {
|
|
return weights[i].bone_idx < p_vert.weights[i].bone_idx;
|
|
}
|
|
|
|
if (weights[i].weight != p_vert.weights[i].weight) {
|
|
return weights[i].weight < p_vert.weights[i].weight;
|
|
}
|
|
}
|
|
} else {
|
|
return weights.size() < p_vert.weights.size();
|
|
}
|
|
}
|
|
|
|
return (color < p_vert.color);
|
|
} else {
|
|
return (uv2 < p_vert.uv2);
|
|
}
|
|
} else {
|
|
return (uv < p_vert.uv);
|
|
}
|
|
} else {
|
|
return (normal < p_vert.normal);
|
|
}
|
|
} else {
|
|
return vertex < p_vert.vertex;
|
|
}
|
|
} else {
|
|
return uid < p_vert.uid;
|
|
}
|
|
}
|
|
|
|
Vertex() {}
|
|
};
|
|
|
|
struct Node {
|
|
enum Type {
|
|
TYPE_NODE,
|
|
TYPE_JOINT,
|
|
TYPE_SKELETON, //this bone is not collada, it's added afterwards as optimization
|
|
TYPE_LIGHT,
|
|
TYPE_CAMERA,
|
|
TYPE_GEOMETRY
|
|
};
|
|
|
|
struct XForm {
|
|
enum Op {
|
|
OP_ROTATE,
|
|
OP_SCALE,
|
|
OP_TRANSLATE,
|
|
OP_MATRIX,
|
|
OP_VISIBILITY
|
|
};
|
|
|
|
String id;
|
|
Op op = OP_ROTATE;
|
|
Vector<float> data;
|
|
};
|
|
|
|
Type type = TYPE_NODE;
|
|
|
|
String name;
|
|
String id;
|
|
String empty_draw_type;
|
|
bool noname = false;
|
|
Vector<XForm> xform_list;
|
|
Transform3D default_transform;
|
|
Transform3D post_transform;
|
|
Vector<Node *> children;
|
|
|
|
Node *parent = nullptr;
|
|
|
|
Transform3D compute_transform(const Collada &state) const;
|
|
Transform3D get_global_transform() const;
|
|
Transform3D get_transform() const;
|
|
|
|
bool ignore_anim = false;
|
|
|
|
Node() {}
|
|
virtual ~Node() {
|
|
for (int i = 0; i < children.size(); i++) {
|
|
memdelete(children[i]);
|
|
}
|
|
};
|
|
};
|
|
|
|
struct NodeSkeleton : public Node {
|
|
NodeSkeleton() { type = TYPE_SKELETON; }
|
|
};
|
|
|
|
struct NodeJoint : public Node {
|
|
NodeSkeleton *owner = nullptr;
|
|
String sid;
|
|
NodeJoint() {
|
|
type = TYPE_JOINT;
|
|
}
|
|
};
|
|
|
|
struct NodeGeometry : public Node {
|
|
bool controller = false;
|
|
String source;
|
|
|
|
struct Material {
|
|
String target;
|
|
};
|
|
|
|
HashMap<String, Material> material_map;
|
|
Vector<String> skeletons;
|
|
|
|
NodeGeometry() { type = TYPE_GEOMETRY; }
|
|
};
|
|
|
|
struct NodeCamera : public Node {
|
|
String camera;
|
|
|
|
NodeCamera() { type = TYPE_CAMERA; }
|
|
};
|
|
|
|
struct NodeLight : public Node {
|
|
String light;
|
|
|
|
NodeLight() { type = TYPE_LIGHT; }
|
|
};
|
|
|
|
struct VisualScene {
|
|
String name;
|
|
Vector<Node *> root_nodes;
|
|
|
|
~VisualScene() {
|
|
for (int i = 0; i < root_nodes.size(); i++) {
|
|
memdelete(root_nodes[i]);
|
|
}
|
|
}
|
|
};
|
|
|
|
struct AnimationClip {
|
|
String name;
|
|
float begin = 0;
|
|
float end = 1;
|
|
Vector<String> tracks;
|
|
|
|
AnimationClip() {}
|
|
};
|
|
|
|
struct AnimationTrack {
|
|
String id;
|
|
String target;
|
|
String param;
|
|
String component;
|
|
bool property = false;
|
|
|
|
enum InterpolationType {
|
|
INTERP_LINEAR,
|
|
INTERP_BEZIER
|
|
};
|
|
|
|
struct Key {
|
|
enum Type {
|
|
TYPE_FLOAT,
|
|
TYPE_MATRIX
|
|
};
|
|
|
|
float time = 0;
|
|
Vector<float> data;
|
|
Point2 in_tangent;
|
|
Point2 out_tangent;
|
|
InterpolationType interp_type = INTERP_LINEAR;
|
|
|
|
Key() {}
|
|
};
|
|
|
|
Vector<float> get_value_at_time(float p_time) const;
|
|
|
|
Vector<Key> keys;
|
|
|
|
AnimationTrack() {}
|
|
};
|
|
|
|
/****************/
|
|
/* IMPORT STATE */
|
|
/****************/
|
|
|
|
struct State {
|
|
int import_flags = 0;
|
|
|
|
float unit_scale = 1.0;
|
|
Vector3::Axis up_axis = Vector3::AXIS_Y;
|
|
bool z_up = false;
|
|
|
|
struct Version {
|
|
int major = 0, minor = 0, rev = 0;
|
|
|
|
bool operator<(const Version &p_ver) const { return (major == p_ver.major) ? ((minor == p_ver.minor) ? (rev < p_ver.rev) : minor < p_ver.minor) : major < p_ver.major; }
|
|
Version(int p_major = 0, int p_minor = 0, int p_rev = 0) {
|
|
major = p_major;
|
|
minor = p_minor;
|
|
rev = p_rev;
|
|
}
|
|
} version;
|
|
|
|
HashMap<String, CameraData> camera_data_map;
|
|
HashMap<String, MeshData> mesh_data_map;
|
|
HashMap<String, LightData> light_data_map;
|
|
HashMap<String, CurveData> curve_data_map;
|
|
|
|
HashMap<String, String> mesh_name_map;
|
|
HashMap<String, String> morph_name_map;
|
|
HashMap<String, String> morph_ownership_map;
|
|
HashMap<String, SkinControllerData> skin_controller_data_map;
|
|
HashMap<String, MorphControllerData> morph_controller_data_map;
|
|
|
|
HashMap<String, Image> image_map;
|
|
HashMap<String, Material> material_map;
|
|
HashMap<String, Effect> effect_map;
|
|
|
|
HashMap<String, VisualScene> visual_scene_map;
|
|
HashMap<String, Node *> scene_map;
|
|
RBSet<String> idref_joints;
|
|
HashMap<String, String> sid_to_node_map;
|
|
//RBMap<String,NodeJoint*> bone_map;
|
|
|
|
HashMap<String, Transform3D> bone_rest_map;
|
|
|
|
String local_path;
|
|
String root_visual_scene;
|
|
String root_physics_scene;
|
|
|
|
Vector<AnimationClip> animation_clips;
|
|
Vector<AnimationTrack> animation_tracks;
|
|
HashMap<String, Vector<int>> referenced_tracks;
|
|
HashMap<String, Vector<int>> by_id_tracks;
|
|
|
|
float animation_length = 0;
|
|
|
|
State() {}
|
|
} state;
|
|
|
|
Error load(const String &p_path, int p_flags = 0);
|
|
|
|
Collada();
|
|
|
|
Transform3D fix_transform(const Transform3D &p_transform);
|
|
|
|
Transform3D get_root_transform() const;
|
|
|
|
int get_uv_channel(String p_name);
|
|
|
|
private: // private stuff
|
|
HashMap<String, int> channel_map;
|
|
|
|
void _parse_asset(XMLParser &parser);
|
|
void _parse_image(XMLParser &parser);
|
|
void _parse_material(XMLParser &parser);
|
|
void _parse_effect_material(XMLParser &parser, Effect &effect, String &id);
|
|
void _parse_effect(XMLParser &parser);
|
|
void _parse_camera(XMLParser &parser);
|
|
void _parse_light(XMLParser &parser);
|
|
void _parse_animation_clip(XMLParser &parser);
|
|
|
|
void _parse_mesh_geometry(XMLParser &parser, String p_id, String p_name);
|
|
void _parse_curve_geometry(XMLParser &parser, String p_id, String p_name);
|
|
|
|
void _parse_skin_controller(XMLParser &parser, String p_id);
|
|
void _parse_morph_controller(XMLParser &parser, String p_id);
|
|
void _parse_controller(XMLParser &parser);
|
|
|
|
Node *_parse_visual_instance_geometry(XMLParser &parser);
|
|
Node *_parse_visual_instance_camera(XMLParser &parser);
|
|
Node *_parse_visual_instance_light(XMLParser &parser);
|
|
|
|
Node *_parse_visual_node_instance_data(XMLParser &parser);
|
|
Node *_parse_visual_scene_node(XMLParser &parser);
|
|
void _parse_visual_scene(XMLParser &parser);
|
|
|
|
void _parse_animation(XMLParser &parser);
|
|
void _parse_scene(XMLParser &parser);
|
|
void _parse_library(XMLParser &parser);
|
|
|
|
Variant _parse_param(XMLParser &parser);
|
|
Vector<float> _read_float_array(XMLParser &parser);
|
|
Vector<String> _read_string_array(XMLParser &parser);
|
|
Transform3D _read_transform(XMLParser &parser);
|
|
String _read_empty_draw_type(XMLParser &parser);
|
|
|
|
void _joint_set_owner(Collada::Node *p_node, NodeSkeleton *p_owner);
|
|
void _create_skeletons(Collada::Node **p_node, NodeSkeleton *p_skeleton = nullptr);
|
|
void _find_morph_nodes(VisualScene *p_vscene, Node *p_node);
|
|
bool _remove_node(Node *p_parent, Node *p_node);
|
|
void _remove_node(VisualScene *p_vscene, Node *p_node);
|
|
void _merge_skeletons2(VisualScene *p_vscene);
|
|
void _merge_skeletons(VisualScene *p_vscene, Node *p_node);
|
|
bool _optimize_skeletons(VisualScene *p_vscene, Node *p_node);
|
|
|
|
bool _move_geometry_to_skeletons(VisualScene *p_vscene, Node *p_node, List<Node *> *p_mgeom);
|
|
|
|
void _optimize();
|
|
};
|
|
|
|
#endif // COLLADA_H
|