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         bool export_hair;
  67         bool export_particles;
  68
  69         bool apply_subdiv;
  70         bool use_subdiv_schema;
  71         bool export_child_hairs;
  72         bool export_ogawa;
  73         bool pack_uv;
  74         bool triangulate;
  75
  76         int quad_method;
  77         int ngon_method;
  78
  79         bool do_convert_axis;
  80         float convert_matrix[3][3];
  81 };
  82
  83 class AbcExporter {
  84         ExportSettings &m_settings;
  85
  86         const char *m_filename;
  87
  88         unsigned int m_trans_sampling_index, m_shape_sampling_index;
  89
  90         Scene *m_scene;
  91
  92         ArchiveWriter *m_writer;
  93
  94         /* mapping from name to transform writer */
  95         typedef std::map<std::string, AbcTransformWriter *> m_xforms_type;
  96         m_xforms_type m_xforms;
  97
  98         std::vector<AbcObjectWriter *> m_shapes;
  99
 100 public:
 101         AbcExporter(Scene *scene, const char *filename, ExportSettings &settings);
 102         ~AbcExporter();
 103
 104         void operator()(Main *bmain, float &progress, bool &was_canceled);
 105
 106 private:
 107         void getShutterSamples(double step, bool time_relative, std::vector<double> &samples);
 108
 109         Alembic::Abc::TimeSamplingPtr createTimeSampling(double step);
 110
 111         void getFrameSet(double step, std::set<double> &frames);
 112
 113         void createTransformWritersHierarchy(EvaluationContext *eval_ctx);
 114         AbcTransformWriter * createTransformWriter(Object *ob,  Object *parent, Object *dupliObParent);
 115         void exploreTransform(EvaluationContext *eval_ctx, Object *ob, Object *parent, Object *dupliObParent = NULL);
 116         void exploreObject(EvaluationContext *eval_ctx, Object *ob, Object *dupliObParent);
 117         void createShapeWriters(EvaluationContext *eval_ctx);
 118         void createShapeWriter(Object *ob, Object *dupliObParent);
 119         void createParticleSystemsWriters(Object *ob, AbcTransformWriter *xform);
 120
 121         AbcTransformWriter *getXForm(const std::string &name);
 122
 123         void setCurrentFrame(Main *bmain, double t);
 124 };
 125
 126 #endif  /* __ABC_EXPORTER_H__ */