OpenGL: Get rid of PRIM_QUADS usage in manipulators code
[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 struct SceneLayer;
42 struct Base;
43
44 struct ExportSettings {
45         ExportSettings();
46
47         Scene *scene;
48         SceneLayer *sl;  // Scene layer to export; all its objects will be exported, unless selected_only=true
49         SimpleLogger logger;
50
51         bool selected_only;
52         bool visible_layers_only;
53         bool renderable_only;
54
55         double frame_start, frame_end;
56         double frame_step_xform;
57         double frame_step_shape;
58         double shutter_open;
59         double shutter_close;
60         float global_scale;
61
62         bool flatten_hierarchy;
63
64         bool export_normals;
65         bool export_uvs;
66         bool export_vcols;
67         bool export_face_sets;
68         bool export_vweigths;
69
70         bool apply_subdiv;
71         bool use_subdiv_schema;
72         bool export_child_hairs;
73         bool export_ogawa;
74         bool pack_uv;
75         bool triangulate;
76
77         int quad_method;
78         int ngon_method;
79
80         bool do_convert_axis;
81         float convert_matrix[3][3];
82 };
83
84 class AbcExporter {
85         ExportSettings &m_settings;
86
87         const char *m_filename;
88
89         unsigned int m_trans_sampling_index, m_shape_sampling_index;
90
91         Scene *m_scene;
92
93         ArchiveWriter *m_writer;
94
95         std::map<std::string, AbcTransformWriter *> m_xforms;
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         void createTransformWritersFlat();
113         void createTransformWriter(Object *ob,  Object *parent, Object *dupliObParent);
114         void exploreTransform(EvaluationContext *eval_ctx, Base *ob_base, Object *parent, Object *dupliObParent);
115         void exploreObject(EvaluationContext *eval_ctx, Base *ob_base, Object *dupliObParent);
116         void createShapeWriters(EvaluationContext *eval_ctx);
117         void createShapeWriter(Base *ob_base, Object *dupliObParent);
118
119         AbcTransformWriter *getXForm(const std::string &name);
120
121         void setCurrentFrame(Main *bmain, double t);
122 };
123
124 #endif  /* __ABC_EXPORTER_H__ */