diff options
Diffstat (limited to 'editor/import/collada.h')
-rw-r--r-- | editor/import/collada.h | 647 |
1 files changed, 647 insertions, 0 deletions
diff --git a/editor/import/collada.h b/editor/import/collada.h new file mode 100644 index 0000000000..b74332fb22 --- /dev/null +++ b/editor/import/collada.h @@ -0,0 +1,647 @@ +/*************************************************************************/ +/* collada.h */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* https://godotengine.org */ +/*************************************************************************/ +/* Copyright (c) 2007-2020 Juan Linietsky, Ariel Manzur. */ +/* Copyright (c) 2014-2020 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/io/xml_parser.h" +#include "core/map.h" +#include "core/project_settings.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; + Map<String, Variant> params; + + struct Channel { + + int uv_idx; + String texture; + Color color; + Channel() { uv_idx = 0; } + }; + + Channel diffuse, specular, emission, bump; + float shininess; + bool found_double_sided; + bool double_sided; + bool unshaded; + + String get_texture_path(const String &p_source, Collada &state) const; + + Effect() { + diffuse.color = Color(1, 1, 1, 1); + double_sided = true; + found_double_sided = false; + shininess = 40; + unshaded = false; + } + }; + + struct CameraData { + + enum Mode { + MODE_PERSPECTIVE, + MODE_ORTHOGONAL + }; + + Mode mode; + + union { + struct { + float x_fov; + float y_fov; + } perspective; + struct { + float x_mag; + float y_mag; + } orthogonal; + }; + + float aspect; + float z_near; + float z_far; + + CameraData() : + mode(MODE_PERSPECTIVE), + aspect(1), + z_near(0.1), + z_far(100) { + perspective.x_fov = 0; + perspective.y_fov = 0; + } + }; + + struct LightData { + + enum Mode { + MODE_AMBIENT, + MODE_DIRECTIONAL, + MODE_OMNI, + MODE_SPOT + }; + + Mode mode; + + Color color; + + float constant_att; + float linear_att; + float quad_att; + + float spot_angle; + float spot_exp; + + LightData() : + mode(MODE_AMBIENT), + color(Color(1, 1, 1, 1)), + constant_att(0), + linear_att(0), + quad_att(0), + spot_angle(45), + spot_exp(1) { + } + }; + + struct MeshData { + + String name; + struct Source { + + Vector<float> array; + int stride; + }; + + Map<String, Source> sources; + + struct Vertices { + + Map<String, String> sources; + }; + + Map<String, Vertices> vertices; + + struct Primitives { + + struct SourceRef { + + String source; + int offset; + }; + + String material; + Map<String, SourceRef> sources; + Vector<float> polygons; + Vector<float> indices; + int count; + int vertex_size; + }; + + Vector<Primitives> primitives; + + bool found_double_sided; + bool double_sided; + + MeshData() { + found_double_sided = false; + double_sided = true; + } + }; + + struct CurveData { + + String name; + bool closed; + + struct Source { + + Vector<String> sarray; + Vector<float> array; + int stride; + }; + + Map<String, Source> sources; + + Map<String, String> control_vertices; + + CurveData() { + + closed = false; + } + }; + struct SkinControllerData { + + String base; + bool use_idrefs; + + Transform bind_shape; + + struct Source { + + Vector<String> sarray; //maybe for names + Vector<float> array; + int stride; + Source() { + stride = 1; + } + }; + + Map<String, Source> sources; + + struct Joints { + + Map<String, String> sources; + } joints; + + struct Weights { + + struct SourceRef { + + String source; + int offset; + }; + + String material; + Map<String, SourceRef> sources; + Vector<float> sets; + Vector<float> indices; + int count; + } weights; + + Map<String, Transform> bone_rest_map; + + SkinControllerData() { use_idrefs = false; } + }; + + struct MorphControllerData { + + String mesh; + String mode; + + struct Source { + + int stride; + Vector<String> sarray; //maybe for names + Vector<float> array; + Source() { stride = 1; } + }; + + Map<String, Source> sources; + + Map<String, String> targets; + MorphControllerData() {} + }; + + struct Vertex { + + int idx; + Vector3 vertex; + Vector3 normal; + Vector3 uv; + Vector3 uv2; + Plane tangent; + Color color; + int uid; + struct Weight { + int bone_idx; + float weight; + 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(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.empty() || !p_vert.weights.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() { + uid = 0; + idx = 0; + } + }; + 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; + Vector<float> data; + }; + + Type type; + + String name; + String id; + String empty_draw_type; + bool noname; + Vector<XForm> xform_list; + Transform default_transform; + Transform post_transform; + Vector<Node *> children; + + Node *parent; + + Transform compute_transform(Collada &state) const; + Transform get_global_transform() const; + Transform get_transform() const; + + bool ignore_anim; + + Node() { + noname = false; + type = TYPE_NODE; + parent = nullptr; + ignore_anim = false; + } + 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; + String sid; + NodeJoint() { + type = TYPE_JOINT; + owner = nullptr; + } + }; + + struct NodeGeometry : public Node { + + bool controller; + String source; + + struct Material { + String target; + }; + + Map<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; + float end; + Vector<String> tracks; + + AnimationClip() { + begin = 0; + end = 1; + } + }; + + struct AnimationTrack { + + String id; + String target; + String param; + String component; + bool property; + + enum InterpolationType { + INTERP_LINEAR, + INTERP_BEZIER + }; + + struct Key { + + enum Type { + TYPE_FLOAT, + TYPE_MATRIX + }; + + float time; + Vector<float> data; + Point2 in_tangent; + Point2 out_tangent; + InterpolationType interp_type; + + Key() { interp_type = INTERP_LINEAR; } + }; + + Vector<float> get_value_at_time(float p_time) const; + + Vector<Key> keys; + + AnimationTrack() { property = false; } + }; + + /****************/ + /* IMPORT STATE */ + /****************/ + + struct State { + + int import_flags; + + float unit_scale; + Vector3::Axis up_axis; + bool z_up; + + struct Version { + + int major, minor, rev; + + 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; + + Map<String, CameraData> camera_data_map; + Map<String, MeshData> mesh_data_map; + Map<String, LightData> light_data_map; + Map<String, CurveData> curve_data_map; + + Map<String, String> mesh_name_map; + Map<String, String> morph_name_map; + Map<String, String> morph_ownership_map; + Map<String, SkinControllerData> skin_controller_data_map; + Map<String, MorphControllerData> morph_controller_data_map; + + Map<String, Image> image_map; + Map<String, Material> material_map; + Map<String, Effect> effect_map; + + Map<String, VisualScene> visual_scene_map; + Map<String, Node *> scene_map; + Set<String> idref_joints; + Map<String, String> sid_to_node_map; + //Map<String,NodeJoint*> bone_map; + + Map<String, Transform> bone_rest_map; + + String local_path; + String root_visual_scene; + String root_physics_scene; + + Vector<AnimationClip> animation_clips; + Vector<AnimationTrack> animation_tracks; + Map<String, Vector<int>> referenced_tracks; + Map<String, Vector<int>> by_id_tracks; + + float animation_length; + + State() : + import_flags(0), + unit_scale(1.0), + up_axis(Vector3::AXIS_Y), + animation_length(0) { + } + } state; + + Error load(const String &p_path, int p_flags = 0); + + Collada(); + + Transform fix_transform(const Transform &p_transform); + + Transform get_root_transform() const; + + int get_uv_channel(String p_name); + +private: // private stuff + Map<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); + Transform _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 |