mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-23 22:29:41 +08:00
2fe9860bb9
This way we can actually write the result to the correct stream rather than opening a random file separately (which was placeholder). Contributes to issue CURA-4872.
35 lines
1.9 KiB
Python
35 lines
1.9 KiB
Python
#Copyright (c) 2018 Ultimaker B.V.
|
|
#Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from charon.VirtualFile import VirtualFile #To open UFP files.
|
|
from charon.OpenMode import OpenMode #To indicate that we want to write to UFP files.
|
|
from io import StringIO #For converting g-code to bytes.
|
|
import os.path #To get the placeholder kitty icon.
|
|
|
|
from UM.Mesh.MeshWriter import MeshWriter #The writer we need to implement.
|
|
from UM.PluginRegistry import PluginRegistry #To get the g-code writer.
|
|
|
|
class UFPWriter(MeshWriter):
|
|
def write(self, stream, nodes, mode = MeshWriter.OutputMode.BinaryMode):
|
|
archive = VirtualFile()
|
|
archive.openStream(stream, "application/x-ufp", OpenMode.WriteOnly)
|
|
|
|
#Store the g-code from the scene.
|
|
archive.addContentType(extension = "gcode", mime_type = "text/x-gcode")
|
|
gcode_textio = StringIO() #We have to convert the g-code into bytes.
|
|
PluginRegistry.getInstance().getPluginObject("GCodeWriter").write(gcode_textio, None)
|
|
gcode = archive.getStream("/3D/model.gcode")
|
|
gcode.write(gcode_textio.getvalue().encode("UTF-8"))
|
|
gcode.close()
|
|
archive.addRelation(virtual_path = "/3D/model.gcode", relation_type = "http://schemas.ultimaker.org/package/2018/relationships/gcode")
|
|
|
|
#Store the thumbnail.
|
|
#TODO: Generate the thumbnail image. Below is just a placeholder.
|
|
archive.addContentType(extension = "png", mime_type = "image/png")
|
|
thumbnail = archive.getStream("/Metadata/thumbnail.png")
|
|
thumbnail.write(open(os.path.join(os.path.dirname(__file__), "kitten.png"), "rb").read())
|
|
thumbnail.close()
|
|
archive.addRelation(virtual_path = "/Metadata/thumbnail.png", relation_type = "http://schemas.openxmlformats.org/package/2006/relationships/metadata/thumbnail", origin = "/3D/model.gcode")
|
|
|
|
archive.close()
|
|
return True |