Merge branch 'master' into blender2.8
[blender.git] / source / blender / alembic / intern / abc_customdata.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  * The Original Code is Copyright (C) 2016 K√©vin Dietrich.
19  * All rights reserved.
20  *
21  * ***** END GPL LICENSE BLOCK *****
22  *
23  */
24
25 #ifndef __ABC_CUSTOMDATA_H__
26 #define __ABC_CUSTOMDATA_H__
27
28 #include <Alembic/Abc/All.h>
29 #include <Alembic/AbcGeom/All.h>
30
31 struct CustomData;
32 struct MLoop;
33 struct MLoopUV;
34 struct MPoly;
35 struct MVert;
36
37 using Alembic::Abc::ICompoundProperty;
38 using Alembic::Abc::OCompoundProperty;
39
40 struct UVSample {
41         std::vector<Imath::V2f> uvs;
42         std::vector<uint32_t> indices;
43 };
44
45 struct CDStreamConfig {
46         MLoop *mloop;
47         int totloop;
48
49         MPoly *mpoly;
50         int totpoly;
51
52         MVert *mvert;
53         int totvert;
54
55         MLoopUV *mloopuv;
56
57         CustomData *loopdata;
58
59         bool pack_uvs;
60
61         /* TODO(kevin): might need a better way to handle adding and/or updating
62          * custom datas such that it updates the custom data holder and its pointers
63          * properly. */
64         void *user_data;
65         void *(*add_customdata_cb)(void *user_data, const char *name, int data_type);
66
67         float weight;
68         float time;
69         Alembic::AbcGeom::index_t index;
70         Alembic::AbcGeom::index_t ceil_index;
71
72         CDStreamConfig()
73             : mloop(NULL)
74             , totloop(0)
75             , mpoly(NULL)
76             , totpoly(0)
77             , totvert(0)
78             , pack_uvs(false)
79             , user_data(NULL)
80             , add_customdata_cb(NULL)
81             , weight(0.0f)
82             , time(0.0f)
83             , index(0)
84             , ceil_index(0)
85         {}
86 };
87
88 /* Get the UVs for the main UV property on a OSchema.
89  * Returns the name of the UV layer.
90  *
91  * For now the active layer is used, maybe needs a better way to choose this. */
92 const char *get_uv_sample(UVSample &sample, const CDStreamConfig &config, CustomData *data);
93
94 void write_custom_data(const OCompoundProperty &prop,
95                        const CDStreamConfig &config,
96                        CustomData *data,
97                        int data_type);
98
99 void read_custom_data(const std::string & iobject_full_name,
100                       const ICompoundProperty &prop,
101                       const CDStreamConfig &config,
102                       const Alembic::Abc::ISampleSelector &iss);
103
104 #endif  /* __ABC_CUSTOMDATA_H__ */