1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# Sdcard file looping support
#
# Copyright (C) 2021 Jason S. McMullan <jason.mcmullan@gmail.com>
#
# This file may be distributed under the terms of the GNU GPLv3 license.
import logging
class SDCardLoop:
def __init__(self, config):
printer = config.get_printer()
self.sdcard = printer.load_object(config, "virtual_sdcard")
self.gcode = printer.lookup_object("gcode")
self.gcode.register_command(
"SDCARD_LOOP_BEGIN",
self.cmd_SDCARD_LOOP_BEGIN,
desc=self.cmd_SDCARD_LOOP_BEGIN_help,
)
self.gcode.register_command(
"SDCARD_LOOP_END",
self.cmd_SDCARD_LOOP_END,
desc=self.cmd_SDCARD_LOOP_END_help,
)
self.gcode.register_command(
"SDCARD_LOOP_DESIST",
self.cmd_SDCARD_LOOP_DESIST,
desc=self.cmd_SDCARD_LOOP_DESIST_help,
)
self.loop_stack = []
cmd_SDCARD_LOOP_BEGIN_help = "Begins a looped section in the SD file."
def cmd_SDCARD_LOOP_BEGIN(self, gcmd):
count = gcmd.get_int("COUNT", minval=0)
if not self.loop_begin(count):
raise gcmd.error("Only permitted in SD file.")
cmd_SDCARD_LOOP_END_help = "Ends a looped section in the SD file."
def cmd_SDCARD_LOOP_END(self, gcmd):
if not self.loop_end():
raise gcmd.error("Only permitted in SD file.")
cmd_SDCARD_LOOP_DESIST_help = "Stops iterating the current loop stack."
def cmd_SDCARD_LOOP_DESIST(self, gcmd):
if not self.loop_desist():
raise gcmd.error("Only permitted outside of a SD file.")
def loop_begin(self, count):
if not self.sdcard.is_cmd_from_sd():
# Can only run inside of an SD file
return False
self.loop_stack.append((count, self.sdcard.get_file_position()))
return True
def loop_end(self):
if not self.sdcard.is_cmd_from_sd():
# Can only run inside of an SD file
return False
# If the stack is empty, no need to skip back
if len(self.loop_stack) == 0:
return True
# Get iteration count and return position
count, position = self.loop_stack.pop()
if count == 0: # Infinite loop
self.sdcard.set_file_position(position)
self.loop_stack.append((0, position))
elif count == 1: # Last repeat
# Nothing to do
pass
else:
# At the next opportunity, seek back to the start of the loop
self.sdcard.set_file_position(position)
# Decrement the count by 1, and add the position back to the stack
self.loop_stack.append((count - 1, position))
return True
def loop_desist(self):
if self.sdcard.is_cmd_from_sd():
# Can only run outside of an SD file
return False
logging.info("Desisting existing SD loops")
self.loop_stack = []
return True
def load_config(config):
return SDCardLoop(config)
|
