This commit is contained in:
Diego Prado Gesto 2018-01-26 14:14:59 +01:00
commit a0ebb5205b
5 changed files with 105 additions and 4 deletions

View File

@ -0,0 +1,68 @@
import zipfile
from io import StringIO
from UM.Resources import Resources
from UM.Mesh.MeshWriter import MeshWriter
from UM.Logger import Logger
from UM.PluginRegistry import PluginRegistry
MYPY = False
try:
if not MYPY:
import xml.etree.cElementTree as ET
except ImportError:
Logger.log("w", "Unable to load cElementTree, switching to slower version")
import xml.etree.ElementTree as ET
class UCPWriter(MeshWriter):
def __init__(self):
super().__init__()
self._namespaces = {
"content-types": "http://schemas.openxmlformats.org/package/2006/content-types",
"relationships": "http://schemas.openxmlformats.org/package/2006/relationships",
}
def write(self, stream, nodes, mode = MeshWriter.OutputMode.BinaryMode):
self._archive = None # Reset archive
archive = zipfile.ZipFile(stream, "w", compression=zipfile.ZIP_DEFLATED)
gcode_file = zipfile.ZipInfo("3D/model.gcode")
gcode_file.compress_type = zipfile.ZIP_DEFLATED
# Create content types file
content_types_file = zipfile.ZipInfo("[Content_Types].xml")
content_types_file.compress_type = zipfile.ZIP_DEFLATED
content_types = ET.Element("Types", xmlns=self._namespaces["content-types"])
rels_type = ET.SubElement(content_types, "Default", Extension="rels",
ContentType="application/vnd.openxmlformats-package.relationships+xml")
gcode_type = ET.SubElement(content_types, "Default", Extension="gcode",
ContentType="text/x-gcode")
image_type = ET.SubElement(content_types, "Default", Extension="png",
ContentType="image/png")
# Create _rels/.rels file
relations_file = zipfile.ZipInfo("_rels/.rels")
relations_file.compress_type = zipfile.ZIP_DEFLATED
relations_element = ET.Element("Relationships", xmlns=self._namespaces["relationships"])
thumbnail_relation_element = ET.SubElement(relations_element, "Relationship", Target="/Metadata/thumbnail.png", Id="rel0",
Type="http://schemas.openxmlformats.org/package/2006/relationships/metadata/thumbnail")
model_relation_element = ET.SubElement(relations_element, "Relationship", Target="/3D/model.gcode",
Id="rel1",
Type="http://schemas.ultimaker.org/package/2018/relationships/gcode")
gcode_string = StringIO()
PluginRegistry.getInstance().getPluginObject("GCodeWriter").write(gcode_string, None)
archive.write(Resources.getPath(Resources.Images, "cura-icon.png"), "Metadata/thumbnail.png")
archive.writestr(gcode_file, gcode_string.getvalue())
archive.writestr(content_types_file, b'<?xml version="1.0" encoding="UTF-8"?> \n' + ET.tostring(content_types))
archive.writestr(relations_file, b'<?xml version="1.0" encoding="UTF-8"?> \n' + ET.tostring(relations_element))
archive.close()

View File

@ -0,0 +1,25 @@
# Copyright (c) 2017 Ultimaker B.V.
# Uranium is released under the terms of the LGPLv3 or higher.
from . import UCPWriter
from UM.i18n import i18nCatalog
i18n_catalog = i18nCatalog("cura")
def getMetaData():
return {
"mesh_writer": {
"output": [
{
"mime_type": "application/x-ucp",
"mode": UCPWriter.UCPWriter.OutputMode.BinaryMode,
"extension": "UCP",
"description": i18n_catalog.i18nc("@item:inlistbox", "UCP File (WIP)")
}
]
}
}
def register(app):
return { "mesh_writer": UCPWriter.UCPWriter() }

View File

@ -0,0 +1,8 @@
{
"name": "UCP Writer",
"author": "Ultimaker B.V.",
"version": "1.0.0",
"description": "Provides support for writing UCP files.",
"api": 4,
"i18n-catalog": "cura"
}

View File

@ -10,7 +10,7 @@ if MYPY:
from cura.PrinterOutput.PrinterOutputModel import PrinterOutputModel
class USBPrinterOuptutController(PrinterOutputController):
class USBPrinterOutputController(PrinterOutputController):
def __init__(self, output_device):
super().__init__(output_device)

View File

@ -12,7 +12,7 @@ from cura.PrinterOutput.PrinterOutputModel import PrinterOutputModel
from cura.PrinterOutput.PrintJobOutputModel import PrintJobOutputModel
from .AutoDetectBaudJob import AutoDetectBaudJob
from .USBPrinterOutputController import USBPrinterOuptutController
from .USBPrinterOutputController import USBPrinterOutputController
from .avr_isp import stk500v2, intelHex
from PyQt5.QtCore import pyqtSlot, pyqtSignal, pyqtProperty
@ -237,7 +237,7 @@ class USBPrinterOutputDevice(PrinterOutputDevice):
container_stack = Application.getInstance().getGlobalContainerStack()
num_extruders = container_stack.getProperty("machine_extruder_count", "value")
# Ensure that a printer is created.
self._printers = [PrinterOutputModel(output_controller=USBPrinterOuptutController(self), number_of_extruders=num_extruders)]
self._printers = [PrinterOutputModel(output_controller=USBPrinterOutputController(self), number_of_extruders=num_extruders)]
self._printers[0].updateName(container_stack.getName())
self.setConnectionState(ConnectionState.connected)
self._update_thread.start()
@ -364,7 +364,7 @@ class USBPrinterOutputDevice(PrinterOutputDevice):
elapsed_time = int(time() - self._print_start_time)
print_job = self._printers[0].activePrintJob
if print_job is None:
print_job = PrintJobOutputModel(output_controller = USBPrinterOuptutController(self), name= Application.getInstance().getPrintInformation().jobName)
print_job = PrintJobOutputModel(output_controller = USBPrinterOutputController(self), name= Application.getInstance().getPrintInformation().jobName)
print_job.updateState("printing")
self._printers[0].updateActivePrintJob(print_job)