mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-20 09:39:09 +08:00
90 lines
3.9 KiB
Python
90 lines
3.9 KiB
Python
from PyQt5.QtCore import QTimer
|
|
|
|
from UM.Scene.SceneNode import SceneNode
|
|
from UM.Scene.Iterator.BreadthFirstIterator import BreadthFirstIterator
|
|
from UM.Operations.TranslateOperation import TranslateOperation
|
|
from UM.Operations.ScaleToBoundsOperation import ScaleToBoundsOperation
|
|
from UM.Math.Float import Float
|
|
from UM.Math.Vector import Vector
|
|
from UM.Math.AxisAlignedBox import AxisAlignedBox
|
|
from UM.Application import Application
|
|
|
|
from PlatformPhysicsOperation import PlatformPhysicsOperation
|
|
from ConvexHullJob import ConvexHullJob
|
|
|
|
import time
|
|
import threading
|
|
|
|
class PlatformPhysics:
|
|
def __init__(self, controller, volume):
|
|
super().__init__()
|
|
self._controller = controller
|
|
self._controller.getScene().sceneChanged.connect(self._onSceneChanged)
|
|
self._build_volume = volume
|
|
|
|
self._change_timer = QTimer()
|
|
self._change_timer.setInterval(100)
|
|
self._change_timer.setSingleShot(True)
|
|
self._change_timer.timeout.connect(self._onChangeTimerFinished)
|
|
|
|
def _onSceneChanged(self, source):
|
|
self._change_timer.start()
|
|
|
|
def _onChangeTimerFinished(self):
|
|
root = self._controller.getScene().getRoot()
|
|
for node in BreadthFirstIterator(root):
|
|
if node is root or type(node) is not SceneNode:
|
|
continue
|
|
|
|
bbox = node.getBoundingBox()
|
|
if not bbox or not bbox.isValid():
|
|
continue
|
|
|
|
# Mark the node as outside the build volume if the bounding box test fails.
|
|
if self._build_volume.getBoundingBox().intersectsBox(bbox) != AxisAlignedBox.IntersectionResult.FullIntersection:
|
|
node._outside_buildarea = True
|
|
else:
|
|
node._outside_buildarea = False
|
|
|
|
# Move the node upwards if the bottom is below the build platform.
|
|
move_vector = Vector()
|
|
if not Float.fuzzyCompare(bbox.bottom, 0.0):
|
|
move_vector.setY(-bbox.bottom)
|
|
|
|
# If there is no convex hull for the node, start calculating it and continue.
|
|
if not hasattr(node, '_convex_hull'):
|
|
if not hasattr(node, '_convex_hull_job'):
|
|
job = ConvexHullJob(node)
|
|
job.start()
|
|
node._convex_hull_job = job
|
|
else:
|
|
# Check for collisions between convex hulls
|
|
for other_node in BreadthFirstIterator(root):
|
|
# Ignore root, ourselves and anything that is not a normal SceneNode.
|
|
if other_node is root or type(other_node) is not SceneNode or other_node is node:
|
|
continue
|
|
|
|
# Ignore nodes that do not have the right properties set.
|
|
if not hasattr(other_node, '_convex_hull') or not other_node.getBoundingBox():
|
|
continue
|
|
|
|
# Check to see if the bounding boxes intersect. If not, we can ignore the node as there is no way the hull intersects.
|
|
if node.getBoundingBox().intersectsBox(other_node.getBoundingBox()) == AxisAlignedBox.IntersectionResult.NoIntersection:
|
|
continue
|
|
|
|
# Get the overlap distance for both convex hulls. If this returns None, there is no intersection.
|
|
overlap = node._convex_hull.intersectsPolygon(other_node._convex_hull)
|
|
if overlap is None:
|
|
continue
|
|
|
|
move_vector.setX(-overlap[0])
|
|
move_vector.setZ(-overlap[1])
|
|
|
|
if move_vector != Vector():
|
|
op = PlatformPhysicsOperation(node, move_vector)
|
|
op.push()
|
|
|
|
if node.getBoundingBox().intersectsBox(self._build_volume.getBoundingBox()) == AxisAlignedBox.IntersectionResult.FullIntersection:
|
|
op = ScaleToBoundsOperation(node, self._build_volume.getBoundingBox())
|
|
op.push()
|