3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-07-07 06:57:28 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

The entire machine is now saved to 3mf file when saving workspace

Browse source 
CURA-1263
This commit is contained in:
Jaime van Kessel 2016-11-10 11:27:25 +01:00
parent 54040d4c99
commit 61d1199abf
1 changed files with 36 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

50
plugins/3MFWriter/ThreeMFWorkspaceWriter.py
View file

@ -1,6 +1,8 @@
from UM.Workspace.WorkspaceWriter import WorkspaceWriter from UM.Workspace.WorkspaceWriter import WorkspaceWriter
from UM.Application import Application from UM.Application import Application
from UM.Preferences import Preferences from UM.Preferences import Preferences
from UM.Settings.ContainerRegistry import ContainerRegistry
from cura.Settings.ExtruderManager import ExtruderManager
import zipfile import zipfile
from io import StringIO from io import StringIO
@ -20,30 +22,50 @@ class ThreeMFWorkspaceWriter(WorkspaceWriter):
mesh_writer.write(stream, nodes, mode) mesh_writer.write(stream, nodes, mode)
archive = mesh_writer.getArchive() archive = mesh_writer.getArchive()
# Add global container stack data to the archive.
global_container_stack = Application.getInstance().getGlobalContainerStack() global_container_stack = Application.getInstance().getGlobalContainerStack()
global_stack_file = zipfile.ZipInfo("Cura/%s.stack.cfg" % global_container_stack.getId())
global_stack_file.compress_type = zipfile.ZIP_DEFLATED
archive.writestr(global_stack_file, global_container_stack.serialize())
# Write user changes to the archive. # Add global container stack data to the archive.
global_user_instance_container = global_container_stack.getTop() self._writeContainerToArchive(global_container_stack, archive)
global_user_instance_file = zipfile.ZipInfo("Cura/%s.inst.cfg" % global_user_instance_container.getId())
global_user_instance_container.compress_type = zipfile.ZIP_DEFLATED
archive.writestr(global_user_instance_file, global_user_instance_container.serialize())
# Write quality changes to the archive. # Also write all containers in the stack to the file
global_quality_changes = global_container_stack.findContainer({"type": "quality_changes"}) for container in global_container_stack.getContainers():
global_quality_changes_file = zipfile.ZipInfo("Cura/%s.inst.cfg" % global_quality_changes.getId()) self._writeContainerToArchive(container, archive)
global_quality_changes.compress_type = zipfile.ZIP_DEFLATED
archive.writestr(global_quality_changes_file, global_quality_changes.serialize()) # Check if the machine has extruders and save all that data as well.
for extruder_stack in ExtruderManager.getInstance().getMachineExtruders(global_container_stack.getId()):
self._writeContainerToArchive(extruder_stack, archive)
for container in extruder_stack.getContainers():
self._writeContainerToArchive(container, archive)
# Write preferences to archive # Write preferences to archive
preferences_file = zipfile.ZipInfo("Cura/preferences.cfg") preferences_file = zipfile.ZipInfo("Cura/preferences.cfg")
preferences_string = StringIO() preferences_string = StringIO()
Preferences.getInstance().writeToFile(preferences_string) Preferences.getInstance().writeToFile(preferences_string)
archive.writestr(preferences_file, preferences_string.getvalue()) archive.writestr(preferences_file, preferences_string.getvalue())
# Close the archive & reset states. # Close the archive & reset states.
archive.close() archive.close()
mesh_writer.setStoreArchive(False) mesh_writer.setStoreArchive(False)
return True return True
@staticmethod
def _writeContainerToArchive(container, archive):
if type(container) == type(ContainerRegistry.getInstance().getEmptyInstanceContainer()):
return # Empty file, do nothing.
file_suffix = ContainerRegistry.getMimeTypeForContainer(type(container)).suffixes[0]
# Some containers have a base file, which should then be the file to use.
base_file = container.getMetaDataEntry("base_file", None)
if base_file:
container = ContainerRegistry.getInstance().findContainers(id = base_file)[0]
file_name = "Cura/%s.%s" % (container.getId(), file_suffix)
if file_name in archive.namelist():
return # File was already saved, no need to do it again.
file_in_archive = zipfile.ZipInfo(file_name)
file_in_archive.compress_type = zipfile.ZIP_DEFLATED
archive.writestr(file_in_archive, container.serialize())