Fix test for UFE lights from MayaUSD

Autodesk · Nov 16, 2023 · 584bb35 · 584bb35
1 parent debf297
commit 584bb35
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/test/lib/mayaUsd/render/mayaToHydra/cpp/testMayaUsdUfeItems.cpp b/test/lib/mayaUsd/render/mayaToHydra/cpp/testMayaUsdUfeItems.cpp
@@ -29,14 +29,9 @@ using namespace MayaHydra;
 
 bool IsUfeLight(const HdSceneIndexBasePtr& sceneIndex, const SdfPath& primPath)
 {
-    bool isUfeLightPrim = primPath.GetElementString().find("ufeLightProxy") < std::string::npos;
-    if (!isUfeLightPrim) {
-        return false;
-    }
-    // The shape prim is used to display the light's wireframe and is the only prim we want to allow
-    // for UFE lights
-    bool isShapePrim = primPath.GetElementString().find("ufeLightProxyShape") < std::string::npos;
-    return !isShapePrim;
+    HdSceneIndexPrim prim = sceneIndex->GetPrim(primPath);
+    return HdPrimTypeIsLight(prim.primType)
+        && primPath.GetElementString().find("ufeLightProxy") < std::string::npos;
 }
 
 TEST(MayaUsdUfeItems, skipUsdUfeLights)