mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-21 21:29:41 +08:00
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
import os.path
|
|
|
|
from UM.Qt.QtApplication import QtApplication
|
|
from cura.Stages.CuraStage import CuraStage
|
|
|
|
from typing import TYPE_CHECKING, Optional
|
|
|
|
if TYPE_CHECKING:
|
|
from UM.View.View import View
|
|
|
|
|
|
class PreviewStage(CuraStage):
|
|
def __init__(self, application: QtApplication, parent = None) -> None:
|
|
super().__init__(parent)
|
|
self._application = application
|
|
self._application.engineCreatedSignal.connect(self._engineCreated)
|
|
self._previously_active_view = None # type: Optional[View]
|
|
|
|
def onStageSelected(self) -> None:
|
|
self._previously_active_view = self._application.getController().getActiveView()
|
|
|
|
def onStageDeselected(self) -> None:
|
|
if self._previously_active_view is not None:
|
|
self._application.getController().setActiveView(self._previously_active_view.getPluginId())
|
|
self._previously_active_view = None
|
|
|
|
def _engineCreated(self) -> None:
|
|
plugin_path = self._application.getPluginRegistry().getPluginPath(self.getPluginId())
|
|
if plugin_path is not None:
|
|
menu_component_path = os.path.join(plugin_path, "PreviewMenu.qml")
|
|
self.addDisplayComponent("menu", menu_component_path)
|