CuraApplication.py 91 KB

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