def testExport(self):
"""
The test scene has multiple face set connections to materials. Make sure
the export code has merged the indices correctly.
"""
usdFile = os.path.abspath('UsdExportGeomSubset.usda')
cmds.usdExport(mergeTransformAndShape=True,
file=usdFile,
shadingMode='useRegistry')
stage = Usd.Stage.Open(usdFile)
expected = [
('/pPlane1/initialShadingGroup', [
1,
]),
('/pPlane1/blinn1SG', [
2,
]),
('/pPlane1/phong1SG', [0, 3]),
('/pPlane2/blinn1SG', [0, 3]),
('/pPlane2/phong1SG', [1, 2]),
]
for subset_path, indices in expected:
subset = UsdGeom.Subset(stage.GetPrimAtPath(subset_path))
self.assertEqual(subset.GetElementTypeAttr().Get(),
UsdGeom.Tokens.face)
self.assertEqual(subset.GetIndicesAttr().Get(),
Vt.IntArray(indices))
def test_RoundTrip(self):
usdFile = 'original.usda'
abcFile = 'converted.abc'
self.assertTrue(UsdAbc._WriteAlembic(usdFile, abcFile))
stage = Usd.Stage.Open(abcFile)
prim1 = stage.GetPrimAtPath('/cube1')
self.assertTrue(prim1.IsValid())
faceset1prim1 = stage.GetPrimAtPath('/cube1/faceset1')
faceset2prim1 = stage.GetPrimAtPath('/cube1/faceset2')
self.assertTrue(faceset1prim1.IsValid())
self.assertTrue(faceset2prim1.IsValid())
faceset1_1 = UsdGeom.Subset(faceset1prim1)
faceset2_1 = UsdGeom.Subset(faceset2prim1)
self.assertEqual(faceset1_1.GetElementTypeAttr().Get(), 'face')
self.assertEqual(faceset2_1.GetElementTypeAttr().Get(), 'face')
# Validate the indices for faceset1 (which is animated)
indices = faceset1_1.GetIndicesAttr()
timeSamples = indices.GetTimeSamples()
expectedTimeSamples = [0.0, 1.0]
self.assertEqual(len(timeSamples), len(expectedTimeSamples))
for c, e in zip(timeSamples, expectedTimeSamples):
self.assertTrue(Gf.IsClose(c, e, 1e-5))
expectedFaceIndices = {0.0: [0, 3, 5], 1.0: [3]}
for time, expectedValue in expectedFaceIndices.items():
faceIndices = indices.Get(time)
for c, e in zip(faceIndices, expectedValue):
self.assertEqual(c, e)
# Validate the indices for faceset2.
indices = faceset2_1.GetIndicesAttr()
timeSamples = indices.GetTimeSamples()
# The reason we expect a timesample here even though original.usda
# had only a default is that currently (unfortunately) schema properties
# do not get the metadata that allows defaults to roundtrip back to
# defaults, and it would take some (unrelated to FaceSets) work to
# change the writer to make them do so.
self.assertEqual(len(timeSamples), 1)
expectedFaceIndices = {0.0: [1, 2, 4]}
for time, expectedValue in expectedFaceIndices.items():
faceIndices = indices.Get(time)
for c, e in zip(faceIndices, expectedValue):
self.assertEqual(c, e)
# Initialize checks for /cube2
prim2 = stage.GetPrimAtPath('/cube2')
self.assertTrue(prim2.IsValid())
faceset1prim2 = stage.GetPrimAtPath('/cube2/faceset1')
faceset2prim2 = stage.GetPrimAtPath('/cube2/faceset2')
faceset3prim2 = stage.GetPrimAtPath('/cube2/faceset3')
self.assertTrue(faceset1prim2.IsValid())
self.assertTrue(faceset2prim2.IsValid())
self.assertTrue(faceset3prim2.IsValid())
faceset1_2 = UsdGeom.Subset(faceset1prim2)
faceset2_2 = UsdGeom.Subset(faceset2prim2)
faceset3_2 = UsdGeom.Subset(faceset3prim2)
# Initialize checks for /cube3
prim3 = stage.GetPrimAtPath('/cube3')
self.assertTrue(prim3.IsValid())
faceset1prim3 = stage.GetPrimAtPath('/cube3/faceset')
self.assertTrue(faceset1prim3.IsValid())
faceset1_3 = UsdGeom.Subset(faceset1prim3)
# Check the round-tripping of familyTypes.
imageable1 = UsdGeom.Imageable(prim1)
imageable2 = UsdGeom.Imageable(prim2)
imageable3 = UsdGeom.Imageable(prim3)
# In cube1 we've used "partition" as the familyType. This should be
# converted to "nonOverlapping" as it more closely matches the Alembic
# definition of "partition".
self.assertEqual(faceset1_1.GetFamilyNameAttr().Get(), 'materialBind')
self.assertEqual(faceset2_1.GetFamilyNameAttr().Get(), 'materialBind')
self.assertEqual(
UsdGeom.Subset.GetFamilyType(imageable1, "materialBind"),
"nonOverlapping")
# In cube2 we've used "unrestricted". This should come across directly.
self.assertEqual(faceset1_2.GetFamilyNameAttr().Get(), 'materialBind')
self.assertEqual(faceset2_2.GetFamilyNameAttr().Get(), 'materialBind')
self.assertEqual(
UsdGeom.Subset.GetFamilyType(imageable2, "materialBind"),
"unrestricted")
# We've also added another familyName in addition. We should see this
# being lost in the round-trip and converted to "materialBind".
self.assertEqual(faceset3_2.GetFamilyNameAttr().Get(), 'materialBind')
# In cube3, no familyName or familyType has been specified. Upon
# round-tripping, a default value of "unrestricted" will appear on the
# the default "materialBind" familyName.
self.assertEqual(faceset1_3.GetFamilyNameAttr().Get(), 'materialBind')
self.assertEqual(
UsdGeom.Subset.GetFamilyType(imageable3, "materialBind"),
"unrestricted")
