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
|
/* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright 2021 Tangent Animation. All rights reserved. */
#include "usd_reader_volume.h"
#include "BKE_object.h"
#include "BKE_volume.h"
#include "DNA_object_types.h"
#include "DNA_volume_types.h"
#include <pxr/usd/usdVol/openVDBAsset.h>
#include <pxr/usd/usdVol/volume.h>
#include <iostream>
namespace usdtokens {
static const pxr::TfToken density("density", pxr::TfToken::Immortal);
}
namespace blender::io::usd {
void USDVolumeReader::create_object(Main *bmain, const double /* motionSampleTime */)
{
Volume *volume = (Volume *)BKE_volume_add(bmain, name_.c_str());
object_ = BKE_object_add_only_object(bmain, OB_VOLUME, name_.c_str());
object_->data = volume;
}
void USDVolumeReader::read_object_data(Main *bmain, const double motionSampleTime)
{
if (!volume_) {
return;
}
Volume *volume = static_cast<Volume *>(object_->data);
if (!volume) {
return;
}
pxr::UsdVolVolume::FieldMap fields = volume_.GetFieldPaths();
for (pxr::UsdVolVolume::FieldMap::const_iterator it = fields.begin(); it != fields.end(); ++it) {
pxr::UsdPrim fieldPrim = prim_.GetStage()->GetPrimAtPath(it->second);
if (!fieldPrim.IsA<pxr::UsdVolOpenVDBAsset>()) {
continue;
}
pxr::UsdVolOpenVDBAsset fieldBase(fieldPrim);
pxr::UsdAttribute fieldNameAttr = fieldBase.GetFieldNameAttr();
if (fieldNameAttr.IsAuthored()) {
pxr::TfToken fieldName;
fieldNameAttr.Get(&fieldName, motionSampleTime);
/* A Blender volume creates density by default. */
if (fieldName != usdtokens::density) {
BKE_volume_grid_add(volume, fieldName.GetString().c_str(), VOLUME_GRID_FLOAT);
}
}
pxr::UsdAttribute filepathAttr = fieldBase.GetFilePathAttr();
if (filepathAttr.IsAuthored()) {
pxr::SdfAssetPath fp;
filepathAttr.Get(&fp, motionSampleTime);
if (filepathAttr.ValueMightBeTimeVarying()) {
std::vector<double> filePathTimes;
filepathAttr.GetTimeSamples(&filePathTimes);
if (!filePathTimes.empty()) {
int start = static_cast<int>(filePathTimes.front());
int end = static_cast<int>(filePathTimes.back());
volume->is_sequence = static_cast<char>(true);
volume->frame_start = start;
volume->frame_duration = (end - start) + 1;
}
}
std::string filepath = fp.GetResolvedPath();
strcpy(volume->filepath, filepath.c_str());
}
}
USDXformReader::read_object_data(bmain, motionSampleTime);
}
} // namespace blender::io::usd
|