mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 21:59:37 +08:00
102 lines
4.1 KiB
Python
102 lines
4.1 KiB
Python
# Copyright (c) 2015 Ultimaker B.V.
|
|
# Cura is released under the terms of the AGPLv3 or higher.
|
|
|
|
import re #Regular expressions for parsing escape characters in the settings.
|
|
import json
|
|
|
|
from UM.Settings.InstanceContainer import InstanceContainer
|
|
from UM.Logger import Logger
|
|
from UM.i18n import i18nCatalog
|
|
catalog = i18nCatalog("cura")
|
|
|
|
from cura.ProfileReader import ProfileReader
|
|
|
|
## A class that reads profile data from g-code files.
|
|
#
|
|
# It reads the profile data from g-code files and stores it in a new profile.
|
|
# This class currently does not process the rest of the g-code in any way.
|
|
class GCodeProfileReader(ProfileReader):
|
|
## The file format version of the serialized g-code.
|
|
#
|
|
# It can only read settings with the same version as the version it was
|
|
# written with. If the file format is changed in a way that breaks reverse
|
|
# compatibility, increment this version number!
|
|
version = 3
|
|
|
|
## Dictionary that defines how characters are escaped when embedded in
|
|
# g-code.
|
|
#
|
|
# Note that the keys of this dictionary are regex strings. The values are
|
|
# not.
|
|
escape_characters = {
|
|
re.escape("\\\\"): "\\", #The escape character.
|
|
re.escape("\\n"): "\n", #Newlines. They break off the comment.
|
|
re.escape("\\r"): "\r" #Carriage return. Windows users may need this for visualisation in their editors.
|
|
}
|
|
|
|
## Initialises the g-code reader as a profile reader.
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
## Reads a g-code file, loading the profile from it.
|
|
#
|
|
# \param file_name The name of the file to read the profile from.
|
|
# \return The profile that was in the specified file, if any. If the
|
|
# specified file was no g-code or contained no parsable profile, \code
|
|
# None \endcode is returned.
|
|
def read(self, file_name):
|
|
if file_name.split(".")[-1] != "gcode":
|
|
return None
|
|
|
|
prefix = ";SETTING_" + str(GCodeProfileReader.version) + " "
|
|
prefix_length = len(prefix)
|
|
|
|
# Loading all settings from the file.
|
|
# They are all at the end, but Python has no reverse seek any more since Python3.
|
|
# TODO: Consider moving settings to the start?
|
|
serialized = "" # Will be filled with the serialized profile.
|
|
try:
|
|
with open(file_name) as f:
|
|
for line in f:
|
|
if line.startswith(prefix):
|
|
# Remove the prefix and the newline from the line and add it to the rest.
|
|
serialized += line[prefix_length : -1]
|
|
except IOError as e:
|
|
Logger.log("e", "Unable to open file %s for reading: %s", file_name, str(e))
|
|
return None
|
|
|
|
serialized = unescapeGcodeComment(serialized)
|
|
Logger.log("i", "Serialized the following from %s: %s" %(file_name, repr(serialized)))
|
|
|
|
json_data = json.loads(serialized)
|
|
|
|
profile_strings = [json_data["global_quality"]]
|
|
profile_strings.extend(json_data.get("extruder_quality", []))
|
|
|
|
return [readQualityProfileFromString(profile_string) for profile_string in profile_strings]
|
|
|
|
## Unescape a string which has been escaped for use in a gcode comment.
|
|
#
|
|
# \param string The string to unescape.
|
|
# \return \type{str} The unscaped string.
|
|
def unescapeGcodeComment(string):
|
|
# Un-escape the serialized profile.
|
|
pattern = re.compile("|".join(GCodeProfileReader.escape_characters.keys()))
|
|
|
|
# Perform the replacement with a regular expression.
|
|
return pattern.sub(lambda m: GCodeProfileReader.escape_characters[re.escape(m.group(0))], string)
|
|
|
|
## Read in a profile from a serialized string.
|
|
#
|
|
# \param profile_string The profile data in serialized form.
|
|
# \return \type{Profile} the resulting Profile object or None if it could not be read.
|
|
def readQualityProfileFromString(profile_string):
|
|
# Create an empty profile - the id and name will be changed by the ContainerRegistry
|
|
profile = InstanceContainer("")
|
|
try:
|
|
profile.deserialize(profile_string)
|
|
except Exception as e: # Not a valid g-code file.
|
|
Logger.log("e", "Unable to serialise the profile: %s", str(e))
|
|
return None
|
|
return profile
|