1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
/* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright 2016 Kévin Dietrich. All rights reserved. */
#pragma once
/** \file
* \ingroup balembic
*/
#include <Alembic/Abc/All.h>
#include <Alembic/AbcGeom/All.h>
#include <map>
#include "BKE_attribute.h"
#include "BLI_listbase_wrapper.hh"
struct CacheAttributeMapping;
struct CustomData;
struct ID;
struct MLoop;
struct MLoopUV;
struct MPoly;
struct MVert;
struct Mesh;
using Alembic::Abc::ICompoundProperty;
using Alembic::Abc::OCompoundProperty;
namespace blender::io::alembic {
class AttributeSelector;
struct UVSample {
std::vector<Imath::V2f> uvs;
std::vector<uint32_t> indices;
};
struct CDStreamConfig {
MLoop *mloop;
int totloop;
MPoly *mpoly;
int totpoly;
MVert *mvert;
int totvert;
MLoopUV *mloopuv;
CustomData *loopdata;
bool pack_uvs;
/* NOTE: the mesh is mostly used for iterating over loops for loop attributes (UVs, MCol, etc.).
* It would be nice to remove it, in favor of a more generic way to iterate valid attribute
* indices.
*/
Mesh *mesh;
ID *id;
float weight;
float time;
int timesample_index;
bool use_vertex_interpolation;
Alembic::AbcGeom::index_t index;
Alembic::AbcGeom::index_t ceil_index;
const char **modifier_error_message;
DomainInfo domain_info[ATTR_DOMAIN_NUM];
/* For error reporting when reading vertex colors. */
std::string iobject_full_name;
const AttributeSelector *attr_selector;
/* Alembic needs Blender to keep references to C++ objects (the destructors finalize the writing
* to ABC). The following fields are all used to keep these references. */
/* Mapping from UV map name to its ABC property, for the 2nd and subsequent UV maps; the primary
* UV map is kept alive by the Alembic mesh sample itself. */
std::map<std::string, Alembic::AbcGeom::OV2fGeomParam> abc_uv_maps;
/* ORCO coordinates, aka Generated Coordinates. */
Alembic::AbcGeom::OV3fGeomParam abc_orco;
/* Mapping from vertex color layer name to its Alembic color data. */
std::map<std::string, Alembic::AbcGeom::OC4fGeomParam> abc_vertex_colors;
CDStreamConfig()
: mloop(NULL),
totloop(0),
mpoly(NULL),
totpoly(0),
totvert(0),
pack_uvs(false),
mesh(NULL),
weight(0.0f),
time(0.0f),
index(0),
ceil_index(0),
modifier_error_message(NULL),
attr_selector(nullptr)
{
}
};
/* Get the UVs for the main UV property on a OSchema.
* Returns the name of the UV layer.
*
* For now the active layer is used, maybe needs a better way to choose this. */
const char *get_uv_sample(UVSample &sample, const CDStreamConfig &config, CustomData *data);
void write_generated_coordinates(const OCompoundProperty &prop, CDStreamConfig &config);
void write_custom_data(const OCompoundProperty &prop,
CDStreamConfig &config,
CustomData *data,
int data_type);
class AttributeSelector {
/* Name of the velocity attribute, it is ignored since we deal with separately. */
std::string velocity_attribute = "";
int read_flags = 0;
ListBaseWrapper<const CacheAttributeMapping> mappings;
public:
AttributeSelector(ListBase *mappings_) : mappings(mappings_)
{
}
void set_velocity_attribute(const char *name);
void set_read_flags(int flags);
const CacheAttributeMapping *get_mapping(const std::string &attr_name) const;
const std::string &velocity_name() const;
bool uvs_requested() const;
bool vertex_colors_requested() const;
bool original_coordinates_requested() const;
bool select_attribute(const std::string &attr_name) const;
};
void read_arbitrary_attributes(const CDStreamConfig &config,
const ICompoundProperty &schema,
const Alembic::AbcGeom::IV2fGeomParam &primary_uvs,
const Alembic::Abc::ISampleSelector &sample_sel,
float velocity_scale);
bool has_animated_attributes(const ICompoundProperty &arb_geom_params);
} // namespace blender::io::alembic
|
