mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 05:39:37 +08:00
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
# Created by Wayne Porter
|
|
|
|
from ..Script import Script
|
|
|
|
class InsertAtLayerChange(Script):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def getSettingDataString(self):
|
|
return """{
|
|
"name": "Insert at layer change",
|
|
"key": "InsertAtLayerChange",
|
|
"metadata": {},
|
|
"version": 2,
|
|
"settings":
|
|
{
|
|
"insert_loc":
|
|
{
|
|
"label": "When to insert",
|
|
"description": "Whether to insert code before or after layer change.",
|
|
"type": "enum",
|
|
"options": {"before": "Before", "after": "After"},
|
|
"default_value": "before"
|
|
},
|
|
"gcode_to_add":
|
|
{
|
|
"label": "GCODE to insert.",
|
|
"description": "GCODE to add before or after layer change.",
|
|
"type": "str",
|
|
"default_value": ""
|
|
}
|
|
}
|
|
}"""
|
|
|
|
def execute(self, data):
|
|
in_layer = False
|
|
gcode_to_add = self.getSettingValueByKey("gcode_to_add") + "\n"
|
|
for layer in data:
|
|
# Check that a layer is being printed
|
|
lines = layer.split("\n")
|
|
if ";LAYER:" in lines[0]:
|
|
in_layer = True
|
|
else:
|
|
in_layer = False
|
|
|
|
if in_layer:
|
|
index = data.index(layer)
|
|
if self.getSettingValueByKey("insert_loc") == "before":
|
|
layer = gcode_to_add + layer
|
|
else:
|
|
layer = layer + gcode_to_add
|
|
|
|
data[index] = layer
|
|
|
|
return data
|