mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 21:59:37 +08:00
db15bc84cf
This spelling is in Ultimaker's style guide. We use g-code, and capitalise the G if it's at the start of a sentence or header. Pretty good score, considering there are literally thousands of user- or log-visible strings mentioning g-code across Cura.
78 lines
3.0 KiB
Python
Executable File
78 lines
3.0 KiB
Python
Executable File
# Copyright (c) 2017 Aleph Objects, Inc.
|
|
# Copyright (c) 2020 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from typing import Optional, Union, List, TYPE_CHECKING
|
|
|
|
from UM.FileHandler.FileReader import FileReader
|
|
from UM.Mesh.MeshReader import MeshReader
|
|
from UM.i18n import i18nCatalog
|
|
from UM.Application import Application
|
|
from UM.MimeTypeDatabase import MimeTypeDatabase, MimeType
|
|
|
|
catalog = i18nCatalog("cura")
|
|
|
|
from .FlavorParser import FlavorParser
|
|
from . import MarlinFlavorParser, RepRapFlavorParser
|
|
|
|
if TYPE_CHECKING:
|
|
from UM.Scene.SceneNode import SceneNode
|
|
from cura.Scene.CuraSceneNode import CuraSceneNode
|
|
|
|
|
|
# Class for loading and parsing G-code files
|
|
class GCodeReader(MeshReader):
|
|
_flavor_default = "Marlin"
|
|
_flavor_keyword = ";FLAVOR:"
|
|
_flavor_readers_dict = {"RepRap" : RepRapFlavorParser.RepRapFlavorParser(),
|
|
"Marlin" : MarlinFlavorParser.MarlinFlavorParser()}
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
MimeTypeDatabase.addMimeType(
|
|
MimeType(
|
|
name = "application/x-cura-gcode-file",
|
|
comment = "Cura G-code File",
|
|
suffixes = ["gcode"]
|
|
)
|
|
)
|
|
self._supported_extensions = [".gcode", ".g"]
|
|
|
|
self._flavor_reader = None # type: Optional[FlavorParser]
|
|
|
|
Application.getInstance().getPreferences().addPreference("gcodereader/show_caution", True)
|
|
|
|
def preReadFromStream(self, stream, *args, **kwargs):
|
|
for line in stream.split("\n"):
|
|
if line[:len(self._flavor_keyword)] == self._flavor_keyword:
|
|
try:
|
|
self._flavor_reader = self._flavor_readers_dict[line[len(self._flavor_keyword):].rstrip()]
|
|
return FileReader.PreReadResult.accepted
|
|
except:
|
|
# If there is no entry in the dictionary for this flavor, just skip and select the by-default flavor
|
|
break
|
|
|
|
# If no flavor is found in the GCode, then we use the by-default
|
|
self._flavor_reader = self._flavor_readers_dict[self._flavor_default]
|
|
return FileReader.PreReadResult.accepted
|
|
|
|
# PreRead is used to get the correct flavor. If not, Marlin is set by default
|
|
def preRead(self, file_name, *args, **kwargs):
|
|
with open(file_name, "r", encoding = "utf-8") as file:
|
|
file_data = file.read()
|
|
return self.preReadFromStream(file_data, args, kwargs)
|
|
|
|
def readFromStream(self, stream: str, filename: str) -> Optional["CuraSceneNode"]:
|
|
if self._flavor_reader is None:
|
|
return None
|
|
return self._flavor_reader.processGCodeStream(stream, filename)
|
|
|
|
def _read(self, file_name: str) -> Union["SceneNode", List["SceneNode"]]:
|
|
with open(file_name, "r", encoding = "utf-8") as file:
|
|
file_data = file.read()
|
|
result = [] # type: List[SceneNode]
|
|
node = self.readFromStream(file_data, file_name)
|
|
if node is not None:
|
|
result.append(node)
|
|
return result
|