mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-21 12:09:13 +08:00
378 lines
15 KiB
Python
378 lines
15 KiB
Python
# Copyright (c) 2017 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
import configparser #To parse preference files.
|
|
import io #To serialise the preference files afterwards.
|
|
import os
|
|
from urllib.parse import quote_plus
|
|
|
|
from UM.Resources import Resources
|
|
from UM.VersionUpgrade import VersionUpgrade #We're inheriting from this.
|
|
|
|
from cura.CuraApplication import CuraApplication
|
|
|
|
|
|
# a list of all legacy "Not Supported" quality profiles
|
|
_OLD_NOT_SUPPORTED_PROFILES = [
|
|
"um2p_pp_0.25_normal",
|
|
"um2p_tpu_0.8_normal",
|
|
"um3_bb0.4_ABS_Fast_Print",
|
|
"um3_bb0.4_ABS_Superdraft_Print",
|
|
"um3_bb0.4_CPEP_Fast_Print",
|
|
"um3_bb0.4_CPEP_Superdraft_Print",
|
|
"um3_bb0.4_CPE_Fast_Print",
|
|
"um3_bb0.4_CPE_Superdraft_Print",
|
|
"um3_bb0.4_Nylon_Fast_Print",
|
|
"um3_bb0.4_Nylon_Superdraft_Print",
|
|
"um3_bb0.4_PC_Fast_Print",
|
|
"um3_bb0.4_PLA_Fast_Print",
|
|
"um3_bb0.4_PLA_Superdraft_Print",
|
|
"um3_bb0.4_PP_Fast_Print",
|
|
"um3_bb0.4_PP_Superdraft_Print",
|
|
"um3_bb0.4_TPU_Fast_Print",
|
|
"um3_bb0.4_TPU_Superdraft_Print",
|
|
"um3_bb0.8_ABS_Fast_Print",
|
|
"um3_bb0.8_ABS_Superdraft_Print",
|
|
"um3_bb0.8_CPEP_Fast_Print",
|
|
"um3_bb0.8_CPEP_Superdraft_Print",
|
|
"um3_bb0.8_CPE_Fast_Print",
|
|
"um3_bb0.8_CPE_Superdraft_Print",
|
|
"um3_bb0.8_Nylon_Fast_Print",
|
|
"um3_bb0.8_Nylon_Superdraft_Print",
|
|
"um3_bb0.8_PC_Fast_Print",
|
|
"um3_bb0.8_PC_Superdraft_Print",
|
|
"um3_bb0.8_PLA_Fast_Print",
|
|
"um3_bb0.8_PLA_Superdraft_Print",
|
|
"um3_bb0.8_PP_Fast_Print",
|
|
"um3_bb0.8_PP_Superdraft_Print",
|
|
"um3_bb0.8_TPU_Fast_print",
|
|
"um3_bb0.8_TPU_Superdraft_Print",
|
|
]
|
|
|
|
|
|
# Some containers have their specific empty containers, those need to be set correctly.
|
|
_EMPTY_CONTAINER_DICT = {
|
|
"1": "empty_quality_changes",
|
|
"2": "empty_quality",
|
|
"3": "empty_material",
|
|
"4": "empty_variant",
|
|
}
|
|
|
|
|
|
class VersionUpgrade30to31(VersionUpgrade):
|
|
## Gets the version number from a CFG file in Uranium's 3.0 format.
|
|
#
|
|
# Since the format may change, this is implemented for the 3.0 format only
|
|
# and needs to be included in the version upgrade system rather than
|
|
# globally in Uranium.
|
|
#
|
|
# \param serialised The serialised form of a CFG file.
|
|
# \return The version number stored in the CFG file.
|
|
# \raises ValueError The format of the version number in the file is
|
|
# incorrect.
|
|
# \raises KeyError The format of the file is incorrect.
|
|
def getCfgVersion(self, serialised):
|
|
parser = configparser.ConfigParser(interpolation = None)
|
|
parser.read_string(serialised)
|
|
format_version = int(parser.get("general", "version")) #Explicitly give an exception when this fails. That means that the file format is not recognised.
|
|
setting_version = int(parser.get("metadata", "setting_version", fallback = 0))
|
|
return format_version * 1000000 + setting_version
|
|
|
|
## Upgrades a preferences file from version 3.0 to 3.1.
|
|
#
|
|
# \param serialised The serialised form of a preferences file.
|
|
# \param filename The name of the file to upgrade.
|
|
def upgradePreferences(self, serialised, filename):
|
|
parser = configparser.ConfigParser(interpolation=None)
|
|
parser.read_string(serialised)
|
|
|
|
# Update version numbers
|
|
if "general" not in parser:
|
|
parser["general"] = {}
|
|
parser["general"]["version"] = "5"
|
|
if "metadata" not in parser:
|
|
parser["metadata"] = {}
|
|
parser["metadata"]["setting_version"] = "4"
|
|
|
|
# Re-serialise the file.
|
|
output = io.StringIO()
|
|
parser.write(output)
|
|
return [filename], [output.getvalue()]
|
|
|
|
## Upgrades the given instance container file from version 3.0 to 3.1.
|
|
#
|
|
# \param serialised The serialised form of the container file.
|
|
# \param filename The name of the file to upgrade.
|
|
def upgradeInstanceContainer(self, serialised, filename):
|
|
parser = configparser.ConfigParser(interpolation=None)
|
|
parser.read_string(serialised)
|
|
|
|
for each_section in ("general", "metadata"):
|
|
if not parser.has_section(each_section):
|
|
parser.add_section(each_section)
|
|
|
|
# Copy global quality changes to extruder quality changes for single extrusion machines
|
|
if parser["metadata"]["type"] == "quality_changes":
|
|
all_quality_changes = self._getSingleExtrusionMachineQualityChanges(parser)
|
|
# Note that DO NOT!!! use the quality_changes returned from _getSingleExtrusionMachineQualityChanges().
|
|
# Those are loaded from the hard drive which are original files that haven't been upgraded yet.
|
|
# NOTE 2: The number can be 0 or 1 depends on whether you are loading it from the qualities folder or
|
|
# from a project file. When you load from a project file, the custom profile may not be in cura
|
|
# yet, so you will get 0.
|
|
if len(all_quality_changes) <= 1 and not parser.has_option("metadata", "extruder"):
|
|
self._createExtruderQualityChangesForSingleExtrusionMachine(filename, parser)
|
|
|
|
if parser["metadata"]["type"] == "definition_changes":
|
|
if parser["general"]["definition"] == "custom":
|
|
|
|
# We are only interested in machine_nozzle_size
|
|
if parser.has_option("values", "machine_nozzle_size"):
|
|
machine_nozzle_size = parser["values"]["machine_nozzle_size"]
|
|
|
|
definition_name = parser["general"]["name"]
|
|
machine_extruders = self._getSingleExtrusionMachineExtruders(definition_name)
|
|
|
|
#For single extuder machine we nee only first extruder
|
|
if len(machine_extruders) !=0:
|
|
if self._updateSingleExtuderDefinitionFile(machine_extruders, machine_nozzle_size):
|
|
parser.remove_option("values", "machine_nozzle_size")
|
|
|
|
# Update version numbers
|
|
parser["general"]["version"] = "2"
|
|
parser["metadata"]["setting_version"] = "4"
|
|
|
|
# Re-serialise the file.
|
|
output = io.StringIO()
|
|
parser.write(output)
|
|
return [filename], [output.getvalue()]
|
|
|
|
|
|
## Upgrades a container stack from version 3.0 to 3.1.
|
|
#
|
|
# \param serialised The serialised form of a container stack.
|
|
# \param filename The name of the file to upgrade.
|
|
def upgradeStack(self, serialised, filename):
|
|
parser = configparser.ConfigParser(interpolation=None)
|
|
parser.read_string(serialised)
|
|
|
|
for each_section in ("general", "metadata"):
|
|
if not parser.has_section(each_section):
|
|
parser.add_section(each_section)
|
|
|
|
# change "not supported" quality profiles to empty because they no longer exist
|
|
if parser.has_section("containers"):
|
|
if parser.has_option("containers", "2"):
|
|
quality_profile_id = parser["containers"]["2"]
|
|
if quality_profile_id in _OLD_NOT_SUPPORTED_PROFILES:
|
|
parser["containers"]["2"] = "empty_quality"
|
|
|
|
# fix empty containers
|
|
for key, specific_empty_container in _EMPTY_CONTAINER_DICT.items():
|
|
if parser.has_option("containers", key) and parser["containers"][key] == "empty":
|
|
parser["containers"][key] = specific_empty_container
|
|
|
|
# Update version numbers
|
|
if "general" not in parser:
|
|
parser["general"] = {}
|
|
parser["general"]["version"] = "3"
|
|
|
|
if "metadata" not in parser:
|
|
parser["metadata"] = {}
|
|
parser["metadata"]["setting_version"] = "4"
|
|
|
|
# Re-serialise the file.
|
|
output = io.StringIO()
|
|
parser.write(output)
|
|
return [filename], [output.getvalue()]
|
|
|
|
def _getSingleExtrusionMachineQualityChanges(self, quality_changes_container):
|
|
quality_changes_dir = Resources.getPath(CuraApplication.ResourceTypes.QualityInstanceContainer)
|
|
quality_changes_containers = []
|
|
|
|
for item in os.listdir(quality_changes_dir):
|
|
file_path = os.path.join(quality_changes_dir, item)
|
|
if not os.path.isfile(file_path):
|
|
continue
|
|
|
|
parser = configparser.ConfigParser(interpolation = None)
|
|
try:
|
|
parser.read([file_path])
|
|
except:
|
|
# skip, it is not a valid stack file
|
|
continue
|
|
|
|
if not parser.has_option("metadata", "type"):
|
|
continue
|
|
if "quality_changes" != parser["metadata"]["type"]:
|
|
continue
|
|
|
|
if not parser.has_option("general", "name"):
|
|
continue
|
|
if quality_changes_container["general"]["name"] != parser["general"]["name"]:
|
|
continue
|
|
|
|
quality_changes_containers.append(parser)
|
|
|
|
return quality_changes_containers
|
|
|
|
def _getSingleExtrusionMachineExtruders(self, definition_name):
|
|
|
|
machine_instances_dir = Resources.getPath(CuraApplication.ResourceTypes.MachineStack)
|
|
|
|
machine_instances = []
|
|
|
|
#Find all machine instances
|
|
for item in os.listdir(machine_instances_dir):
|
|
file_path = os.path.join(machine_instances_dir, item)
|
|
if not os.path.isfile(file_path):
|
|
continue
|
|
|
|
parser = configparser.ConfigParser(interpolation=None)
|
|
try:
|
|
parser.read([file_path])
|
|
except:
|
|
# skip, it is not a valid stack file
|
|
continue
|
|
|
|
if not parser.has_option("metadata", "type"):
|
|
continue
|
|
if "machine" != parser["metadata"]["type"]:
|
|
continue
|
|
|
|
if not parser.has_option("general", "id"):
|
|
continue
|
|
|
|
machine_instances.append(parser)
|
|
|
|
#Find for extruders
|
|
extruders_instances_dir = Resources.getPath(CuraApplication.ResourceTypes.ExtruderStack)
|
|
#"machine",[extruders]
|
|
extruder_instances_per_machine = {}
|
|
|
|
#Find all custom extruders for founded machines
|
|
for item in os.listdir(extruders_instances_dir):
|
|
file_path = os.path.join(extruders_instances_dir, item)
|
|
if not os.path.isfile(file_path):
|
|
continue
|
|
|
|
parser = configparser.ConfigParser(interpolation=None)
|
|
try:
|
|
parser.read([file_path])
|
|
except:
|
|
# skip, it is not a valid stack file
|
|
continue
|
|
|
|
if not parser.has_option("metadata", "type"):
|
|
continue
|
|
if "extruder_train" != parser["metadata"]["type"]:
|
|
continue
|
|
|
|
if not parser.has_option("metadata", "machine"):
|
|
continue
|
|
if not parser.has_option("metadata", "position"):
|
|
continue
|
|
|
|
|
|
for machine_instace in machine_instances:
|
|
|
|
machine_id = machine_instace["general"]["id"]
|
|
if machine_id != parser["metadata"]["machine"]:
|
|
continue
|
|
|
|
if machine_id + "_settings" != definition_name:
|
|
continue
|
|
|
|
if extruder_instances_per_machine.get(machine_id) is None:
|
|
extruder_instances_per_machine.update({machine_id:[]})
|
|
|
|
extruder_instances_per_machine.get(machine_id).append(parser)
|
|
#the extruder can be related only to one machine
|
|
break
|
|
|
|
return extruder_instances_per_machine
|
|
|
|
#Find extruder defition at index 0 and update its values
|
|
def _updateSingleExtuderDefinitionFile(self, extruder_instances_per_machine, machine_nozzle_size):
|
|
|
|
defintion_instances_dir = Resources.getPath(CuraApplication.ResourceTypes.DefinitionChangesContainer)
|
|
|
|
for item in os.listdir(defintion_instances_dir):
|
|
file_path = os.path.join(defintion_instances_dir, item)
|
|
if not os.path.isfile(file_path):
|
|
continue
|
|
|
|
parser = configparser.ConfigParser(interpolation=None)
|
|
try:
|
|
parser.read([file_path])
|
|
except:
|
|
# skip, it is not a valid stack file
|
|
continue
|
|
|
|
if not parser.has_option("general", "name"):
|
|
continue
|
|
name = parser["general"]["name"]
|
|
custom_extruder_at_0_position = None
|
|
for machine_extruders in extruder_instances_per_machine:
|
|
for extruder_instance in extruder_instances_per_machine[machine_extruders]:
|
|
|
|
if extruder_instance["general"]["id"] + "_settings" == name:
|
|
defition_position = extruder_instance["metadata"]["position"]
|
|
|
|
if defition_position == "0":
|
|
custom_extruder_at_0_position = extruder_instance
|
|
break
|
|
if custom_extruder_at_0_position is not None:
|
|
break
|
|
|
|
#If not null, then parsed file is for first extuder and then can be updated. I need to update only
|
|
# first, because this update for single extuder machine
|
|
if custom_extruder_at_0_position is not None:
|
|
|
|
#Add new value
|
|
parser["values"]["machine_nozzle_size"] = machine_nozzle_size
|
|
|
|
definition_output = io.StringIO()
|
|
parser.write(definition_output)
|
|
|
|
with open(file_path, "w") as f:
|
|
f.write(definition_output.getvalue())
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def _createExtruderQualityChangesForSingleExtrusionMachine(self, filename, global_quality_changes):
|
|
suffix = "_" + quote_plus(global_quality_changes["general"]["name"].lower())
|
|
machine_name = os.path.os.path.basename(filename).replace(".inst.cfg", "").replace(suffix, "")
|
|
|
|
# Why is this here?!
|
|
# When we load a .curaprofile file the deserialize will trigger a version upgrade, creating a dangling file.
|
|
# This file can be recognized by it's lack of a machine name in the target filename.
|
|
# So when we detect that situation here, we don't create the file and return.
|
|
if machine_name == "":
|
|
return
|
|
|
|
new_filename = machine_name + "_" + "fdmextruder" + suffix
|
|
|
|
extruder_quality_changes_parser = configparser.ConfigParser(interpolation = None)
|
|
extruder_quality_changes_parser.add_section("general")
|
|
extruder_quality_changes_parser["general"]["version"] = str(2)
|
|
extruder_quality_changes_parser["general"]["name"] = global_quality_changes["general"]["name"]
|
|
extruder_quality_changes_parser["general"]["definition"] = global_quality_changes["general"]["definition"]
|
|
|
|
extruder_quality_changes_parser.add_section("metadata")
|
|
extruder_quality_changes_parser["metadata"]["quality_type"] = global_quality_changes["metadata"]["quality_type"]
|
|
extruder_quality_changes_parser["metadata"]["type"] = global_quality_changes["metadata"]["type"]
|
|
extruder_quality_changes_parser["metadata"]["setting_version"] = str(4)
|
|
extruder_quality_changes_parser["metadata"]["extruder"] = "fdmextruder"
|
|
|
|
extruder_quality_changes_output = io.StringIO()
|
|
extruder_quality_changes_parser.write(extruder_quality_changes_output)
|
|
extruder_quality_changes_filename = quote_plus(new_filename) + ".inst.cfg"
|
|
|
|
quality_changes_dir = Resources.getPath(CuraApplication.ResourceTypes.QualityInstanceContainer)
|
|
|
|
with open(os.path.join(quality_changes_dir, extruder_quality_changes_filename), "w") as f:
|
|
f.write(extruder_quality_changes_output.getvalue())
|