CuraApplication.py 79 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730
  1. # Copyright (c) 2018 Ultimaker B.V.
  2. # Cura is released under the terms of the LGPLv3 or higher.
  3. import os
  4. import sys
  5. import time
  6. from typing import cast, TYPE_CHECKING
  7. import numpy
  8. from PyQt5.QtCore import QObject, QTimer, QUrl, pyqtSignal, pyqtProperty, QEvent, Q_ENUMS
  9. from PyQt5.QtGui import QColor, QIcon
  10. from PyQt5.QtWidgets import QMessageBox
  11. from PyQt5.QtQml import qmlRegisterUncreatableType, qmlRegisterSingletonType, qmlRegisterType
  12. from UM.Scene.SceneNode import SceneNode
  13. from UM.Scene.Camera import Camera
  14. from UM.Math.Vector import Vector
  15. from UM.Math.Quaternion import Quaternion
  16. from UM.Math.AxisAlignedBox import AxisAlignedBox
  17. from UM.Math.Matrix import Matrix
  18. from UM.Platform import Platform
  19. from UM.Resources import Resources
  20. from UM.Scene.ToolHandle import ToolHandle
  21. from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator
  22. from UM.Mesh.ReadMeshJob import ReadMeshJob
  23. from UM.Logger import Logger
  24. from UM.Preferences import Preferences
  25. from UM.Qt.QtApplication import QtApplication #The class we're inheriting from.
  26. from UM.View.SelectionPass import SelectionPass #For typing.
  27. from UM.Scene.Selection import Selection
  28. from UM.Scene.GroupDecorator import GroupDecorator
  29. from UM.Settings.ContainerStack import ContainerStack
  30. from UM.Settings.InstanceContainer import InstanceContainer
  31. from UM.Settings.Validator import Validator
  32. from UM.Message import Message
  33. from UM.i18n import i18nCatalog
  34. from UM.Workspace.WorkspaceReader import WorkspaceReader
  35. from UM.Decorators import deprecated
  36. from UM.Operations.AddSceneNodeOperation import AddSceneNodeOperation
  37. from UM.Operations.RemoveSceneNodeOperation import RemoveSceneNodeOperation
  38. from UM.Operations.GroupedOperation import GroupedOperation
  39. from UM.Operations.SetTransformOperation import SetTransformOperation
  40. from cura.Arranging.Arrange import Arrange
  41. from cura.Arranging.ArrangeObjectsJob import ArrangeObjectsJob
  42. from cura.Arranging.ArrangeObjectsAllBuildPlatesJob import ArrangeObjectsAllBuildPlatesJob
  43. from cura.Arranging.ShapeArray import ShapeArray
  44. from cura.MultiplyObjectsJob import MultiplyObjectsJob
  45. from cura.Scene.ConvexHullDecorator import ConvexHullDecorator
  46. from cura.Operations.SetParentOperation import SetParentOperation
  47. from cura.Scene.SliceableObjectDecorator import SliceableObjectDecorator
  48. from cura.Scene.BlockSlicingDecorator import BlockSlicingDecorator
  49. from cura.Scene.BuildPlateDecorator import BuildPlateDecorator
  50. from cura.Scene.CuraSceneNode import CuraSceneNode
  51. from cura.Scene.CuraSceneController import CuraSceneController
  52. from UM.Settings.SettingDefinition import SettingDefinition, DefinitionPropertyType
  53. from UM.Settings.ContainerRegistry import ContainerRegistry
  54. from UM.Settings.SettingFunction import SettingFunction
  55. from cura.Settings.MachineNameValidator import MachineNameValidator
  56. from cura.Machines.Models.BuildPlateModel import BuildPlateModel
  57. from cura.Machines.Models.NozzleModel import NozzleModel
  58. from cura.Machines.Models.QualityProfilesDropDownMenuModel import QualityProfilesDropDownMenuModel
  59. from cura.Machines.Models.CustomQualityProfilesDropDownMenuModel import CustomQualityProfilesDropDownMenuModel
  60. from cura.Machines.Models.MultiBuildPlateModel import MultiBuildPlateModel
  61. from cura.Machines.Models.MaterialManagementModel import MaterialManagementModel
  62. from cura.Machines.Models.GenericMaterialsModel import GenericMaterialsModel
  63. from cura.Machines.Models.BrandMaterialsModel import BrandMaterialsModel
  64. from cura.Machines.Models.QualityManagementModel import QualityManagementModel
  65. from cura.Machines.Models.QualitySettingsModel import QualitySettingsModel
  66. from cura.Machines.Models.MachineManagementModel import MachineManagementModel
  67. from cura.Machines.Models.SettingVisibilityPresetsModel import SettingVisibilityPresetsModel
  68. from cura.Machines.MachineErrorChecker import MachineErrorChecker
  69. from cura.Settings.SettingInheritanceManager import SettingInheritanceManager
  70. from cura.Settings.SimpleModeSettingsManager import SimpleModeSettingsManager
  71. from cura.Machines.VariantManager import VariantManager
  72. from .SingleInstance import SingleInstance
  73. from .AutoSave import AutoSave
  74. from . import PlatformPhysics
  75. from . import BuildVolume
  76. from . import CameraAnimation
  77. from . import PrintInformation
  78. from . import CuraActions
  79. from cura.Scene import ZOffsetDecorator
  80. from . import CuraSplashScreen
  81. from . import CameraImageProvider
  82. from . import MachineActionManager
  83. from cura.TaskManagement.OnExitCallbackManager import OnExitCallbackManager
  84. from cura.Settings.MachineManager import MachineManager
  85. from cura.Settings.ExtruderManager import ExtruderManager
  86. from cura.Settings.UserChangesModel import UserChangesModel
  87. from cura.Settings.ExtrudersModel import ExtrudersModel
  88. from cura.Settings.MaterialSettingsVisibilityHandler import MaterialSettingsVisibilityHandler
  89. from cura.Settings.ContainerManager import ContainerManager
  90. from cura.Settings.SidebarCustomMenuItemsModel import SidebarCustomMenuItemsModel
  91. import cura.Settings.cura_empty_instance_containers
  92. from cura.ObjectsModel import ObjectsModel
  93. from UM.FlameProfiler import pyqtSlot
  94. if TYPE_CHECKING:
  95. from plugins.SliceInfoPlugin.SliceInfo import SliceInfo
  96. numpy.seterr(all = "ignore")
  97. try:
  98. from cura.CuraVersion import CuraVersion, CuraBuildType, CuraDebugMode, CuraSDKVersion
  99. except ImportError:
  100. CuraVersion = "master" # [CodeStyle: Reflecting imported value]
  101. CuraBuildType = ""
  102. CuraDebugMode = False
  103. CuraSDKVersion = ""
  104. class CuraApplication(QtApplication):
  105. # SettingVersion represents the set of settings available in the machine/extruder definitions.
  106. # You need to make sure that this version number needs to be increased if there is any non-backwards-compatible
  107. # changes of the settings.
  108. SettingVersion = 5
  109. Created = False
  110. class ResourceTypes:
  111. QmlFiles = Resources.UserType + 1
  112. Firmware = Resources.UserType + 2
  113. QualityInstanceContainer = Resources.UserType + 3
  114. QualityChangesInstanceContainer = Resources.UserType + 4
  115. MaterialInstanceContainer = Resources.UserType + 5
  116. VariantInstanceContainer = Resources.UserType + 6
  117. UserInstanceContainer = Resources.UserType + 7
  118. MachineStack = Resources.UserType + 8
  119. ExtruderStack = Resources.UserType + 9
  120. DefinitionChangesContainer = Resources.UserType + 10
  121. SettingVisibilityPreset = Resources.UserType + 11
  122. Q_ENUMS(ResourceTypes)
  123. def __init__(self, *args, **kwargs):
  124. super().__init__(name = "cura",
  125. version = CuraVersion,
  126. buildtype = CuraBuildType,
  127. is_debug_mode = CuraDebugMode,
  128. tray_icon_name = "cura-icon-32.png",
  129. **kwargs)
  130. self.default_theme = "cura-light"
  131. self._boot_loading_time = time.time()
  132. self._on_exit_callback_manager = OnExitCallbackManager(self)
  133. # Variables set from CLI
  134. self._files_to_open = []
  135. self._use_single_instance = False
  136. self._trigger_early_crash = False # For debug only
  137. self._single_instance = None
  138. self._cura_package_manager = None
  139. self._machine_action_manager = None
  140. self.empty_container = None
  141. self.empty_definition_changes_container = None
  142. self.empty_variant_container = None
  143. self.empty_material_container = None
  144. self.empty_quality_container = None
  145. self.empty_quality_changes_container = None
  146. self._variant_manager = None
  147. self._material_manager = None
  148. self._quality_manager = None
  149. self._machine_manager = None
  150. self._extruder_manager = None
  151. self._container_manager = None
  152. self._object_manager = None
  153. self._build_plate_model = None
  154. self._multi_build_plate_model = None
  155. self._setting_visibility_presets_model = None
  156. self._setting_inheritance_manager = None
  157. self._simple_mode_settings_manager = None
  158. self._cura_scene_controller = None
  159. self._machine_error_checker = None
  160. self._quality_profile_drop_down_menu_model = None
  161. self._custom_quality_profile_drop_down_menu_model = None
  162. self._physics = None
  163. self._volume = None
  164. self._output_devices = {}
  165. self._print_information = None
  166. self._previous_active_tool = None
  167. self._platform_activity = False
  168. self._scene_bounding_box = AxisAlignedBox.Null
  169. self._center_after_select = False
  170. self._camera_animation = None
  171. self._cura_actions = None
  172. self.started = False
  173. self._message_box_callback = None
  174. self._message_box_callback_arguments = []
  175. self._preferred_mimetype = ""
  176. self._i18n_catalog = None
  177. self._currently_loading_files = []
  178. self._non_sliceable_extensions = []
  179. self._additional_components = {} # Components to add to certain areas in the interface
  180. self._open_file_queue = [] # A list of files to open (after the application has started)
  181. self._update_platform_activity_timer = None
  182. self._need_to_show_user_agreement = True
  183. self._sidebar_custom_menu_items = [] # type: list # Keeps list of custom menu items for the side bar
  184. self._plugins_loaded = False
  185. # Backups
  186. self._auto_save = None
  187. self._save_data_enabled = True
  188. from cura.Settings.CuraContainerRegistry import CuraContainerRegistry
  189. self._container_registry_class = CuraContainerRegistry
  190. from cura.CuraPackageManager import CuraPackageManager
  191. self._package_manager_class = CuraPackageManager
  192. # Adds command line options to the command line parser. This should be called after the application is created and
  193. # before the pre-start.
  194. def addCommandLineOptions(self):
  195. super().addCommandLineOptions()
  196. self._cli_parser.add_argument("--help", "-h",
  197. action = "store_true",
  198. default = False,
  199. help = "Show this help message and exit.")
  200. self._cli_parser.add_argument("--single-instance",
  201. dest = "single_instance",
  202. action = "store_true",
  203. default = False)
  204. # >> For debugging
  205. # Trigger an early crash, i.e. a crash that happens before the application enters its event loop.
  206. self._cli_parser.add_argument("--trigger-early-crash",
  207. dest = "trigger_early_crash",
  208. action = "store_true",
  209. default = False,
  210. help = "FOR TESTING ONLY. Trigger an early crash to show the crash dialog.")
  211. self._cli_parser.add_argument("file", nargs = "*", help = "Files to load after starting the application.")
  212. def parseCliOptions(self):
  213. super().parseCliOptions()
  214. if self._cli_args.help:
  215. self._cli_parser.print_help()
  216. sys.exit(0)
  217. self._use_single_instance = self._cli_args.single_instance
  218. self._trigger_early_crash = self._cli_args.trigger_early_crash
  219. for filename in self._cli_args.file:
  220. self._files_to_open.append(os.path.abspath(filename))
  221. def initialize(self) -> None:
  222. self.__addExpectedResourceDirsAndSearchPaths() # Must be added before init of super
  223. super().initialize()
  224. self.__sendCommandToSingleInstance()
  225. self.__initializeSettingDefinitionsAndFunctions()
  226. self.__addAllResourcesAndContainerResources()
  227. self.__addAllEmptyContainers()
  228. self.__setLatestResouceVersionsForVersionUpgrade()
  229. self._machine_action_manager = MachineActionManager.MachineActionManager(self)
  230. self._machine_action_manager.initialize()
  231. self.change_log_url = "https://ultimaker.com/ultimaker-cura-latest-features"
  232. def __sendCommandToSingleInstance(self):
  233. self._single_instance = SingleInstance(self, self._files_to_open)
  234. # If we use single instance, try to connect to the single instance server, send commands, and then exit.
  235. # If we cannot find an existing single instance server, this is the only instance, so just keep going.
  236. if self._use_single_instance:
  237. if self._single_instance.startClient():
  238. Logger.log("i", "Single instance commands were sent, exiting")
  239. sys.exit(0)
  240. # Adds expected directory names and search paths for Resources.
  241. def __addExpectedResourceDirsAndSearchPaths(self):
  242. # this list of dir names will be used by UM to detect an old cura directory
  243. for dir_name in ["extruders", "machine_instances", "materials", "plugins", "quality", "quality_changes", "user", "variants"]:
  244. Resources.addExpectedDirNameInData(dir_name)
  245. Resources.addSearchPath(os.path.join(self._app_install_dir, "share", "cura", "resources"))
  246. if not hasattr(sys, "frozen"):
  247. resource_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "resources")
  248. Resources.addSearchPath(resource_path)
  249. # Adds custom property types, settings types, and extra operators (functions) that need to be registered in
  250. # SettingDefinition and SettingFunction.
  251. def __initializeSettingDefinitionsAndFunctions(self):
  252. # Need to do this before ContainerRegistry tries to load the machines
  253. SettingDefinition.addSupportedProperty("settable_per_mesh", DefinitionPropertyType.Any, default = True, read_only = True)
  254. SettingDefinition.addSupportedProperty("settable_per_extruder", DefinitionPropertyType.Any, default = True, read_only = True)
  255. # this setting can be changed for each group in one-at-a-time mode
  256. SettingDefinition.addSupportedProperty("settable_per_meshgroup", DefinitionPropertyType.Any, default = True, read_only = True)
  257. SettingDefinition.addSupportedProperty("settable_globally", DefinitionPropertyType.Any, default = True, read_only = True)
  258. # From which stack the setting would inherit if not defined per object (handled in the engine)
  259. # AND for settings which are not settable_per_mesh:
  260. # which extruder is the only extruder this setting is obtained from
  261. SettingDefinition.addSupportedProperty("limit_to_extruder", DefinitionPropertyType.Function, default = "-1", depends_on = "value")
  262. # For settings which are not settable_per_mesh and not settable_per_extruder:
  263. # A function which determines the glabel/meshgroup value by looking at the values of the setting in all (used) extruders
  264. SettingDefinition.addSupportedProperty("resolve", DefinitionPropertyType.Function, default = None, depends_on = "value")
  265. SettingDefinition.addSettingType("extruder", None, str, Validator)
  266. SettingDefinition.addSettingType("optional_extruder", None, str, None)
  267. SettingDefinition.addSettingType("[int]", None, str, None)
  268. SettingFunction.registerOperator("extruderValues", ExtruderManager.getExtruderValues)
  269. SettingFunction.registerOperator("extruderValue", ExtruderManager.getExtruderValue)
  270. SettingFunction.registerOperator("resolveOrValue", ExtruderManager.getResolveOrValue)
  271. SettingFunction.registerOperator("defaultExtruderPosition", ExtruderManager.getDefaultExtruderPosition)
  272. # Adds all resources and container related resources.
  273. def __addAllResourcesAndContainerResources(self) -> None:
  274. Resources.addStorageType(self.ResourceTypes.QualityInstanceContainer, "quality")
  275. Resources.addStorageType(self.ResourceTypes.QualityChangesInstanceContainer, "quality_changes")
  276. Resources.addStorageType(self.ResourceTypes.VariantInstanceContainer, "variants")
  277. Resources.addStorageType(self.ResourceTypes.MaterialInstanceContainer, "materials")
  278. Resources.addStorageType(self.ResourceTypes.UserInstanceContainer, "user")
  279. Resources.addStorageType(self.ResourceTypes.ExtruderStack, "extruders")
  280. Resources.addStorageType(self.ResourceTypes.MachineStack, "machine_instances")
  281. Resources.addStorageType(self.ResourceTypes.DefinitionChangesContainer, "definition_changes")
  282. Resources.addStorageType(self.ResourceTypes.SettingVisibilityPreset, "setting_visibility")
  283. self._container_registry.addResourceType(self.ResourceTypes.QualityInstanceContainer, "quality")
  284. self._container_registry.addResourceType(self.ResourceTypes.QualityChangesInstanceContainer, "quality_changes")
  285. self._container_registry.addResourceType(self.ResourceTypes.VariantInstanceContainer, "variant")
  286. self._container_registry.addResourceType(self.ResourceTypes.MaterialInstanceContainer, "material")
  287. self._container_registry.addResourceType(self.ResourceTypes.UserInstanceContainer, "user")
  288. self._container_registry.addResourceType(self.ResourceTypes.ExtruderStack, "extruder_train")
  289. self._container_registry.addResourceType(self.ResourceTypes.MachineStack, "machine")
  290. self._container_registry.addResourceType(self.ResourceTypes.DefinitionChangesContainer, "definition_changes")
  291. Resources.addType(self.ResourceTypes.QmlFiles, "qml")
  292. Resources.addType(self.ResourceTypes.Firmware, "firmware")
  293. # Adds all empty containers.
  294. def __addAllEmptyContainers(self) -> None:
  295. # Add empty variant, material and quality containers.
  296. # Since they are empty, they should never be serialized and instead just programmatically created.
  297. # We need them to simplify the switching between materials.
  298. self.empty_container = cura.Settings.cura_empty_instance_containers.empty_container
  299. self._container_registry.addContainer(
  300. cura.Settings.cura_empty_instance_containers.empty_definition_changes_container)
  301. self.empty_definition_changes_container = cura.Settings.cura_empty_instance_containers.empty_definition_changes_container
  302. self._container_registry.addContainer(cura.Settings.cura_empty_instance_containers.empty_variant_container)
  303. self.empty_variant_container = cura.Settings.cura_empty_instance_containers.empty_variant_container
  304. self._container_registry.addContainer(cura.Settings.cura_empty_instance_containers.empty_material_container)
  305. self.empty_material_container = cura.Settings.cura_empty_instance_containers.empty_material_container
  306. self._container_registry.addContainer(cura.Settings.cura_empty_instance_containers.empty_quality_container)
  307. self.empty_quality_container = cura.Settings.cura_empty_instance_containers.empty_quality_container
  308. self._container_registry.addContainer(cura.Settings.cura_empty_instance_containers.empty_quality_changes_container)
  309. self.empty_quality_changes_container = cura.Settings.cura_empty_instance_containers.empty_quality_changes_container
  310. # Initializes the version upgrade manager with by providing the paths for each resource type and the latest
  311. # versions.
  312. def __setLatestResouceVersionsForVersionUpgrade(self):
  313. self._version_upgrade_manager.setCurrentVersions(
  314. {
  315. ("quality", InstanceContainer.Version * 1000000 + self.SettingVersion): (self.ResourceTypes.QualityInstanceContainer, "application/x-uranium-instancecontainer"),
  316. ("quality_changes", InstanceContainer.Version * 1000000 + self.SettingVersion): (self.ResourceTypes.QualityChangesInstanceContainer, "application/x-uranium-instancecontainer"),
  317. ("machine_stack", ContainerStack.Version * 1000000 + self.SettingVersion): (self.ResourceTypes.MachineStack, "application/x-cura-globalstack"),
  318. ("extruder_train", ContainerStack.Version * 1000000 + self.SettingVersion): (self.ResourceTypes.ExtruderStack, "application/x-cura-extruderstack"),
  319. ("preferences", Preferences.Version * 1000000 + self.SettingVersion): (Resources.Preferences, "application/x-uranium-preferences"),
  320. ("user", InstanceContainer.Version * 1000000 + self.SettingVersion): (self.ResourceTypes.UserInstanceContainer, "application/x-uranium-instancecontainer"),
  321. ("definition_changes", InstanceContainer.Version * 1000000 + self.SettingVersion): (self.ResourceTypes.DefinitionChangesContainer, "application/x-uranium-instancecontainer"),
  322. ("variant", InstanceContainer.Version * 1000000 + self.SettingVersion): (self.ResourceTypes.VariantInstanceContainer, "application/x-uranium-instancecontainer"),
  323. }
  324. )
  325. # Runs preparations that needs to be done before the starting process.
  326. def startSplashWindowPhase(self):
  327. super().startSplashWindowPhase()
  328. self.setWindowIcon(QIcon(Resources.getPath(Resources.Images, "cura-icon.png")))
  329. self.setRequiredPlugins([
  330. # Misc.:
  331. "ConsoleLogger",
  332. "CuraEngineBackend",
  333. "UserAgreement",
  334. "FileLogger",
  335. "XmlMaterialProfile",
  336. "Toolbox",
  337. "PrepareStage",
  338. "MonitorStage",
  339. "LocalFileOutputDevice",
  340. "LocalContainerProvider",
  341. # Views:
  342. "SimpleView",
  343. "SimulationView",
  344. "SolidView",
  345. # Readers & Writers:
  346. "GCodeWriter",
  347. "STLReader",
  348. # Tools:
  349. "CameraTool",
  350. "MirrorTool",
  351. "RotateTool",
  352. "ScaleTool",
  353. "SelectionTool",
  354. "TranslateTool",
  355. ])
  356. self._i18n_catalog = i18nCatalog("cura")
  357. self._update_platform_activity_timer = QTimer()
  358. self._update_platform_activity_timer.setInterval(500)
  359. self._update_platform_activity_timer.setSingleShot(True)
  360. self._update_platform_activity_timer.timeout.connect(self.updatePlatformActivity)
  361. self.getController().getScene().sceneChanged.connect(self.updatePlatformActivityDelayed)
  362. self.getController().toolOperationStopped.connect(self._onToolOperationStopped)
  363. self.getController().contextMenuRequested.connect(self._onContextMenuRequested)
  364. self.getCuraSceneController().activeBuildPlateChanged.connect(self.updatePlatformActivityDelayed)
  365. self.showSplashMessage(self._i18n_catalog.i18nc("@info:progress", "Loading machines..."))
  366. with self._container_registry.lockFile():
  367. self._container_registry.loadAllMetadata()
  368. # set the setting version for Preferences
  369. preferences = self.getPreferences()
  370. preferences.addPreference("metadata/setting_version", 0)
  371. preferences.setValue("metadata/setting_version", self.SettingVersion) #Don't make it equal to the default so that the setting version always gets written to the file.
  372. preferences.addPreference("cura/active_mode", "simple")
  373. preferences.addPreference("cura/categories_expanded", "")
  374. preferences.addPreference("cura/jobname_prefix", True)
  375. preferences.addPreference("cura/select_models_on_load", False)
  376. preferences.addPreference("view/center_on_select", False)
  377. preferences.addPreference("mesh/scale_to_fit", False)
  378. preferences.addPreference("mesh/scale_tiny_meshes", True)
  379. preferences.addPreference("cura/dialog_on_project_save", True)
  380. preferences.addPreference("cura/asked_dialog_on_project_save", False)
  381. preferences.addPreference("cura/choice_on_profile_override", "always_ask")
  382. preferences.addPreference("cura/choice_on_open_project", "always_ask")
  383. preferences.addPreference("cura/use_multi_build_plate", False)
  384. preferences.addPreference("cura/currency", "€")
  385. preferences.addPreference("cura/material_settings", "{}")
  386. preferences.addPreference("view/invert_zoom", False)
  387. preferences.addPreference("view/filter_current_build_plate", False)
  388. preferences.addPreference("cura/sidebar_collapsed", False)
  389. self._need_to_show_user_agreement = not self.getPreferences().getValue("general/accepted_user_agreement")
  390. for key in [
  391. "dialog_load_path", # dialog_save_path is in LocalFileOutputDevicePlugin
  392. "dialog_profile_path",
  393. "dialog_material_path"]:
  394. preferences.addPreference("local_file/%s" % key, os.path.expanduser("~/"))
  395. preferences.setDefault("local_file/last_used_type", "text/x-gcode")
  396. self.applicationShuttingDown.connect(self.saveSettings)
  397. self.engineCreatedSignal.connect(self._onEngineCreated)
  398. self.globalContainerStackChanged.connect(self._onGlobalContainerChanged)
  399. self._onGlobalContainerChanged()
  400. self.getCuraSceneController().setActiveBuildPlate(0) # Initialize
  401. CuraApplication.Created = True
  402. def _onEngineCreated(self):
  403. self._qml_engine.addImageProvider("camera", CameraImageProvider.CameraImageProvider())
  404. @pyqtProperty(bool)
  405. def needToShowUserAgreement(self):
  406. return self._need_to_show_user_agreement
  407. def setNeedToShowUserAgreement(self, set_value = True):
  408. self._need_to_show_user_agreement = set_value
  409. # DO NOT call this function to close the application, use checkAndExitApplication() instead which will perform
  410. # pre-exit checks such as checking for in-progress USB printing, etc.
  411. def closeApplication(self):
  412. Logger.log("i", "Close application")
  413. main_window = self.getMainWindow()
  414. if main_window is not None:
  415. main_window.close()
  416. else:
  417. self.exit(0)
  418. # This function first performs all upon-exit checks such as USB printing that is in progress.
  419. # Use this to close the application.
  420. @pyqtSlot()
  421. def checkAndExitApplication(self) -> None:
  422. self._on_exit_callback_manager.resetCurrentState()
  423. self._on_exit_callback_manager.triggerNextCallback()
  424. @pyqtSlot(result = bool)
  425. def getIsAllChecksPassed(self) -> bool:
  426. return self._on_exit_callback_manager.getIsAllChecksPassed()
  427. def getOnExitCallbackManager(self) -> "OnExitCallbackManager":
  428. return self._on_exit_callback_manager
  429. def triggerNextExitCheck(self) -> None:
  430. self._on_exit_callback_manager.triggerNextCallback()
  431. showConfirmExitDialog = pyqtSignal(str, arguments = ["message"])
  432. def setConfirmExitDialogCallback(self, callback):
  433. self._confirm_exit_dialog_callback = callback
  434. @pyqtSlot(bool)
  435. def callConfirmExitDialogCallback(self, yes_or_no: bool):
  436. self._confirm_exit_dialog_callback(yes_or_no)
  437. ## Signal to connect preferences action in QML
  438. showPreferencesWindow = pyqtSignal()
  439. ## Show the preferences window
  440. @pyqtSlot()
  441. def showPreferences(self):
  442. self.showPreferencesWindow.emit()
  443. ## A reusable dialogbox
  444. #
  445. showMessageBox = pyqtSignal(str, str, str, str, int, int, arguments = ["title", "text", "informativeText", "detailedText", "buttons", "icon"])
  446. def messageBox(self, title, text, informativeText = "", detailedText = "", buttons = QMessageBox.Ok, icon = QMessageBox.NoIcon, callback = None, callback_arguments = []):
  447. self._message_box_callback = callback
  448. self._message_box_callback_arguments = callback_arguments
  449. self.showMessageBox.emit(title, text, informativeText, detailedText, buttons, icon)
  450. showDiscardOrKeepProfileChanges = pyqtSignal()
  451. def discardOrKeepProfileChanges(self):
  452. has_user_interaction = False
  453. choice = self.getPreferences().getValue("cura/choice_on_profile_override")
  454. if choice == "always_discard":
  455. # don't show dialog and DISCARD the profile
  456. self.discardOrKeepProfileChangesClosed("discard")
  457. elif choice == "always_keep":
  458. # don't show dialog and KEEP the profile
  459. self.discardOrKeepProfileChangesClosed("keep")
  460. elif not self._is_headless:
  461. # ALWAYS ask whether to keep or discard the profile
  462. self.showDiscardOrKeepProfileChanges.emit()
  463. has_user_interaction = True
  464. return has_user_interaction
  465. @pyqtSlot(str)
  466. def discardOrKeepProfileChangesClosed(self, option):
  467. global_stack = self.getGlobalContainerStack()
  468. if option == "discard":
  469. for extruder in global_stack.extruders.values():
  470. extruder.userChanges.clear()
  471. global_stack.userChanges.clear()
  472. # if the user decided to keep settings then the user settings should be re-calculated and validated for errors
  473. # before slicing. To ensure that slicer uses right settings values
  474. elif option == "keep":
  475. for extruder in global_stack.extruders.values():
  476. extruder.userChanges.update()
  477. global_stack.userChanges.update()
  478. @pyqtSlot(int)
  479. def messageBoxClosed(self, button):
  480. if self._message_box_callback:
  481. self._message_box_callback(button, *self._message_box_callback_arguments)
  482. self._message_box_callback = None
  483. self._message_box_callback_arguments = []
  484. showPrintMonitor = pyqtSignal(bool, arguments = ["show"])
  485. def setSaveDataEnabled(self, enabled: bool) -> None:
  486. self._save_data_enabled = enabled
  487. # Cura has multiple locations where instance containers need to be saved, so we need to handle this differently.
  488. def saveSettings(self):
  489. if not self.started or not self._save_data_enabled:
  490. # Do not do saving during application start or when data should not be saved on quit.
  491. return
  492. ContainerRegistry.getInstance().saveDirtyContainers()
  493. self.savePreferences()
  494. def saveStack(self, stack):
  495. ContainerRegistry.getInstance().saveContainer(stack)
  496. @pyqtSlot(str, result = QUrl)
  497. def getDefaultPath(self, key):
  498. default_path = self.getPreferences().getValue("local_file/%s" % key)
  499. return QUrl.fromLocalFile(default_path)
  500. @pyqtSlot(str, str)
  501. def setDefaultPath(self, key, default_path):
  502. self.getPreferences().setValue("local_file/%s" % key, QUrl(default_path).toLocalFile())
  503. ## Handle loading of all plugin types (and the backend explicitly)
  504. # \sa PluginRegistry
  505. def _loadPlugins(self):
  506. self._plugin_registry.addType("profile_reader", self._addProfileReader)
  507. self._plugin_registry.addType("profile_writer", self._addProfileWriter)
  508. if Platform.isLinux():
  509. lib_suffixes = {"", "64", "32", "x32"} #A few common ones on different distributions.
  510. else:
  511. lib_suffixes = {""}
  512. for suffix in lib_suffixes:
  513. self._plugin_registry.addPluginLocation(os.path.join(QtApplication.getInstallPrefix(), "lib" + suffix, "cura"))
  514. if not hasattr(sys, "frozen"):
  515. self._plugin_registry.addPluginLocation(os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "plugins"))
  516. self._plugin_registry.loadPlugin("ConsoleLogger")
  517. self._plugin_registry.loadPlugin("CuraEngineBackend")
  518. self._plugin_registry.loadPlugins()
  519. if self.getBackend() is None:
  520. raise RuntimeError("Could not load the backend plugin!")
  521. self._plugins_loaded = True
  522. def run(self):
  523. super().run()
  524. container_registry = self._container_registry
  525. Logger.log("i", "Initializing variant manager")
  526. self._variant_manager = VariantManager(container_registry)
  527. self._variant_manager.initialize()
  528. Logger.log("i", "Initializing material manager")
  529. from cura.Machines.MaterialManager import MaterialManager
  530. self._material_manager = MaterialManager(container_registry, parent = self)
  531. self._material_manager.initialize()
  532. Logger.log("i", "Initializing quality manager")
  533. from cura.Machines.QualityManager import QualityManager
  534. self._quality_manager = QualityManager(container_registry, parent = self)
  535. self._quality_manager.initialize()
  536. Logger.log("i", "Initializing machine manager")
  537. self._machine_manager = MachineManager(self)
  538. Logger.log("i", "Initializing container manager")
  539. self._container_manager = ContainerManager(self)
  540. Logger.log("i", "Initializing machine error checker")
  541. self._machine_error_checker = MachineErrorChecker(self)
  542. self._machine_error_checker.initialize()
  543. # Check if we should run as single instance or not. If so, set up a local socket server which listener which
  544. # coordinates multiple Cura instances and accepts commands.
  545. if self._use_single_instance:
  546. self.__setUpSingleInstanceServer()
  547. # Setup scene and build volume
  548. root = self.getController().getScene().getRoot()
  549. self._volume = BuildVolume.BuildVolume(self, root)
  550. Arrange.build_volume = self._volume
  551. # initialize info objects
  552. self._print_information = PrintInformation.PrintInformation(self)
  553. self._cura_actions = CuraActions.CuraActions(self)
  554. # Initialize setting visibility presets model
  555. self._setting_visibility_presets_model = SettingVisibilityPresetsModel(self)
  556. default_visibility_profile = self._setting_visibility_presets_model.getItem(0)
  557. self.getPreferences().setDefault("general/visible_settings", ";".join(default_visibility_profile["settings"]))
  558. # Detect in which mode to run and execute that mode
  559. if self._is_headless:
  560. self.runWithoutGUI()
  561. else:
  562. self.runWithGUI()
  563. self.started = True
  564. self.initializationFinished.emit()
  565. Logger.log("d", "Booting Cura took %s seconds", time.time() - self._boot_loading_time)
  566. # For now use a timer to postpone some things that need to be done after the application and GUI are
  567. # initialized, for example opening files because they may show dialogs which can be closed due to incomplete
  568. # GUI initialization.
  569. self._post_start_timer = QTimer(self)
  570. self._post_start_timer.setInterval(1000)
  571. self._post_start_timer.setSingleShot(True)
  572. self._post_start_timer.timeout.connect(self._onPostStart)
  573. self._post_start_timer.start()
  574. self._auto_save = AutoSave(self)
  575. self._auto_save.initialize()
  576. self.exec_()
  577. def __setUpSingleInstanceServer(self):
  578. if self._use_single_instance:
  579. self._single_instance.startServer()
  580. def _onPostStart(self):
  581. for file_name in self._files_to_open:
  582. self.callLater(self._openFile, file_name)
  583. for file_name in self._open_file_queue: # Open all the files that were queued up while plug-ins were loading.
  584. self.callLater(self._openFile, file_name)
  585. initializationFinished = pyqtSignal()
  586. ## Run Cura without GUI elements and interaction (server mode).
  587. def runWithoutGUI(self):
  588. self.closeSplash()
  589. ## Run Cura with GUI (desktop mode).
  590. def runWithGUI(self):
  591. self.showSplashMessage(self._i18n_catalog.i18nc("@info:progress", "Setting up scene..."))
  592. controller = self.getController()
  593. t = controller.getTool("TranslateTool")
  594. if t:
  595. t.setEnabledAxis([ToolHandle.XAxis, ToolHandle.YAxis, ToolHandle.ZAxis])
  596. Selection.selectionChanged.connect(self.onSelectionChanged)
  597. # Set default background color for scene
  598. self.getRenderer().setBackgroundColor(QColor(245, 245, 245))
  599. # Initialize platform physics
  600. self._physics = PlatformPhysics.PlatformPhysics(controller, self._volume)
  601. # Initialize camera
  602. root = controller.getScene().getRoot()
  603. camera = Camera("3d", root)
  604. diagonal = self.getBuildVolume().getDiagonalSize()
  605. if diagonal < 1: #No printer added yet. Set a default camera distance for normal-sized printers.
  606. diagonal = 375
  607. camera.setPosition(Vector(-80, 250, 700) * diagonal / 375)
  608. camera.setPerspective(True)
  609. camera.lookAt(Vector(0, 0, 0))
  610. controller.getScene().setActiveCamera("3d")
  611. # Initialize camera tool
  612. camera_tool = controller.getTool("CameraTool")
  613. camera_tool.setOrigin(Vector(0, 100, 0))
  614. camera_tool.setZoomRange(0.1, 2000)
  615. # Initialize camera animations
  616. self._camera_animation = CameraAnimation.CameraAnimation()
  617. self._camera_animation.setCameraTool(self.getController().getTool("CameraTool"))
  618. self.showSplashMessage(self._i18n_catalog.i18nc("@info:progress", "Loading interface..."))
  619. # Initialize QML engine
  620. self.setMainQml(Resources.getPath(self.ResourceTypes.QmlFiles, "Cura.qml"))
  621. self._qml_import_paths.append(Resources.getPath(self.ResourceTypes.QmlFiles))
  622. self.initializeEngine()
  623. # Initialize UI state
  624. controller.setActiveStage("PrepareStage")
  625. controller.setActiveView("SolidView")
  626. controller.setCameraTool("CameraTool")
  627. controller.setSelectionTool("SelectionTool")
  628. # Hide the splash screen
  629. self.closeSplash()
  630. @pyqtSlot(result = QObject)
  631. def getSettingVisibilityPresetsModel(self, *args) -> SettingVisibilityPresetsModel:
  632. return self._setting_visibility_presets_model
  633. def getMachineErrorChecker(self, *args) -> MachineErrorChecker:
  634. return self._machine_error_checker
  635. def getMachineManager(self, *args) -> MachineManager:
  636. if self._machine_manager is None:
  637. self._machine_manager = MachineManager(self)
  638. return self._machine_manager
  639. def getExtruderManager(self, *args):
  640. if self._extruder_manager is None:
  641. self._extruder_manager = ExtruderManager()
  642. return self._extruder_manager
  643. def getVariantManager(self, *args):
  644. return self._variant_manager
  645. @pyqtSlot(result = QObject)
  646. def getMaterialManager(self, *args):
  647. return self._material_manager
  648. @pyqtSlot(result = QObject)
  649. def getQualityManager(self, *args):
  650. return self._quality_manager
  651. def getObjectsModel(self, *args):
  652. if self._object_manager is None:
  653. self._object_manager = ObjectsModel.createObjectsModel()
  654. return self._object_manager
  655. @pyqtSlot(result = QObject)
  656. def getMultiBuildPlateModel(self, *args):
  657. if self._multi_build_plate_model is None:
  658. self._multi_build_plate_model = MultiBuildPlateModel(self)
  659. return self._multi_build_plate_model
  660. @pyqtSlot(result = QObject)
  661. def getBuildPlateModel(self, *args):
  662. if self._build_plate_model is None:
  663. self._build_plate_model = BuildPlateModel(self)
  664. return self._build_plate_model
  665. def getCuraSceneController(self, *args):
  666. if self._cura_scene_controller is None:
  667. self._cura_scene_controller = CuraSceneController.createCuraSceneController()
  668. return self._cura_scene_controller
  669. def getSettingInheritanceManager(self, *args):
  670. if self._setting_inheritance_manager is None:
  671. self._setting_inheritance_manager = SettingInheritanceManager.createSettingInheritanceManager()
  672. return self._setting_inheritance_manager
  673. ## Get the machine action manager
  674. # We ignore any *args given to this, as we also register the machine manager as qml singleton.
  675. # It wants to give this function an engine and script engine, but we don't care about that.
  676. def getMachineActionManager(self, *args):
  677. return self._machine_action_manager
  678. def getSimpleModeSettingsManager(self, *args):
  679. if self._simple_mode_settings_manager is None:
  680. self._simple_mode_settings_manager = SimpleModeSettingsManager()
  681. return self._simple_mode_settings_manager
  682. ## Handle Qt events
  683. def event(self, event):
  684. if event.type() == QEvent.FileOpen:
  685. if self._plugins_loaded:
  686. self._openFile(event.file())
  687. else:
  688. self._open_file_queue.append(event.file())
  689. return super().event(event)
  690. def getAutoSave(self):
  691. return self._auto_save
  692. ## Get print information (duration / material used)
  693. def getPrintInformation(self):
  694. return self._print_information
  695. def getQualityProfilesDropDownMenuModel(self, *args, **kwargs):
  696. if self._quality_profile_drop_down_menu_model is None:
  697. self._quality_profile_drop_down_menu_model = QualityProfilesDropDownMenuModel(self)
  698. return self._quality_profile_drop_down_menu_model
  699. def getCustomQualityProfilesDropDownMenuModel(self, *args, **kwargs):
  700. if self._custom_quality_profile_drop_down_menu_model is None:
  701. self._custom_quality_profile_drop_down_menu_model = CustomQualityProfilesDropDownMenuModel(self)
  702. return self._custom_quality_profile_drop_down_menu_model
  703. ## Registers objects for the QML engine to use.
  704. #
  705. # \param engine The QML engine.
  706. def registerObjects(self, engine):
  707. super().registerObjects(engine)
  708. # global contexts
  709. engine.rootContext().setContextProperty("Printer", self)
  710. engine.rootContext().setContextProperty("CuraApplication", self)
  711. engine.rootContext().setContextProperty("PrintInformation", self._print_information)
  712. engine.rootContext().setContextProperty("CuraActions", self._cura_actions)
  713. engine.rootContext().setContextProperty("CuraSDKVersion", CuraSDKVersion)
  714. qmlRegisterUncreatableType(CuraApplication, "Cura", 1, 0, "ResourceTypes", "Just an Enum type")
  715. qmlRegisterSingletonType(CuraSceneController, "Cura", 1, 0, "SceneController", self.getCuraSceneController)
  716. qmlRegisterSingletonType(ExtruderManager, "Cura", 1, 0, "ExtruderManager", self.getExtruderManager)
  717. qmlRegisterSingletonType(MachineManager, "Cura", 1, 0, "MachineManager", self.getMachineManager)
  718. qmlRegisterSingletonType(SettingInheritanceManager, "Cura", 1, 0, "SettingInheritanceManager", self.getSettingInheritanceManager)
  719. qmlRegisterSingletonType(SimpleModeSettingsManager, "Cura", 1, 0, "SimpleModeSettingsManager", self.getSimpleModeSettingsManager)
  720. qmlRegisterSingletonType(MachineActionManager.MachineActionManager, "Cura", 1, 0, "MachineActionManager", self.getMachineActionManager)
  721. qmlRegisterSingletonType(ObjectsModel, "Cura", 1, 0, "ObjectsModel", self.getObjectsModel)
  722. qmlRegisterType(BuildPlateModel, "Cura", 1, 0, "BuildPlateModel")
  723. qmlRegisterType(MultiBuildPlateModel, "Cura", 1, 0, "MultiBuildPlateModel")
  724. qmlRegisterType(InstanceContainer, "Cura", 1, 0, "InstanceContainer")
  725. qmlRegisterType(ExtrudersModel, "Cura", 1, 0, "ExtrudersModel")
  726. qmlRegisterType(GenericMaterialsModel, "Cura", 1, 0, "GenericMaterialsModel")
  727. qmlRegisterType(BrandMaterialsModel, "Cura", 1, 0, "BrandMaterialsModel")
  728. qmlRegisterType(MaterialManagementModel, "Cura", 1, 0, "MaterialManagementModel")
  729. qmlRegisterType(QualityManagementModel, "Cura", 1, 0, "QualityManagementModel")
  730. qmlRegisterType(MachineManagementModel, "Cura", 1, 0, "MachineManagementModel")
  731. qmlRegisterSingletonType(QualityProfilesDropDownMenuModel, "Cura", 1, 0,
  732. "QualityProfilesDropDownMenuModel", self.getQualityProfilesDropDownMenuModel)
  733. qmlRegisterSingletonType(CustomQualityProfilesDropDownMenuModel, "Cura", 1, 0,
  734. "CustomQualityProfilesDropDownMenuModel", self.getCustomQualityProfilesDropDownMenuModel)
  735. qmlRegisterType(NozzleModel, "Cura", 1, 0, "NozzleModel")
  736. qmlRegisterType(MaterialSettingsVisibilityHandler, "Cura", 1, 0, "MaterialSettingsVisibilityHandler")
  737. qmlRegisterType(SettingVisibilityPresetsModel, "Cura", 1, 0, "SettingVisibilityPresetsModel")
  738. qmlRegisterType(QualitySettingsModel, "Cura", 1, 0, "QualitySettingsModel")
  739. qmlRegisterType(MachineNameValidator, "Cura", 1, 0, "MachineNameValidator")
  740. qmlRegisterType(UserChangesModel, "Cura", 1, 0, "UserChangesModel")
  741. qmlRegisterSingletonType(ContainerManager, "Cura", 1, 0, "ContainerManager", ContainerManager.getInstance)
  742. qmlRegisterType(SidebarCustomMenuItemsModel, "Cura", 1, 0, "SidebarCustomMenuItemsModel")
  743. # As of Qt5.7, it is necessary to get rid of any ".." in the path for the singleton to work.
  744. actions_url = QUrl.fromLocalFile(os.path.abspath(Resources.getPath(CuraApplication.ResourceTypes.QmlFiles, "Actions.qml")))
  745. qmlRegisterSingletonType(actions_url, "Cura", 1, 0, "Actions")
  746. for path in Resources.getAllResourcesOfType(CuraApplication.ResourceTypes.QmlFiles):
  747. type_name = os.path.splitext(os.path.basename(path))[0]
  748. if type_name in ("Cura", "Actions"):
  749. continue
  750. # Ignore anything that is not a QML file.
  751. if not path.endswith(".qml"):
  752. continue
  753. qmlRegisterType(QUrl.fromLocalFile(path), "Cura", 1, 0, type_name)
  754. def onSelectionChanged(self):
  755. if Selection.hasSelection():
  756. if self.getController().getActiveTool():
  757. # If the tool has been disabled by the new selection
  758. if not self.getController().getActiveTool().getEnabled():
  759. # Default
  760. self.getController().setActiveTool("TranslateTool")
  761. else:
  762. if self._previous_active_tool:
  763. self.getController().setActiveTool(self._previous_active_tool)
  764. if not self.getController().getActiveTool().getEnabled():
  765. self.getController().setActiveTool("TranslateTool")
  766. self._previous_active_tool = None
  767. else:
  768. # Default
  769. self.getController().setActiveTool("TranslateTool")
  770. if self.getPreferences().getValue("view/center_on_select"):
  771. self._center_after_select = True
  772. else:
  773. if self.getController().getActiveTool():
  774. self._previous_active_tool = self.getController().getActiveTool().getPluginId()
  775. self.getController().setActiveTool(None)
  776. def _onToolOperationStopped(self, event):
  777. if self._center_after_select and Selection.getSelectedObject(0) is not None:
  778. self._center_after_select = False
  779. self._camera_animation.setStart(self.getController().getTool("CameraTool").getOrigin())
  780. self._camera_animation.setTarget(Selection.getSelectedObject(0).getWorldPosition())
  781. self._camera_animation.start()
  782. def _onGlobalContainerChanged(self):
  783. if self._global_container_stack is not None:
  784. machine_file_formats = [file_type.strip() for file_type in self._global_container_stack.getMetaDataEntry("file_formats").split(";")]
  785. new_preferred_mimetype = ""
  786. if machine_file_formats:
  787. new_preferred_mimetype = machine_file_formats[0]
  788. if new_preferred_mimetype != self._preferred_mimetype:
  789. self._preferred_mimetype = new_preferred_mimetype
  790. self.preferredOutputMimetypeChanged.emit()
  791. requestAddPrinter = pyqtSignal()
  792. activityChanged = pyqtSignal()
  793. sceneBoundingBoxChanged = pyqtSignal()
  794. preferredOutputMimetypeChanged = pyqtSignal()
  795. @pyqtProperty(bool, notify = activityChanged)
  796. def platformActivity(self):
  797. return self._platform_activity
  798. @pyqtProperty(str, notify=preferredOutputMimetypeChanged)
  799. def preferredOutputMimetype(self):
  800. return self._preferred_mimetype
  801. @pyqtProperty(str, notify = sceneBoundingBoxChanged)
  802. def getSceneBoundingBoxString(self):
  803. return self._i18n_catalog.i18nc("@info 'width', 'depth' and 'height' are variable names that must NOT be translated; just translate the format of ##x##x## mm.", "%(width).1f x %(depth).1f x %(height).1f mm") % {'width' : self._scene_bounding_box.width.item(), 'depth': self._scene_bounding_box.depth.item(), 'height' : self._scene_bounding_box.height.item()}
  804. def updatePlatformActivityDelayed(self, node = None):
  805. if node is not None and (node.getMeshData() is not None or node.callDecoration("getLayerData")):
  806. self._update_platform_activity_timer.start()
  807. ## Update scene bounding box for current build plate
  808. def updatePlatformActivity(self, node = None):
  809. count = 0
  810. scene_bounding_box = None
  811. is_block_slicing_node = False
  812. active_build_plate = self.getMultiBuildPlateModel().activeBuildPlate
  813. print_information = self.getPrintInformation()
  814. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  815. if (
  816. not issubclass(type(node), CuraSceneNode) or
  817. (not node.getMeshData() and not node.callDecoration("getLayerData")) or
  818. (node.callDecoration("getBuildPlateNumber") != active_build_plate)):
  819. continue
  820. if node.callDecoration("isBlockSlicing"):
  821. is_block_slicing_node = True
  822. count += 1
  823. # After clicking the Undo button, if the build plate empty the project name needs to be set
  824. if print_information.baseName == '':
  825. print_information.setBaseName(node.getName())
  826. if not scene_bounding_box:
  827. scene_bounding_box = node.getBoundingBox()
  828. else:
  829. other_bb = node.getBoundingBox()
  830. if other_bb is not None:
  831. scene_bounding_box = scene_bounding_box + node.getBoundingBox()
  832. if print_information:
  833. print_information.setPreSliced(is_block_slicing_node)
  834. if not scene_bounding_box:
  835. scene_bounding_box = AxisAlignedBox.Null
  836. if repr(self._scene_bounding_box) != repr(scene_bounding_box):
  837. self._scene_bounding_box = scene_bounding_box
  838. self.sceneBoundingBoxChanged.emit()
  839. self._platform_activity = True if count > 0 else False
  840. self.activityChanged.emit()
  841. # Remove all selected objects from the scene.
  842. @pyqtSlot()
  843. @deprecated("Moved to CuraActions", "2.6")
  844. def deleteSelection(self):
  845. if not self.getController().getToolsEnabled():
  846. return
  847. removed_group_nodes = []
  848. op = GroupedOperation()
  849. nodes = Selection.getAllSelectedObjects()
  850. for node in nodes:
  851. op.addOperation(RemoveSceneNodeOperation(node))
  852. group_node = node.getParent()
  853. if group_node and group_node.callDecoration("isGroup") and group_node not in removed_group_nodes:
  854. remaining_nodes_in_group = list(set(group_node.getChildren()) - set(nodes))
  855. if len(remaining_nodes_in_group) == 1:
  856. removed_group_nodes.append(group_node)
  857. op.addOperation(SetParentOperation(remaining_nodes_in_group[0], group_node.getParent()))
  858. op.addOperation(RemoveSceneNodeOperation(group_node))
  859. op.push()
  860. ## Remove an object from the scene.
  861. # Note that this only removes an object if it is selected.
  862. @pyqtSlot("quint64")
  863. @deprecated("Use deleteSelection instead", "2.6")
  864. def deleteObject(self, object_id):
  865. if not self.getController().getToolsEnabled():
  866. return
  867. node = self.getController().getScene().findObject(object_id)
  868. if not node and object_id != 0: # Workaround for tool handles overlapping the selected object
  869. node = Selection.getSelectedObject(0)
  870. if node:
  871. op = GroupedOperation()
  872. op.addOperation(RemoveSceneNodeOperation(node))
  873. group_node = node.getParent()
  874. if group_node:
  875. # Note that at this point the node has not yet been deleted
  876. if len(group_node.getChildren()) <= 2 and group_node.callDecoration("isGroup"):
  877. op.addOperation(SetParentOperation(group_node.getChildren()[0], group_node.getParent()))
  878. op.addOperation(RemoveSceneNodeOperation(group_node))
  879. op.push()
  880. ## Create a number of copies of existing object.
  881. # \param object_id
  882. # \param count number of copies
  883. # \param min_offset minimum offset to other objects.
  884. @pyqtSlot("quint64", int)
  885. @deprecated("Use CuraActions::multiplySelection", "2.6")
  886. def multiplyObject(self, object_id, count, min_offset = 8):
  887. node = self.getController().getScene().findObject(object_id)
  888. if not node:
  889. node = Selection.getSelectedObject(0)
  890. while node.getParent() and node.getParent().callDecoration("isGroup"):
  891. node = node.getParent()
  892. job = MultiplyObjectsJob([node], count, min_offset)
  893. job.start()
  894. return
  895. ## Center object on platform.
  896. @pyqtSlot("quint64")
  897. @deprecated("Use CuraActions::centerSelection", "2.6")
  898. def centerObject(self, object_id):
  899. node = self.getController().getScene().findObject(object_id)
  900. if not node and object_id != 0: # Workaround for tool handles overlapping the selected object
  901. node = Selection.getSelectedObject(0)
  902. if not node:
  903. return
  904. if node.getParent() and node.getParent().callDecoration("isGroup"):
  905. node = node.getParent()
  906. if node:
  907. op = SetTransformOperation(node, Vector())
  908. op.push()
  909. ## Select all nodes containing mesh data in the scene.
  910. @pyqtSlot()
  911. def selectAll(self):
  912. if not self.getController().getToolsEnabled():
  913. return
  914. Selection.clear()
  915. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  916. if not isinstance(node, SceneNode):
  917. continue
  918. if not node.getMeshData() and not node.callDecoration("isGroup"):
  919. continue # Node that doesnt have a mesh and is not a group.
  920. if node.getParent() and node.getParent().callDecoration("isGroup") or node.getParent().callDecoration("isSliceable"):
  921. continue # Grouped nodes don't need resetting as their parent (the group) is resetted)
  922. if not node.isSelectable():
  923. continue # i.e. node with layer data
  924. if not node.callDecoration("isSliceable") and not node.callDecoration("isGroup"):
  925. continue # i.e. node with layer data
  926. Selection.add(node)
  927. ## Reset all translation on nodes with mesh data.
  928. @pyqtSlot()
  929. def resetAllTranslation(self):
  930. Logger.log("i", "Resetting all scene translations")
  931. nodes = []
  932. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  933. if not isinstance(node, SceneNode):
  934. continue
  935. if not node.getMeshData() and not node.callDecoration("isGroup"):
  936. continue # Node that doesnt have a mesh and is not a group.
  937. if node.getParent() and node.getParent().callDecoration("isGroup"):
  938. continue # Grouped nodes don't need resetting as their parent (the group) is resetted)
  939. if not node.isSelectable():
  940. continue # i.e. node with layer data
  941. nodes.append(node)
  942. if nodes:
  943. op = GroupedOperation()
  944. for node in nodes:
  945. # Ensure that the object is above the build platform
  946. node.removeDecorator(ZOffsetDecorator.ZOffsetDecorator)
  947. if node.getBoundingBox():
  948. center_y = node.getWorldPosition().y - node.getBoundingBox().bottom
  949. else:
  950. center_y = 0
  951. op.addOperation(SetTransformOperation(node, Vector(0, center_y, 0)))
  952. op.push()
  953. ## Reset all transformations on nodes with mesh data.
  954. @pyqtSlot()
  955. def resetAll(self):
  956. Logger.log("i", "Resetting all scene transformations")
  957. nodes = []
  958. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  959. if not isinstance(node, SceneNode):
  960. continue
  961. if not node.getMeshData() and not node.callDecoration("isGroup"):
  962. continue # Node that doesnt have a mesh and is not a group.
  963. if node.getParent() and node.getParent().callDecoration("isGroup"):
  964. continue # Grouped nodes don't need resetting as their parent (the group) is resetted)
  965. if not node.callDecoration("isSliceable") and not node.callDecoration("isGroup"):
  966. continue # i.e. node with layer data
  967. nodes.append(node)
  968. if nodes:
  969. op = GroupedOperation()
  970. for node in nodes:
  971. # Ensure that the object is above the build platform
  972. node.removeDecorator(ZOffsetDecorator.ZOffsetDecorator)
  973. if node.getBoundingBox():
  974. center_y = node.getWorldPosition().y - node.getBoundingBox().bottom
  975. else:
  976. center_y = 0
  977. op.addOperation(SetTransformOperation(node, Vector(0, center_y, 0), Quaternion(), Vector(1, 1, 1)))
  978. op.push()
  979. ## Arrange all objects.
  980. @pyqtSlot()
  981. def arrangeObjectsToAllBuildPlates(self):
  982. nodes = []
  983. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  984. if not isinstance(node, SceneNode):
  985. continue
  986. if not node.getMeshData() and not node.callDecoration("isGroup"):
  987. continue # Node that doesnt have a mesh and is not a group.
  988. if node.getParent() and node.getParent().callDecoration("isGroup"):
  989. continue # Grouped nodes don't need resetting as their parent (the group) is resetted)
  990. if not node.callDecoration("isSliceable") and not node.callDecoration("isGroup"):
  991. continue # i.e. node with layer data
  992. # Skip nodes that are too big
  993. if node.getBoundingBox().width < self._volume.getBoundingBox().width or node.getBoundingBox().depth < self._volume.getBoundingBox().depth:
  994. nodes.append(node)
  995. job = ArrangeObjectsAllBuildPlatesJob(nodes)
  996. job.start()
  997. self.getCuraSceneController().setActiveBuildPlate(0) # Select first build plate
  998. # Single build plate
  999. @pyqtSlot()
  1000. def arrangeAll(self):
  1001. nodes = []
  1002. active_build_plate = self.getMultiBuildPlateModel().activeBuildPlate
  1003. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  1004. if not isinstance(node, SceneNode):
  1005. continue
  1006. if not node.getMeshData() and not node.callDecoration("isGroup"):
  1007. continue # Node that doesnt have a mesh and is not a group.
  1008. if node.getParent() and node.getParent().callDecoration("isGroup"):
  1009. continue # Grouped nodes don't need resetting as their parent (the group) is resetted)
  1010. if not node.isSelectable():
  1011. continue # i.e. node with layer data
  1012. if not node.callDecoration("isSliceable") and not node.callDecoration("isGroup"):
  1013. continue # i.e. node with layer data
  1014. if node.callDecoration("getBuildPlateNumber") == active_build_plate:
  1015. # Skip nodes that are too big
  1016. if node.getBoundingBox().width < self._volume.getBoundingBox().width or node.getBoundingBox().depth < self._volume.getBoundingBox().depth:
  1017. nodes.append(node)
  1018. self.arrange(nodes, fixed_nodes = [])
  1019. ## Arrange a set of nodes given a set of fixed nodes
  1020. # \param nodes nodes that we have to place
  1021. # \param fixed_nodes nodes that are placed in the arranger before finding spots for nodes
  1022. def arrange(self, nodes, fixed_nodes):
  1023. min_offset = self.getBuildVolume().getEdgeDisallowedSize() + 2 # Allow for some rounding errors
  1024. job = ArrangeObjectsJob(nodes, fixed_nodes, min_offset = max(min_offset, 8))
  1025. job.start()
  1026. ## Reload all mesh data on the screen from file.
  1027. @pyqtSlot()
  1028. def reloadAll(self):
  1029. Logger.log("i", "Reloading all loaded mesh data.")
  1030. nodes = []
  1031. has_merged_nodes = False
  1032. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  1033. if not isinstance(node, CuraSceneNode) or not node.getMeshData() :
  1034. if node.getName() == "MergedMesh":
  1035. has_merged_nodes = True
  1036. continue
  1037. nodes.append(node)
  1038. if not nodes:
  1039. return
  1040. for node in nodes:
  1041. file_name = node.getMeshData().getFileName()
  1042. if file_name:
  1043. job = ReadMeshJob(file_name)
  1044. job._node = node
  1045. job.finished.connect(self._reloadMeshFinished)
  1046. if has_merged_nodes:
  1047. job.finished.connect(self.updateOriginOfMergedMeshes)
  1048. job.start()
  1049. else:
  1050. Logger.log("w", "Unable to reload data because we don't have a filename.")
  1051. ## Get logging data of the backend engine
  1052. # \returns \type{string} Logging data
  1053. @pyqtSlot(result = str)
  1054. def getEngineLog(self):
  1055. log = ""
  1056. for entry in self.getBackend().getLog():
  1057. log += entry.decode()
  1058. return log
  1059. @pyqtSlot("QStringList")
  1060. def setExpandedCategories(self, categories):
  1061. categories = list(set(categories))
  1062. categories.sort()
  1063. joined = ";".join(categories)
  1064. if joined != self.getPreferences().getValue("cura/categories_expanded"):
  1065. self.getPreferences().setValue("cura/categories_expanded", joined)
  1066. self.expandedCategoriesChanged.emit()
  1067. expandedCategoriesChanged = pyqtSignal()
  1068. @pyqtProperty("QStringList", notify = expandedCategoriesChanged)
  1069. def expandedCategories(self):
  1070. return self.getPreferences().getValue("cura/categories_expanded").split(";")
  1071. @pyqtSlot()
  1072. def mergeSelected(self):
  1073. self.groupSelected()
  1074. try:
  1075. group_node = Selection.getAllSelectedObjects()[0]
  1076. except Exception as e:
  1077. Logger.log("e", "mergeSelected: Exception: %s", e)
  1078. return
  1079. meshes = [node.getMeshData() for node in group_node.getAllChildren() if node.getMeshData()]
  1080. # Compute the center of the objects
  1081. object_centers = []
  1082. # Forget about the translation that the original objects have
  1083. zero_translation = Matrix(data=numpy.zeros(3))
  1084. for mesh, node in zip(meshes, group_node.getChildren()):
  1085. transformation = node.getLocalTransformation()
  1086. transformation.setTranslation(zero_translation)
  1087. transformed_mesh = mesh.getTransformed(transformation)
  1088. center = transformed_mesh.getCenterPosition()
  1089. if center is not None:
  1090. object_centers.append(center)
  1091. if object_centers and len(object_centers) > 0:
  1092. middle_x = sum([v.x for v in object_centers]) / len(object_centers)
  1093. middle_y = sum([v.y for v in object_centers]) / len(object_centers)
  1094. middle_z = sum([v.z for v in object_centers]) / len(object_centers)
  1095. offset = Vector(middle_x, middle_y, middle_z)
  1096. else:
  1097. offset = Vector(0, 0, 0)
  1098. # Move each node to the same position.
  1099. for mesh, node in zip(meshes, group_node.getChildren()):
  1100. transformation = node.getLocalTransformation()
  1101. transformation.setTranslation(zero_translation)
  1102. transformed_mesh = mesh.getTransformed(transformation)
  1103. # Align the object around its zero position
  1104. # and also apply the offset to center it inside the group.
  1105. node.setPosition(-transformed_mesh.getZeroPosition() - offset)
  1106. # Use the previously found center of the group bounding box as the new location of the group
  1107. group_node.setPosition(group_node.getBoundingBox().center)
  1108. group_node.setName("MergedMesh") # add a specific name to distinguish this node
  1109. ## Updates origin position of all merged meshes
  1110. # \param jobNode \type{Job} empty object which passed which is required by JobQueue
  1111. def updateOriginOfMergedMeshes(self, jobNode):
  1112. group_nodes = []
  1113. for node in DepthFirstIterator(self.getController().getScene().getRoot()):
  1114. if isinstance(node, CuraSceneNode) and node.getName() == "MergedMesh":
  1115. #checking by name might be not enough, the merged mesh should has "GroupDecorator" decorator
  1116. for decorator in node.getDecorators():
  1117. if isinstance(decorator, GroupDecorator):
  1118. group_nodes.append(node)
  1119. break
  1120. for group_node in group_nodes:
  1121. meshes = [node.getMeshData() for node in group_node.getAllChildren() if node.getMeshData()]
  1122. # Compute the center of the objects
  1123. object_centers = []
  1124. # Forget about the translation that the original objects have
  1125. zero_translation = Matrix(data=numpy.zeros(3))
  1126. for mesh, node in zip(meshes, group_node.getChildren()):
  1127. transformation = node.getLocalTransformation()
  1128. transformation.setTranslation(zero_translation)
  1129. transformed_mesh = mesh.getTransformed(transformation)
  1130. center = transformed_mesh.getCenterPosition()
  1131. if center is not None:
  1132. object_centers.append(center)
  1133. if object_centers and len(object_centers) > 0:
  1134. middle_x = sum([v.x for v in object_centers]) / len(object_centers)
  1135. middle_y = sum([v.y for v in object_centers]) / len(object_centers)
  1136. middle_z = sum([v.z for v in object_centers]) / len(object_centers)
  1137. offset = Vector(middle_x, middle_y, middle_z)
  1138. else:
  1139. offset = Vector(0, 0, 0)
  1140. # Move each node to the same position.
  1141. for mesh, node in zip(meshes, group_node.getChildren()):
  1142. transformation = node.getLocalTransformation()
  1143. transformation.setTranslation(zero_translation)
  1144. transformed_mesh = mesh.getTransformed(transformation)
  1145. # Align the object around its zero position
  1146. # and also apply the offset to center it inside the group.
  1147. node.setPosition(-transformed_mesh.getZeroPosition() - offset)
  1148. # Use the previously found center of the group bounding box as the new location of the group
  1149. group_node.setPosition(group_node.getBoundingBox().center)
  1150. @pyqtSlot()
  1151. def groupSelected(self):
  1152. # Create a group-node
  1153. group_node = CuraSceneNode()
  1154. group_decorator = GroupDecorator()
  1155. group_node.addDecorator(group_decorator)
  1156. group_node.addDecorator(ConvexHullDecorator())
  1157. group_node.addDecorator(BuildPlateDecorator(self.getMultiBuildPlateModel().activeBuildPlate))
  1158. group_node.setParent(self.getController().getScene().getRoot())
  1159. group_node.setSelectable(True)
  1160. center = Selection.getSelectionCenter()
  1161. group_node.setPosition(center)
  1162. group_node.setCenterPosition(center)
  1163. # Remove nodes that are directly parented to another selected node from the selection so they remain parented
  1164. selected_nodes = Selection.getAllSelectedObjects().copy()
  1165. for node in selected_nodes:
  1166. if node.getParent() in selected_nodes and not node.getParent().callDecoration("isGroup"):
  1167. Selection.remove(node)
  1168. # Move selected nodes into the group-node
  1169. Selection.applyOperation(SetParentOperation, group_node)
  1170. # Deselect individual nodes and select the group-node instead
  1171. for node in group_node.getChildren():
  1172. Selection.remove(node)
  1173. Selection.add(group_node)
  1174. @pyqtSlot()
  1175. def ungroupSelected(self):
  1176. selected_objects = Selection.getAllSelectedObjects().copy()
  1177. for node in selected_objects:
  1178. if node.callDecoration("isGroup"):
  1179. op = GroupedOperation()
  1180. group_parent = node.getParent()
  1181. children = node.getChildren().copy()
  1182. for child in children:
  1183. # Ungroup only 1 level deep
  1184. if child.getParent() != node:
  1185. continue
  1186. # Set the parent of the children to the parent of the group-node
  1187. op.addOperation(SetParentOperation(child, group_parent))
  1188. # Add all individual nodes to the selection
  1189. Selection.add(child)
  1190. op.push()
  1191. # Note: The group removes itself from the scene once all its children have left it,
  1192. # see GroupDecorator._onChildrenChanged
  1193. def _createSplashScreen(self):
  1194. if self._is_headless:
  1195. return None
  1196. return CuraSplashScreen.CuraSplashScreen()
  1197. def _onActiveMachineChanged(self):
  1198. pass
  1199. fileLoaded = pyqtSignal(str)
  1200. fileCompleted = pyqtSignal(str)
  1201. def _reloadMeshFinished(self, job):
  1202. # TODO; This needs to be fixed properly. We now make the assumption that we only load a single mesh!
  1203. job_result = job.getResult()
  1204. if len(job_result) == 0:
  1205. Logger.log("e", "Reloading the mesh failed.")
  1206. return
  1207. mesh_data = job_result[0].getMeshData()
  1208. if not mesh_data:
  1209. Logger.log("w", "Could not find a mesh in reloaded node.")
  1210. return
  1211. job._node.setMeshData(mesh_data)
  1212. def _openFile(self, filename):
  1213. self.readLocalFile(QUrl.fromLocalFile(filename))
  1214. def _addProfileReader(self, profile_reader):
  1215. # TODO: Add the profile reader to the list of plug-ins that can be used when importing profiles.
  1216. pass
  1217. def _addProfileWriter(self, profile_writer):
  1218. pass
  1219. @pyqtSlot("QSize")
  1220. def setMinimumWindowSize(self, size):
  1221. main_window = self.getMainWindow()
  1222. if main_window:
  1223. main_window.setMinimumSize(size)
  1224. def getBuildVolume(self):
  1225. return self._volume
  1226. additionalComponentsChanged = pyqtSignal(str, arguments = ["areaId"])
  1227. @pyqtProperty("QVariantMap", notify = additionalComponentsChanged)
  1228. def additionalComponents(self):
  1229. return self._additional_components
  1230. ## Add a component to a list of components to be reparented to another area in the GUI.
  1231. # The actual reparenting is done by the area itself.
  1232. # \param area_id \type{str} Identifying name of the area to which the component should be reparented
  1233. # \param component \type{QQuickComponent} The component that should be reparented
  1234. @pyqtSlot(str, "QVariant")
  1235. def addAdditionalComponent(self, area_id, component):
  1236. if area_id not in self._additional_components:
  1237. self._additional_components[area_id] = []
  1238. self._additional_components[area_id].append(component)
  1239. self.additionalComponentsChanged.emit(area_id)
  1240. @pyqtSlot(str)
  1241. def log(self, msg):
  1242. Logger.log("d", msg)
  1243. openProjectFile = pyqtSignal(QUrl, arguments = ["project_file"]) # Emitted when a project file is about to open.
  1244. @pyqtSlot(QUrl, bool)
  1245. def readLocalFile(self, file, skip_project_file_check = False):
  1246. if not file.isValid():
  1247. return
  1248. scene = self.getController().getScene()
  1249. for node in DepthFirstIterator(scene.getRoot()):
  1250. if node.callDecoration("isBlockSlicing"):
  1251. self.deleteAll()
  1252. break
  1253. if not skip_project_file_check and self.checkIsValidProjectFile(file):
  1254. self.callLater(self.openProjectFile.emit, file)
  1255. return
  1256. if self.getPreferences().getValue("cura/select_models_on_load"):
  1257. Selection.clear()
  1258. f = file.toLocalFile()
  1259. extension = os.path.splitext(f)[1]
  1260. extension = extension.lower()
  1261. filename = os.path.basename(f)
  1262. if len(self._currently_loading_files) > 0:
  1263. # If a non-slicable file is already being loaded, we prevent loading of any further non-slicable files
  1264. if extension in self._non_sliceable_extensions:
  1265. message = Message(
  1266. self._i18n_catalog.i18nc("@info:status",
  1267. "Only one G-code file can be loaded at a time. Skipped importing {0}",
  1268. filename), title = self._i18n_catalog.i18nc("@info:title", "Warning"))
  1269. message.show()
  1270. return
  1271. # If file being loaded is non-slicable file, then prevent loading of any other files
  1272. extension = os.path.splitext(self._currently_loading_files[0])[1]
  1273. extension = extension.lower()
  1274. if extension in self._non_sliceable_extensions:
  1275. message = Message(
  1276. self._i18n_catalog.i18nc("@info:status",
  1277. "Can't open any other file if G-code is loading. Skipped importing {0}",
  1278. filename), title = self._i18n_catalog.i18nc("@info:title", "Error"))
  1279. message.show()
  1280. return
  1281. self._currently_loading_files.append(f)
  1282. if extension in self._non_sliceable_extensions:
  1283. self.deleteAll(only_selectable = False)
  1284. job = ReadMeshJob(f)
  1285. job.finished.connect(self._readMeshFinished)
  1286. job.start()
  1287. def _readMeshFinished(self, job):
  1288. nodes = job.getResult()
  1289. file_name = job.getFileName()
  1290. file_name_lower = file_name.lower()
  1291. file_extension = file_name_lower.split(".")[-1]
  1292. self._currently_loading_files.remove(file_name)
  1293. self.fileLoaded.emit(file_name)
  1294. target_build_plate = self.getMultiBuildPlateModel().activeBuildPlate
  1295. root = self.getController().getScene().getRoot()
  1296. fixed_nodes = []
  1297. for node_ in DepthFirstIterator(root):
  1298. if node_.callDecoration("isSliceable") and node_.callDecoration("getBuildPlateNumber") == target_build_plate:
  1299. fixed_nodes.append(node_)
  1300. global_container_stack = self.getGlobalContainerStack()
  1301. machine_width = global_container_stack.getProperty("machine_width", "value")
  1302. machine_depth = global_container_stack.getProperty("machine_depth", "value")
  1303. arranger = Arrange.create(x = machine_width, y = machine_depth, fixed_nodes = fixed_nodes)
  1304. min_offset = 8
  1305. default_extruder_position = self.getMachineManager().defaultExtruderPosition
  1306. default_extruder_id = self._global_container_stack.extruders[default_extruder_position].getId()
  1307. select_models_on_load = self.getPreferences().getValue("cura/select_models_on_load")
  1308. for original_node in nodes:
  1309. # Create a CuraSceneNode just if the original node is not that type
  1310. if isinstance(original_node, CuraSceneNode):
  1311. node = original_node
  1312. else:
  1313. node = CuraSceneNode()
  1314. node.setMeshData(original_node.getMeshData())
  1315. #Setting meshdata does not apply scaling.
  1316. if(original_node.getScale() != Vector(1.0, 1.0, 1.0)):
  1317. node.scale(original_node.getScale())
  1318. node.setSelectable(True)
  1319. node.setName(os.path.basename(file_name))
  1320. self.getBuildVolume().checkBoundsAndUpdate(node)
  1321. is_non_sliceable = "." + file_extension in self._non_sliceable_extensions
  1322. if is_non_sliceable:
  1323. self.callLater(lambda: self.getController().setActiveView("SimulationView"))
  1324. block_slicing_decorator = BlockSlicingDecorator()
  1325. node.addDecorator(block_slicing_decorator)
  1326. else:
  1327. sliceable_decorator = SliceableObjectDecorator()
  1328. node.addDecorator(sliceable_decorator)
  1329. scene = self.getController().getScene()
  1330. # If there is no convex hull for the node, start calculating it and continue.
  1331. if not node.getDecorator(ConvexHullDecorator):
  1332. node.addDecorator(ConvexHullDecorator())
  1333. for child in node.getAllChildren():
  1334. if not child.getDecorator(ConvexHullDecorator):
  1335. child.addDecorator(ConvexHullDecorator())
  1336. if file_extension != "3mf":
  1337. if node.callDecoration("isSliceable"):
  1338. # Only check position if it's not already blatantly obvious that it won't fit.
  1339. if node.getBoundingBox() is None or self._volume.getBoundingBox() is None or node.getBoundingBox().width < self._volume.getBoundingBox().width or node.getBoundingBox().depth < self._volume.getBoundingBox().depth:
  1340. # Find node location
  1341. offset_shape_arr, hull_shape_arr = ShapeArray.fromNode(node, min_offset = min_offset)
  1342. # If a model is to small then it will not contain any points
  1343. if offset_shape_arr is None and hull_shape_arr is None:
  1344. Message(self._i18n_catalog.i18nc("@info:status", "The selected model was too small to load."),
  1345. title=self._i18n_catalog.i18nc("@info:title", "Warning")).show()
  1346. return
  1347. # Step is for skipping tests to make it a lot faster. it also makes the outcome somewhat rougher
  1348. arranger.findNodePlacement(node, offset_shape_arr, hull_shape_arr, step = 10)
  1349. # This node is deep copied from some other node which already has a BuildPlateDecorator, but the deepcopy
  1350. # of BuildPlateDecorator produces one that's associated with build plate -1. So, here we need to check if
  1351. # the BuildPlateDecorator exists or not and always set the correct build plate number.
  1352. build_plate_decorator = node.getDecorator(BuildPlateDecorator)
  1353. if build_plate_decorator is None:
  1354. build_plate_decorator = BuildPlateDecorator(target_build_plate)
  1355. node.addDecorator(build_plate_decorator)
  1356. build_plate_decorator.setBuildPlateNumber(target_build_plate)
  1357. op = AddSceneNodeOperation(node, scene.getRoot())
  1358. op.push()
  1359. node.callDecoration("setActiveExtruder", default_extruder_id)
  1360. scene.sceneChanged.emit(node)
  1361. if select_models_on_load:
  1362. Selection.add(node)
  1363. self.fileCompleted.emit(file_name)
  1364. def addNonSliceableExtension(self, extension):
  1365. self._non_sliceable_extensions.append(extension)
  1366. @pyqtSlot(str, result=bool)
  1367. def checkIsValidProjectFile(self, file_url):
  1368. """
  1369. Checks if the given file URL is a valid project file.
  1370. """
  1371. file_path = QUrl(file_url).toLocalFile()
  1372. workspace_reader = self.getWorkspaceFileHandler().getReaderForFile(file_path)
  1373. if workspace_reader is None:
  1374. return False # non-project files won't get a reader
  1375. try:
  1376. result = workspace_reader.preRead(file_path, show_dialog=False)
  1377. return result == WorkspaceReader.PreReadResult.accepted
  1378. except Exception as e:
  1379. Logger.logException("e", "Could not check file %s", file_url)
  1380. return False
  1381. def _onContextMenuRequested(self, x: float, y: float) -> None:
  1382. # Ensure we select the object if we request a context menu over an object without having a selection.
  1383. if not Selection.hasSelection():
  1384. node = self.getController().getScene().findObject(cast(SelectionPass, self.getRenderer().getRenderPass("selection")).getIdAtPosition(x, y))
  1385. if node:
  1386. parent = node.getParent()
  1387. while(parent and parent.callDecoration("isGroup")):
  1388. node = parent
  1389. parent = node.getParent()
  1390. Selection.add(node)
  1391. @pyqtSlot()
  1392. def showMoreInformationDialogForAnonymousDataCollection(self):
  1393. cast(SliceInfo, self._plugin_registry.getPluginObject("SliceInfoPlugin")).showMoreInfoDialog()
  1394. def addSidebarCustomMenuItem(self, menu_item: dict) -> None:
  1395. self._sidebar_custom_menu_items.append(menu_item)
  1396. def getSidebarCustomMenuItems(self) -> list:
  1397. return self._sidebar_custom_menu_items