Changeset View
Changeset View
Standalone View
Standalone View
source/blender/io/usd/intern/usd_writer_mesh.cc
| Show All 20 Lines | |||||
| #include <pxr/usd/usdGeom/mesh.h> | #include <pxr/usd/usdGeom/mesh.h> | ||||
| #include <pxr/usd/usdShade/material.h> | #include <pxr/usd/usdShade/material.h> | ||||
| #include <pxr/usd/usdShade/materialBindingAPI.h> | #include <pxr/usd/usdShade/materialBindingAPI.h> | ||||
| #include "BLI_assert.h" | #include "BLI_assert.h" | ||||
| #include "BLI_math_vector.h" | #include "BLI_math_vector.h" | ||||
| #include "BKE_attribute.h" | |||||
| #include "BKE_customdata.h" | #include "BKE_customdata.h" | ||||
| #include "BKE_lib_id.h" | #include "BKE_lib_id.h" | ||||
| #include "BKE_material.h" | #include "BKE_material.h" | ||||
| #include "BKE_mesh.h" | #include "BKE_mesh.h" | ||||
| #include "BKE_modifier.h" | #include "BKE_modifier.h" | ||||
| #include "BKE_object.h" | #include "BKE_object.h" | ||||
| #include "DEG_depsgraph.h" | #include "DEG_depsgraph.h" | ||||
| ▲ Show 20 Lines • Show All 367 Lines • ▼ Show 20 Lines | void USDGenericMeshWriter::write_normals(const Mesh *mesh, pxr::UsdGeomMesh usd_mesh) | ||||
| pxr::UsdAttribute attr_normals = usd_mesh.CreateNormalsAttr(pxr::VtValue(), true); | pxr::UsdAttribute attr_normals = usd_mesh.CreateNormalsAttr(pxr::VtValue(), true); | ||||
| if (!attr_normals.HasValue()) { | if (!attr_normals.HasValue()) { | ||||
| attr_normals.Set(loop_normals, pxr::UsdTimeCode::Default()); | attr_normals.Set(loop_normals, pxr::UsdTimeCode::Default()); | ||||
| } | } | ||||
| usd_value_writer_.SetAttribute(attr_normals, pxr::VtValue(loop_normals), timecode); | usd_value_writer_.SetAttribute(attr_normals, pxr::VtValue(loop_normals), timecode); | ||||
| usd_mesh.SetNormalsInterpolation(pxr::UsdGeomTokens->faceVarying); | usd_mesh.SetNormalsInterpolation(pxr::UsdGeomTokens->faceVarying); | ||||
| } | } | ||||
| void USDGenericMeshWriter::write_surface_velocity(Object *object, | void USDGenericMeshWriter::write_surface_velocity(Object *object, | ||||
kevindietrich: `object` is not used anymore, which also gives a compile warning. | |||||
| const Mesh *mesh, | const Mesh *mesh, | ||||
| pxr::UsdGeomMesh usd_mesh) | pxr::UsdGeomMesh usd_mesh) | ||||
| { | { | ||||
| /* Only velocities from the fluid simulation are exported. This is the most important case, | /* Export velocity attribute output by fluid sim, sequence cache modifier | ||||
| * though, as the baked mesh changes topology all the time, and thus computing the velocities | * and geometry nodes. */ | ||||
| * at import time in a post-processing step is hard. */ | CustomDataLayer *velocity_layer = BKE_id_attribute_find( | ||||
| ModifierData *md = BKE_modifiers_findby_type(object, eModifierType_Fluidsim); | &mesh->id, "velocity", CD_PROP_FLOAT3, ATTR_DOMAIN_POINT); | ||||
| if (md == nullptr) { | |||||
| return; | |||||
| } | |||||
| /* Check that the fluid sim modifier is enabled and has useful data. */ | if (velocity_layer == nullptr) { | ||||
| const bool use_render = (DEG_get_mode(usd_export_context_.depsgraph) == DAG_EVAL_RENDER); | |||||
| const ModifierMode required_mode = use_render ? eModifierMode_Render : eModifierMode_Realtime; | |||||
| const Scene *scene = DEG_get_evaluated_scene(usd_export_context_.depsgraph); | |||||
| if (!BKE_modifier_is_enabled(scene, md, required_mode)) { | |||||
| return; | |||||
| } | |||||
| FluidsimModifierData *fsmd = reinterpret_cast<FluidsimModifierData *>(md); | |||||
| if (!fsmd->fss || fsmd->fss->type != OB_FLUIDSIM_DOMAIN) { | |||||
| return; | |||||
| } | |||||
| FluidsimSettings *fss = fsmd->fss; | |||||
| if (!fss->meshVelocities) { | |||||
| return; | return; | ||||
| } | } | ||||
| const float(*velocities)[3] = reinterpret_cast<float(*)[3]>(velocity_layer->data); | |||||
| /* Export per-vertex velocity vectors. */ | /* Export per-vertex velocity vectors. */ | ||||
| pxr::VtVec3fArray usd_velocities; | pxr::VtVec3fArray usd_velocities; | ||||
| usd_velocities.reserve(mesh->totvert); | usd_velocities.reserve(mesh->totvert); | ||||
| FluidVertexVelocity *mesh_velocities = fss->meshVelocities; | for (int vertex_idx = 0, totvert = mesh->totvert; vertex_idx < totvert; ++vertex_idx) { | ||||
| for (int vertex_idx = 0, totvert = mesh->totvert; vertex_idx < totvert; | usd_velocities.push_back(pxr::GfVec3f(velocities[vertex_idx])); | ||||
| ++vertex_idx, ++mesh_velocities) { | |||||
| usd_velocities.push_back(pxr::GfVec3f(mesh_velocities->vel)); | |||||
| } | } | ||||
| pxr::UsdTimeCode timecode = get_export_time_code(); | pxr::UsdTimeCode timecode = get_export_time_code(); | ||||
| usd_mesh.CreateVelocitiesAttr().Set(usd_velocities, timecode); | usd_mesh.CreateVelocitiesAttr().Set(usd_velocities, timecode); | ||||
| } | } | ||||
| USDMeshWriter::USDMeshWriter(const USDExporterContext &ctx) : USDGenericMeshWriter(ctx) | USDMeshWriter::USDMeshWriter(const USDExporterContext &ctx) : USDGenericMeshWriter(ctx) | ||||
| { | { | ||||
| } | } | ||||
| Mesh *USDMeshWriter::get_export_mesh(Object *object_eval, bool & /*r_needsfree*/) | Mesh *USDMeshWriter::get_export_mesh(Object *object_eval, bool & /*r_needsfree*/) | ||||
| { | { | ||||
| return BKE_object_get_evaluated_mesh(object_eval); | return BKE_object_get_evaluated_mesh(object_eval); | ||||
| } | } | ||||
| } // namespace blender::io::usd | } // namespace blender::io::usd | ||||
object is not used anymore, which also gives a compile warning.