|
| 1 | +import unittest |
| 2 | + |
| 3 | +import os.path |
| 4 | +import tempfile |
| 5 | +from distutils.dir_util import copy_tree |
| 6 | + |
| 7 | +import fixturesUtils |
| 8 | +import testUtils |
| 9 | +import mayaUsd.lib |
| 10 | +import mayaUsd.ufe |
| 11 | +from maya import cmds |
| 12 | +from pxr import Sdf, Usd |
| 13 | + |
| 14 | +from testComponentCreatorBase import _ComponentCreatorTestBase |
| 15 | + |
| 16 | +class SaveToComponentTestCase(_ComponentCreatorTestBase, unittest.TestCase): |
| 17 | + """ |
| 18 | + Tests for usd_component_creator_plugin.create_component.add_to_component_from_nodes |
| 19 | + and then saving and reloading. |
| 20 | + """ |
| 21 | + |
| 22 | + _tempFolder = None |
| 23 | + |
| 24 | + @classmethod |
| 25 | + def setUpClass(cls): |
| 26 | + fixturesUtils.readOnlySetUpClass(__file__, initializeStandalone=False) |
| 27 | + |
| 28 | + testFolderName = "component3Variants" |
| 29 | + fromDirectory = testUtils.getTestScene(testFolderName) |
| 30 | + toDirectory = os.path.join(tempfile.gettempdir(), 'SaveToComponentTestCase') |
| 31 | + copy_tree(fromDirectory, toDirectory) |
| 32 | + SaveToComponentTestCase._tempFolder = toDirectory |
| 33 | + |
| 34 | + @classmethod |
| 35 | + def tearDownClass(cls): |
| 36 | + if SaveToComponentTestCase._tempFolder: |
| 37 | + if os.path.exists(SaveToComponentTestCase._tempFolder): |
| 38 | + import shutil |
| 39 | + shutil.rmtree(SaveToComponentTestCase._tempFolder) |
| 40 | + cls._resetDefaultTemplate() |
| 41 | + return super().tearDownClass() |
| 42 | + |
| 43 | + def _findVariantSet(self, desc, withName): |
| 44 | + variantSetMap = desc.GetVariantSets() |
| 45 | + self.assertGreaterEqual(len(variantSetMap), 1, "There must be at least one variant set") |
| 46 | + for variantSetName, variantSet in variantSetMap.items(): |
| 47 | + if withName not in variantSetName: |
| 48 | + continue |
| 49 | + return variantSet |
| 50 | + return None |
| 51 | + |
| 52 | + def setUp(self): |
| 53 | + self._setUpCC() |
| 54 | + # Clear the variant editor state so there is no lingering component from a |
| 55 | + # previous test. |
| 56 | + self._resetDefaultTemplate() |
| 57 | + return super().setUp() |
| 58 | + |
| 59 | + def testSaveAndReload(self): |
| 60 | + """ |
| 61 | + Open a Maya scene containing a USD stage of a component. |
| 62 | + Modify all variants of the component by replacing its data with a new node. |
| 63 | + Save the scene. Re-open the scene and verify that the component is still |
| 64 | + valid and has the new data. |
| 65 | + """ |
| 66 | + before = self._snapshotProxyShapes() |
| 67 | + |
| 68 | + mayaSceneFilePath = os.path.join(SaveToComponentTestCase._tempFolder, "repro-3530.ma") |
| 69 | + cmds.file(mayaSceneFilePath, open=True) |
| 70 | + |
| 71 | + proxy = self._findNewProxyShape(before) |
| 72 | + self.assertIsNotNone(proxy) |
| 73 | + stage = mayaUsd.ufe.getStage(proxy) |
| 74 | + self.assertIsNotNone(stage) |
| 75 | + desc = self._getDescFromStage(stage) |
| 76 | + self.assertIsNotNone(desc, 'Could not get ComponentDescription from the Maya scene') |
| 77 | + |
| 78 | + first_vs = self._findVariantSet(desc, 'variant_set_1') |
| 79 | + self.assertTrue(first_vs) |
| 80 | + self.assertGreaterEqual(len(first_vs.GetVariants()), 3, "There must be at least three variants in the set {}".format(first_vs.GetName())) |
| 81 | + |
| 82 | + variantsMap = first_vs.GetVariants() |
| 83 | + self.assertGreaterEqual(len(variantsMap), 3, "There must be at least three variants") |
| 84 | + |
| 85 | + for variantName in variantsMap.keys(): |
| 86 | + polyCubeName = cmds.polyCube(name='pCubeExtra')[0] |
| 87 | + self.assertIn('pCubeExtra', polyCubeName) |
| 88 | + |
| 89 | + # The following code is equivalent to add_to_component_from_nodes, |
| 90 | + # but that function does not work in maya batch mode and we need to run this test in batch mode |
| 91 | + # because the Maya save would pop-up a dialog in interactive mode. |
| 92 | + # |
| 93 | + # from usd_component_creator_plugin import add_to_component_from_nodes |
| 94 | + # result = add_to_component_from_nodes( |
| 95 | + # [polyCubeName], |
| 96 | + # [(first_vs.name, variantName)], |
| 97 | + # is_replacing=True, |
| 98 | + # component_desc=desc) |
| 99 | + # self.assertTrue(result) |
| 100 | + |
| 101 | + from AdskUsdComponentCreator import CreateFromFileCommand |
| 102 | + from usd_component_creator_plugin import ExportNodesCommand, execute_ufe_command, AddComponentToManagerCommand, UfeCommandWrapper |
| 103 | + from usd_component_creator_plugin.create_component import _generate_unique_temp_filename, _update_options_for_node, _update_options_for_purpose_from_nodes |
| 104 | + |
| 105 | + component_desc = desc |
| 106 | + nodes = [polyCubeName] |
| 107 | + variant_selections = [(first_vs.name, variantName)] |
| 108 | + is_replacing = True |
| 109 | + export_options = None |
| 110 | + purpose = None |
| 111 | + |
| 112 | + options = component_desc.GetOptions().Clone() |
| 113 | + _update_options_for_node(options, nodes[0], nodes) |
| 114 | + options.component_variants = variant_selections if variant_selections else [] |
| 115 | + options.replace_variant_content = is_replacing |
| 116 | + # Never change the default variant when appending to an existing component. |
| 117 | + options.is_default_variant = False |
| 118 | + |
| 119 | + input_usd_filename = _generate_unique_temp_filename(options) |
| 120 | + export_cmd = ExportNodesCommand(nodes, input_usd_filename, export_options) |
| 121 | + execute_ufe_command(export_cmd) |
| 122 | + |
| 123 | + creation_options = _update_options_for_purpose_from_nodes(options, purpose, input_usd_filename) |
| 124 | + creation_options.Validate() |
| 125 | + |
| 126 | + delete_input_file = False |
| 127 | + create_cmd = CreateFromFileCommand(component_desc, input_usd_filename, creation_options, delete_input_file) |
| 128 | + self.assertTrue(UfeCommandWrapper.executeWithUndo(create_cmd)) |
| 129 | + |
| 130 | + add_cmd = AddComponentToManagerCommand(create_cmd) |
| 131 | + execute_ufe_command(add_cmd) |
| 132 | + |
| 133 | + |
| 134 | + updated_desc = self._getDescFromStage(stage) |
| 135 | + self.assertIsNotNone(updated_desc, 'Could not get updated ComponentDescription') |
| 136 | + |
| 137 | + # Save the file. Make sure the USD edits will go to a USD file. |
| 138 | + cmds.optionVar(intValue=('mayaUsd_ConfirmExistingFileSave', 0)) |
| 139 | + saveLocation = 1 |
| 140 | + cmds.optionVar(intValue=(mayaUsd.lib.OptionVarTokens.SerializedUsdEditsLocation, saveLocation)) |
| 141 | + cmds.file(save=True, force=True) |
| 142 | + |
| 143 | + cmds.file(new=True, force=True) |
| 144 | + cmds.file(mayaSceneFilePath, open=True) |
| 145 | + |
| 146 | + proxy = self._findNewProxyShape(before) |
| 147 | + self.assertIsNotNone(proxy) |
| 148 | + stage = mayaUsd.ufe.getStage(proxy) |
| 149 | + self.assertIsNotNone(stage) |
| 150 | + desc = self._getDescFromStage(stage) |
| 151 | + self.assertIsNotNone(desc, 'Could not get ComponentDescription from the Maya scene') |
| 152 | + |
| 153 | + first_vs = self._findVariantSet(desc, 'variant_set_1') |
| 154 | + self.assertTrue(first_vs) |
| 155 | + |
| 156 | + variantsMap = first_vs.GetVariants() |
| 157 | + self.assertGreaterEqual(len(variantsMap), 3, "There must be at least three variants") |
| 158 | + |
| 159 | + variant_to_cube_map = { |
| 160 | + 'pPlane1': 'pCubeExtra', |
| 161 | + 'pPlane2': 'pCubeExtra1', |
| 162 | + 'pPlane3': 'pCubeExtra2', |
| 163 | + } |
| 164 | + |
| 165 | + found_cubes = set() |
| 166 | + |
| 167 | + for variant in variantsMap.values(): |
| 168 | + primPath = desc.root_prim_path |
| 169 | + prim = stage.GetPrimAtPath(primPath) |
| 170 | + self.assertTrue(prim) |
| 171 | + prim.GetVariantSet(first_vs.GetName()).SetVariantSelection(variant.GetName()) |
| 172 | + |
| 173 | + stage.Reload() |
| 174 | + |
| 175 | + expected_cube_name = variant_to_cube_map.get(variant.GetName()) |
| 176 | + self.assertIsNotNone(expected_cube_name, "No expected cube name for variant {}".format(variant.GetName())) |
| 177 | + |
| 178 | + geoPrim = stage.GetPrimAtPath(Sdf.Path('/root/geo')) |
| 179 | + for child in geoPrim.GetChildren(): |
| 180 | + if child.GetTypeName() == 'Mesh': |
| 181 | + self.assertEqual(expected_cube_name, child.GetName(), f"The variant {variant.GetName()} should have a cube named {expected_cube_name}") |
| 182 | + found_cubes.add(child.GetName()) |
| 183 | + |
| 184 | + self.assertEqual(len(found_cubes), len(variantsMap), "There should be one cube for each variant") |
| 185 | + |
| 186 | + |
| 187 | +if __name__ == '__main__': |
| 188 | + fixturesUtils.runTests(globals()) |
| 189 | + |
0 commit comments