ArrangeObjectsJob.py 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
  1. # Copyright (c) 2019 Ultimaker B.V.
  2. # Cura is released under the terms of the LGPLv3 or higher.
  3. from UM.Application import Application
  4. from UM.Job import Job
  5. from UM.Scene.SceneNode import SceneNode
  6. from UM.Math.Vector import Vector
  7. from UM.Operations.TranslateOperation import TranslateOperation
  8. from UM.Operations.GroupedOperation import GroupedOperation
  9. from UM.Logger import Logger
  10. from UM.Message import Message
  11. from UM.i18n import i18nCatalog
  12. i18n_catalog = i18nCatalog("cura")
  13. from cura.Scene.ZOffsetDecorator import ZOffsetDecorator
  14. from cura.Arranging.Arrange import Arrange
  15. from cura.Arranging.ShapeArray import ShapeArray
  16. from typing import List
  17. class ArrangeObjectsJob(Job):
  18. def __init__(self, nodes: List[SceneNode], fixed_nodes: List[SceneNode], min_offset = 8) -> None:
  19. super().__init__()
  20. self._nodes = nodes
  21. self._fixed_nodes = fixed_nodes
  22. self._min_offset = min_offset
  23. def run(self):
  24. status_message = Message(i18n_catalog.i18nc("@info:status", "Finding new location for objects"),
  25. lifetime = 0,
  26. dismissable=False,
  27. progress = 0,
  28. title = i18n_catalog.i18nc("@info:title", "Finding Location"))
  29. status_message.show()
  30. global_container_stack = Application.getInstance().getGlobalContainerStack()
  31. machine_width = global_container_stack.getProperty("machine_width", "value")
  32. machine_depth = global_container_stack.getProperty("machine_depth", "value")
  33. arranger = Arrange.create(x = machine_width, y = machine_depth, fixed_nodes = self._fixed_nodes, min_offset = self._min_offset)
  34. # Build set to exclude children (those get arranged together with the parents).
  35. included_as_child = set()
  36. for node in self._nodes:
  37. included_as_child.update(node.getAllChildren())
  38. # Collect nodes to be placed
  39. nodes_arr = [] # fill with (size, node, offset_shape_arr, hull_shape_arr)
  40. for node in self._nodes:
  41. if node in included_as_child:
  42. continue
  43. offset_shape_arr, hull_shape_arr = ShapeArray.fromNode(node, min_offset = self._min_offset, include_children = True)
  44. if offset_shape_arr is None:
  45. Logger.log("w", "Node [%s] could not be converted to an array for arranging...", str(node))
  46. continue
  47. nodes_arr.append((offset_shape_arr.arr.shape[0] * offset_shape_arr.arr.shape[1], node, offset_shape_arr, hull_shape_arr))
  48. # Sort the nodes with the biggest area first.
  49. nodes_arr.sort(key=lambda item: item[0])
  50. nodes_arr.reverse()
  51. # Place nodes one at a time
  52. start_priority = 0
  53. last_priority = start_priority
  54. last_size = None
  55. grouped_operation = GroupedOperation()
  56. found_solution_for_all = True
  57. not_fit_count = 0
  58. for idx, (size, node, offset_shape_arr, hull_shape_arr) in enumerate(nodes_arr):
  59. # For performance reasons, we assume that when a location does not fit,
  60. # it will also not fit for the next object (while what can be untrue).
  61. if last_size == size: # This optimization works if many of the objects have the same size
  62. start_priority = last_priority
  63. else:
  64. start_priority = 0
  65. best_spot = arranger.bestSpot(hull_shape_arr, start_prio = start_priority)
  66. x, y = best_spot.x, best_spot.y
  67. node.removeDecorator(ZOffsetDecorator)
  68. if node.getBoundingBox():
  69. center_y = node.getWorldPosition().y - node.getBoundingBox().bottom
  70. else:
  71. center_y = 0
  72. if x is not None: # We could find a place
  73. last_size = size
  74. last_priority = best_spot.priority
  75. arranger.place(x, y, offset_shape_arr) # take place before the next one
  76. grouped_operation.addOperation(TranslateOperation(node, Vector(x, center_y, y), set_position = True))
  77. else:
  78. Logger.log("d", "Arrange all: could not find spot!")
  79. found_solution_for_all = False
  80. grouped_operation.addOperation(TranslateOperation(node, Vector(200, center_y, -not_fit_count * 20), set_position = True))
  81. not_fit_count += 1
  82. status_message.setProgress((idx + 1) / len(nodes_arr) * 100)
  83. Job.yieldThread()
  84. grouped_operation.push()
  85. status_message.hide()
  86. if not found_solution_for_all:
  87. no_full_solution_message = Message(i18n_catalog.i18nc("@info:status", "Unable to find a location within the build volume for all objects"),
  88. title = i18n_catalog.i18nc("@info:title", "Can't Find Location"))
  89. no_full_solution_message.show()
  90. self.finished.emit(self)