38d85cb7ebcb4240ed332e0c1f8d4240fde3c02c
[blender.git] / source / blender / collada / ControllerExporter.h
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  */
16
17 /** \file
18  * \ingroup collada
19  */
20
21 #ifndef __CONTROLLEREXPORTER_H__
22 #define __CONTROLLEREXPORTER_H__
23
24 #include <list>
25 #include <string>
26 //#include <vector>
27
28 #include "COLLADASWStreamWriter.h"
29 #include "COLLADASWLibraryControllers.h"
30 #include "COLLADASWInstanceController.h"
31 #include "COLLADASWInputList.h"
32 #include "COLLADASWNode.h"
33 #include "COLLADASWExtraTechnique.h"
34
35 #include "DNA_armature_types.h"
36 #include "DNA_listBase.h"
37 #include "DNA_mesh_types.h"
38 #include "DNA_object_types.h"
39 #include "DNA_constraint_types.h"
40 #include "DNA_scene_types.h"
41 #include "DNA_key_types.h"
42
43 #include "TransformWriter.h"
44 #include "InstanceWriter.h"
45
46 #include "ExportSettings.h"
47
48 #include "BKE_key.h"
49
50 struct Depsgraph;
51 class SceneExporter;
52
53 class ControllerExporter : public COLLADASW::LibraryControllers,
54                            protected TransformWriter,
55                            protected InstanceWriter {
56  public:
57   // XXX exporter writes wrong data for shared armatures.  A separate
58   // controller should be written for each armature-mesh binding how do
59   // we make controller ids then?
60   ControllerExporter(BlenderContext &blender_context,
61                      COLLADASW::StreamWriter *sw,
62                      const ExportSettings *export_settings)
63       : COLLADASW::LibraryControllers(sw),
64         blender_context(blender_context),
65         export_settings(export_settings)
66   {
67   }
68
69   bool is_skinned_mesh(Object *ob);
70
71   bool add_instance_controller(Object *ob);
72
73   void export_controllers();
74
75   void operator()(Object *ob);
76
77  private:
78   BlenderContext &blender_context;
79   const ExportSettings *export_settings;
80
81 #if 0
82   std::vector<Object *> written_armatures;
83
84   bool already_written(Object *ob_arm);
85
86   void wrote(Object *ob_arm);
87
88   void find_objects_using_armature(Object *ob_arm, std::vector<Object *> &objects, Scene *sce);
89 #endif
90
91   std::string get_controller_id(Object *ob_arm, Object *ob);
92
93   std::string get_controller_id(Key *key, Object *ob);
94
95   // ob should be of type OB_MESH
96   // both args are required
97   void export_skin_controller(Object *ob, Object *ob_arm);
98
99   void export_morph_controller(Object *ob, Key *key);
100
101   void add_joints_element(ListBase *defbase,
102                           const std::string &joints_source_id,
103                           const std::string &inv_bind_mat_source_id);
104
105   void add_bind_shape_mat(Object *ob);
106
107   std::string add_morph_targets(Key *key, Object *ob);
108
109   std::string add_morph_weights(Key *key, Object *ob);
110
111   void add_weight_extras(Key *key);
112
113   std::string add_joints_source(Object *ob_arm,
114                                 ListBase *defbase,
115                                 const std::string &controller_id);
116
117   std::string add_inv_bind_mats_source(Object *ob_arm,
118                                        ListBase *defbase,
119                                        const std::string &controller_id);
120
121   Bone *get_bone_from_defgroup(Object *ob_arm, bDeformGroup *def);
122
123   bool is_bone_defgroup(Object *ob_arm, bDeformGroup *def);
124
125   std::string add_weights_source(Mesh *me,
126                                  const std::string &controller_id,
127                                  const std::list<float> &weights);
128
129   void add_vertex_weights_element(const std::string &weights_source_id,
130                                   const std::string &joints_source_id,
131                                   const std::list<int> &vcount,
132                                   const std::list<int> &joints);
133
134   void write_bone_URLs(COLLADASW::InstanceController &ins, Object *ob_arm, Bone *bone);
135 };
136
137 #endif