cadd6940e9d1b2b814e33e92ab0c93516597bcf5
[blender.git] / source / blender / collada / AnimationExporter.h
1 /*
2  * $Id: DocumentExporter.cpp 36898 2011-05-25 17:14:31Z phabtar $
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * Contributor(s): Chingiz Dyussenov, Arystanbek Dyussenov, Jan Diederich, Tod Liverseed.
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 #include <stdlib.h>
26 #include <stdio.h>
27 #include <math.h>
28 extern "C" 
29 {
30 #include "DNA_scene_types.h"
31 #include "DNA_object_types.h"
32 #include "DNA_anim_types.h"
33 #include "DNA_action_types.h"
34 #include "DNA_curve_types.h"
35 #include "DNA_lamp_types.h"
36 #include "DNA_camera_types.h"
37 #include "DNA_armature_types.h"
38 #include "DNA_material_types.h"
39
40 #include "BKE_DerivedMesh.h"
41 #include "BKE_fcurve.h"
42 #include "BKE_animsys.h"
43 #ifdef NAN_BUILDINFO
44 extern char build_rev[];
45 #endif
46 }
47
48 #include "MEM_guardedalloc.h"
49
50 #include "BKE_action.h" // pose functions
51 #include "BKE_armature.h"
52 #include "BKE_object.h"
53
54 #include "BLI_math.h"
55 #include "BLI_string.h"
56 #include "BLI_listbase.h"
57
58 #include "RNA_access.h"
59
60 #include "COLLADASWSource.h"
61 #include "COLLADASWInstanceGeometry.h"
62 #include "COLLADASWInputList.h"
63 #include "COLLADASWPrimitves.h"
64 #include "COLLADASWVertices.h"
65 #include "COLLADASWLibraryAnimations.h"
66 #include "COLLADASWParamTemplate.h"
67 #include "COLLADASWParamBase.h"
68 #include "COLLADASWSampler.h"
69 #include "COLLADASWConstants.h"
70 #include "COLLADASWBaseInputElement.h"
71
72 #include "EffectExporter.h"
73
74 #include "collada_internal.h"
75
76 #include <vector>
77 #include <algorithm> // std::find
78
79 class AnimationExporter: COLLADASW::LibraryAnimations
80 {
81 private:
82         Scene *scene;
83         COLLADASW::StreamWriter *sw;
84
85 public:
86
87         AnimationExporter(COLLADASW::StreamWriter *sw): COLLADASW::LibraryAnimations(sw) { this->sw = sw; }
88     
89
90         void exportAnimations(Scene *sce);
91
92         // called for each exported object
93         void operator() (Object *ob); 
94         
95 protected:
96
97         void dae_animation(Object* ob, FCurve *fcu, char* transformName , bool is_param, Material *ma = NULL);
98
99         void bake_bone_animation(Object *ob_arm, Bone *bone);
100
101         void write_bone_animation(Object *ob_arm, Bone *bone);
102
103         void sample_and_write_bone_animation(Object *ob_arm, Bone *bone, int transform_type);
104
105         void sample_and_bake_bone_animation(Object *ob_arm, Bone *bone);
106
107         void sample_animation(float *v, std::vector<float> &frames, int type, Bone *bone, Object *ob_arm, bPoseChannel *pChan);
108
109         void sample_animation(float *v, std::vector<float> &frames, Bone *bone, Object *ob_arm, bPoseChannel *pChan);
110
111         // dae_bone_animation -> add_bone_animation
112         // (blend this into dae_bone_animation)
113         void dae_bone_animation(std::vector<float> &fra, float *v, int tm_type, int axis, std::string ob_name, std::string bone_name);
114     
115         void dae_baked_animation(std::vector<float> &fra, float *values, std::string ob_name, std::string bone_name);
116
117         float convert_time(float frame);
118
119         float convert_angle(float angle);
120
121         std::string get_semantic_suffix(COLLADASW::InputSemantic::Semantics semantic);  
122
123         void add_source_parameters(COLLADASW::SourceBase::ParameterNameList& param,
124                                                            COLLADASW::InputSemantic::Semantics semantic, bool is_rot, const char *axis , bool transform);
125         
126     void get_source_values(BezTriple *bezt, COLLADASW::InputSemantic::Semantics semantic, bool rotation, float *values, int *length);
127         
128     float * get_eul_source_for_quat(Object *ob );
129
130         std::string create_source_from_fcurve(COLLADASW::InputSemantic::Semantics semantic, FCurve *fcu, const std::string& anim_id, const char *axis_name);
131
132         std::string create_source_from_array(COLLADASW::InputSemantic::Semantics semantic, float *v, int tot, bool is_rot, const std::string& anim_id, const char *axis_name);
133
134         std::string create_source_from_vector(COLLADASW::InputSemantic::Semantics semantic, std::vector<float> &fra, bool is_rot, const std::string& anim_id, const char *axis_name);
135
136         std::string create_xyz_source(float *v, int tot, const std::string& anim_id);
137
138         std::string create_4x4_source(float *v, int tot, const std::string& anim_id);
139
140         std::string create_interpolation_source(FCurve *fcu, const std::string& anim_id, const char *axis_name, bool *has_tangents);
141
142         std::string fake_interpolation_source(int tot, const std::string& anim_id, const char *axis_name);
143         // for rotation, axis name is always appended and the value of append_axis is ignored
144         std::string get_transform_sid(char *rna_path, int tm_type, const char *axis_name, bool append_axis);
145         
146         void find_frames(Object *ob, std::vector<float> &fra, const char *prefix, const char *tm_name);
147
148         void find_all_frames(Object *ob, std::vector<float> &fra);
149         
150         void find_rotation_frames(Object *ob, std::vector<float> &fra, const char *prefix, int rotmode);
151         
152         // enable fcurves driving a specific bone, disable all the rest
153         // if bone_name = NULL enable all fcurves
154         void enable_fcurves(bAction *act, char *bone_name);
155         
156         bool hasAnimations(Scene *sce);
157         
158         char* extract_transform_name(char *rna_path);
159
160         std::string getObjectBoneName ( Object *ob,const FCurve * fcu);
161 };