StartSliceJob.py 8.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197
  1. # Copyright (c) 2015 Ultimaker B.V.
  2. # Cura is released under the terms of the AGPLv3 or higher.
  3. import numpy
  4. from string import Formatter
  5. import traceback
  6. from UM.Application import Application
  7. from UM.Settings.DefinitionContainer import DefinitionContainer
  8. from UM.Job import Job
  9. from UM.Logger import Logger
  10. from UM.Scene.SceneNode import SceneNode
  11. from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator
  12. from cura.OneAtATimeIterator import OneAtATimeIterator
  13. ## Formatter class that handles token expansion in start/end gcod
  14. class GcodeStartEndFormatter(Formatter):
  15. def get_value(self, key, args, kwargs): # [CodeStyle: get_value is an overridden function from the Formatter class]
  16. if isinstance(key, str):
  17. try:
  18. return kwargs[key]
  19. except KeyError:
  20. Logger.log("w", "Unable to replace '%s' placeholder in start/end gcode", key)
  21. return "{" + key + "}"
  22. else:
  23. Logger.log("w", "Incorrectly formatted placeholder '%s' in start/end gcode", key)
  24. return "{" + str(key) + "}"
  25. ## Job that handles sending the current scene data to CuraEngine
  26. class StartSliceJob(Job):
  27. def __init__(self, socket):
  28. super().__init__()
  29. self._scene = Application.getInstance().getController().getScene()
  30. self._socket = socket
  31. self._stack = Application.getInstance().getGlobalContainerStack()
  32. ## Runs the job that initiates the slicing.
  33. def run(self):
  34. self._scene.acquireLock()
  35. # Remove old layer data.
  36. for node in DepthFirstIterator(self._scene.getRoot()):
  37. if node.callDecoration("getLayerData"):
  38. node.getParent().removeChild(node)
  39. break
  40. # Get the objects in their groups to print.
  41. object_groups = []
  42. if Application.getInstance().getGlobalContainerStack().getProperty("print_sequence", "value") == "one_at_a_time":
  43. for node in OneAtATimeIterator(self._scene.getRoot()):
  44. temp_list = []
  45. # Node can't be printed, so don't bother sending it.
  46. if getattr(node, "_outside_buildarea", False):
  47. continue
  48. children = node.getAllChildren()
  49. children.append(node)
  50. for child_node in children:
  51. if type(child_node) is SceneNode and child_node.getMeshData() and child_node.getMeshData().getVertices() is not None:
  52. temp_list.append(child_node)
  53. if temp_list:
  54. object_groups.append(temp_list)
  55. Job.yieldThread()
  56. if len(object_groups) == 0:
  57. Logger.log("w", "No objects suitable for one at a time found, or no correct order found")
  58. else:
  59. temp_list = []
  60. for node in DepthFirstIterator(self._scene.getRoot()):
  61. if type(node) is SceneNode and node.getMeshData() and node.getMeshData().getVertices() is not None:
  62. if not getattr(node, "_outside_buildarea", False):
  63. temp_list.append(node)
  64. Job.yieldThread()
  65. if temp_list:
  66. object_groups.append(temp_list)
  67. self._scene.releaseLock()
  68. if not object_groups:
  69. return
  70. self._sendGlobalSettings()
  71. slice_message = self._socket.createMessage("cura.proto.Slice")
  72. for group in object_groups:
  73. group_message = slice_message.addRepeatedMessage("object_lists")
  74. if group[0].getParent().callDecoration("isGroup"):
  75. self._handlePerObjectSettings(group[0].getParent(), group_message)
  76. for current_object in group:
  77. mesh_data = current_object.getMeshData().getTransformed(current_object.getWorldTransformation())
  78. obj = group_message.addRepeatedMessage("objects")
  79. obj.id = id(current_object)
  80. verts = numpy.array(mesh_data.getVertices())
  81. verts[:,[1,2]] = verts[:,[2,1]]
  82. verts[:,1] *= -1
  83. obj.vertices = verts
  84. self._handlePerObjectSettings(current_object, obj)
  85. Job.yieldThread()
  86. Logger.log("d", "Sending data to engine for slicing.")
  87. self._socket.sendMessage(slice_message)
  88. Logger.log("d", "Sending data to engine is completed")
  89. self.setResult(True)
  90. def _expandGcodeTokens(self, key, value, settings):
  91. try:
  92. # any setting can be used as a token
  93. fmt = GcodeStartEndFormatter()
  94. return str(fmt.format(value, **settings)).encode("utf-8")
  95. except:
  96. Logger.log("w", "Unabled to do token replacement on start/end gcode %s", traceback.format_exc())
  97. return str(value).encode("utf-8")
  98. ## Sends all global settings to the engine.
  99. #
  100. # The settings are taken from the global stack. This does not include any
  101. # per-extruder settings or per-object settings.
  102. def _sendGlobalSettings(self):
  103. message = self._socket.createMessage("cura.proto.SettingList")
  104. settings = self._getAllSettingValues() #Get all the settings to send.
  105. print(settings)
  106. start_gcode = settings["machine_start_gcode"]
  107. settings["material_bed_temp_prepend"] = "{material_bed_temperature}" not in start_gcode #Pre-compute material_bed_temp_prepend and material_print_temp_prepend.
  108. settings["material_print_temp_prepend"] = "{material_print_temperature}" not in start_gcode
  109. for key, value in settings.items(): #Add all submessages for each individual setting.
  110. setting_message = message.addRepeatedMessage("settings")
  111. setting_message.name = key
  112. if key == "machine_start_gcode" or key == "machine_end_gcode": #If it's a g-code message, use special formatting.
  113. setting_message.value = self._expandGcodeTokens(key, value, settings)
  114. else:
  115. setting_message.value = str(value).encode("utf-8")
  116. self._socket.sendMessage(message)
  117. def _handlePerObjectSettings(self, node, message):
  118. profile = node.callDecoration("getProfile")
  119. if profile:
  120. for key, value in profile.getAllSettingValues().items():
  121. setting = message.addRepeatedMessage("settings")
  122. setting.name = key
  123. setting.value = str(value).encode()
  124. Job.yieldThread()
  125. object_settings = node.callDecoration("getAllSettingValues")
  126. if not object_settings:
  127. return
  128. for key, value in object_settings.items():
  129. setting = message.addRepeatedMessage("settings")
  130. setting.name = key
  131. setting.value = str(value).encode()
  132. Job.yieldThread()
  133. ## Gets the current values for all child definitions of a definition.
  134. #
  135. # To be clear, it looks up all child definitions, and returns the CURRENT
  136. # VALUE of the keys of these definitions according to the global stack. It
  137. # doesn't return the "value" property in the definition.
  138. #
  139. # \param definition The setting definition to get the child settings of.
  140. def _getAllChildSettingValues(self, definition):
  141. setting_values = { }
  142. for child in definition.children:
  143. setting_values[child.key] = self._stack.getProperty(child.key, "value")
  144. for key, value in self._getAllChildSettingValues(child).items():
  145. setting_values[key] = value
  146. return setting_values
  147. ## Gets all setting values as a dictionary.
  148. #
  149. # \return A dictionary with the setting keys as keys and the setting
  150. # values as values.
  151. def _getAllSettingValues(self):
  152. setting_values = {}
  153. definition_containers = [container for container in self._stack.getContainers() if container.__class__ == DefinitionContainer] #To get all keys, get all definitions from all definition containers.
  154. for definition_container in definition_containers:
  155. for definition in definition_container.definitions:
  156. setting_values[definition.key] = self._stack.getProperty(definition.key, "value")
  157. for key, value in self._getAllChildSettingValues(definition).items():
  158. setting_values[key] = value
  159. return setting_values