Brian Herrera
90 lines
4.0 KiB
Python
Executable File
90 lines
4.0 KiB
Python
Executable File
"""
|
|
All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
its licensors.
|
|
|
|
For complete copyright and license terms please see the LICENSE at the root of this
|
|
distribution (the "License"). All use of this software is governed by the License,
|
|
or, if provided, by the license below or the license accompanying this file. Do not
|
|
remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
"""
|
|
|
|
# Tests the Entity Selection Python API while the Editor is running
|
|
|
|
import azlmbr.bus as bus
|
|
import azlmbr.editor as editor
|
|
from azlmbr.entity import EntityId
|
|
|
|
def createTestEntities(count):
|
|
testEntityIds = []
|
|
|
|
for i in range(count):
|
|
entityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', EntityId())
|
|
testEntityIds.append(entityId)
|
|
print("Entity " + str(i) + " created.")
|
|
editor.EditorEntityAPIBus(bus.Event, 'SetName', entityId, "TestEntity")
|
|
|
|
return testEntityIds
|
|
|
|
# Open a level (any level should work)
|
|
editor.EditorToolsApplicationRequestBus(bus.Broadcast, 'OpenLevelNoPrompt', 'WaterSample')
|
|
|
|
|
|
# Create new test entities at the root level
|
|
numTestEntities = 10
|
|
testEntityIds = createTestEntities(numTestEntities)
|
|
|
|
# Select all test entities
|
|
editor.ToolsApplicationRequestBus(bus.Broadcast, 'SetSelectedEntities', testEntityIds)
|
|
|
|
# Get all test entities selected and check if any entity selected
|
|
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
|
|
areAnyEntitiesSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'AreAnyEntitiesSelected')
|
|
|
|
if(len(testEntityIds) == len(selectedTestEntityIds) and areAnyEntitiesSelected):
|
|
print("SUCCESS: All Test Entities Selected")
|
|
|
|
# Mark first test entity deselected
|
|
editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntityDeselected', testEntityIds[0])
|
|
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
|
|
isSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'IsSelected', testEntityIds[0])
|
|
|
|
if(len(testEntityIds) - 1 == len(selectedTestEntityIds) and not isSelected):
|
|
print("SUCCESS: One Test Entity Marked as DeSelected")
|
|
|
|
# Mark first test entity selected
|
|
editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntitySelected', testEntityIds[0])
|
|
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
|
|
isSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'IsSelected', testEntityIds[0])
|
|
|
|
if(len(testEntityIds) == len(selectedTestEntityIds) and isSelected):
|
|
print("SUCCESS: One New Test Entity Marked as Selected")
|
|
|
|
# Mark first half of test entities as deselected
|
|
halfNumTestEntities = len(testEntityIds)//2
|
|
|
|
editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntitiesDeselected', testEntityIds[:halfNumTestEntities])
|
|
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
|
|
|
|
if(len(testEntityIds) - halfNumTestEntities == len(selectedTestEntityIds)):
|
|
print("SUCCESS: Half of Test Entity Marked as DeSelected")
|
|
|
|
# Mark first half test entity as selected
|
|
editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntitiesSelected', testEntityIds[:halfNumTestEntities])
|
|
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
|
|
|
|
if(len(testEntityIds) == len(selectedTestEntityIds)):
|
|
print("SUCCESS: All Test Entities Marked as Selected")
|
|
|
|
# Clear all test entities selected
|
|
editor.ToolsApplicationRequestBus(bus.Broadcast, 'SetSelectedEntities', [])
|
|
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
|
|
areAnyEntitiesSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'AreAnyEntitiesSelected')
|
|
|
|
if(len(selectedTestEntityIds) == 0 and not areAnyEntitiesSelected):
|
|
print("SUCCESS: Clear All Test Entities Selected")
|
|
|
|
|
|
# Close Editor
|
|
editor.EditorToolsApplicationRequestBus(bus.Broadcast, 'ExitNoPrompt')
|