Alembic import: fix scene min/max time computation to take objects with
[blender.git] / source / blender / alembic / intern / abc_util.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_UTIL_H__
24 #define __ABC_UTIL_H__
25
26 #include <Alembic/Abc/All.h>
27 #include <Alembic/AbcGeom/All.h>
28
29 #ifdef _MSC_VER
30 #  define ABC_INLINE static __forceinline
31 #else
32 #  define ABC_INLINE static inline
33 #endif
34
35 using Alembic::Abc::chrono_t;
36
37 class ImportSettings;
38
39 struct ID;
40 struct Object;
41
42 std::string get_id_name(ID *id);
43 std::string get_id_name(Object *ob);
44 std::string get_object_dag_path_name(Object *ob, Object *dupli_parent);
45
46 bool object_selected(Object *ob);
47 bool parent_selected(Object *ob);
48
49 Imath::M44d convert_matrix(float mat[4][4]);
50 void create_transform_matrix(float r_mat[4][4]);
51 void create_transform_matrix(Object *obj, float transform_mat[4][4]);
52
53 void split(const std::string &s, const char delim, std::vector<std::string> &tokens);
54
55 template<class TContainer>
56 bool begins_with(const TContainer &input, const TContainer &match)
57 {
58         return input.size() >= match.size()
59                 && std::equal(match.begin(), match.end(), input.begin());
60 }
61
62 void create_input_transform(const Alembic::AbcGeom::ISampleSelector &sample_sel,
63                             const Alembic::AbcGeom::IXform &ixform, Object *ob,
64                             float r_mat[4][4], float scale, bool has_alembic_parent = false);
65
66 template <typename Schema>
67 void get_min_max_time_ex(const Schema &schema, chrono_t &min, chrono_t &max)
68 {
69         const Alembic::Abc::TimeSamplingPtr &time_samp = schema.getTimeSampling();
70
71         if (!schema.isConstant()) {
72                 const size_t num_samps = schema.getNumSamples();
73
74                 if (num_samps > 0) {
75                         const chrono_t min_time = time_samp->getSampleTime(0);
76                         min = std::min(min, min_time);
77
78                         const chrono_t max_time = time_samp->getSampleTime(num_samps - 1);
79                         max = std::max(max, max_time);
80                 }
81         }
82 }
83
84 template <typename Schema>
85 void get_min_max_time(const Alembic::AbcGeom::IObject &object, const Schema &schema, chrono_t &min, chrono_t &max)
86 {
87         get_min_max_time_ex(schema, min, max);
88
89         Alembic::AbcGeom::IXform parent(object.getParent(), Alembic::AbcGeom::kWrapExisting);
90         get_min_max_time_ex(parent.getSchema(), min, max);
91 }
92
93 bool has_property(const Alembic::Abc::ICompoundProperty &prop, const std::string &name);
94
95 /* ************************** */
96
97 /* TODO(kevin): for now keeping these transformations hardcoded to make sure
98  * everything works properly, and also because Alembic is almost exclusively
99  * used in Y-up software, but eventually they'll be set by the user in the UI
100  * like other importers/exporters do, to support other axis. */
101
102 /* Copy from Y-up to Z-up. */
103
104 ABC_INLINE void copy_yup_zup(float zup[3], const float yup[3])
105 {
106         zup[0] = yup[0];
107         zup[1] = -yup[2];
108         zup[2] = yup[1];
109 }
110
111 ABC_INLINE void copy_yup_zup(short zup[3], const short yup[3])
112 {
113         zup[0] = yup[0];
114         zup[1] = -yup[2];
115         zup[2] = yup[1];
116 }
117
118 /* Copy from Z-up to Y-up. */
119
120 ABC_INLINE void copy_zup_yup(float yup[3], const float zup[3])
121 {
122         yup[0] = zup[0];
123         yup[1] = zup[2];
124         yup[2] = -zup[1];
125 }
126
127 ABC_INLINE void copy_zup_yup(short yup[3], const short zup[3])
128 {
129         yup[0] = zup[0];
130         yup[1] = zup[2];
131         yup[2] = -zup[1];
132 }
133
134 #endif  /* __ABC_UTIL_H__ */