mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-21 21:29:41 +08:00
bc18b078ab
Instead of only writing g-code, the Removable Drive output device will now try to write a file format that the current machine supports. It just picks the first one it finds. Contributes to issue CURA-611.
109 lines
4.8 KiB
Python
109 lines
4.8 KiB
Python
import os.path
|
|
|
|
from UM.Application import Application
|
|
from UM.Logger import Logger
|
|
from UM.Message import Message
|
|
from UM.Mesh.WriteMeshJob import WriteMeshJob
|
|
from UM.Mesh.MeshWriter import MeshWriter
|
|
from UM.Scene.Iterator.BreadthFirstIterator import BreadthFirstIterator
|
|
from UM.OutputDevice.OutputDevice import OutputDevice
|
|
from UM.OutputDevice import OutputDeviceError
|
|
from UM.Preferences import Preferences
|
|
|
|
from UM.i18n import i18nCatalog
|
|
catalog = i18nCatalog("cura")
|
|
|
|
class RemovableDriveOutputDevice(OutputDevice):
|
|
def __init__(self, device_id, device_name):
|
|
super().__init__(device_id)
|
|
|
|
self.setName(device_name)
|
|
self.setShortDescription(catalog.i18nc("@action:button", "Save to Removable Drive"))
|
|
self.setDescription(catalog.i18nc("@item:inlistbox", "Save to Removable Drive {0}").format(device_name))
|
|
self.setIconName("save_sd")
|
|
self.setPriority(1)
|
|
|
|
self._writing = False
|
|
|
|
def requestWrite(self, node, file_name = None):
|
|
if self._writing:
|
|
raise OutputDeviceError.DeviceBusyError()
|
|
|
|
file_formats = Application.getInstance().getMeshFileHandler().getSupportedFileTypesWrite() #Formats supported by this application.
|
|
machine_file_formats = Application.getInstance().getMachineManager().getActiveMachineInstance().getMachineDefinition().getFileFormats()
|
|
for file_format in file_formats:
|
|
if file_format["mime_type"] in machine_file_formats:
|
|
writer = Application.getInstance().getMeshFileHandler().getWriterByMimeType(file_format["mime_type"])
|
|
extension = file_format["extension"]
|
|
break #We have a valid mesh writer, supported by the machine. Just pick the first one.
|
|
else:
|
|
Logger.log("e", "None of the file formats supported by this machine are supported by the application!")
|
|
raise OutputDeviceError.WriteRequestFailedError()
|
|
|
|
if file_name == None:
|
|
for n in BreadthFirstIterator(node):
|
|
if n.getMeshData():
|
|
file_name = n.getName()
|
|
if file_name:
|
|
break
|
|
|
|
if not file_name:
|
|
Logger.log("e", "Could not determine a proper file name when trying to write to %s, aborting", self.getName())
|
|
raise OutputDeviceError.WriteRequestFailedError()
|
|
|
|
if extension: #Not empty string.
|
|
extension = "." + extension
|
|
file_name = os.path.join(self.getId(), os.path.splitext(file_name)[0] + extension)
|
|
|
|
try:
|
|
Logger.log("d", "Writing to %s", file_name)
|
|
stream = open(file_name, "wt")
|
|
job = WriteMeshJob(writer, stream, node, MeshWriter.OutputMode.TextMode)
|
|
job.setFileName(file_name)
|
|
job.progress.connect(self._onProgress)
|
|
job.finished.connect(self._onFinished)
|
|
|
|
message = Message(catalog.i18nc("@info:progress", "Saving to Removable Drive <filename>{0}</filename>").format(self.getName()), 0, False, -1)
|
|
message.show()
|
|
|
|
self.writeStarted.emit(self)
|
|
|
|
job._message = message
|
|
self._writing = True
|
|
job.start()
|
|
except PermissionError as e:
|
|
Logger.log("e", "Permission denied when trying to write to %s: %s", file_name, str(e))
|
|
raise OutputDeviceError.PermissionDeniedError() from e
|
|
except OSError as e:
|
|
Logger.log("e", "Operating system would not let us write to %s: %s", file_name, str(e))
|
|
raise OutputDeviceError.WriteRequestFailedError() from e
|
|
|
|
def _onProgress(self, job, progress):
|
|
if hasattr(job, "_message"):
|
|
job._message.setProgress(progress)
|
|
self.writeProgress.emit(self, progress)
|
|
|
|
def _onFinished(self, job):
|
|
if hasattr(job, "_message"):
|
|
job._message.hide()
|
|
job._message = None
|
|
|
|
self._writing = False
|
|
self.writeFinished.emit(self)
|
|
if job.getResult():
|
|
message = Message(catalog.i18nc("@info:status", "Saved to Removable Drive {0} as {1}").format(self.getName(), os.path.basename(job.getFileName())))
|
|
message.addAction("eject", catalog.i18nc("@action:button", "Eject"), "eject", catalog.i18nc("@action", "Eject removable device {0}").format(self.getName()))
|
|
message.actionTriggered.connect(self._onActionTriggered)
|
|
message.show()
|
|
self.writeSuccess.emit(self)
|
|
else:
|
|
message = Message(catalog.i18nc("@info:status", "Could not save to removable drive {0}: {1}").format(self.getName(), str(job.getError())))
|
|
message.show()
|
|
self.writeError.emit(self)
|
|
job.getStream().close()
|
|
|
|
def _onActionTriggered(self, message, action):
|
|
if action == "eject":
|
|
Application.getInstance().getOutputDeviceManager().getOutputDevicePlugin("RemovableDriveOutputDevice").ejectDevice(self)
|
|
|