You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
3.6 KiB
Python
85 lines
3.6 KiB
Python
"""
|
|
Copyright (c) Contributors to the Open 3D Engine Project
|
|
|
|
SPDX-License-Identifier: Apache-2.0 OR MIT
|
|
"""
|
|
|
|
# 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')
|