mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-20 20:59:48 +08:00
cab2e14728
in the GCode and then the right flavor handler is responsible for reading the code. At this moment just the Griffin and RepRap flavors are taken into account.
40 lines
1.5 KiB
Python
Executable File
40 lines
1.5 KiB
Python
Executable File
# Copyright (c) 2017 Aleph Objects, Inc.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
from UM.FileHandler.FileReader import FileReader
|
|
from UM.Mesh.MeshReader import MeshReader
|
|
from UM.i18n import i18nCatalog
|
|
from UM.Preferences import Preferences
|
|
|
|
catalog = i18nCatalog("cura")
|
|
from . import GriffinFlavor, RepRapFlavor
|
|
|
|
# Class for loading and parsing G-code files
|
|
class GCodeReader(MeshReader):
|
|
|
|
_flavor_keyword = ";FLAVOR:"
|
|
_flavor_readers_dict = {"Griffin" : GriffinFlavor.GriffinFlavor(),
|
|
"RepRap" : RepRapFlavor.RepRapFlavor()}
|
|
|
|
def __init__(self):
|
|
super(GCodeReader, self).__init__()
|
|
self._supported_extensions = [".gcode", ".g"]
|
|
self._flavor_reader = None
|
|
|
|
Preferences.getInstance().addPreference("gcodereader/show_caution", True)
|
|
|
|
# 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") as file:
|
|
for line in file:
|
|
if line[:len(self._flavor_keyword)] == self._flavor_keyword:
|
|
self._flavor_reader = self._flavor_readers_dict[line[len(self._flavor_keyword):].rstrip()]
|
|
return FileReader.PreReadResult.accepted
|
|
|
|
# If no flavor is found in the GCode, then we use the by-default
|
|
self._flavor_reader = self._flavor_readers_dict["Griffin"]
|
|
return FileReader.PreReadResult.accepted
|
|
|
|
def read(self, file_name):
|
|
return self._flavor_reader.processGCodeFile(file_name)
|