1 files changed, 140 insertions, 0 deletions
diff --git a/sources/pyside6/tests/Qt3DExtras/qt3dextras_test.py b/sources/pyside6/tests/Qt3DExtras/qt3dextras_test.py
new file mode 100644
index 000000000..23e2d6e7e
--- /dev/null
+++ b/sources/pyside6/tests/Qt3DExtras/qt3dextras_test.py
@@ -0,0 +1,140 @@
+#!/usr/bin/python
+# Copyright (C) 2022 The Qt Company Ltd.
+# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
+
+'''Test cases for Q3DExtras'''
+
+import os
+import sys
+import unittest
+
+from pathlib import Path
+sys.path.append(os.fspath(Path(__file__).resolve().parents[1]))
+from init_paths import init_test_paths
+init_test_paths(False)
+
+from helper.usesqapplication import UsesQApplication
+
+from PySide6.QtCore import (Property, QObject, QPropertyAnimation, QTimer, Signal, Slot)
+from PySide6.QtGui import (QGuiApplication, QMatrix4x4, QQuaternion, QVector3D, QWindow)
+from PySide6.Qt3DCore import (Qt3DCore)
+from PySide6.Qt3DRender import (Qt3DRender)
+from PySide6.Qt3DExtras import (Qt3DExtras)
+
+
+class OrbitTransformController(QObject):
+    def __init__(self, parent):
+        super().__init__(parent)
+        self._target = None
+        self._matrix = QMatrix4x4()
+        self._radius = 1
+        self._angle = 0
+
+    def setTarget(self, t):
+        self._target = t
+
+    def getTarget(self):
+        return self._target
+
+    def setRadius(self, radius):
+        if self._radius != radius:
+            self._radius = radius
+            self.updateMatrix()
+            self.radiusChanged.emit()
+
+    def getRadius(self):
+        return self._radius
+
+    def setAngle(self, angle):
+        if self._angle != angle:
+            self._angle = angle
+            self.updateMatrix()
+            self.angleChanged.emit()
+
+    def getAngle(self):
+        return self._angle
+
+    def updateMatrix(self):
+        self._matrix.setToIdentity()
+        self._matrix.rotate(self._angle, QVector3D(0, 1, 0))
+        self._matrix.translate(self._radius, 0, 0)
+        if self._target is not None:
+            self._target.setMatrix(self._matrix)
+
+    angleChanged = Signal()
+    radiusChanged = Signal()
+    angle = Property(float, getAngle, setAngle, notify=angleChanged)
+    radius = Property(float, getRadius, setRadius, notify=radiusChanged)
+
+
+class Window(Qt3DExtras.Qt3DWindow):
+    def __init__(self):
+        super().__init__()
+
+        # Camera
+        self.camera().lens().setPerspectiveProjection(45, 16 / 9, 0.1, 1000)
+        self.camera().setPosition(QVector3D(0, 0, 40))
+        self.camera().setViewCenter(QVector3D(0, 0, 0))
+
+        # For camera controls
+        self.createScene()
+        self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
+        self.camController.setLinearSpeed(50)
+        self.camController.setLookSpeed(180)
+        self.camController.setCamera(self.camera())
+
+        self.setRootEntity(self.rootEntity)
+
+    def createScene(self):
+        # Root entity
+        self.rootEntity = Qt3DCore.QEntity()
+
+        # Material
+        self.material = Qt3DExtras.QPhongMaterial(self.rootEntity)
+
+        # Torus
+        self.torusEntity = Qt3DCore.QEntity(self.rootEntity)
+        self.torusMesh = Qt3DExtras.QTorusMesh()
+        self.torusMesh.setRadius(5)
+        self.torusMesh.setMinorRadius(1)
+        self.torusMesh.setRings(100)
+        self.torusMesh.setSlices(20)
+
+        self.torusTransform = Qt3DCore.QTransform()
+        self.torusTransform.setScale3D(QVector3D(1.5, 1, 0.5))
+        self.torusTransform.setRotation(QQuaternion.fromAxisAndAngle(QVector3D(1, 0, 0), 45))
+
+        self.torusEntity.addComponent(self.torusMesh)
+        self.torusEntity.addComponent(self.torusTransform)
+        self.torusEntity.addComponent(self.material)
+
+        # Sphere
+        self.sphereEntity = Qt3DCore.QEntity(self.rootEntity)
+        self.sphereMesh = Qt3DExtras.QSphereMesh()
+        self.sphereMesh.setRadius(3)
+
+        self.sphereTransform = Qt3DCore.QTransform()
+        self.controller = OrbitTransformController(self.sphereTransform)
+        self.controller.setTarget(self.sphereTransform)
+        self.controller.setRadius(20)
+
+        self.sphereEntity.addComponent(self.sphereMesh)
+        self.sphereEntity.addComponent(self.sphereTransform)
+        self.sphereEntity.addComponent(self.material)
+
+
+class Qt3DExtrasTestCase(UsesQApplication):
+    '''Tests related to Q3DExtras'''
+
+    def test3DExtras(self):
+
+        window = Window()
+        window.show()
+        while not window.isExposed():
+            self.app.processEvents()
+        QTimer.singleShot(2000, self.app.quit)
+        self.app.exec()
+
+
+if __name__ == '__main__':
+    unittest.main()