73b7af280d9c09f50821638c67d14b26d6f22b80
[blender.git] / source / blender / alembic / intern / abc_exporter.h
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * Contributor(s): Esteban Tovagliari, Cedric Paille, Kevin Dietrich
19  *
20  * ***** END GPL LICENSE BLOCK *****
21  */
22
23 #ifndef __ABC_EXPORTER_H__
24 #define __ABC_EXPORTER_H__
25
26 #include <Alembic/Abc/All.h>
27 #include <map>
28 #include <set>
29 #include <vector>
30
31 #include "abc_util.h"
32
33 class AbcObjectWriter;
34 class AbcTransformWriter;
35 class ArchiveWriter;
36
37 struct EvaluationContext;
38 struct Main;
39 struct Object;
40 struct Scene;
41
42 struct ExportSettings {
43         ExportSettings();
44
45         Scene *scene;
46         SimpleLogger logger;
47
48         bool selected_only;
49         bool visible_layers_only;
50         bool renderable_only;
51
52         double frame_start, frame_end;
53         double frame_step_xform;
54         double frame_step_shape;
55         double shutter_open;
56         double shutter_close;
57         float global_scale;
58
59         bool flatten_hierarchy;
60
61         bool export_normals;
62         bool export_uvs;
63         bool export_vcols;
64         bool export_face_sets;
65         bool export_vweigths;
66
67         bool apply_subdiv;
68         bool use_subdiv_schema;
69         bool export_child_hairs;
70         bool export_ogawa;
71         bool pack_uv;
72         bool triangulate;
73
74         int quad_method;
75         int ngon_method;
76
77         bool do_convert_axis;
78         float convert_matrix[3][3];
79 };
80
81 class AbcExporter {
82         ExportSettings &m_settings;
83
84         const char *m_filename;
85
86         unsigned int m_trans_sampling_index, m_shape_sampling_index;
87
88         Scene *m_scene;
89
90         ArchiveWriter *m_writer;
91
92         /* mapping from name to transform writer */
93         typedef std::map<std::string, AbcTransformWriter *> m_xforms_type;
94         m_xforms_type m_xforms;
95
96         std::vector<AbcObjectWriter *> m_shapes;
97
98 public:
99         AbcExporter(Scene *scene, const char *filename, ExportSettings &settings);
100         ~AbcExporter();
101
102         void operator()(Main *bmain, float &progress, bool &was_canceled);
103
104 private:
105         void getShutterSamples(double step, bool time_relative, std::vector<double> &samples);
106
107         Alembic::Abc::TimeSamplingPtr createTimeSampling(double step);
108
109         void getFrameSet(double step, std::set<double> &frames);
110
111         void createTransformWritersHierarchy(EvaluationContext *eval_ctx);
112         AbcTransformWriter * createTransformWriter(Object *ob,  Object *parent, Object *dupliObParent);
113         void exploreTransform(EvaluationContext *eval_ctx, Object *ob, Object *parent, Object *dupliObParent = NULL);
114         void exploreObject(EvaluationContext *eval_ctx, Object *ob, Object *dupliObParent);
115         void createShapeWriters(EvaluationContext *eval_ctx);
116         void createShapeWriter(Object *ob, Object *dupliObParent);
117
118         AbcTransformWriter *getXForm(const std::string &name);
119
120         void setCurrentFrame(Main *bmain, double t);
121 };
122
123 #endif  /* __ABC_EXPORTER_H__ */