godot/tools/editor/io_plugins/editor_import_collada.cpp

2445 lines
66 KiB
C++
Raw Normal View History

2014-02-10 02:10:30 +01:00
/*************************************************************************/
/* editor_import_collada.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* http://www.godotengine.org */
/*************************************************************************/
2016-01-01 14:50:53 +01:00
/* Copyright (c) 2007-2016 Juan Linietsky, Ariel Manzur. */
2014-02-10 02:10:30 +01:00
/* */
/* 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 "editor_import_collada.h"
#include "collada/collada.h"
#include "scene/3d/spatial.h"
#include "scene/3d/skeleton.h"
#include "scene/3d/path.h"
#include "scene/3d/camera.h"
#include "scene/3d/light.h"
#include "scene/animation/animation_player.h"
#include "scene/3d/mesh_instance.h"
#include "scene/resources/animation.h"
#include "scene/resources/packed_scene.h"
#include "os/os.h"
#include "tools/editor/editor_node.h"
#include <iostream>
2014-02-10 02:10:30 +01:00
struct ColladaImport {
Collada collada;
Spatial *scene;
Vector<Ref<Animation> > animations;
struct NodeMap {
//String path;
Spatial *node;
int bone;
List<int> anim_tracks;
NodeMap() { node=NULL; bone=-1; }
};
bool found_ambient;
Color ambient;
bool found_directional;
bool force_make_tangents;
bool apply_mesh_xform_to_vertices;
float bake_fps;
2014-02-10 02:10:30 +01:00
Map<String,NodeMap> node_map; //map from collada node to engine node
Map<String,String> node_name_map; //map from collada node to engine node
2014-02-10 02:10:30 +01:00
Map<String, Ref<Mesh> > mesh_cache;
Map<String, Ref<Curve3D> > curve_cache;
Map<String, Ref<Material> > material_cache;
Map<Collada::Node*,Skeleton*> skeleton_map;
2014-02-10 02:10:30 +01:00
Map< Skeleton*, Map< String, int> > skeleton_bone_map;
Set<String> valid_animated_nodes;
Vector<int> valid_animated_properties;
Map<String,bool> bones_with_animation;
Error _populate_skeleton(Skeleton *p_skeleton,Collada::Node *p_node, int &r_bone, int p_parent);
Error _create_scene_skeletons(Collada::Node *p_node);
2014-02-10 02:10:30 +01:00
Error _create_scene(Collada::Node *p_node, Spatial *p_parent);
Error _create_resources(Collada::Node *p_node);
Error _create_material(const String& p_material);
Error _create_mesh_surfaces(bool p_optimize, Ref<Mesh>& p_mesh, const Map<String,Collada::NodeGeometry::Material>& p_material_map, const Collada::MeshData &meshdata, const Transform& p_local_xform, const Vector<int> &bone_remap, const Collada::SkinControllerData *p_skin_data, const Collada::MorphControllerData *p_morph_data, Vector<Ref<Mesh> > p_morph_meshes=Vector<Ref<Mesh> >(), bool p_for_morph=false);
2014-02-10 02:10:30 +01:00
Error load(const String& p_path, int p_flags, bool p_force_make_tangents=false);
void _fix_param_animation_tracks();
void create_animation(int p_clip,bool p_make_tracks_in_all_bones);
void create_animations(bool p_make_tracks_in_all_bones);
2014-02-10 02:10:30 +01:00
Set<String> tracks_in_clips;
Vector<String> missing_textures;
void _pre_process_lights(Collada::Node *p_node);
ColladaImport() {
found_ambient=false;
found_directional=false;
force_make_tangents=false;
apply_mesh_xform_to_vertices=true;
bake_fps=15;
2014-02-10 02:10:30 +01:00
}
};
Error ColladaImport::_populate_skeleton(Skeleton *p_skeleton,Collada::Node *p_node, int &r_bone, int p_parent) {
if (p_node->type!=Collada::Node::TYPE_JOINT)
return OK;
Collada::NodeJoint *joint = static_cast<Collada::NodeJoint*>(p_node);
print_line("populating joint "+joint->name);
2014-02-10 02:10:30 +01:00
p_skeleton->add_bone(p_node->name);
if (p_parent>=0)
p_skeleton->set_bone_parent(r_bone,p_parent);
NodeMap nm;
nm.node=p_skeleton;
nm.bone = r_bone;
node_map[p_node->id]=nm;
node_name_map[p_node->name]=p_node->id;
2014-02-10 02:10:30 +01:00
skeleton_bone_map[p_skeleton][joint->sid]=r_bone;
if (collada.state.bone_rest_map.has(joint->sid)) {
p_skeleton->set_bone_rest(r_bone,collada.fix_transform(collada.state.bone_rest_map[joint->sid]));
//should map this bone to something for animation?
} else {
print_line("no rest: "+joint->sid);
WARN_PRINT("Joint has no rest..");
}
int id = r_bone++;
for(int i=0;i<p_node->children.size();i++) {
Error err = _populate_skeleton(p_skeleton,p_node->children[i],r_bone,id);
if (err)
return err;
}
return OK;
}
void ColladaImport::_pre_process_lights(Collada::Node *p_node) {
if (p_node->type==Collada::Node::TYPE_LIGHT) {
Collada::NodeLight *light=static_cast<Collada::NodeLight*>(p_node);
if (collada.state.light_data_map.has(light->light)) {
Collada::LightData &ld = collada.state.light_data_map[light->light];
if (ld.mode==Collada::LightData::MODE_AMBIENT) {
found_ambient=true;
ambient=ld.color;
}
if (ld.mode==Collada::LightData::MODE_DIRECTIONAL) {
found_directional=true;
}
}
}
for(int i=0;i<p_node->children.size();i++)
_pre_process_lights(p_node->children[i]);
}
Error ColladaImport::_create_scene_skeletons(Collada::Node *p_node) {
if (p_node->type==Collada::Node::TYPE_SKELETON) {
Skeleton *sk = memnew( Skeleton );
int bone = 0;
for(int i=0;i<p_node->children.size();i++) {
_populate_skeleton(sk,p_node->children[i],bone,-1);
}
sk->localize_rests(); //after creating skeleton, rests must be localized...!
skeleton_map[p_node]=sk;
}
for(int i=0;i<p_node->children.size();i++) {
Error err = _create_scene_skeletons(p_node->children[i]);
if (err)
return err;
}
return OK;
}
2014-02-10 02:10:30 +01:00
Error ColladaImport::_create_scene(Collada::Node *p_node, Spatial *p_parent) {
Spatial * node=NULL;
switch(p_node->type) {
case Collada::Node::TYPE_NODE: {
node = memnew( Spatial );
} break;
case Collada::Node::TYPE_JOINT: {
return OK; // do nothing
} break;
case Collada::Node::TYPE_LIGHT: {
//node = memnew( Light)
Collada::NodeLight *light = static_cast<Collada::NodeLight*>(p_node);
if (collada.state.light_data_map.has(light->light)) {
Collada::LightData &ld = collada.state.light_data_map[light->light];
if (ld.mode==Collada::LightData::MODE_AMBIENT) {
if (found_directional)
return OK; //do nothing not needed
if (!bool(GLOBAL_DEF("collada/use_ambient",false)))
return OK;
//well, it's an ambient light..
Light *l = memnew( DirectionalLight );
// l->set_color(Light::COLOR_AMBIENT,ld.color);
// l->set_color(Light::COLOR_DIFFUSE,Color(0,0,0));
// l->set_color(Light::COLOR_SPECULAR,Color(0,0,0));
2014-02-10 02:10:30 +01:00
node = l;
} else if (ld.mode==Collada::LightData::MODE_DIRECTIONAL) {
//well, it's an ambient light..
Light *l = memnew( DirectionalLight );
//if (found_ambient) //use it here
// l->set_color(Light::COLOR_AMBIENT,ambient);
2014-02-10 02:10:30 +01:00
// l->set_color(Light::COLOR_DIFFUSE,ld.color);
// l->set_color(Light::COLOR_SPECULAR,Color(1,1,1));
2014-02-10 02:10:30 +01:00
node = l;
} else {
Light *l;
if (ld.mode==Collada::LightData::MODE_OMNI)
l=memnew( OmniLight );
else {
l=memnew( SpotLight );
// l->set_parameter(Light::PARAM_SPOT_ANGLE,ld.spot_angle);
// l->set_parameter(Light::PARAM_SPOT_ATTENUATION,ld.spot_exp);
2014-02-10 02:10:30 +01:00
}
//
// l->set_color(Light::COLOR_DIFFUSE,ld.color);
// l->set_color(Light::COLOR_SPECULAR,Color(1,1,1));
// l->approximate_opengl_attenuation(ld.constant_att,ld.linear_att,ld.quad_att);
2014-02-10 02:10:30 +01:00
node=l;
}
} else {
node = memnew( Spatial );
}
} break;
case Collada::Node::TYPE_CAMERA: {
Collada::NodeCamera *cam = static_cast<Collada::NodeCamera*>(p_node);
Camera *camera = memnew( Camera );
if (collada.state.camera_data_map.has(cam->camera)) {
const Collada::CameraData &cd = collada.state.camera_data_map[cam->camera];
switch(cd.mode) {
case Collada::CameraData::MODE_ORTHOGONAL: {
2015-03-03 18:39:13 +01:00
if (cd.orthogonal.y_mag) {
2014-02-10 02:10:30 +01:00
2015-03-03 18:39:13 +01:00
camera->set_keep_aspect_mode(Camera::KEEP_HEIGHT);
camera->set_orthogonal(cd.orthogonal.y_mag*2.0 ,cd.z_near,cd.z_far);
2014-02-10 02:10:30 +01:00
2015-03-03 18:39:13 +01:00
} else if (!cd.orthogonal.y_mag && cd.orthogonal.x_mag) {
2014-02-10 02:10:30 +01:00
2015-03-03 18:39:13 +01:00
camera->set_keep_aspect_mode(Camera::KEEP_WIDTH);
camera->set_orthogonal(cd.orthogonal.x_mag*2.0,cd.z_near,cd.z_far);
2014-02-10 02:10:30 +01:00
}
} break;
case Collada::CameraData::MODE_PERSPECTIVE: {
if (cd.perspective.y_fov) {
camera->set_perspective(cd.perspective.y_fov,cd.z_near,cd.z_far);
} else if (!cd.perspective.y_fov && cd.perspective.x_fov) {
camera->set_perspective(cd.perspective.x_fov / cd.aspect,cd.z_near,cd.z_far);
}
} break;
}
}
node=camera;
} break;
case Collada::Node::TYPE_GEOMETRY: {
Collada::NodeGeometry *ng = static_cast<Collada::NodeGeometry*>(p_node);
if (collada.state.curve_data_map.has(ng->source)) {
node = memnew( Path );
} else {
//mesh since nothing else
node = memnew( MeshInstance );
node->cast_to<MeshInstance>()->set_flag(GeometryInstance::FLAG_USE_BAKED_LIGHT,true);
2014-02-10 02:10:30 +01:00
}
} break;
case Collada::Node::TYPE_SKELETON: {
ERR_FAIL_COND_V(!skeleton_map.has(p_node),ERR_CANT_CREATE);
Skeleton *sk = skeleton_map[p_node];
2014-02-10 02:10:30 +01:00
node=sk;
} break;
}
if (p_node->name!="")
node->set_name(p_node->name);
NodeMap nm;
nm.node=node;
node_map[p_node->id]=nm;
node_name_map[p_node->name]=p_node->id;
2014-02-10 02:10:30 +01:00
Transform xf = p_node->default_transform;
xf = collada.fix_transform( xf ) * p_node->post_transform;
node->set_transform(xf);
p_parent->add_child(node);
node->set_owner(scene);
if (p_node->empty_draw_type!="") {
node->set_meta("empty_draw_type", Variant(p_node->empty_draw_type));
}
2014-02-10 02:10:30 +01:00
for(int i=0;i<p_node->children.size();i++) {
Error err = _create_scene(p_node->children[i],node);
if (err)
return err;
}
return OK;
}
Error ColladaImport::_create_material(const String& p_target) {
ERR_FAIL_COND_V(material_cache.has(p_target),ERR_ALREADY_EXISTS);
ERR_FAIL_COND_V(!collada.state.material_map.has(p_target),ERR_INVALID_PARAMETER);
Collada::Material &src_mat=collada.state.material_map[p_target];
ERR_FAIL_COND_V(!collada.state.effect_map.has(src_mat.instance_effect),ERR_INVALID_PARAMETER);
Collada::Effect &effect=collada.state.effect_map[src_mat.instance_effect];
Ref<FixedSpatialMaterial> material= memnew( FixedSpatialMaterial );
2014-02-10 02:10:30 +01:00
if (src_mat.name!="")
material->set_name(src_mat.name);
else if (effect.name!="")
material->set_name(effect.name);
2014-02-10 02:10:30 +01:00
// DIFFUSE
if (effect.diffuse.texture!="") {
String texfile = effect.get_texture_path(effect.diffuse.texture,collada);
if (texfile!="") {
Ref<Texture> texture = ResourceLoader::load(texfile,"Texture");
if (texture.is_valid()) {
// material->set_texture(FixedSpatialMaterial::PARAM_DIFFUSE,texture);
// material->set_parameter(FixedSpatialMaterial::PARAM_DIFFUSE,Color(1,1,1,1));
2014-02-10 02:10:30 +01:00
} else {
missing_textures.push_back(texfile.get_file());
}
}
} else {
// material->set_parameter(FixedSpatialMaterial::PARAM_DIFFUSE,effect.diffuse.color);
2014-02-10 02:10:30 +01:00
}
// SPECULAR
if (effect.specular.texture!="") {
String texfile = effect.get_texture_path(effect.specular.texture,collada);
if (texfile!="") {
Ref<Texture> texture = ResourceLoader::load(texfile,"Texture");
if (texture.is_valid()) {
// material->set_texture(FixedSpatialMaterial::PARAM_SPECULAR,texture);
// material->set_parameter(FixedSpatialMaterial::PARAM_SPECULAR,Color(1,1,1,1));
2014-02-10 02:10:30 +01:00
} else {
missing_textures.push_back(texfile.get_file());
}
}
} else {
// material->set_parameter(FixedSpatialMaterial::PARAM_SPECULAR,effect.specular.color);
2014-02-10 02:10:30 +01:00
}
// EMISSION
if (effect.emission.texture!="") {
String texfile = effect.get_texture_path(effect.emission.texture,collada);
if (texfile!="") {
Ref<Texture> texture = ResourceLoader::load(texfile,"Texture");
if (texture.is_valid()) {
// material->set_texture(FixedSpatialMaterial::PARAM_EMISSION,texture);
// material->set_parameter(FixedSpatialMaterial::PARAM_EMISSION,Color(1,1,1,1));
2014-02-10 02:10:30 +01:00
}else {
// missing_textures.push_back(texfile.get_file());
2014-02-10 02:10:30 +01:00
}
}
} else {
// material->set_parameter(FixedSpatialMaterial::PARAM_EMISSION,effect.emission.color);
2014-02-10 02:10:30 +01:00
}
// NORMAL
if (effect.bump.texture!="") {
String texfile = effect.get_texture_path(effect.bump.texture,collada);
if (texfile!="") {
Ref<Texture> texture = ResourceLoader::load(texfile,"Texture");
if (texture.is_valid()) {
// material->set_texture(FixedSpatialMaterial::PARAM_NORMAL,texture);
2014-02-10 02:10:30 +01:00
}else {
// missing_textures.push_back(texfile.get_file());
2014-02-10 02:10:30 +01:00
}
}
}
// material->set_parameter(FixedSpatialMaterial::PARAM_SPECULAR_EXP,effect.shininess);
// material->set_flag(Material::FLAG_DOUBLE_SIDED,effect.double_sided);
// material->set_flag(Material::FLAG_UNSHADED,effect.unshaded);
2014-02-10 02:10:30 +01:00
2014-02-10 02:10:30 +01:00
material_cache[p_target]=material;
return OK;
}
static void _generate_normals(const DVector<int>& p_indices,const DVector<Vector3>& p_vertices,DVector<Vector3>&r_normals) {
r_normals.resize(p_vertices.size());
DVector<Vector3>::Write narrayw = r_normals.write();
int iacount=p_indices.size()/3;
DVector<int>::Read index_arrayr = p_indices.read();
DVector<Vector3>::Read vertex_arrayr = p_vertices.read();
for(int idx=0;idx<iacount;idx++) {
Vector3 v[3]={
vertex_arrayr[index_arrayr[idx*3+0]],
vertex_arrayr[index_arrayr[idx*3+1]],
vertex_arrayr[index_arrayr[idx*3+2]]
};
Vector3 normal = Plane(v[0],v[1],v[2]).normal;
narrayw[index_arrayr[idx*3+0]]+=normal;
narrayw[index_arrayr[idx*3+1]]+=normal;
narrayw[index_arrayr[idx*3+2]]+=normal;
}
int vlen=p_vertices.size();
for(int idx=0;idx<vlen;idx++) {
narrayw[idx].normalize();
}
}
static void _generate_tangents_and_binormals(const DVector<int>& p_indices,const DVector<Vector3>& p_vertices,const DVector<Vector3>& p_uvs,const DVector<Vector3>& p_normals,DVector<real_t>&r_tangents) {
int vlen=p_vertices.size();
Vector<Vector3> tangents;
tangents.resize(vlen);
Vector<Vector3> binormals;
binormals.resize(vlen);
int iacount=p_indices.size()/3;
DVector<int>::Read index_arrayr = p_indices.read();
DVector<Vector3>::Read vertex_arrayr = p_vertices.read();
DVector<Vector3>::Read narrayr = p_normals.read();
DVector<Vector3>::Read uvarrayr = p_uvs.read();
for(int idx=0;idx<iacount;idx++) {
Vector3 v1 = vertex_arrayr[ index_arrayr[idx*3+0] ];
Vector3 v2 = vertex_arrayr[ index_arrayr[idx*3+1] ];
Vector3 v3 = vertex_arrayr[ index_arrayr[idx*3+2] ];
Vector3 w1 = uvarrayr[ index_arrayr[idx*3+0] ];
Vector3 w2 = uvarrayr[ index_arrayr[idx*3+1] ];
Vector3 w3 = uvarrayr[ index_arrayr[idx*3+2] ];
real_t x1 = v2.x - v1.x;
real_t x2 = v3.x - v1.x;
real_t y1 = v2.y - v1.y;
real_t y2 = v3.y - v1.y;
real_t z1 = v2.z - v1.z;
real_t z2 = v3.z - v1.z;
real_t s1 = w2.x - w1.x;
real_t s2 = w3.x - w1.x;
real_t t1 = w2.y - w1.y;
real_t t2 = w3.y - w1.y;
real_t r = (s1 * t2 - s2 * t1);
Vector3 tangent;
Vector3 binormal;
if (r==0) {
binormal=Vector3();
tangent=Vector3();
} else {
tangent = Vector3((t2 * x1 - t1 * x2) * r, (t2 * y1 - t1 * y2) * r,
(t2 * z1 - t1 * z2) * r).normalized();
2014-02-10 02:10:30 +01:00
binormal = Vector3((s1 * x2 - s2 * x1) * r, (s1 * y2 - s2 * y1) * r,
(s1 * z2 - s2 * z1) * r).normalized();
2014-02-10 02:10:30 +01:00
}
tangents[ index_arrayr[idx*3+0] ]+=tangent;
binormals[ index_arrayr[idx*3+0] ]+=binormal;
tangents[ index_arrayr[idx*3+1] ]+=tangent;
binormals[ index_arrayr[idx*3+1] ]+=binormal;
tangents[ index_arrayr[idx*3+2] ]+=tangent;
binormals[ index_arrayr[idx*3+2] ]+=binormal;
//print_line(itos(idx)+" tangent: "+tangent);
//print_line(itos(idx)+" binormal: "+binormal);
2014-02-10 02:10:30 +01:00
}
r_tangents.resize(vlen*4);
DVector<real_t>::Write tarrayw = r_tangents.write();
for(int idx=0;idx<vlen;idx++) {
Vector3 tangent = tangents[idx];
Vector3 bingen = narrayr[idx].cross(tangent);
float dir;
if (bingen.dot(binormals[idx]) < 0 )
dir=-1.0;
else
dir=+1.0;
tarrayw[idx*4+0]=tangent.x;
tarrayw[idx*4+1]=tangent.y;
tarrayw[idx*4+2]=tangent.z;
tarrayw[idx*4+3]=dir;
}
}
Error ColladaImport::_create_mesh_surfaces(bool p_optimize,Ref<Mesh>& p_mesh,const Map<String,Collada::NodeGeometry::Material>& p_material_map,const Collada::MeshData &meshdata,const Transform& p_local_xform,const Vector<int> &bone_remap, const Collada::SkinControllerData *skin_controller, const Collada::MorphControllerData *p_morph_data,Vector<Ref<Mesh> > p_morph_meshes,bool p_for_morph) {
2014-02-10 02:10:30 +01:00
bool local_xform_mirror=p_local_xform.basis.determinant() < 0;
if (p_morph_data) {
2014-02-10 02:10:30 +01:00
//add morphie target
ERR_FAIL_COND_V( !p_morph_data->targets.has("MORPH_TARGET"), ERR_INVALID_DATA );
String mt = p_morph_data->targets["MORPH_TARGET"];
ERR_FAIL_COND_V( !p_morph_data->sources.has(mt), ERR_INVALID_DATA);
int morph_targets = p_morph_data->sources[mt].sarray.size();
for(int i=0;i<morph_targets;i++) {
String target = p_morph_data->sources[mt].sarray[i];
ERR_FAIL_COND_V( !collada.state.mesh_data_map.has(target), ERR_INVALID_DATA );
String name = collada.state.mesh_data_map[target].name;
p_mesh->add_morph_target(name);
}
if (p_morph_data->mode=="RELATIVE")
p_mesh->set_morph_target_mode(Mesh::MORPH_MODE_RELATIVE);
else if (p_morph_data->mode=="NORMALIZED")
p_mesh->set_morph_target_mode(Mesh::MORPH_MODE_NORMALIZED);
}
int surface=0;
for(int p_i = 0; p_i < meshdata.primitives.size(); p_i ++ ) {
const Collada::MeshData::Primitives& p = meshdata.primitives[p_i];
/* VERTEX SOURCE */
ERR_FAIL_COND_V(!p.sources.has("VERTEX"),ERR_INVALID_DATA);
String vertex_src_id = p.sources["VERTEX"].source;
int vertex_ofs=p.sources["VERTEX"].offset;
ERR_FAIL_COND_V(!meshdata.vertices.has(vertex_src_id),ERR_INVALID_DATA);
ERR_FAIL_COND_V(!meshdata.vertices[vertex_src_id].sources.has("POSITION"),ERR_INVALID_DATA);
String position_src_id = meshdata.vertices[vertex_src_id].sources["POSITION"];
ERR_FAIL_COND_V(!meshdata.sources.has(position_src_id),ERR_INVALID_DATA);
const Collada::MeshData::Source *vertex_src=&meshdata.sources[position_src_id];
/* NORMAL SOURCE */
const Collada::MeshData::Source *normal_src=NULL;
int normal_ofs=0;
if (p.sources.has("NORMAL")) {
String normal_source_id = p.sources["NORMAL"].source;
normal_ofs = p.sources["NORMAL"].offset;
ERR_FAIL_COND_V( !meshdata.sources.has(normal_source_id),ERR_INVALID_DATA);
normal_src=&meshdata.sources[normal_source_id];
}
const Collada::MeshData::Source *binormal_src=NULL;
int binormal_ofs=0;
if (p.sources.has("TEXBINORMAL")) {
String binormal_source_id = p.sources["TEXBINORMAL"].source;
binormal_ofs = p.sources["TEXBINORMAL"].offset;
ERR_FAIL_COND_V( !meshdata.sources.has(binormal_source_id),ERR_INVALID_DATA);
binormal_src=&meshdata.sources[binormal_source_id];
}
const Collada::MeshData::Source *tangent_src=NULL;
int tangent_ofs=0;
if (p.sources.has("TEXTANGENT")) {
String tangent_source_id = p.sources["TEXTANGENT"].source;
tangent_ofs = p.sources["TEXTANGENT"].offset;
ERR_FAIL_COND_V( !meshdata.sources.has(tangent_source_id),ERR_INVALID_DATA);
tangent_src=&meshdata.sources[tangent_source_id];
}
const Collada::MeshData::Source *uv_src=NULL;
int uv_ofs=0;
if (p.sources.has("TEXCOORD0")) {
String uv_source_id = p.sources["TEXCOORD0"].source;
uv_ofs = p.sources["TEXCOORD0"].offset;
ERR_FAIL_COND_V( !meshdata.sources.has(uv_source_id),ERR_INVALID_DATA);
uv_src=&meshdata.sources[uv_source_id];
}
const Collada::MeshData::Source *uv2_src=NULL;
int uv2_ofs=0;
if (p.sources.has("TEXCOORD1")) {
String uv2_source_id = p.sources["TEXCOORD1"].source;
uv2_ofs = p.sources["TEXCOORD1"].offset;
ERR_FAIL_COND_V( !meshdata.sources.has(uv2_source_id),ERR_INVALID_DATA);
uv2_src=&meshdata.sources[uv2_source_id];
}
const Collada::MeshData::Source *color_src=NULL;
int color_ofs=0;
if (p.sources.has("COLOR")) {
2014-02-10 02:10:30 +01:00
String color_source_id = p.sources["COLOR"].source;
color_ofs = p.sources["COLOR"].offset;
ERR_FAIL_COND_V( !meshdata.sources.has(color_source_id), ERR_INVALID_DATA );
color_src=&meshdata.sources[color_source_id];
}
//find largest source..
2015-09-04 04:24:55 +02:00
/************************/
/* ADD WEIGHTS IF EXIST */
/************************/
Map<int,Vector<Collada::Vertex::Weight> > pre_weights;
bool has_weights=false;
if (skin_controller) {
const Collada::SkinControllerData::Source *weight_src=NULL;
int weight_ofs=0;
if (skin_controller->weights.sources.has("WEIGHT")) {
String weight_id = skin_controller->weights.sources["WEIGHT"].source;
weight_ofs = skin_controller->weights.sources["WEIGHT"].offset;
if (skin_controller->sources.has(weight_id)) {
weight_src = &skin_controller->sources[weight_id];
}
}
int joint_ofs=0;
if (skin_controller->weights.sources.has("JOINT")) {
joint_ofs = skin_controller->weights.sources["JOINT"].offset;
}
//should be OK, given this was pre-checked.
int index_ofs=0;
int wstride = skin_controller->weights.sources.size();
for(int w_i=0;w_i<skin_controller->weights.sets.size();w_i++) {
int amount = skin_controller->weights.sets[w_i];
Vector<Collada::Vertex::Weight> weights;
for (int a_i=0;a_i<amount;a_i++) {
Collada::Vertex::Weight w;
int read_from = index_ofs+a_i*wstride;
ERR_FAIL_INDEX_V(read_from+wstride-1,skin_controller->weights.indices.size(),ERR_INVALID_DATA);
int weight_index = skin_controller->weights.indices[read_from+weight_ofs];
ERR_FAIL_INDEX_V(weight_index,weight_src->array.size(),ERR_INVALID_DATA);
w.weight = weight_src->array[weight_index];
int bone_index = skin_controller->weights.indices[read_from+joint_ofs];
if (bone_index==-1)
continue; //ignore this weight (refers to bind shape)
ERR_FAIL_INDEX_V(bone_index,bone_remap.size(),ERR_INVALID_DATA);
w.bone_idx=bone_remap[bone_index];
weights.push_back(w);
}
/* FIX WEIGHTS */
weights.sort();
if (weights.size()>4) {
//cap to 4 and make weights add up 1
weights.resize(4);
}
//make sure weights allways add up to 1
float total=0;
for(int i=0;i<weights.size();i++)
total+=weights[i].weight;
if (total)
for(int i=0;i<weights.size();i++)
weights[i].weight/=total;
if (weights.size()==0 || total==0) { //if nothing, add a weight to bone 0
//no weights assigned
Collada::Vertex::Weight w;
w.bone_idx=0;
w.weight=1.0;
weights.clear();
weights.push_back(w);
}
pre_weights[w_i]=weights;
/*
for(Set<int>::Element *E=vertex_map[w_i].front();E;E=E->next()) {
int dst = E->get();
ERR_EXPLAIN("invalid vertex index in array");
ERR_FAIL_INDEX_V(dst,vertex_array.size(),ERR_INVALID_DATA);
vertex_array[dst].weights=weights;
}*/
index_ofs+=wstride*amount;
}
//vertices need to be localized
has_weights=true;
}
2014-02-10 02:10:30 +01:00
Set<Collada::Vertex> vertex_set; //vertex set will be the vertices
List<int> indices_list; //indices will be the indices
2015-09-04 04:24:55 +02:00
//Map<int,Set<int> > vertex_map; //map vertices (for setting skinning/morph)
2014-02-10 02:10:30 +01:00
/**************************/
/* CREATE PRIMITIVE ARRAY */
/**************************/
// The way collada uses indices is more optimal, and friendlier with 3D modelling sofware,
// because it can index everything, not only vertices (similar to how the WII works).
// This is, however, more incompatible with standard video cards, so arrays must be converted.
// Must convert to GL/DX format.
int _prim_ofs=0;
int vertidx=0;
2014-02-10 02:10:30 +01:00
for(int p_i=0;p_i<p.count;p_i++) {
int amount;
if (p.polygons.size()) {
ERR_FAIL_INDEX_V(p_i,p.polygons.size(),ERR_INVALID_DATA);
amount=p.polygons[p_i];
} else {
amount=3; //triangles;
}
//COLLADA_PRINT("amount: "+itos(amount));
int prev2[2]={0,0};
for(int j=0;j<amount;j++) {
int src=_prim_ofs;
//_prim_ofs+=p.sources.size()
ERR_FAIL_INDEX_V(src,p.indices.size(),ERR_INVALID_DATA);
Collada::Vertex vertex;
if (!p_optimize)
vertex.uid=vertidx++;
2014-02-10 02:10:30 +01:00
2015-09-04 04:24:55 +02:00
2014-02-10 02:10:30 +01:00
int vertex_index=p.indices[src+vertex_ofs]; //used for index field (later used by controllers)
int vertex_pos = (vertex_src->stride?vertex_src->stride:3) * vertex_index;
ERR_FAIL_INDEX_V(vertex_pos,vertex_src->array.size(),ERR_INVALID_DATA);
vertex.vertex=Vector3(vertex_src->array[vertex_pos+0],vertex_src->array[vertex_pos+1],vertex_src->array[vertex_pos+2]);
2015-09-04 04:24:55 +02:00
if (pre_weights.has(vertex_index)) {
vertex.weights=pre_weights[vertex_index];
}
2014-02-10 02:10:30 +01:00
if (normal_src) {
int normal_pos = (normal_src->stride?normal_src->stride:3) * p.indices[src+normal_ofs];
ERR_FAIL_INDEX_V(normal_pos,normal_src->array.size(),ERR_INVALID_DATA);
vertex.normal=Vector3(normal_src->array[normal_pos+0],normal_src->array[normal_pos+1],normal_src->array[normal_pos+2]);
vertex.normal=vertex.normal.snapped(0.001);
if (tangent_src && binormal_src) {
int binormal_pos = (binormal_src->stride?binormal_src->stride:3) * p.indices[src+binormal_ofs];
ERR_FAIL_INDEX_V(binormal_pos,binormal_src->array.size(),ERR_INVALID_DATA);
Vector3 binormal =Vector3(binormal_src->array[binormal_pos+0],binormal_src->array[binormal_pos+1],binormal_src->array[binormal_pos+2]);
int tangent_pos = (tangent_src->stride?tangent_src->stride:3) * p.indices[src+tangent_ofs];
ERR_FAIL_INDEX_V(tangent_pos,tangent_src->array.size(),ERR_INVALID_DATA);
Vector3 tangent =Vector3(tangent_src->array[tangent_pos+0],tangent_src->array[tangent_pos+1],tangent_src->array[tangent_pos+2]);
vertex.tangent.normal=tangent;
vertex.tangent.d= vertex.normal.cross(tangent).dot(binormal) > 0 ? 1 : -1;
2014-02-10 02:10:30 +01:00
}
}
if (uv_src) {
int uv_pos = (uv_src->stride?uv_src->stride:2) * p.indices[src+uv_ofs];
ERR_FAIL_INDEX_V(uv_pos,uv_src->array.size(),ERR_INVALID_DATA);
vertex.uv=Vector3(uv_src->array[uv_pos+0],1.0-uv_src->array[uv_pos+1],0);
}
if (uv2_src) {
int uv2_pos = (uv2_src->stride?uv2_src->stride:2) * p.indices[src+uv2_ofs];
ERR_FAIL_INDEX_V(uv2_pos,uv2_src->array.size(),ERR_INVALID_DATA);
vertex.uv2=Vector3(uv2_src->array[uv2_pos+0],1.0-uv2_src->array[uv2_pos+1],0);
}
if (color_src) {
int color_pos = (color_src->stride?color_src->stride:3) * p.indices[src+color_ofs]; // colors are RGB in collada..
ERR_FAIL_INDEX_V(color_pos,color_src->array.size(),ERR_INVALID_DATA);
vertex.color=Color(color_src->array[color_pos+0],color_src->array[color_pos+1],color_src->array[color_pos+2],(color_src->stride>3)?color_src->array[color_pos+3]:1.0);
}
#ifndef NO_UP_AXIS_SWAP
if (collada.state.up_axis==Vector3::AXIS_Z) {
SWAP( vertex.vertex.z, vertex.vertex.y );
vertex.vertex.z = -vertex.vertex.z;
SWAP( vertex.normal.z, vertex.normal.y );
vertex.normal.z = -vertex.normal.z;
SWAP( vertex.tangent.normal.z, vertex.tangent.normal.y );
vertex.tangent.normal.z = -vertex.tangent.normal.z;
2014-02-10 02:10:30 +01:00
}
#endif
vertex.fix_unit_scale(collada);
int index=0;
//COLLADA_PRINT("vertex: "+vertex.vertex);
if (vertex_set.has(vertex)) {
index=vertex_set.find(vertex)->get().idx;
} else {
index=vertex_set.size();
vertex.idx=index;
vertex_set.insert(vertex);
}
2015-09-04 04:24:55 +02:00
/* if (!vertex_map.has(vertex_index))
2014-02-10 02:10:30 +01:00
vertex_map[vertex_index]=Set<int>();
2015-09-04 04:24:55 +02:00
vertex_map[vertex_index].insert(index); //should be outside..*/
2014-02-10 02:10:30 +01:00
//build triangles if needed
if (j==0)
prev2[0]=index;
if (j>=2) {
//insert indices in reverse order (collada uses CCW as frontface)
if (local_xform_mirror) {
indices_list.push_back(prev2[0]);
indices_list.push_back(prev2[1]);
indices_list.push_back(index);
} else {
indices_list.push_back(prev2[0]);
indices_list.push_back(index);
indices_list.push_back(prev2[1]);
}
}
prev2[1]=index;
_prim_ofs+=p.vertex_size;
}
}
Vector<Collada::Vertex> vertex_array; //there we go, vertex array
vertex_array.resize(vertex_set.size());
for(Set<Collada::Vertex>::Element *F=vertex_set.front();F;F=F->next()) {
vertex_array[F->get().idx]=F->get();
}
2015-09-04 04:24:55 +02:00
if (has_weights) {
2014-02-10 02:10:30 +01:00
2015-09-04 04:24:55 +02:00
//if skeleton, localize
2014-02-10 02:10:30 +01:00
Transform local_xform = p_local_xform;
for(int i=0;i<vertex_array.size();i++) {
vertex_array[i].vertex=local_xform.xform(vertex_array[i].vertex);
vertex_array[i].normal=local_xform.basis.xform(vertex_array[i].normal).normalized();
vertex_array[i].tangent.normal=local_xform.basis.xform(vertex_array[i].tangent.normal).normalized();
if (local_xform_mirror) {
//i shouldn't do this? wtf?
//vertex_array[i].normal*=-1.0;
//vertex_array[i].tangent.normal*=-1.0;
}
}
}
2015-09-04 04:24:55 +02:00
2014-02-10 02:10:30 +01:00
DVector<int> index_array;
index_array.resize(indices_list.size());
DVector<int>::Write index_arrayw = index_array.write();
int iidx=0;
for(List<int>::Element *F=indices_list.front();F;F=F->next()) {
index_arrayw[iidx++]=F->get();
}
index_arrayw=DVector<int>::Write();
/*****************/
/* MAKE SURFACES */
/*****************/
{
Ref<FixedSpatialMaterial> material;
2014-02-10 02:10:30 +01:00
//find material
Mesh::PrimitiveType primitive=Mesh::PRIMITIVE_TRIANGLES;
{
if (p_material_map.has(p.material)) {
String target=p_material_map[p.material].target;
if (!material_cache.has(target)) {
Error err = _create_material(target);
if (!err)
material=material_cache[target];
} else
material=material_cache[target];
} else if (p.material!=""){
print_line("Warning, unreferenced material in geometry instance: "+p.material);
}
}
DVector<Vector3> final_vertex_array;
DVector<Vector3> final_normal_array;
DVector<float> final_tangent_array;
DVector<Color> final_color_array;
DVector<Vector3> final_uv_array;
DVector<Vector3> final_uv2_array;
DVector<int> final_bone_array;
2014-02-10 02:10:30 +01:00
DVector<float> final_weight_array;
uint32_t final_format=0;
//create format
final_format=Mesh::ARRAY_FORMAT_VERTEX|Mesh::ARRAY_FORMAT_INDEX;
if (normal_src) {
final_format|=Mesh::ARRAY_FORMAT_NORMAL;
if (uv_src && binormal_src && tangent_src) {
final_format|=Mesh::ARRAY_FORMAT_TANGENT;
}
}
if (color_src)
final_format|=Mesh::ARRAY_FORMAT_COLOR;
if (uv_src)
final_format|=Mesh::ARRAY_FORMAT_TEX_UV;
if (uv2_src)
final_format|=Mesh::ARRAY_FORMAT_TEX_UV2;
if (has_weights) {
final_format|=Mesh::ARRAY_FORMAT_WEIGHTS;
final_format|=Mesh::ARRAY_FORMAT_BONES;
}
//set arrays
int vlen = vertex_array.size();
{ //vertices
DVector<Vector3> varray;
varray.resize(vertex_array.size());
DVector<Vector3>::Write varrayw = varray.write();
for(int k=0;k<vlen;k++)
varrayw[k]=vertex_array[k].vertex;
varrayw = DVector<Vector3>::Write();
final_vertex_array=varray;
}
if (uv_src) { //compute uv first, may be needed for computing tangent/bionrmal
DVector<Vector3> uvarray;
uvarray.resize(vertex_array.size());
DVector<Vector3>::Write uvarrayw = uvarray.write();
for(int k=0;k<vlen;k++) {
uvarrayw[k]=vertex_array[k].uv;
}
uvarrayw = DVector<Vector3>::Write();
final_uv_array=uvarray;
}
if (uv2_src) { //compute uv first, may be needed for computing tangent/bionrmal
DVector<Vector3> uv2array;
uv2array.resize(vertex_array.size());
DVector<Vector3>::Write uv2arrayw = uv2array.write();
for(int k=0;k<vlen;k++) {
uv2arrayw[k]=vertex_array[k].uv2;
2014-02-10 02:10:30 +01:00
}
uv2arrayw = DVector<Vector3>::Write();
final_uv2_array=uv2array;
}
if (normal_src) {
DVector<Vector3> narray;
narray.resize(vertex_array.size());
DVector<Vector3>::Write narrayw = narray.write();
for(int k=0;k<vlen;k++) {
narrayw[k]=vertex_array[k].normal;
}
narrayw = DVector<Vector3>::Write();
final_normal_array=narray;
//DVector<Vector3> altnaray;
//_generate_normals(index_array,final_vertex_array,altnaray);
//for(int i=0;i<altnaray.size();i++)
// print_line(rtos(altnaray[i].dot(final_normal_array[i])));
} else if (primitive==Mesh::PRIMITIVE_TRIANGLES) {
//generate normals (even if unused later)
_generate_normals(index_array,final_vertex_array,final_normal_array);
if (OS::get_singleton()->is_stdout_verbose())
print_line("Collada: Triangle mesh lacks normals, so normals were generated.");
final_format|=Mesh::ARRAY_FORMAT_NORMAL;
}
if (final_normal_array.size() && uv_src && binormal_src && tangent_src && !force_make_tangents) {
DVector<real_t> tarray;
tarray.resize(vertex_array.size()*4);
DVector<real_t>::Write tarrayw = tarray.write();
for(int k=0;k<vlen;k++) {
tarrayw[k*4+0]=vertex_array[k].tangent.normal.x;
tarrayw[k*4+1]=vertex_array[k].tangent.normal.y;
tarrayw[k*4+2]=vertex_array[k].tangent.normal.z;
tarrayw[k*4+3]=vertex_array[k].tangent.d;
}
tarrayw = DVector<real_t>::Write();
final_tangent_array=tarray;
} else if (final_normal_array.size() && primitive==Mesh::PRIMITIVE_TRIANGLES && final_uv_array.size() && (force_make_tangents || (material.is_valid()))){
2014-02-10 02:10:30 +01:00
//if this uses triangles, there are uvs and the material is using a normalmap, generate tangents and binormals, because they WILL be needed
//generate binormals/tangents
_generate_tangents_and_binormals(index_array,final_vertex_array,final_uv_array,final_normal_array,final_tangent_array);
final_format|=Mesh::ARRAY_FORMAT_TANGENT;
if (OS::get_singleton()->is_stdout_verbose())
print_line("Collada: Triangle mesh lacks tangents (And normalmap was used), so tangents were generated.");
}
if (color_src) {
DVector<Color> colorarray;
colorarray.resize(vertex_array.size());
DVector<Color>::Write colorarrayw = colorarray.write();
for(int k=0;k<vlen;k++) {
colorarrayw[k]=vertex_array[k].color;
}
colorarrayw = DVector<Color>::Write();
final_color_array=colorarray;
}
if (has_weights) {
DVector<float> weightarray;
DVector<int> bonearray;
2014-02-10 02:10:30 +01:00
weightarray.resize(vertex_array.size()*4);
DVector<float>::Write weightarrayw = weightarray.write();
bonearray.resize(vertex_array.size()*4);
DVector<int>::Write bonearrayw = bonearray.write();
2014-02-10 02:10:30 +01:00
for(int k=0;k<vlen;k++) {
float sum=0;
for(int l=0;l<VS::ARRAY_WEIGHTS_SIZE;l++) {
if (l<vertex_array[k].weights.size()) {
weightarrayw[k*VS::ARRAY_WEIGHTS_SIZE+l]=vertex_array[k].weights[l].weight;
sum+=weightarrayw[k*VS::ARRAY_WEIGHTS_SIZE+l];
bonearrayw[k*VS::ARRAY_WEIGHTS_SIZE+l]=int(vertex_array[k].weights[l].bone_idx);
2014-02-10 02:10:30 +01:00
//COLLADA_PRINT(itos(k)+": "+rtos(bonearrayw[k*VS::ARRAY_WEIGHTS_SIZE+l])+":"+rtos(weightarray[k*VS::ARRAY_WEIGHTS_SIZE+l]));
} else {
weightarrayw[k*VS::ARRAY_WEIGHTS_SIZE+l]=0;
bonearrayw[k*VS::ARRAY_WEIGHTS_SIZE+l]=0;
}
}
// if (sum<0.8)
// COLLADA_PRINT("ERROR SUMMING INDEX "+itos(k)+" had weights: "+itos(vertex_array[k].weights.size()));
}
weightarrayw = DVector<float>::Write();
bonearrayw = DVector<int>::Write();
2014-02-10 02:10:30 +01:00
final_weight_array = weightarray;
final_bone_array = bonearray;
}
////////////////////////////
// FINALLY CREATE SUFRACE //
////////////////////////////
Array d;
d.resize(VS::ARRAY_MAX);
d[Mesh::ARRAY_INDEX]=index_array;
d[Mesh::ARRAY_VERTEX]=final_vertex_array;
if (final_normal_array.size())
d[Mesh::ARRAY_NORMAL]=final_normal_array;
if (final_tangent_array.size())
d[Mesh::ARRAY_TANGENT]=final_tangent_array;
if (final_uv_array.size())
d[Mesh::ARRAY_TEX_UV]=final_uv_array;
if (final_uv2_array.size())
d[Mesh::ARRAY_TEX_UV2]=final_uv2_array;
if (final_color_array.size())
d[Mesh::ARRAY_COLOR]=final_color_array;
if (final_weight_array.size())
d[Mesh::ARRAY_WEIGHTS]=final_weight_array;
if (final_bone_array.size())
d[Mesh::ARRAY_BONES]=final_bone_array;
Array mr;
////////////////////////////
// THEN THE MORPH TARGETS //
////////////////////////////
#if 0
2014-02-10 02:10:30 +01:00
if (p_morph_data) {
//add morphie target
ERR_FAIL_COND_V( !p_morph_data->targets.has("MORPH_TARGET"), ERR_INVALID_DATA );
String mt = p_morph_data->targets["MORPH_TARGET"];
ERR_FAIL_COND_V( !p_morph_data->sources.has(mt), ERR_INVALID_DATA);
int morph_targets = p_morph_data->sources[mt].sarray.size();
mr.resize(morph_targets);
for(int j=0;j<morph_targets;j++) {
Array mrt;
mrt.resize(VS::ARRAY_MAX);
String target = p_morph_data->sources[mt].sarray[j];
ERR_FAIL_COND_V( !collada.state.mesh_data_map.has(target), ERR_INVALID_DATA );
String name = collada.state.mesh_data_map[target].name;
Collada::MeshData &md = collada.state.mesh_data_map[target];
// collada in itself supports morphing everything. However, the spec is unclear and no examples or exporters that
// morph anything but "POSITIONS" seem to exit. Because of this, normals and binormals/tangents have to be regenerated here,
// which may result in inaccurate (but most of the time good enough) results.
DVector<Vector3> vertices;
vertices.resize(vlen);
ERR_FAIL_COND_V( md.vertices.size() != 1, ERR_INVALID_DATA);
String vertex_src_id=md.vertices.front()->key();
ERR_FAIL_COND_V(!md.vertices[vertex_src_id].sources.has("POSITION"),ERR_INVALID_DATA);
String position_src_id = md.vertices[vertex_src_id].sources["POSITION"];
ERR_FAIL_COND_V(!md.sources.has(position_src_id),ERR_INVALID_DATA);
const Collada::MeshData::Source *m=&md.sources[position_src_id];
ERR_FAIL_COND_V( m->array.size() != vertex_src->array.size(), ERR_INVALID_DATA);
int stride=m->stride;
if (stride==0)
stride=3;
//read vertices from morph target
DVector<Vector3>::Write vertw = vertices.write();
for(int m_i=0;m_i<m->array.size()/stride;m_i++) {
int pos = m_i*stride;
Vector3 vtx( m->array[pos+0], m->array[pos+1], m->array[pos+2] );
#ifndef NO_UP_AXIS_SWAP
if (collada.state.up_axis==Vector3::AXIS_Z) {
SWAP( vtx.z, vtx.y );
vtx.z = -vtx.z;
}
#endif
Collada::Vertex vertex;
vertex.vertex=vtx;
vertex.fix_unit_scale(collada);
vtx=vertex.vertex;
vtx = p_local_xform.xform(vtx);
if (vertex_map.has(m_i)) { //vertex may no longer be here, don't bother converting
for (Set<int> ::Element *E=vertex_map[m_i].front() ; E; E=E->next() ) {
vertw[E->get()]=vtx;
}
}
}
//vertices are in place, now generate everything else
vertw = DVector<Vector3>::Write();
DVector<Vector3> normals;
DVector<float> tangents;
print_line("vertex source id: "+vertex_src_id);
if(md.vertices[vertex_src_id].sources.has("NORMAL")){
2016-03-09 00:00:52 +01:00
//has normals
normals.resize(vlen);
//std::cout << "has normals" << std::endl;
String normal_src_id = md.vertices[vertex_src_id].sources["NORMAL"];
//std::cout << "normals source: "<< normal_src_id.utf8().get_data() <<std::endl;
ERR_FAIL_COND_V(!md.sources.has(normal_src_id),ERR_INVALID_DATA);
2014-02-10 02:10:30 +01:00
const Collada::MeshData::Source *m=&md.sources[normal_src_id];
ERR_FAIL_COND_V( m->array.size() != vertex_src->array.size(), ERR_INVALID_DATA);
int stride=m->stride;
if (stride==0)
stride=3;
2016-03-09 00:00:52 +01:00
//read normals from morph target
DVector<Vector3>::Write vertw = normals.write();
for(int m_i=0;m_i<m->array.size()/stride;m_i++) {
int pos = m_i*stride;
Vector3 vtx( m->array[pos+0], m->array[pos+1], m->array[pos+2] );
#ifndef NO_UP_AXIS_SWAP
if (collada.state.up_axis==Vector3::AXIS_Z) {
SWAP( vtx.z, vtx.y );
vtx.z = -vtx.z;
}
#endif
Collada::Vertex vertex;
vertex.vertex=vtx;
vertex.fix_unit_scale(collada);
vtx=vertex.vertex;
vtx = p_local_xform.xform(vtx);
if (vertex_map.has(m_i)) { //vertex may no longer be here, don't bother converting
for (Set<int> ::Element *E=vertex_map[m_i].front() ; E; E=E->next() ) {
vertw[E->get()]=vtx;
}
}
}
2016-03-09 00:00:52 +01:00
print_line("using built-in normals");
}else{
print_line("generating normals");
_generate_normals(index_array,vertices,normals);//no normals
}
2014-02-10 02:10:30 +01:00
if (final_tangent_array.size() && final_uv_array.size()) {
_generate_tangents_and_binormals(index_array,vertices,final_uv_array,normals,tangents);
}
mrt[Mesh::ARRAY_VERTEX]=vertices;
mrt[Mesh::ARRAY_NORMAL]=normals;
if (tangents.size())
mrt[Mesh::ARRAY_TANGENT]=tangents;
if (final_uv_array.size())
mrt[Mesh::ARRAY_TEX_UV]=final_uv_array;
if (final_uv2_array.size())
mrt[Mesh::ARRAY_TEX_UV2]=final_uv2_array;
if (final_color_array.size())
mrt[Mesh::ARRAY_COLOR]=final_color_array;
mr[j]=mrt;
}
}
#endif
for(int mi=0;mi<p_morph_meshes.size();mi++) {
2015-04-24 20:09:35 +02:00
// print_line("want surface "+itos(mi)+" has "+itos(p_morph_meshes[mi]->get_surface_count()));
Array a = p_morph_meshes[mi]->surface_get_arrays(surface);
//add valid weight and bone arrays if they exist, TODO check if they are unique to shape (generally not)
if (final_weight_array.size())
a[Mesh::ARRAY_WEIGHTS]=final_weight_array;
if (final_bone_array.size())
a[Mesh::ARRAY_BONES]=final_bone_array;
a[Mesh::ARRAY_INDEX]=Variant();
//a.resize(Mesh::ARRAY_MAX); //no need for index
mr.push_back(a);
}
2014-02-10 02:10:30 +01:00
p_mesh->add_surface_from_arrays(Mesh::PRIMITIVE_TRIANGLES,d,mr,p_for_morph?0:Mesh::ARRAY_COMPRESS_DEFAULT);
2014-02-10 02:10:30 +01:00
if (material.is_valid()) {
p_mesh->surface_set_material(surface, material);
p_mesh->surface_set_name(surface, material->get_name());
}
}
/*****************/
/* FIND MATERIAL */
/*****************/
surface++;
}
return OK;
}
Error ColladaImport::_create_resources(Collada::Node *p_node) {
if (p_node->type==Collada::Node::TYPE_GEOMETRY && node_map.has(p_node->id)) {
Spatial * node=node_map[p_node->id].node;
Collada::NodeGeometry *ng = static_cast<Collada::NodeGeometry*>(p_node);
if (node->cast_to<Path>()) {
Path *path = node->cast_to<Path>();
String curve = ng->source;
if (curve_cache.has(ng->source)) {
path->set_curve(curve_cache[ng->source]);
} else {
Ref<Curve3D> c = memnew( Curve3D );
const Collada::CurveData &cd = collada.state.curve_data_map[ng->source];
ERR_FAIL_COND_V( !cd.control_vertices.has("POSITION") , ERR_INVALID_DATA);
ERR_FAIL_COND_V( !cd.control_vertices.has("IN_TANGENT") , ERR_INVALID_DATA);
ERR_FAIL_COND_V( !cd.control_vertices.has("OUT_TANGENT") , ERR_INVALID_DATA);
ERR_FAIL_COND_V( !cd.control_vertices.has("INTERPOLATION") , ERR_INVALID_DATA);
ERR_FAIL_COND_V( !cd.sources.has(cd.control_vertices["POSITION"] ) , ERR_INVALID_DATA);
const Collada::CurveData::Source &vertices = cd.sources[ cd.control_vertices["POSITION"] ];
ERR_FAIL_COND_V( vertices.stride!=3, ERR_INVALID_DATA );
ERR_FAIL_COND_V( !cd.sources.has(cd.control_vertices["IN_TANGENT"] ) , ERR_INVALID_DATA);
const Collada::CurveData::Source &in_tangents = cd.sources[ cd.control_vertices["IN_TANGENT"] ];
ERR_FAIL_COND_V( in_tangents.stride!=3 , ERR_INVALID_DATA);
ERR_FAIL_COND_V( !cd.sources.has(cd.control_vertices["OUT_TANGENT"] ), ERR_INVALID_DATA );
const Collada::CurveData::Source &out_tangents = cd.sources[ cd.control_vertices["OUT_TANGENT"] ];
ERR_FAIL_COND_V( out_tangents.stride!=3, ERR_INVALID_DATA );
ERR_FAIL_COND_V( !cd.sources.has(cd.control_vertices["INTERPOLATION"] ), ERR_INVALID_DATA );
const Collada::CurveData::Source &interps = cd.sources[ cd.control_vertices["INTERPOLATION"] ];
ERR_FAIL_COND_V( interps.stride!=1, ERR_INVALID_DATA );
const Collada::CurveData::Source *tilts=NULL;
if (cd.control_vertices.has("TILT") && cd.sources.has(cd.control_vertices["TILT"]))
tilts=&cd.sources[ cd.control_vertices["TILT"] ];
if (tilts) {
print_line("FOUND TILTS!!!");
}
int pc = vertices.array.size()/3;
for(int i=0;i<pc;i++) {
Vector3 pos( vertices.array[i*3+0], vertices.array[i*3+1], vertices.array[i*3+2] );
Vector3 in( in_tangents.array[i*3+0], in_tangents.array[i*3+1], in_tangents.array[i*3+2] );
Vector3 out( out_tangents.array[i*3+0], out_tangents.array[i*3+1], out_tangents.array[i*3+2] );
#ifndef NO_UP_AXIS_SWAP
if (collada.state.up_axis==Vector3::AXIS_Z) {
SWAP(pos.y,pos.z);
pos.z=-pos.z;
SWAP(in.y,in.z);
in.z=-in.z;
SWAP(out.y,out.z);
out.z=-out.z;
}
#endif
pos*=collada.state.unit_scale;
in*=collada.state.unit_scale;
out*=collada.state.unit_scale;
c->add_point(pos,in-pos,out-pos);
if (tilts)
c->set_point_tilt(i,tilts->array[i]);
}
curve_cache[ng->source]=c;
path->set_curve(c);
}
}
if (node->cast_to<MeshInstance>()) {
Collada::NodeGeometry *ng = static_cast<Collada::NodeGeometry*>(p_node);
MeshInstance *mi = node->cast_to<MeshInstance>();
ERR_FAIL_COND_V(!mi,ERR_BUG);
Collada::SkinControllerData *skin=NULL;
Collada::MorphControllerData *morph=NULL;
String meshid;
Transform apply_xform;
Vector<int> bone_remap;
Vector<Ref<Mesh> > morphs;
2014-02-10 02:10:30 +01:00
print_line("mesh: "+String(mi->get_name()));
2014-02-10 02:10:30 +01:00
if (ng->controller) {
print_line("has controller");
String ngsource = ng->source;
if (collada.state.skin_controller_data_map.has(ngsource)) {
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND_V(!collada.state.skin_controller_data_map.has(ngsource),ERR_INVALID_DATA);
skin=&collada.state.skin_controller_data_map[ngsource];
2014-02-10 02:10:30 +01:00
Vector<String> skeletons = ng->skeletons;
ERR_FAIL_COND_V( skeletons.empty(), ERR_INVALID_DATA );
String skname = skeletons[0];
if (!node_map.has(skname)) {
print_line("no node for skeleton "+skname);
}
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND_V( !node_map.has(skname), ERR_INVALID_DATA );
NodeMap nmsk = node_map[skname];
Skeleton *sk = nmsk.node->cast_to<Skeleton>();
ERR_FAIL_COND_V( !sk, ERR_INVALID_DATA );
ERR_FAIL_COND_V( !skeleton_bone_map.has(sk), ERR_INVALID_DATA );
Map<String, int> &bone_remap_map=skeleton_bone_map[sk];
meshid=skin->base;
if (collada.state.morph_controller_data_map.has(meshid)) {
//it's a morph!!
2015-04-24 03:42:52 +02:00
morph = &collada.state.morph_controller_data_map[meshid];
ngsource=meshid;
2014-02-10 02:10:30 +01:00
meshid=morph->mesh;
} else {
ngsource="";
2014-02-10 02:10:30 +01:00
}
if (apply_mesh_xform_to_vertices) {
apply_xform=collada.fix_transform(p_node->default_transform);
node->set_transform(Transform());
} else {
apply_xform=Transform();
}
2014-02-10 02:10:30 +01:00
Collada::SkinControllerData::Source *joint_src=NULL;
ERR_FAIL_COND_V(!skin->weights.sources.has("JOINT"),ERR_INVALID_DATA);
String joint_id = skin->weights.sources["JOINT"].source;
ERR_FAIL_COND_V(!skin->sources.has(joint_id),ERR_INVALID_DATA);
joint_src = &skin->sources[joint_id];
bone_remap.resize(joint_src->sarray.size());
for(int i=0;i<bone_remap.size();i++) {
String str = joint_src->sarray[i];
if (!bone_remap_map.has(str)) {
print_line("bone not found for remap: "+str);
print_line("in skeleton: "+skname);
}
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND_V( !bone_remap_map.has(str), ERR_INVALID_DATA );
bone_remap[i]=bone_remap_map[str];
}
}
if (collada.state.morph_controller_data_map.has(ngsource)) {
print_line("is morph "+ngsource);
2014-02-10 02:10:30 +01:00
//it's a morph!!
morph = &collada.state.morph_controller_data_map[ngsource];
2014-02-10 02:10:30 +01:00
meshid=morph->mesh;
printf("KKmorph: %p\n",morph);
print_line("morph mshid: "+meshid);
Vector<String> targets;
morph->targets.has("MORPH_TARGET");
String target = morph->targets["MORPH_TARGET"];
bool valid=false;
if (morph->sources.has(target)) {
valid=true;
Vector<String> names = morph->sources[target].sarray;
for(int i=0;i<names.size();i++) {
String meshid=names[i];
if (collada.state.mesh_data_map.has(meshid)) {
Ref<Mesh> mesh=Ref<Mesh>(memnew( Mesh ));
const Collada::MeshData &meshdata = collada.state.mesh_data_map[meshid];
Error err = _create_mesh_surfaces(false,mesh,ng->material_map,meshdata,apply_xform,bone_remap,skin,NULL,Vector<Ref<Mesh> >(),true);
ERR_FAIL_COND_V(err,err);
morphs.push_back(mesh);
} else {
valid=false;
}
}
}
if (!valid)
morphs.clear();
ngsource="";
}
if (ngsource!=""){
ERR_EXPLAIN("Controller Instance Source '"+ngsource+"' is neither skin or morph!");
2014-02-10 02:10:30 +01:00
ERR_FAIL_V( ERR_INVALID_DATA );
}
} else {
meshid=ng->source;
}
Ref<Mesh> mesh;
if (mesh_cache.has(meshid)) {
mesh=mesh_cache[meshid];
} else {
if (collada.state.mesh_data_map.has(meshid)) {
//bleh, must ignore invalid
ERR_FAIL_COND_V(!collada.state.mesh_data_map.has(meshid),ERR_INVALID_DATA);
mesh=Ref<Mesh>(memnew( Mesh ));
const Collada::MeshData &meshdata = collada.state.mesh_data_map[meshid];
mesh->set_name( meshdata.name );
Error err = _create_mesh_surfaces(morphs.size()==0,mesh,ng->material_map,meshdata,apply_xform,bone_remap,skin,morph,morphs);
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND_V(err,err);
mesh_cache[meshid]=mesh;
} else {
print_line("Warning, will not import geometry: "+meshid);
}
}
if (!mesh.is_null()) {
mi->set_mesh(mesh);
}
}
}
for(int i=0;i<p_node->children.size();i++) {
Error err = _create_resources(p_node->children[i]);
if (err)
return err;
}
return OK;
}
Error ColladaImport::load(const String& p_path,int p_flags,bool p_force_make_tangents) {
Error err = collada.load(p_path,p_flags);
ERR_FAIL_COND_V(err,err);
force_make_tangents=p_force_make_tangents;
ERR_FAIL_COND_V( !collada.state.visual_scene_map.has( collada.state.root_visual_scene ), ERR_INVALID_DATA );
Collada::VisualScene &vs = collada.state.visual_scene_map[ collada.state.root_visual_scene ];
scene = memnew( Spatial ); // root
//determine what's going on with the lights
for(int i=0;i<vs.root_nodes.size();i++) {
_pre_process_lights(vs.root_nodes[i]);
}
//import scene
for(int i=0;i<vs.root_nodes.size();i++) {
Error err = _create_scene_skeletons(vs.root_nodes[i]);
if (err!=OK) {
memdelete(scene);
ERR_FAIL_COND_V(err,err);
}
}
2014-02-10 02:10:30 +01:00
for(int i=0;i<vs.root_nodes.size();i++) {
Error err = _create_scene(vs.root_nodes[i],scene);
if (err!=OK) {
memdelete(scene);
ERR_FAIL_COND_V(err,err);
}
Error err2 = _create_resources(vs.root_nodes[i]);
if (err2!=OK) {
memdelete(scene);
ERR_FAIL_COND_V(err2,err2);
}
}
//optatively, set unit scale in the root
scene->set_transform(collada.get_root_transform());
return OK;
}
void ColladaImport::_fix_param_animation_tracks() {
for (Map<String,Collada::Node*>::Element *E=collada.state.scene_map.front();E;E=E->next()) {
Collada::Node *n = E->get();
switch(n->type) {
case Collada::Node::TYPE_NODE: {
// ? do nothing
} break;
case Collada::Node::TYPE_JOINT: {
} break;
case Collada::Node::TYPE_SKELETON: {
} break;
case Collada::Node::TYPE_LIGHT: {
} break;
case Collada::Node::TYPE_CAMERA: {
} break;
case Collada::Node::TYPE_GEOMETRY: {
Collada::NodeGeometry *ng = static_cast<Collada::NodeGeometry*>(n);
// test source(s)
String source = ng->source;
while (source!="") {
if (collada.state.skin_controller_data_map.has(source)) {
const Collada::SkinControllerData& skin = collada.state.skin_controller_data_map[source];
//nothing to animate here i think
source=skin.base;
} else if (collada.state.morph_controller_data_map.has(source)) {
2014-02-10 02:10:30 +01:00
const Collada::MorphControllerData& morph = collada.state.morph_controller_data_map[source];
2014-02-10 02:10:30 +01:00
if (morph.targets.has("MORPH_WEIGHT") && morph.targets.has("MORPH_TARGET")) {
2014-02-10 02:10:30 +01:00
String weights = morph.targets["MORPH_WEIGHT"];
String targets = morph.targets["MORPH_TARGET"];
//fails here
2014-02-10 02:10:30 +01:00
if (morph.sources.has(targets) && morph.sources.has(weights)) {
const Collada::MorphControllerData::Source &weight_src=morph.sources[weights];
const Collada::MorphControllerData::Source &target_src=morph.sources[targets];
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND(weight_src.array.size() != target_src.sarray.size());
for(int i=0;i<weight_src.array.size();i++) {
String track_name = weights+"("+itos(i)+")";
String mesh_name = target_src.sarray[i];
if (collada.state.mesh_name_map.has(mesh_name) && collada.state.referenced_tracks.has(track_name)) {
2014-02-10 02:10:30 +01:00
const Vector<int>&rt = collada.state.referenced_tracks[track_name];
for(int rti=0;rti<rt.size();rti++) {
Collada::AnimationTrack *at = &collada.state.animation_tracks[rt[rti]];
at->target=E->key();
at->param="morph/"+collada.state.mesh_name_map[mesh_name];
at->property=true;
//at->param
}
}
}
}
}
source=morph.mesh;
} else {
source=""; // for now nothing else supported
}
}
} break;
}
}
}
void ColladaImport::create_animations(bool p_make_tracks_in_all_bones) {
2014-02-10 02:10:30 +01:00
2014-02-10 02:10:30 +01:00
_fix_param_animation_tracks();
for(int i=0;i<collada.state.animation_clips.size();i++) {
for(int j=0;j<collada.state.animation_clips[i].tracks.size();j++)
tracks_in_clips.insert(collada.state.animation_clips[i].tracks[j]);
}
for(int i=0;i<collada.state.animation_tracks.size();i++) {
Collada::AnimationTrack &at = collada.state.animation_tracks[i];
//print_line("CHANNEL: "+at.target+" PARAM: "+at.param);
String node;
2014-02-10 02:10:30 +01:00
if (!node_map.has(at.target)) {
if (node_name_map.has(at.target)) {
node=node_name_map[at.target];
} else {
print_line("Coudlnt find node: "+at.target);
continue;
}
} else {
node=at.target;
2014-02-10 02:10:30 +01:00
}
if (at.property) {
valid_animated_properties.push_back(i);
} else {
node_map[node].anim_tracks.push_back(i);
valid_animated_nodes.insert(node);
2014-02-10 02:10:30 +01:00
}
}
create_animation(-1,p_make_tracks_in_all_bones);
//print_line("clipcount: "+itos(collada.state.animation_clips.size()));
2014-02-10 02:10:30 +01:00
for(int i=0;i<collada.state.animation_clips.size();i++)
create_animation(i,p_make_tracks_in_all_bones);
2014-02-10 02:10:30 +01:00
}
void ColladaImport::create_animation(int p_clip, bool p_make_tracks_in_all_bones) {
2014-02-10 02:10:30 +01:00
Ref<Animation> animation = Ref<Animation>( memnew( Animation ));
2014-02-10 02:10:30 +01:00
if (p_clip==-1) {
//print_line("default");
2014-02-10 02:10:30 +01:00
animation->set_name("default");
} else {
//print_line("clip name: "+collada.state.animation_clips[p_clip].name);
2014-02-10 02:10:30 +01:00
animation->set_name(collada.state.animation_clips[p_clip].name);
}
for(Map<String,NodeMap>::Element *E=node_map.front();E;E=E->next()) {
if (E->get().bone<0)
continue;
bones_with_animation[E->key()]=false;
}
//store and validate tracks
if (p_clip==-1) {
//main anim
}
Set<int> track_filter;
if (p_clip==-1) {
for(int i=0;i<collada.state.animation_clips.size();i++) {
int tc = collada.state.animation_clips[i].tracks.size();
for(int j=0;j<tc;j++) {
String n = collada.state.animation_clips[i].tracks[j];
if (collada.state.by_id_tracks.has(n)) {
const Vector<int>&ti = collada.state.by_id_tracks[n];
for(int k=0;k<ti.size();k++) {
track_filter.insert(ti[k]);
}
}
}
}
} else {
int tc = collada.state.animation_clips[p_clip].tracks.size();
for(int j=0;j<tc;j++) {
String n = collada.state.animation_clips[p_clip].tracks[j];
if (collada.state.by_id_tracks.has(n)) {
const Vector<int>&ti = collada.state.by_id_tracks[n];
for(int k=0;k<ti.size();k++) {
track_filter.insert(ti[k]);
}
}
}
}
//animation->set_loop(true);
//create animation tracks
Vector<float> base_snapshots;
float f=0;
float snapshot_interval = 1.0/bake_fps; //should be customizable somewhere...
2014-02-10 02:10:30 +01:00
float anim_length=collada.state.animation_length;
if (p_clip>=0 && collada.state.animation_clips[p_clip].end)
anim_length=collada.state.animation_clips[p_clip].end;
Huge Amount of BugFix -=-=-=-=-=-=-=-=-=-=- -Fixes to Collada Exporter (avoid crash situtions) -Fixed to Collada Importer (Fixed Animation Optimizer Bugs) -Fixes to RigidBody/RigidBody2D body_enter/body_exit, was buggy -Fixed ability for RigidBody/RigidBody2D to get contacts reported and bodyin/out in Kinematic mode. -Added proper trigger support for 3D Physics shapes -Changed proper value for Z-Offset in OmniLight -Fixed spot attenuation bug in SpotLight -Fixed some 3D and 2D spatial soudn bugs related to distance attenuation. -Fixed bugs in EventPlayer (channels were muted by default) -Fix in ButtonGroup (get nodes in group are now returned in order) -Fixed Linear->SRGB Conversion, previous algo sucked, new algo works OK -Changed SRGB->Linear conversion to use hardware if supported, improves texture quality a lot -Fixed options for Y-Fov and X-Fov in camera, should be more intuitive. -Fixed bugs related to viewports and transparency Huge Amount of New Stuff: -=-=-=-=-=-=-=-==-=-=-=- -Ability to manually advance an AnimationPlayer that is inactive (with advance() function) -More work in WinRT platform -Added XY normalmap support, imports on this format by default. Reduces normlmap size and enables much nice compression using LATC -Added Anisotropic filter support to textures, can be specified on import -Added support for Non-Square, Isometric and Hexagonal tilemaps in TileMap. -Added Isometric Dungeon demo. -Added simple hexagonal map demo. -Added Truck-Town demo. Shows how most types of joints and vehicles are used. Please somebody make a nicer town, this one is too hardcore. -Added an Object-Picking API to both RigidBody and Area! (and relevant demo)
2014-10-03 05:10:51 +02:00
while(f<anim_length) {
2014-02-10 02:10:30 +01:00
base_snapshots.push_back(f);
2014-02-10 02:10:30 +01:00
f+=snapshot_interval;
if (f>=anim_length) {
base_snapshots.push_back(anim_length);
}
2014-02-10 02:10:30 +01:00
}
//print_line("anim len: "+rtos(anim_length));
2014-02-10 02:10:30 +01:00
animation->set_length(anim_length);
bool tracks_found=false;
2014-02-10 02:10:30 +01:00
for(Set<String>::Element* E=valid_animated_nodes.front();E;E=E->next()) {
// take snapshots
if (!collada.state.scene_map.has(E->get())) {
2014-02-10 02:10:30 +01:00
continue;
}
2014-02-10 02:10:30 +01:00
NodeMap &nm = node_map[E->get()];
String path = scene->get_path_to(nm.node);
if (nm.bone>=0) {
Skeleton *sk = static_cast<Skeleton*>(nm.node);
String name = sk->get_bone_name(nm.bone);
path=path+":"+name;
}
bool found_anim=false;
Collada::Node *cn = collada.state.scene_map[E->get()];
if (cn->ignore_anim) {
2014-02-10 02:10:30 +01:00
continue;
}
animation->add_track(Animation::TYPE_TRANSFORM);
2016-03-09 00:00:52 +01:00
int track = animation->get_track_count() -1;
2014-02-10 02:10:30 +01:00
animation->track_set_path( track , path );
animation->track_set_imported( track , true ); //helps merging later
2014-02-10 02:10:30 +01:00
Vector<float> snapshots = base_snapshots;
if (nm.anim_tracks.size()==1) {
//use snapshot keys from anim track instead, because this was most likely exported baked
Collada::AnimationTrack &at = collada.state.animation_tracks[nm.anim_tracks.front()->get()];
snapshots.clear();
for(int i=0;i<at.keys.size();i++)
snapshots.push_back(at.keys[i].time);
2014-02-10 02:10:30 +01:00
}
for(int i=0;i<snapshots.size();i++) {
2014-02-10 02:10:30 +01:00
for(List<int>::Element *ET=nm.anim_tracks.front();ET;ET=ET->next()) {
//apply tracks
2014-02-10 02:10:30 +01:00
if (p_clip==-1) {
if (track_filter.has(ET->get())) {
2014-02-10 02:10:30 +01:00
continue;
}
2014-02-10 02:10:30 +01:00
} else {
if (!track_filter.has(ET->get()))
continue;
}
found_anim=true;
Collada::AnimationTrack &at = collada.state.animation_tracks[ET->get()];
int xform_idx=-1;
for(int j=0;j<cn->xform_list.size();j++) {
if (cn->xform_list[j].id==at.param) {
xform_idx=j;
break;
}
}
if (xform_idx==-1) {
print_line("couldnt find matching node "+at.target+" xform for track "+at.param);
continue;
}
ERR_CONTINUE(xform_idx==-1);
Vector<float> data = at.get_value_at_time(snapshots[i]);
ERR_CONTINUE(data.empty());
Collada::Node::XForm &xf = cn->xform_list[xform_idx];
if (at.component=="ANGLE") {
ERR_CONTINUE(data.size()!=1);
ERR_CONTINUE(xf.op!=Collada::Node::XForm::OP_ROTATE);
ERR_CONTINUE(xf.data.size()<4);
xf.data[3]=data[0];
} else if (at.component=="X" || at.component=="Y" || at.component=="Z") {
int cn=at.component[0]-'X';
ERR_CONTINUE(cn>=xf.data.size());
ERR_CONTINUE(data.size()>1);
xf.data[cn]=data[0];
} else if (data.size()==xf.data.size()) {
xf.data=data;
} else {
if ( data.size()!=xf.data.size() ) {
print_line("component "+at.component+" datasize "+itos(data.size())+" xfdatasize "+itos(xf.data.size()));
}
ERR_CONTINUE( data.size()!=xf.data.size() );
}
}
Transform xform = cn->compute_transform(collada);
xform = collada.fix_transform(xform) * cn->post_transform;
if (nm.bone>=0) {
//make bone transform relative to rest (in case of skeleton)
Skeleton *sk = nm.node->cast_to<Skeleton>();
if (sk) {
xform = sk->get_bone_rest(nm.bone).affine_inverse() * xform;
} else {
ERR_PRINT("INVALID SKELETON!!!!");
}
}
Quat q = xform.basis;
q.normalize();
Vector3 s = xform.basis.get_scale();
Vector3 l = xform.origin;
animation->transform_track_insert_key(track,snapshots[i],l,q,s);
}
if (nm.bone>=0) {
if (found_anim)
bones_with_animation[E->get()]=true;
}
if (found_anim)
tracks_found=true;
else {
animation->remove_track( track );
}
}
if (p_make_tracks_in_all_bones) {
2014-02-10 02:10:30 +01:00
//some bones may lack animation, but since we don't store pose as a property, we must add keyframes!
for(Map<String,bool>::Element *E=bones_with_animation.front();E;E=E->next()) {
2014-02-10 02:10:30 +01:00
if (E->get())
continue;
2014-02-10 02:10:30 +01:00
//print_line("BONE LACKS ANIM: "+E->key());
2014-02-10 02:10:30 +01:00
NodeMap &nm = node_map[E->key()];
String path = scene->get_path_to(nm.node);
ERR_CONTINUE( nm.bone <0 );
Skeleton *sk = static_cast<Skeleton*>(nm.node);
String name = sk->get_bone_name(nm.bone);
path=path+":"+name;
2014-02-10 02:10:30 +01:00
Collada::Node *cn = collada.state.scene_map[E->key()];
if (cn->ignore_anim) {
print_line("warning, ignoring animation on node: "+path);
continue;
}
2014-02-10 02:10:30 +01:00
animation->add_track(Animation::TYPE_TRANSFORM);
int track = animation->get_track_count() -1;
animation->track_set_path( track , path );
animation->track_set_imported( track , true ); //helps merging later
2014-02-10 02:10:30 +01:00
Transform xform = cn->compute_transform(collada);
xform = collada.fix_transform(xform) * cn->post_transform;
2014-02-10 02:10:30 +01:00
xform = sk->get_bone_rest(nm.bone).affine_inverse() * xform;
2014-02-10 02:10:30 +01:00
Quat q = xform.basis;
q.normalize();
Vector3 s = xform.basis.get_scale();
Vector3 l = xform.origin;
2014-02-10 02:10:30 +01:00
animation->transform_track_insert_key(track,0,l,q,s);
2014-02-10 02:10:30 +01:00
tracks_found=true;
}
2014-02-10 02:10:30 +01:00
}
for(int i=0;i<valid_animated_properties.size();i++) {
int ti = valid_animated_properties[i];
if (p_clip==-1) {
if (track_filter.has(ti))
continue;
} else {
if (!track_filter.has(ti))
continue;
}
Collada::AnimationTrack &at = collada.state.animation_tracks[ ti ];
// take snapshots
if (!collada.state.scene_map.has(at.target))
continue;
NodeMap &nm = node_map[at.target];
String path = scene->get_path_to(nm.node);
animation->add_track(Animation::TYPE_VALUE);
int track = animation->get_track_count() -1;
2014-02-10 02:10:30 +01:00
path = path +":"+at.param;
animation->track_set_path( track , path );
animation->track_set_imported( track , true ); //helps merging later
2014-02-10 02:10:30 +01:00
for(int i=0;i<at.keys.size();i++) {
float time = at.keys[i].time;
Variant value;
Vector<float> data = at.keys[i].data;
if (data.size()==1) {
//push a float
value=data[0];
} else if (data.size()==16) {
//matrix
print_line("value keys for matrices not supported");
} else {
print_line("don't know what to do with this amount of value keys: "+itos(data.size()));
}
animation->track_insert_key(track,time,value);
}
tracks_found=true;
}
if (tracks_found) {
animations.push_back(animation);
}
}
/*********************************************************************************/
/*************************************** SCENE ***********************************/
/*********************************************************************************/
#define DEBUG_ANIMATION
uint32_t EditorSceneImporterCollada::get_import_flags() const {
return IMPORT_SCENE|IMPORT_ANIMATION;
}
void EditorSceneImporterCollada::get_extensions(List<String> *r_extensions) const {
r_extensions->push_back("dae");
}
Node* EditorSceneImporterCollada::import_scene(const String& p_path, uint32_t p_flags,int p_bake_fps, List<String> *r_missing_deps, Error* r_err) {
2014-02-10 02:10:30 +01:00
ColladaImport state;
uint32_t flags=Collada::IMPORT_FLAG_SCENE;
if (p_flags&IMPORT_ANIMATION)
flags|=Collada::IMPORT_FLAG_ANIMATION;
state.bake_fps=p_bake_fps;
2014-02-10 02:10:30 +01:00
Error err = state.load(p_path,flags,p_flags&EditorSceneImporter::IMPORT_GENERATE_TANGENT_ARRAYS);
ERR_FAIL_COND_V(err!=OK,NULL);
if (state.missing_textures.size()) {
//for(int i=0;i<state.missing_textures.size();i++) {
// EditorNode::add_io_error("Texture Not Found: "+state.missing_textures[i]);
// }
if (r_missing_deps) {
2014-02-10 02:10:30 +01:00
for(int i=0;i<state.missing_textures.size();i++) {
//EditorNode::add_io_error("Texture Not Found: "+state.missing_textures[i]);
r_missing_deps->push_back(state.missing_textures[i]);
}
}
2014-02-10 02:10:30 +01:00
}
if (p_flags&IMPORT_ANIMATION) {
state.create_animations(p_flags&IMPORT_ANIMATION_FORCE_ALL_TRACKS_IN_ALL_CLIPS);
2014-02-10 02:10:30 +01:00
AnimationPlayer *ap = memnew( AnimationPlayer );
for(int i=0;i<state.animations.size();i++) {
String name;
if (state.animations[i]->get_name()=="")
name="default";
else
name=state.animations[i]->get_name();
if (p_flags&IMPORT_ANIMATION_DETECT_LOOP) {
if (name.begins_with("loop") || name.ends_with("loop") || name.begins_with("cycle") || name.ends_with("cycle")) {
state.animations[i]->set_loop(true);
}
}
ap->add_animation(name,state.animations[i]);
}
state.scene->add_child(ap);
ap->set_owner(state.scene);
}
return state.scene;
}
Ref<Animation> EditorSceneImporterCollada::import_animation(const String& p_path,uint32_t p_flags) {
ColladaImport state;
Error err = state.load(p_path,Collada::IMPORT_FLAG_ANIMATION,p_flags&EditorSceneImporter::IMPORT_GENERATE_TANGENT_ARRAYS);
ERR_FAIL_COND_V(err!=OK,RES());
state.create_animations(p_flags&EditorSceneImporter::IMPORT_ANIMATION_FORCE_ALL_TRACKS_IN_ALL_CLIPS);
2014-02-10 02:10:30 +01:00
if (state.scene)
memdelete(state.scene);
if (state.animations.size()==0)
return Ref<Animation>();
Ref<Animation> anim=state.animations[0];
anim=state.animations[0];
print_line("Anim Load OK");
String base = p_path.basename().to_lower();
if (p_flags&IMPORT_ANIMATION_DETECT_LOOP) {
if (base.begins_with("loop") || base.ends_with("loop") || base.begins_with("cycle") || base.ends_with("cycle")) {
anim->set_loop(true);
}
}
return anim;
}
EditorSceneImporterCollada::EditorSceneImporterCollada() {
}