c671a26637
For 3.2 USD will be bumped to a newer version with some slight API changes, however since we cannot simultaneously land the libs for all platforms as well as these code changes, we'll have to support both 21.02 and 21.11+ for at least a short period of time making the code slightly more messy than it could have been. Differential Revision: https://developer.blender.org/D14184 Reviewed by: sybren
123 lines
3.9 KiB
C++
123 lines
3.9 KiB
C++
/* SPDX-License-Identifier: GPL-2.0-or-later
|
|
* Copyright 2019 Blender Foundation. All rights reserved. */
|
|
#include "usd_writer_light.h"
|
|
#include "usd_hierarchy_iterator.h"
|
|
|
|
#include <pxr/usd/usdLux/diskLight.h>
|
|
#include <pxr/usd/usdLux/distantLight.h>
|
|
#include <pxr/usd/usdLux/rectLight.h>
|
|
#include <pxr/usd/usdLux/sphereLight.h>
|
|
|
|
#include "BLI_assert.h"
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "DNA_light_types.h"
|
|
#include "DNA_object_types.h"
|
|
|
|
namespace blender::io::usd {
|
|
|
|
USDLightWriter::USDLightWriter(const USDExporterContext &ctx) : USDAbstractWriter(ctx)
|
|
{
|
|
}
|
|
|
|
bool USDLightWriter::is_supported(const HierarchyContext *context) const
|
|
{
|
|
Light *light = static_cast<Light *>(context->object->data);
|
|
return ELEM(light->type, LA_AREA, LA_LOCAL, LA_SUN);
|
|
}
|
|
|
|
void USDLightWriter::do_write(HierarchyContext &context)
|
|
{
|
|
pxr::UsdStageRefPtr stage = usd_export_context_.stage;
|
|
const pxr::SdfPath &usd_path = usd_export_context_.usd_path;
|
|
pxr::UsdTimeCode timecode = get_export_time_code();
|
|
|
|
Light *light = static_cast<Light *>(context.object->data);
|
|
#if PXR_VERSION >= 2111
|
|
pxr::UsdLuxLightAPI usd_light_api;
|
|
#else
|
|
pxr::UsdLuxLight usd_light_api;
|
|
|
|
#endif
|
|
|
|
switch (light->type) {
|
|
case LA_AREA:
|
|
switch (light->area_shape) {
|
|
case LA_AREA_DISK:
|
|
case LA_AREA_ELLIPSE: { /* An ellipse light will deteriorate into a disk light. */
|
|
pxr::UsdLuxDiskLight disk_light = pxr::UsdLuxDiskLight::Define(stage, usd_path);
|
|
disk_light.CreateRadiusAttr().Set(light->area_size, timecode);
|
|
#if PXR_VERSION >= 2111
|
|
usd_light_api = disk_light.LightAPI();
|
|
#else
|
|
usd_light_api = disk_light;
|
|
#endif
|
|
break;
|
|
}
|
|
case LA_AREA_RECT: {
|
|
pxr::UsdLuxRectLight rect_light = pxr::UsdLuxRectLight::Define(stage, usd_path);
|
|
rect_light.CreateWidthAttr().Set(light->area_size, timecode);
|
|
rect_light.CreateHeightAttr().Set(light->area_sizey, timecode);
|
|
#if PXR_VERSION >= 2111
|
|
usd_light_api = rect_light.LightAPI();
|
|
#else
|
|
usd_light_api = rect_light;
|
|
#endif
|
|
break;
|
|
}
|
|
case LA_AREA_SQUARE: {
|
|
pxr::UsdLuxRectLight rect_light = pxr::UsdLuxRectLight::Define(stage, usd_path);
|
|
rect_light.CreateWidthAttr().Set(light->area_size, timecode);
|
|
rect_light.CreateHeightAttr().Set(light->area_size, timecode);
|
|
#if PXR_VERSION >= 2111
|
|
usd_light_api = rect_light.LightAPI();
|
|
#else
|
|
usd_light_api = rect_light;
|
|
#endif
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
case LA_LOCAL: {
|
|
pxr::UsdLuxSphereLight sphere_light = pxr::UsdLuxSphereLight::Define(stage, usd_path);
|
|
sphere_light.CreateRadiusAttr().Set(light->area_size, timecode);
|
|
#if PXR_VERSION >= 2111
|
|
usd_light_api = sphere_light.LightAPI();
|
|
#else
|
|
usd_light_api = sphere_light;
|
|
#endif
|
|
break;
|
|
}
|
|
case LA_SUN: {
|
|
pxr::UsdLuxDistantLight distant_light = pxr::UsdLuxDistantLight::Define(stage, usd_path);
|
|
/* TODO(makowalski): set angle attribute here. */
|
|
#if PXR_VERSION >= 2111
|
|
usd_light_api = distant_light.LightAPI();
|
|
#else
|
|
usd_light_api = distant_light;
|
|
#endif
|
|
break;
|
|
}
|
|
default:
|
|
BLI_assert_msg(0, "is_supported() returned true for unsupported light type");
|
|
}
|
|
|
|
/* Scale factor to get to somewhat-similar illumination. Since the USDViewer had similar
|
|
* over-exposure as Blender Internal with the same values, this code applies the reverse of the
|
|
* versioning code in light_emission_unify(). */
|
|
float usd_intensity;
|
|
if (light->type == LA_SUN) {
|
|
/* Untested, as the Hydra GL viewport of USDViewer doesn't support distant lights. */
|
|
usd_intensity = light->energy;
|
|
}
|
|
else {
|
|
usd_intensity = light->energy / 100.0f;
|
|
}
|
|
usd_light_api.CreateIntensityAttr().Set(usd_intensity, timecode);
|
|
|
|
usd_light_api.CreateColorAttr().Set(pxr::GfVec3f(light->r, light->g, light->b), timecode);
|
|
usd_light_api.CreateSpecularAttr().Set(light->spec_fac, timecode);
|
|
}
|
|
|
|
} // namespace blender::io::usd
|