class OrbitTransformController(QTransform):
targetChanged = pyqtSignal()
angleChanged = pyqtSignal()
radiusChanged = pyqtSignal()
def __init__(self, parent):
QTransform.__init__(self, parent)
self.m_target = QTransform()
self.m_matrix = QMatrix4x4()
self.m_radius = 1.0
self.m_angle = 0.0
def target(self):
return self.m_target
def setTarget(self, target):
if self.m_target == target:
return
self.m_target = target
self.targetChanged.emit()
def setRadius(self, radius):
if fuzzyCompareDouble(radius, self.m_radius):
return
self.m_radius = radius
self.radiusChanged.emit()
def radius(self, ):
return self.m_radius
def setAngle(self, angle):
if fuzzyCompareDouble(angle, self.m_angle):
return
self.m_angle = angle
self.updateMatrix()
self.angleChanged.emit()
def angle(self):
return self.m_angle
def updateMatrix(self, ):
self.m_matrix.setToIdentity()
self.m_matrix.rotate(self.m_angle, QVector3D(0.0, 1.0, 0.0))
self.m_matrix.translate(self.m_radius, 0.0, 0.0)
self.m_target.setMatrix(self.m_matrix)
angle = pyqtProperty(float, fget=angle, fset=setAngle, notify=angleChanged)
radius = pyqtProperty(float,
fget=radius,
fset=setRadius,
notify=radiusChanged)
target = pyqtProperty(QTransform,
fget=target,
fset=setTarget,
notify=angleChanged)
def __init__(self, parent=None):
super(TrefoilKnot, self).__init__(parent)
self.m_mesh = QMesh()
self.m_transform = QTransform()
self.m_theta = 0.0
self.m_phi = 0.0
self.m_position = QVector3D()
self.m_scale = 1.0
self.m_mesh.setSource(QUrl.fromLocalFile('assets/obj/trefoil.obj'))
self.addComponent(self.m_mesh)
self.addComponent(self.m_transform)
class OrbitTransformController(QObject):
def __init__(self, parent):
super(OrbitTransformController, self).__init__(parent)
self.m_target = QTransform()
self.m_matrix = QMatrix4x4()
self.m_radius = 1.0
self.m_angle = 0
def getTarget(self):
return self.m_target
def setTarget(self, target):
if self.m_target != target:
self.m_target = target
self.targetChanged.emit()
def getRadius(self):
return self.m_radius
def setRadius(self, radius):
if not QtCore.qFuzzyCompare(self.m_radius, radius):
self.m_radius = radius
self.updateMatrix()
self.radiusChanged.emit()
def getAngle(self):
return self.m_angle
def setAngle(self, angle):
if not QtCore.qFuzzyCompare(angle, self.m_angle):
self.m_angle = angle
self.updateMatrix()
self.angleChanged.emit()
def updateMatrix(self):
self.m_matrix.setToIdentity()
self.m_matrix.rotate(self.m_angle, QVector3D(0, 1, 0))
self.m_matrix.translate(self.m_radius, 0, 0)
self.m_target.setMatrix(self.m_matrix)
# QSignal
targetChanged = pyqtSignal()
radiusChanged = pyqtSignal()
angleChanged = pyqtSignal()
# Qt properties
target = pyqtProperty(QTransform, fget=getTarget, fset=setTarget)
radius = pyqtProperty(float, fget=getRadius, fset=setRadius)
angle = pyqtProperty(float, fget=getAngle, fset=setAngle)
def add_rgb_cube(self, w, h, d):
"""
Helper method to add an RGB-textured cube to the scene.
Arguments:
w: float, the width of the cube
h: float, the height of the cube
d: float, the depth of the cube
Returns:
the QTransform of the cube
"""
cube_entity = QEntity(self.scene)
cube_mesh = QCuboidMesh()
cube_mesh.setXExtent(w)
cube_mesh.setYExtent(h)
cube_mesh.setZExtent(d)
cube_entity.addComponent(cube_mesh)
cube_transform = QTransform(self.scene)
cube_entity.addComponent(cube_transform)
if not hasattr(self, 'rgb_cube_material'):
# load material definition from QML
# this was the easiest way I could find to create a custom shader in Qt3D...
self.rgb_cube_material = self.load_qml(
os.path.join(os.path.dirname(__file__), 'RGBCubeMaterial.qml'),
self.scene)
cube_entity.addComponent(self.rgb_cube_material)
return cube_transform
def add_light(self, position, intensity=1.0, color=util.hsl(0, 0, 100)):
"""
Helper method to add a simple point light to the scene.
Arguments:
position: QVector3D, the position of the light in the scene
intensity: float, the intensity of the light
color: QColor, the color of the light
"""
light_entity = QEntity(self.scene)
light = QPointLight(light_entity)
light.setColor(color)
light.setIntensity(intensity)
light_transform = QTransform(self.scene)
light_transform.setTranslation(position)
light_entity.addComponent(light)
light_entity.addComponent(light_transform)
def __init__(self, parent=None):
super(RenderableEntity, self).__init__(parent)
self.m_mesh = QMesh()
self.m_transform = QTransform()
self.addComponent(self.m_mesh)
self.addComponent(self.m_transform)
def createScene(self):
# Root entity
rootEntity = QEntity()
light_entity = QEntity(rootEntity)
light = QPointLight(light_entity)
light.setColor(QColor(255, 255, 255))
light.setIntensity(0.6)
trans = QTransform()
trans.setTranslation(QVector3D(0, 0, 2))
light_entity.addComponent(trans)
light_entity.addComponent(light)
# Material
material = QPhongMaterial(rootEntity)
material.setAmbient(QColor(100, 100, 100))
# material.setShininess(0)
self.robot_entity = QEntity(rootEntity)
f1_mesh = QMesh()
f1_mesh.setSource(QUrl('qrc:/assets/' + self.robot_type + '.obj'))
self.robot_entity.addComponent(f1_mesh)
self.robot_entity.addComponent(material)
sphereTransform = QTransform()
controller = OrbitTransformController(sphereTransform)
controller.setTarget(sphereTransform)
controller.setRadius(0.0)
self.sphereRotateTransformAnimation = QPropertyAnimation(
sphereTransform)
self.sphereRotateTransformAnimation.setTargetObject(controller)
self.sphereRotateTransformAnimation.setPropertyName(b"angle")
self.sphereRotateTransformAnimation.setStartValue(0)
self.sphereRotateTransformAnimation.setEndValue(360)
self.sphereRotateTransformAnimation.setDuration(10000)
self.sphereRotateTransformAnimation.setLoopCount(-1)
self.robot_entity.addComponent(sphereTransform)
self.start_animation()
return rootEntity
def create_atom(self, atom: Atom, root_entity, color: QColor):
if atom.element in self._materials_dict:
material = self._materials_dict[atom.element]
else:
material = QDiffuseSpecularMaterial(root_entity)
material.setDiffuse(color)
material.setAmbient(color.darker(200))
self._materials_dict[atom.element] = material
sphere_entity = QEntity(root_entity)
sphere_mesh = QSphereMesh()
sphere_mesh.setRadius(0.5)
sphere_transform = QTransform()
sphere_transform.setTranslation(QVector3D(*atom.coords_cartesian))
sphere_entity.addComponent(sphere_mesh)
sphere_entity.addComponent(sphere_transform)
sphere_entity.addComponent(material)
def createScene(self):
root = QEntity()
# Material
material = QPhongMaterial(root)
material.setAmbient(QColor(100, 100, 0))
material.setDiffuse(QColor(100, 0, 100))
cube_E = QEntity(root)
cube_mesh = QCuboidMesh()
cube_mesh.setXExtent(2)
cube_mesh.setYExtent(2)
cube_mesh.setZExtent(2)
cube_tr = QTransform()
cube_tr.setScale3D(QVector3D(2, 2, 2))
cube_E.addComponent(cube_mesh)
cube_E.addComponent(cube_tr)
cube_E.addComponent(material)
return root
def createScene():
# Root entity.
rootEntity = QEntity()
# Material.
material = QPhongMaterial(rootEntity)
# Torus.
torusEntity = QEntity(rootEntity)
torusMesh = QTorusMesh()
torusMesh.setRadius(5)
torusMesh.setMinorRadius(1)
torusMesh.setRings(100)
torusMesh.setSlices(20)
torusTransform = QTransform()
torusTransform.setScale3D(QVector3D(1.5, 1.0, 0.5))
torusTransform.setRotation(
QQuaternion.fromAxisAndAngle(QVector3D(1.0, 0.0, 0.0), 45.0))
torusEntity.addComponent(torusMesh)
torusEntity.addComponent(torusTransform)
torusEntity.addComponent(material)
# Sphere.
sphereEntity = QEntity(rootEntity)
sphereMesh = QSphereMesh()
sphereMesh.setRadius(3)
sphereEntity.addComponent(sphereMesh)
sphereEntity.addComponent(material)
return rootEntity
def createScene():
# Root entity
rootEntity = QEntity()
# Material
material = QPhongMaterial(rootEntity)
# Torus
torusEntity = QEntity(rootEntity)
# Qt3DExtras.QTorusMesh *
torusMesh = QTorusMesh()
torusMesh.setRadius(5)
torusMesh.setMinorRadius(1)
torusMesh.setRings(100)
torusMesh.setSlices(20)
# Qt3DCore.QTransform *
torusTransform = QTransform()
torusTransform.setScale3D(QVector3D(1.5, 1, 0.5))
torusTransform.setRotation(
QQuaternion.fromAxisAndAngle(QVector3D(1, 0, 0), 45.0))
torusEntity.addComponent(torusMesh)
torusEntity.addComponent(torusTransform)
torusEntity.addComponent(material)
# Sphere
sphereEntity = QEntity(rootEntity)
sphereMesh = QSphereMesh()
sphereMesh.setRadius(3)
# Qt3DCore.QTransform *
sphereTransform = QTransform()
# OrbitTransformController *
controller = OrbitTransformController(sphereTransform)
controller.setTarget(sphereTransform)
controller.setRadius(20.0)
# QPropertyAnimation *
sphereRotateTransformAnimation = QPropertyAnimation(sphereTransform)
sphereRotateTransformAnimation.setTargetObject(controller)
sphereRotateTransformAnimation.setPropertyName(b"angle")
sphereRotateTransformAnimation.setStartValue(0)
sphereRotateTransformAnimation.setEndValue(360)
sphereRotateTransformAnimation.setDuration(10000)
sphereRotateTransformAnimation.setLoopCount(-1)
sphereRotateTransformAnimation.start()
sphereEntity.addComponent(sphereMesh)
sphereEntity.addComponent(sphereTransform)
sphereEntity.addComponent(material)
return rootEntity
def createScene(self):
# root
rootEntity = QSkyboxEntity()
material = QPhongMaterial(rootEntity)
#skybox = QSkyboxEntity(rootEntity)
# torus
cubeEntity = QEntity(rootEntity)
cubeMesh = QCuboidMesh()
cubeTransform = QTransform()
#cubeMaterial = getRgbCubeMaterial(cubeEntity)
cubeMaterial = QPhongMaterial(cubeEntity)
cubeTransform.setTranslation(QVector3D(5.0, -4.0, 0.0))
cubeTransform.setScale(4.0)
cubeTransform.setRotation(QQuaternion.fromAxisAndAngle(QVector3D(1, 0, 0), 45))
cubeEntity.addComponent(cubeMesh)
cubeEntity.addComponent(cubeTransform)
cubeEntity.addComponent(cubeMaterial)
return rootEntity
def add_sphere(self, r):
"""
Helper method to add a sphere to the scene.
Arguments:
r: float, the radius of the sphere
Returns:
the QTransform of the sphere
"""
sphere_entity = QEntity(self.scene)
sphere_mesh = QSphereMesh()
sphere_mesh.setRadius(r)
sphere_entity.addComponent(sphere_mesh)
sphere_transform = QTransform(self.scene)
sphere_entity.addComponent(sphere_transform)
if not hasattr(self, 'sphere_material'):
self.sphere_material = QPhongMaterial(self.scene)
self.sphere_material.setAmbient(util.hsl(0, 0, 50))
sphere_entity.addComponent(self.sphere_material)
return sphere_transform
def add_plane(self, color=util.hsl(0, 0, 50)):
"""
Helper method to add a plane to the scene.
Arguments:
color: QColor, the color of the plane's material
Returns:
the QTransform of the plane
"""
plane_entity = QEntity(self.scene)
plane_mesh = QPlaneMesh()
plane_entity.addComponent(plane_mesh)
plane_transform = QTransform(self.scene)
plane_entity.addComponent(plane_transform)
plane_material = QPhongMaterial(self.scene)
plane_material.setDiffuse(color)
plane_entity.addComponent(plane_material)
return plane_transform
def add_cylinder(self, color=util.hsl(0, 0, 50)):
"""
Helper method to add a cylinder to the scene.
Arguments:
color: QColor, the color of the cylinder's material
Returns:
the QTransform of the cylinder
"""
cylinder_entity = QEntity(self.scene)
cylinder_mesh = QCylinderMesh()
cylinder_entity.addComponent(cylinder_mesh)
cylinder_transform = QTransform(self.scene)
cylinder_entity.addComponent(cylinder_transform)
cylinder_material = QPhongMaterial(self.scene)
cylinder_material.setDiffuse(color)
cylinder_entity.addComponent(cylinder_material)
return cylinder_transform
def add_sphere(self, color=util.hsl(0, 0, 50)):
"""
Helper method to add a sphere to the scene.
Arguments:
color: QColor, the color of the sphere's material
Returns:
the QTransform of the sphere
"""
sphere_entity = QEntity(self.scene)
sphere_mesh = QSphereMesh()
sphere_entity.addComponent(sphere_mesh)
sphere_transform = QTransform(self.scene)
sphere_entity.addComponent(sphere_transform)
sphere_material = QPhongMaterial(self.scene)
sphere_material.setDiffuse(color)
sphere_entity.addComponent(sphere_material)
return sphere_transform
def create_plane(root_entity, vector_1: np.ndarray, vector_2: np.ndarray):
material = QDiffuseSpecularMaterial(root_entity)
material.setDiffuse(QColor(200, 200, 200))
material.setAmbient(QColor(200, 200, 200))
plane_entity = QEntity(root_entity)
plane_mesh = QPlaneMesh()
plane_transform = QTransform()
plane_mesh.setWidth(np.linalg.norm(vector_1))
plane_mesh.setHeight(np.linalg.norm(vector_2))
plane_mesh.setMirrored(False)
translation_vector = (vector_1 + vector_2) / 2
plane_transform.setRotation(_get_quaternion(np.cross(vector_1, vector_2)))
plane_transform.setTranslation(QVector3D(*translation_vector))
plane_entity.addComponent(plane_mesh)
plane_entity.addComponent(plane_transform)
plane_entity.addComponent(material)
def add_path(self, *pts, color=util.hsl(0, 0, 50)):
"""
Helper method to add a path to the scene.
Arguments:
pts: list of QVector3D's, the points in the path
Returns:
a list of the entities added to the scene
"""
path_material = QPhongMaterial(self.scene)
path_material.setDiffuse(color)
# make a bunch of cylinder objects aligned along the path
entities = []
prev_pt = None
for pt in pts:
if prev_pt is not None:
if prev_pt != pt:
# for each adjacent pair of points that are different
# make a cylinder
path_entity = QEntity(self.scene)
path_mesh = QCylinderMesh()
path_mesh.setRadius(0.05) # very thin
path_mesh.setLength((prev_pt - pt).length()) # length is the distance between the points
path_entity.addComponent(path_mesh)
path_transform = QTransform(self.scene)
path_transform.setRotation(QQuaternion.fromDirection(QVector3D(0, 0, -1), prev_pt - pt)) # rotate to point along path
path_transform.setTranslation((pt + prev_pt) / 2) # center between points
path_entity.addComponent(path_transform)
path_entity.addComponent(path_material)
entities.append(path_entity)
prev_pt = pt
return entities
def create_cylinder(root_entity, center, direction):
material = QDiffuseSpecularMaterial(root_entity)
material.setDiffuse(QColor(200, 200, 200, 50))
material.setAmbient(QColor(200, 200, 200, 50))
cylinder_entity = QEntity(root_entity)
cylinder_mesh = QCylinderMesh()
length = np.linalg.norm(direction)
translate_vector = center + direction / 2
cylinder_mesh.setLength(length)
cylinder_mesh.setRadius(0.02)
cylinder_transform = QTransform()
cylinder_transform.setRotation(_get_quaternion(direction))
cylinder_transform.setTranslation(QVector3D(*translate_vector))
cylinder_entity.addComponent(cylinder_mesh)
cylinder_entity.addComponent(cylinder_transform)
cylinder_entity.addComponent(material)
def __init__(self, rootEntity):
super(SceneModifier, self).__init__()
self.m_rootEntity = rootEntity
# Torus shape data.
self.m_torus = QTorusMesh()
self.m_torus.setRadius(1.0)
self.m_torus.setMinorRadius(0.4)
self.m_torus.setRings(100)
self.m_torus.setSlices(20)
# TorusMesh transform.
torusTransform = QTransform()
torusTransform.setScale(2.0)
torusTransform.setRotation(
QQuaternion.fromAxisAndAngle(QVector3D(0.0, 1.0, 0.0), 25.0))
torusTransform.setTranslation(QVector3D(5.0, 4.0, 0.0))
torusMaterial = QPhongMaterial()
torusMaterial.setDiffuse(QColor(0xbeb32b))
# Torus.
self.m_torusEntity = QEntity(self.m_rootEntity)
self.m_torusEntity.addComponent(self.m_torus)
self.m_torusEntity.addComponent(torusMaterial)
self.m_torusEntity.addComponent(torusTransform)
# Cone shape data.
cone = QConeMesh()
cone.setTopRadius(0.5)
cone.setBottomRadius(1)
cone.setLength(3)
cone.setRings(50)
cone.setSlices(20)
# ConeMesh transform.
coneTransform = QTransform()
coneTransform.setScale(1.5)
coneTransform.setRotation(
QQuaternion.fromAxisAndAngle(QVector3D(1.0, 0.0, 0.0), 45.0))
coneTransform.setTranslation(QVector3D(0.0, 4.0, -1.5))
coneMaterial = QPhongMaterial()
coneMaterial.setDiffuse(QColor(0x928327))
# Cone.
self.m_coneEntity = QEntity(self.m_rootEntity)
self.m_coneEntity.addComponent(cone)
self.m_coneEntity.addComponent(coneMaterial)
self.m_coneEntity.addComponent(coneTransform)
# Cylinder shape data.
cylinder = QCylinderMesh()
cylinder.setRadius(1)
cylinder.setLength(3)
cylinder.setRings(100)
cylinder.setSlices(20)
# CylinderMesh transform.
cylinderTransform = QTransform()
cylinderTransform.setScale(1.5)
cylinderTransform.setRotation(
QQuaternion.fromAxisAndAngle(QVector3D(1.0, 0.0, 0.0), 45.0))
cylinderTransform.setTranslation(QVector3D(-5.0, 4.0, -1.5))
cylinderMaterial = QPhongMaterial()
cylinderMaterial.setDiffuse(QColor(0x928327))
# Cylinder.
self.m_cylinderEntity = QEntity(self.m_rootEntity)
self.m_cylinderEntity.addComponent(cylinder)
self.m_cylinderEntity.addComponent(cylinderMaterial)
self.m_cylinderEntity.addComponent(cylinderTransform)
# Cuboid shape data.
cuboid = QCuboidMesh()
# CuboidMesh transform.
cuboidTransform = QTransform()
cuboidTransform.setScale(4.0)
cuboidTransform.setTranslation(QVector3D(5.0, -4.0, 0.0))
cuboidMaterial = QPhongMaterial()
cuboidMaterial.setDiffuse(QColor(0x665423))
# Cuboid.
self.m_cuboidEntity = QEntity(self.m_rootEntity)
self.m_cuboidEntity.addComponent(cuboid)
self.m_cuboidEntity.addComponent(cuboidMaterial)
self.m_cuboidEntity.addComponent(cuboidTransform)
# Plane shape data.
planeMesh = QPlaneMesh()
planeMesh.setWidth(2)
planeMesh.setHeight(2)
# Plane mesh transform.
planeTransform = QTransform()
planeTransform.setScale(1.3)
planeTransform.setRotation(
QQuaternion.fromAxisAndAngle(QVector3D(1.0, 0.0, 0.0), 45.0))
planeTransform.setTranslation(QVector3D(0.0, -4.0, 0.0))
planeMaterial = QPhongMaterial()
planeMaterial.setDiffuse(QColor(0xa69929))
# Plane.
self.m_planeEntity = QEntity(self.m_rootEntity)
self.m_planeEntity.addComponent(planeMesh)
self.m_planeEntity.addComponent(planeMaterial)
self.m_planeEntity.addComponent(planeTransform)
# Sphere shape data.
sphereMesh = QSphereMesh()
sphereMesh.setRings(20)
sphereMesh.setSlices(20)
sphereMesh.setRadius(2)
# Sphere mesh transform.
sphereTransform = QTransform()
sphereTransform.setScale(1.3)
sphereTransform.setTranslation(QVector3D(-5.0, -4.0, 0.0))
sphereMaterial = QPhongMaterial()
sphereMaterial.setDiffuse(QColor(0xa69929))
# Sphere.
self.m_sphereEntity = QEntity(self.m_rootEntity)
self.m_sphereEntity.addComponent(sphereMesh)
self.m_sphereEntity.addComponent(sphereMaterial)
self.m_sphereEntity.addComponent(sphereTransform)
def __init__(self, app):
view = Qt3DWindow()
# view.defaultFramegraph().setClearColor(QColor(0x4d4d4f))
container = QWidget.createWindowContainer(view)
screenSize = view.screen().size()
container.setMinimumSize(QSize(200, 100))
container.setMaximumSize(screenSize)
widget = QWidget()
self.hLayout = QHBoxLayout(widget)
self.vLayout = QVBoxLayout()
self.vLayout.setAlignment(Qt.AlignTop)
self.hLayout.addWidget(container, 1)
self.hLayout.addLayout(self.vLayout)
widget.setWindowTitle("3D Viewer")
aspect = QInputAspect()
view.registerAspect(aspect)
# Root entity.
self.rootEntity = QEntity()
# Camera.
cameraEntity = view.camera()
cameraEntity.lens().setPerspectiveProjection(45.0, 16.0 / 9.0, 0.1,
1000.0)
cameraEntity.setPosition(QVector3D(0.0, -50.0, -0.5))
cameraEntity.setUpVector(QVector3D(0.0, 1.0, 0.0))
cameraEntity.setViewCenter(QVector3D(0.0, 0.0, 0.0))
# Light
lightEntity = QEntity(self.rootEntity)
light = QPointLight(lightEntity)
light.setColor(QColor.fromRgbF(1.0, 1.0, 1.0, 1.0))
light.setIntensity(1)
lightEntity.addComponent(light)
lightTransform = QTransform(lightEntity)
lightTransform.setTranslation(QVector3D(10.0, 40.0, 0.0))
lightEntity.addComponent(lightTransform)
# For camera controls.
camController = QFirstPersonCameraController(self.rootEntity)
camController.setCamera(cameraEntity)
# Set root object of the scene.
view.setRootEntity(self.rootEntity)
modifier = SceneModifier(self.rootEntity)
moveLeft = QPushButton(text="Left")
moveLeft.clicked.connect(modifier.transformLeft)
moveLeft.setAutoRepeat(True)
moveRight = QPushButton(text="Right")
moveRight.clicked.connect(modifier.transformRight)
moveRight.setAutoRepeat(True)
moveUp = QPushButton(text="Up")
moveUp.clicked.connect(modifier.transformUp)
moveUp.setAutoRepeat(True)
moveDown = QPushButton(text="Down")
moveDown.clicked.connect(modifier.transformDown)
moveDown.setAutoRepeat(True)
scaleDown = QPushButton(text="Scale Down")
scaleDown.clicked.connect(modifier.scaleDown)
scaleDown.setAutoRepeat(True)
scaleUp = QPushButton(text="Scale Up")
scaleUp.clicked.connect(modifier.scaleUp)
scaleUp.setAutoRepeat(True)
switchModel = QPushButton(text="Switch Model")
switchModel.clicked.connect(self.modifier.handlePickerPress)
loadModel = QPushButton(text="Load Model")
loadModel.clicked.connect(self.modifier.loadscene)
self.vLayout.addWidget(moveLeft)
self.vLayout.addWidget(moveRight)
self.vLayout.addWidget(moveUp)
self.vLayout.addWidget(moveDown)
self.vLayout.addWidget(scaleUp)
self.vLayout.addWidget(scaleDown)
self.vLayout.addWidget(switchModel)
self.vLayout.addWidget(loadModel)
# Show the window.
widget.show()
widget.resize(1200, 800)
sys.exit(app.exec_())
def __init__(self, parent):
super(OrbitTransformController, self).__init__(parent)
self.m_target = QTransform()
self.m_matrix = QMatrix4x4()
self.m_radius = 1.0
self.m_angle = 0
def create3DWidget(self):
view = Qt3DWindow()
# view.defaultFramegraph().setClearColor(QColor(0x4d4d4f))
container = QWidget.createWindowContainer(view)
screenSize = view.screen().size()
container.setMinimumSize(QSize(200, 100))
container.setMaximumSize(screenSize)
aspect = QInputAspect()
view.registerAspect(aspect)
# Root entity.
self.rootEntity = QEntity()
# Camera.
cameraEntity = view.camera()
cameraEntity.lens().setPerspectiveProjection(45.0, 16.0 / 9.0, 0.1,
1000.0)
cameraEntity.setPosition(QVector3D(0.0, -50.0, -0.5))
cameraEntity.setUpVector(QVector3D(0.0, -1.0, 0.0))
cameraEntity.setViewCenter(QVector3D(0.0, 0.0, 0.0))
# Light
lightEntity = QEntity(self.rootEntity)
light = QPointLight(lightEntity)
light.setColor(QColor.fromRgbF(1.0, 1.0, 1.0, 1.0))
light.setIntensity(1)
lightEntity.addComponent(light)
lightTransform = QTransform(lightEntity)
lightTransform.setTranslation(QVector3D(10.0, -40.0, 0.0))
lightEntity.addComponent(lightTransform)
# For camera controls.
camController = QFirstPersonCameraController(self.rootEntity)
camController.setCamera(cameraEntity)
# Set root object of the scene.
view.setRootEntity(self.rootEntity)
self.modifier = SceneModifier(self.rootEntity)
moveLeft = QPushButton(text="Left")
moveLeft.clicked.connect(self.modifier.transformLeft)
moveLeft.setAutoRepeat(True)
moveRight = QPushButton(text="Right")
moveRight.clicked.connect(self.modifier.transformRight)
moveRight.setAutoRepeat(True)
moveUp = QPushButton(text="Up")
moveUp.clicked.connect(self.modifier.transformUp)
moveUp.setAutoRepeat(True)
moveDown = QPushButton(text="Down")
moveDown.clicked.connect(self.modifier.transformDown)
moveDown.setAutoRepeat(True)
moveX = QPushButton(text="RotateX")
moveX.clicked.connect(self.modifier.rotateX)
moveX.setAutoRepeat(True)
moveY = QPushButton(text="RotateY")
moveY.clicked.connect(self.modifier.rotateY)
moveY.setAutoRepeat(True)
moveZ = QPushButton(text="RotateZ")
moveZ.clicked.connect(self.modifier.rotateZ)
moveZ.setAutoRepeat(True)
scaleDown = QPushButton(text="Scale Down")
scaleDown.clicked.connect(self.modifier.scaleDown)
scaleDown.setAutoRepeat(True)
scaleUp = QPushButton(text="Scale Up")
scaleUp.clicked.connect(self.modifier.scaleUp)
scaleUp.setAutoRepeat(True)
switchModel = QPushButton(text="Switch Model")
switchModel.clicked.connect(self.modifier.handlePickerPress)
loadModel = QPushButton(text="Load Model")
loadModel.clicked.connect(self.modifier.loadscene)
self.vLayout.addWidget(moveLeft)
self.vLayout.addWidget(moveRight)
self.vLayout.addWidget(moveUp)
self.vLayout.addWidget(moveDown)
self.vLayout.addWidget(moveX)
self.vLayout.addWidget(moveY)
self.vLayout.addWidget(moveZ)
self.vLayout.addWidget(scaleUp)
self.vLayout.addWidget(scaleDown)
self.vLayout.addWidget(switchModel)
self.vLayout.addWidget(loadModel)
self.scribbleArea.modifier = self.modifier
return container
class TrefoilKnot(QEntity):
def __init__(self, parent=None):
super(TrefoilKnot, self).__init__(parent)
self.m_mesh = QMesh()
self.m_transform = QTransform()
self.m_theta = 0.0
self.m_phi = 0.0
self.m_position = QVector3D()
self.m_scale = 1.0
self.m_mesh.setSource(QUrl.fromLocalFile('assets/obj/trefoil.obj'))
self.addComponent(self.m_mesh)
self.addComponent(self.m_transform)
def updateTransform(self):
m = QMatrix4x4()
m.translate(self.m_position)
m.rotate(self.m_phi, QVector3D(1.0, 0.0, 0.0))
m.rotate(self.m_phi, QVector3D(0.0, 1.0, 0.0))
m.scale(self.m_scale)
self.m_transform.setMatrix(m)
thetaChanged = pyqtSignal(float)
@pyqtProperty(float, notify=thetaChanged)
def theta(self):
return self.m_theta
@theta.setter
def theta(self, value):
if self.m_theta != value:
self.m_theta = value
self.thetaChanged.emit(value)
self.updateTransform()
phiChanged = pyqtSignal(float)
@pyqtProperty(float, notify=phiChanged)
def phi(self):
return self.m_phi
@phi.setter
def phi(self, value):
if self.m_phi != value:
self.m_phi = value
self.phiChanged.emit(value)
self.updateTransform()
scaleChanged = pyqtSignal(float)
@pyqtProperty(float, notify=scaleChanged)
def scale(self):
return self.m_scale
@scale.setter
def scale(self, value):
if self.m_scale != value:
self.m_scale = value
self.scaleChanged.emit(value)
self.updateTransform()
positionChanged = pyqtSignal(QVector3D)
@pyqtProperty(QVector3D, notify=positionChanged)
def position(self):
return self.m_position
@position.setter
def position(self, value):
if self.m_position != value:
self.m_position = value
self.positionChanged.emit(value)
self.updateTransform()
def __init__(self, parent):
super().__init__(parent)
self.m_target = QTransform()
self.m_matrix = QMatrix4x4()
self.m_radius = 1.0
self.m_angle = 0.0
