Fixed Camera Ortho scale animation import
[blender-staging.git] / source / blender / collada / ArmatureExporter.h
1 /*
2  * $Id$
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  *                 Nathan Letwory
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  */
25
26 /** \file ArmatureExporter.h
27  *  \ingroup collada
28  */
29
30 #ifndef __ARMATUREEXPORTER_H__
31 #define __ARMATUREEXPORTER_H__
32
33 #include <string>
34 //#include <vector>
35
36 #include "COLLADASWStreamWriter.h"
37 #include "COLLADASWLibraryControllers.h"
38 #include "COLLADASWInputList.h"
39 #include "COLLADASWNode.h"
40
41 #include "DNA_armature_types.h"
42 #include "DNA_listBase.h"
43 #include "DNA_mesh_types.h"
44 #include "DNA_object_types.h"
45 #include "DNA_scene_types.h"
46
47 #include "TransformWriter.h"
48 #include "InstanceWriter.h"
49
50 // XXX exporter writes wrong data for shared armatures.  A separate
51 // controller should be written for each armature-mesh binding how do
52 // we make controller ids then?
53 class ArmatureExporter: public COLLADASW::LibraryControllers, protected TransformWriter, protected InstanceWriter
54 {
55 private:
56         Scene *scene;
57
58 public:
59         ArmatureExporter(COLLADASW::StreamWriter *sw);
60
61         // write bone nodes
62         void add_armature_bones(Object *ob_arm, Scene *sce);
63
64         bool is_skinned_mesh(Object *ob);
65
66         void add_instance_controller(Object *ob);
67
68         void export_controllers(Scene *sce, bool export_selected);
69
70         void operator()(Object *ob);
71
72 private:
73
74         UnitConverter converter;
75
76 #if 0
77         std::vector<Object*> written_armatures;
78
79         bool already_written(Object *ob_arm);
80
81         void wrote(Object *ob_arm);
82
83         void find_objects_using_armature(Object *ob_arm, std::vector<Object *>& objects, Scene *sce);
84 #endif
85
86         Object *get_assigned_armature(Object *ob);
87
88         std::string get_joint_sid(Bone *bone, Object *ob_arm);
89
90         // parent_mat is armature-space
91         void add_bone_node(Bone *bone, Object *ob_arm);
92
93         void add_bone_transform(Object *ob_arm, Bone *bone, COLLADASW::Node& node);
94
95         std::string get_controller_id(Object *ob_arm, Object *ob);
96
97         // ob should be of type OB_MESH
98         // both args are required
99         void export_controller(Object* ob, Object *ob_arm);
100
101         void add_joints_element(ListBase *defbase,
102                                                         const std::string& joints_source_id, const std::string& inv_bind_mat_source_id);
103
104         void add_bind_shape_mat(Object *ob);
105
106         std::string add_joints_source(Object *ob_arm, ListBase *defbase, const std::string& controller_id);
107
108         std::string add_inv_bind_mats_source(Object *ob_arm, ListBase *defbase, const std::string& controller_id);
109
110         Bone *get_bone_from_defgroup(Object *ob_arm, bDeformGroup* def);
111
112         bool is_bone_defgroup(Object *ob_arm, bDeformGroup* def);
113
114         std::string add_weights_source(Mesh *me, const std::string& controller_id);
115
116         void add_vertex_weights_element(const std::string& weights_source_id, const std::string& joints_source_id, Mesh *me,
117                                                                         Object *ob_arm, ListBase *defbase);
118 };
119
120 /*
121 struct GeometryFunctor {
122         // f should have
123         // void operator()(Object* ob)
124         template<class Functor>
125         void forEachMeshObjectInScene(Scene *sce, Functor &f)
126         {
127                 
128                 Base *base= (Base*) sce->base.first;
129                 while(base) {
130                         Object *ob = base->object;
131                         
132                         if (ob->type == OB_MESH && ob->data) {
133                                 f(ob);
134                         }
135                         base= base->next;
136                         
137                 }
138         }
139 };*/
140
141 #endif