Collada exporter update
[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  private:
57   BlenderContext &blender_context;
58   BCExportSettings export_settings;
59
60  public:
61   // XXX exporter writes wrong data for shared armatures.  A separate
62   // controller should be written for each armature-mesh binding how do
63   // we make controller ids then?
64   ControllerExporter(BlenderContext &blender_context,
65                      COLLADASW::StreamWriter *sw,
66                      BCExportSettings &export_settings)
67       : COLLADASW::LibraryControllers(sw),
68         blender_context(blender_context),
69         export_settings(export_settings)
70   {
71   }
72
73   bool is_skinned_mesh(Object *ob);
74
75   bool add_instance_controller(Object *ob);
76
77   void export_controllers();
78
79   void operator()(Object *ob);
80
81  private:
82 #if 0
83   std::vector<Object *> written_armatures;
84
85   bool already_written(Object *ob_arm);
86
87   void wrote(Object *ob_arm);
88
89   void find_objects_using_armature(Object *ob_arm, std::vector<Object *> &objects, Scene *sce);
90 #endif
91
92   std::string get_controller_id(Object *ob_arm, Object *ob);
93
94   std::string get_controller_id(Key *key, Object *ob);
95
96   // ob should be of type OB_MESH
97   // both args are required
98   void export_skin_controller(Object *ob, Object *ob_arm);
99
100   void export_morph_controller(Object *ob, Key *key);
101
102   void add_joints_element(ListBase *defbase,
103                           const std::string &joints_source_id,
104                           const std::string &inv_bind_mat_source_id);
105
106   void add_bind_shape_mat(Object *ob);
107
108   std::string add_morph_targets(Key *key, Object *ob);
109
110   std::string add_morph_weights(Key *key, Object *ob);
111
112   void add_weight_extras(Key *key);
113
114   std::string add_joints_source(Object *ob_arm,
115                                 ListBase *defbase,
116                                 const std::string &controller_id);
117
118   std::string add_inv_bind_mats_source(Object *ob_arm,
119                                        ListBase *defbase,
120                                        const std::string &controller_id);
121
122   Bone *get_bone_from_defgroup(Object *ob_arm, bDeformGroup *def);
123
124   bool is_bone_defgroup(Object *ob_arm, bDeformGroup *def);
125
126   std::string add_weights_source(Mesh *me,
127                                  const std::string &controller_id,
128                                  const std::list<float> &weights);
129
130   void add_vertex_weights_element(const std::string &weights_source_id,
131                                   const std::string &joints_source_id,
132                                   const std::list<int> &vcount,
133                                   const std::list<int> &joints);
134
135   void write_bone_URLs(COLLADASW::InstanceController &ins, Object *ob_arm, Bone *bone);
136 };
137
138 #endif