def test_ValueBlocks(self):
src_stage = Usd.Stage.Open('valueBlocks_root.usda')
def replaceWithFoo(layer, s):
return 'foo'
src_layer_stack = src_stage._GetPcpCache().layerStack
layer = Usd.FlattenLayerStack(src_layer_stack, tag='valueBlocks')
print(layer.ExportToString())
result_stage = Usd.Stage.Open(layer)
# verify that value blocks worked
prim = result_stage.GetPrimAtPath('/Human')
a = prim.GetAttribute('a')
self.assertEqual(a.Get(), Vt.IntArray(1, (1,)))
# a strong value block flattens to a value block
b = prim.GetAttribute('b')
self.assertEqual(b.Get(), None)
# a value block will block both defaults and time samples
c = prim.GetAttribute('c')
self.assertEqual(c.Get(), None)
self.assertEqual(c.Get(1), None)
# strong time samples will override a weaker value block
d = prim.GetAttribute('d')
self.assertEqual(d.Get(1), 789)
def test_ValueTypeMismatch(self):
src_stage = Usd.Stage.Open('valueTypeMismatch_root.usda')
src_layer_stack = src_stage._GetPcpCache().layerStack
layer = Usd.FlattenLayerStack(src_layer_stack, tag='valueBlocks')
print(layer.ExportToString())
result_stage = Usd.Stage.Open(layer)
# Verify that the strongest value type prevailed
prim = result_stage.GetPrimAtPath('/p')
a = prim.GetAttribute('x')
self.assertEqual(a.Get(), Vt.IntArray(1, (0, )))
def test_ResolveAssetPathFn(self):
src_stage = Usd.Stage.Open('emptyAssetPaths.usda')
def replaceWithFoo(layer, s):
return 'foo'
src_layer_stack = src_stage._GetPcpCache().layerStack
layer = Usd.FlattenLayerStack(src_layer_stack,
resolveAssetPathFn=replaceWithFoo,
tag='resolveAssetPathFn')
result_stage = Usd.Stage.Open(layer)
# verify that we replaced asset paths with "foo"
prim = result_stage.GetPrimAtPath('/Test')
assetAttr = prim.GetAttribute('a')
self.assertEqual(assetAttr.Get(), Sdf.AssetPath('foo'))
assetArrayAttr = prim.GetAttribute('b')
self.assertEqual(list(assetArrayAttr.Get()),
[Sdf.AssetPath('foo'), Sdf.AssetPath('foo')])
def test_EmptyAssetPaths(self):
src_stage = Usd.Stage.Open('emptyAssetPaths.usda')
src_layer_stack = src_stage._GetPcpCache().layerStack
layer = Usd.FlattenLayerStack(src_layer_stack, tag='emptyAssetPaths')
result_stage = Usd.Stage.Open(layer)
# Verify that empty asset paths do not trigger coding errors during
# flattening.
prim = result_stage.GetPrimAtPath('/Test')
self.assertTrue(prim.HasAuthoredReferences())
primSpec = layer.GetPrimAtPath('/Test')
expectedRefs = Sdf.ReferenceListOp()
expectedRefs.explicitItems = [Sdf.Reference(primPath="/Ref")]
self.assertEqual(primSpec.GetInfo('references'), expectedRefs)
assetAttr = prim.GetAttribute('a')
self.assertEqual(assetAttr.Get(), Sdf.AssetPath())
assetArrayAttr = prim.GetAttribute('b')
self.assertEqual(list(assetArrayAttr.Get()),
[Sdf.AssetPath(), Sdf.AssetPath()])
def test_Basic(self):
src_stage = Usd.Stage.Open('root.usda')
src_layer_stack = src_stage._GetPcpCache().layerStack
layer = Usd.FlattenLayerStack(src_layer_stack, tag="test.usda")
result_stage = Usd.Stage.Open(layer)
# Confirm that the tag makde it into the display name.
self.assertTrue('test.usda' in layer.GetDisplayName())
print('#'*72)
print('Flattened layer:')
print(layer.ExportToString())
print('#'*72)
# Run the same set of queries against the src and dest stages.
# They should yield the same results.
for stage in [src_stage, result_stage]:
p = stage.GetPrimAtPath('/Sphere')
# Composition arcs remain intact
self.assertTrue(p.HasAuthoredReferences())
self.assertTrue(p.HasAuthoredInherits())
self.assertTrue(p.HasAuthoredPayloads())
self.assertTrue(p.HasVariantSets())
# Classes continue to exist
self.assertTrue( stage.GetPrimAtPath('/_class_Sphere') )
# Default in a strong layer wins over timeSamples in weak.
a = p.GetAttribute('defaultOverTimeSamples')
self.assertEqual( a.GetResolveInfo().GetSource(),
Usd.ResolveInfoSourceDefault )
a = p.GetAttribute('timeSamplesOverDefault')
self.assertEqual( a.GetResolveInfo().GetSource(),
Usd.ResolveInfoSourceTimeSamples)
self.assertEqual( a.Get(), 123 ) # default
self.assertEqual( a.Get(1), 1 ) # time sample
# Layer offsets affect time samples
a = p.GetAttribute('timeSamplesAcrossLayerOffset')
self.assertEqual( a.GetTimeSamples(), [-9.0, 0.0] )
# Layer offsets get folded into reference arcs
p = stage.GetPrimAtPath('/Sphere/ChildFromReference')
a = p.GetAttribute('timeSamplesAcrossRef')
self.assertEqual( a.GetTimeSamples(), [-9.0, 0.0] )
# Confirm result of list-editing.
p = stage.GetPrimAtPath('/ListOpTest')
r = p.GetRelationship('foo')
self.assertEqual( r.GetTargets(),
[Sdf.Path('/Pre_root'),
Sdf.Path('/Pre_sub'),
Sdf.Path('/App_sub'),
Sdf.Path('/App_root')])
# Confirm children from across various kinds of arcs.
for childPath in [
'/Sphere/ChildFromPayload',
'/Sphere/ChildFromReference',
'/Sphere/ChildFromNestedVariant']:
self.assertTrue(stage.GetPrimAtPath(childPath))
# Confirm time samples coming from (offset) clips.
p = stage.GetPrimAtPath('/SphereUsingClip_LegacyForm')
a = p.GetAttribute('xformOp:translate')
self.assertEqual( a.GetResolveInfo(5).GetSource(),
Usd.ResolveInfoSourceValueClips )
p = stage.GetPrimAtPath('/SphereUsingClip')
a = p.GetAttribute('xformOp:translate')
self.assertEqual( a.GetResolveInfo(5).GetSource(),
Usd.ResolveInfoSourceValueClips )
# Confirm time code values were resolve across the offset layer
p = stage.GetPrimAtPath('/TimeCodeTest')
# Prim metadata
self.assertEqual(p.GetMetadata("timeCodeTest"), 0.0)
self.assertEqual(p.GetMetadata("timeCodeArrayTest"),
Sdf.TimeCodeArray([0.0, 10.0]))
self.assertEqual(p.GetMetadata("doubleTest"), 10.0)
self.assertEqual(
p.GetMetadataByDictKey("customData", "timeCode"), 0.0)
self.assertEqual(
p.GetMetadataByDictKey("customData", "timeCodeArray"),
Sdf.TimeCodeArray([0.0, 10.0]))
self.assertEqual(
p.GetMetadataByDictKey("customData", "doubleVal"), 10.0)
self.assertEqual(
p.GetMetadataByDictKey("customData", "subDict:timeCode"), 0.0)
self.assertEqual(
p.GetMetadataByDictKey("customData", "subDict:timeCodeArray"),
Sdf.TimeCodeArray([0.0, 10.0]))
self.assertEqual(
p.GetMetadataByDictKey("customData", "subDict:doubleVal"), 10.0)
# Atribute defaults and time samples
a = p.GetAttribute("TimeCode")
self.assertEqual(a.Get(), 0.0)
self.assertEqual(a.GetTimeSamples(), [-10.0, -9.0])
self.assertEqual(a.Get(-10.0), 0.0)
self.assertEqual(a.Get(-9), 10.0)
a = p.GetAttribute("TimeCodeArray")
self.assertEqual(a.Get(), Sdf.TimeCodeArray([0.0, 10.0]))
self.assertEqual(a.GetTimeSamples(), [-10.0, -9.0])
self.assertEqual(a.Get(-10.0), Sdf.TimeCodeArray([0.0, 20.0]))
self.assertEqual(a.Get(-9), Sdf.TimeCodeArray([10.0, 30.0]))
a = p.GetAttribute("Double")
self.assertEqual(a.Get(), 10.0)
self.assertEqual(a.GetTimeSamples(), [-10.0, -9.0])
self.assertEqual(a.Get(-10.0), 10.0)
self.assertEqual(a.Get(-9), 20.0)
# Variant sets and selections
p = stage.GetPrimAtPath('/assetWithVariant')
self.assertEqual(
p.GetVariantSets().GetNames(),
['variantSet1', 'variantSet2'])
self.assertFalse(stage.GetPrimAtPath('/assetWithVariant/a'))
self.assertTrue(stage.GetPrimAtPath('/assetWithVariant/b'))
self.assertFalse(stage.GetPrimAtPath('/assetWithVariant/c'))
self.assertTrue(stage.GetPrimAtPath('/assetWithVariant/d'))
# Confirm relative asset path handling in the output stage.
p = result_stage.GetPrimAtPath('/Sphere')
a = p.GetAttribute('relativePath')
# It should have become an absolute path.
self.assertTrue(os.path.isabs(a.Get().path))
# Check arrays of paths.
a = p.GetAttribute('relativePathVec')
for path in a.Get():
self.assertTrue(os.path.isabs(path.path) )
# Confirm Sdf-level result of list-editing.
p = layer.GetPrimAtPath('/ListOpTest')
targets = p.relationships['foo'].targetPathList
self.assertEqual(list(targets.prependedItems),
['/Pre_root', '/Pre_sub'])
self.assertEqual(list(targets.appendedItems),
['/App_sub', '/App_root'])
self.assertEqual(list(targets.deletedItems),
['/Del_sub', '/Del_root'])
# Confirm offsets have been folded into value clips.
p1 = layer.GetPrimAtPath('/SphereUsingClip_LegacyForm')
p2 = layer.GetPrimAtPath('/SphereUsingClip')
self.assertEqual( p1.GetInfo('clipActive'),
Vt.Vec2dArray(1, [(-9.0, 0.0)]) )
self.assertEqual( p1.GetInfo('clipTimes'),
Vt.Vec2dArray(2, [(-9.0, 1), (0.0, 10)]) )
self.assertEqual( p2.GetInfo('clips')['default']['active'],
Vt.Vec2dArray(1, [(-9.0, 0)]) )
self.assertEqual( p2.GetInfo('clips')['default']['times'],
Vt.Vec2dArray(2, [(-9.0, 1), (0.0, 10)]) )
# Confirm nested variant sets still exist
self.assertTrue(layer.GetObjectAtPath(
'/Sphere{vset_1=default}{vset_2=default}ChildFromNestedVariant'))
