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
|
/*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
* The Original Code is Copyright (C) 2020 Blender Foundation.
* All rights reserved.
*/
#pragma once
#include "IO_abstract_hierarchy_iterator.h"
#include "abc_custom_props.h"
#include "abc_hierarchy_iterator.h"
#include <Alembic/Abc/OObject.h>
#include <vector>
#include "DEG_depsgraph_query.h"
#include "DNA_material_types.h"
struct IDProperty;
struct Object;
namespace blender::io::alembic {
class ABCAbstractWriter : public AbstractHierarchyWriter {
protected:
const ABCWriterConstructorArgs args_;
bool frame_has_been_written_;
bool is_animated_;
uint32_t timesample_index_;
Imath::Box3d bounding_box_;
/* Visibility of this writer's data in Alembic. */
Alembic::Abc::OCharProperty abc_visibility_;
/* Optional writer for custom properties. */
std::unique_ptr<CustomPropertiesExporter> custom_props_;
public:
explicit ABCAbstractWriter(const ABCWriterConstructorArgs &args);
virtual void write(HierarchyContext &context) override;
/* Returns true if the data to be written is actually supported. This would, for example, allow a
* hypothetical camera writer accept a perspective camera but reject an orthogonal one.
*
* Returning false from a transform writer will prevent the object and all its descendants from
* being exported. Returning false from a data writer (object data, hair, or particles) will
* only prevent that data from being written (and thus cause the object to be exported as an
* Empty). */
virtual bool is_supported(const HierarchyContext *context) const;
uint32_t timesample_index() const;
const Imath::Box3d &bounding_box() const;
/* Called by AlembicHierarchyCreator after checking that the data is supported via
* is_supported(). */
virtual void create_alembic_objects(const HierarchyContext *context) = 0;
virtual Alembic::Abc::OObject get_alembic_object() const = 0;
/* Return the Alembic object's CompoundProperty that'll contain the custom properties.
*
* This function is called whenever there are custom properties to be written to Alembic. It
* should call abc_schema_prop_for_custom_props() with the writer's Alembic schema object.
*
* If custom properties are not supported by a specific subclass, it should return an empty
* OCompoundProperty() and override ensure_custom_properties_exporter() to do nothing.
*/
virtual Alembic::Abc::OCompoundProperty abc_prop_for_custom_props() = 0;
protected:
virtual void do_write(HierarchyContext &context) = 0;
virtual void update_bounding_box(Object *object);
/* Return ID properties of whatever ID datablock is written by this writer. Defaults to the
* properties of the object data. Can return nullptr if no custom properties are to be written.
*/
virtual const IDProperty *get_id_properties(const HierarchyContext &context) const;
virtual void ensure_custom_properties_exporter(const HierarchyContext &context);
void write_visibility(const HierarchyContext &context);
/* Return the Alembic schema's compound property, which will be used for writing custom
* properties.
*
* This can return either abc_schema.getUserProperties() or abc_schema.getArbGeomParams(). The
* former only holds values similar to Blender's custom properties, whereas the latter can also
* specify that certain custom properties vary per mesh component (so per face, vertex, etc.). As
* such, .userProperties is more suitable for custom properties. However, Maya, Houdini use
* .arbGeomParams for custom data.
*
* Because of this, the code uses this templated function so that there is one place that
* determines where custom properties are exporter to.
*/
template<typename T>
Alembic::Abc::OCompoundProperty abc_schema_prop_for_custom_props(T abc_schema)
{
return abc_schema.getUserProperties();
}
};
} // namespace blender::io::alembic
|