def test_signal():
test = SignalReceiver()
signal = Signal(type = Signal.Direct)
signal.connect(test.slot)
signal.emit()
assert test.getEmitCount() == 1
def test_connectWhilePostponed():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
with postponeSignals(signal):
signal.connect(test.slot) # This won't do anything, as we're postponing at the moment!
signal.emit()
assert test.getEmitCount() == 0 # The connection was never made, so we should get 0
def test_signal():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
signal.emit()
assert test.getEmitCount() == 1
def test_deepCopy():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
copied_signal = deepcopy(signal)
copied_signal.emit()
# Even though the original signal was not called, the copied one should have the same result
assert test.getEmitCount() == 1
def test_disconnectWhilePostponed():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
with postponeSignals(signal):
signal.disconnect(test.slot) # This won't do anything, as we're postponing at the moment!
signal.disconnectAll() # Same holds true for the disconnect all
signal.emit()
assert test.getEmitCount() == 1 # Despite attempting to disconnect, this didn't happen because of the postpone
def test_deepCopy():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
copied_signal = deepcopy(signal)
copied_signal.emit()
# Even though the original signal was not called, the copied one should have the same result
assert test.getEmitCount() == 1
def test_connectWhilePostponed():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
with postponeSignals(signal):
signal.connect(
test.slot
) # This won't do anything, as we're postponing at the moment!
signal.emit()
assert test.getEmitCount(
) == 0 # The connection was never made, so we should get 0
def test_postponeEmitCompressSingle():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
with postponeSignals(signal, compress=CompressTechnique.CompressSingle):
signal.emit()
assert test.getEmitCount() == 0 # as long as we're in this context, nothing should happen!
signal.emit()
assert test.getEmitCount() == 0
assert test.getEmitCount() == 1
def test_signalWithFlameProfiler():
with patch("UM.Signal._recordSignalNames", MagicMock(return_value=True)):
FlameProfiler.record_profile = True
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
signal.emit()
assert test.getEmitCount() == 1
FlameProfiler.record_profile = False
def test_doubleSignalWithFlameProfiler():
FlameProfiler.record_profile = True
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal2 = Signal(type=Signal.Direct)
signal.connect(test.slot)
signal2.connect(signal)
signal2.emit()
assert test.getEmitCount() == 1
FlameProfiler.record_profile = False
def test_signalWithFlameProfiler():
with patch("UM.Signal._recordSignalNames", MagicMock(return_value = True)):
FlameProfiler.record_profile = True
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
signal.emit()
assert test.getEmitCount() == 1
FlameProfiler.record_profile = False
def test_doubleSignalWithFlameProfiler():
FlameProfiler.record_profile = True
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal2 = Signal(type=Signal.Direct)
signal.connect(test.slot)
signal2.connect(signal)
signal2.emit()
assert test.getEmitCount() == 1
FlameProfiler.record_profile = False
def test_postponeEmitCompressSingle():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
with postponeSignals(signal, compress=CompressTechnique.CompressSingle):
signal.emit()
assert test.getEmitCount(
) == 0 # as long as we're in this context, nothing should happen!
signal.emit()
assert test.getEmitCount() == 0
assert test.getEmitCount() == 1
def test_postponeEmitCompressPerParameterValue():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
with postponeSignals(signal, compress=CompressTechnique.CompressPerParameterValue):
signal.emit("ZOMG")
assert test.getEmitCount() == 0 # as long as we're in this context, nothing should happen!
signal.emit("ZOMG")
assert test.getEmitCount() == 0
signal.emit("BEEP")
# We got 3 signal emits, but 2 of them were the same, so we end up with 2 unique emits.
assert test.getEmitCount() == 2
def test_disconnectWhilePostponed():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
with postponeSignals(signal):
signal.disconnect(
test.slot
) # This won't do anything, as we're postponing at the moment!
signal.disconnectAll() # Same holds true for the disconnect all
signal.emit()
assert test.getEmitCount(
) == 1 # Despite attempting to disconnect, this didn't happen because of the postpone
def test_postponeEmitCompressPerParameterValue():
test = SignalReceiver()
signal = Signal(type=Signal.Direct)
signal.connect(test.slot)
with postponeSignals(signal,
compress=CompressTechnique.CompressPerParameterValue):
signal.emit("ZOMG")
assert test.getEmitCount(
) == 0 # as long as we're in this context, nothing should happen!
signal.emit("ZOMG")
assert test.getEmitCount() == 0
signal.emit("BEEP")
# We got 3 signal emits, but 2 of them were the same, so we end up with 2 unique emits.
assert test.getEmitCount() == 2
def test_connectSelf():
signal = Signal(type=Signal.Direct)
signal.connect(signal)
signal.emit(
) # If they are connected, this crashes with a max recursion depth error
def test_connectSelf():
signal = Signal(type=Signal.Direct)
signal.connect(signal)
signal.emit() # If they are connected, this crashes with a max recursion depth error
class MultiSlicePlugin(QObject, Extension):
"""
A plugin for Ultimaker Cura that allows the user to load, slice, and export .gcode files
for a series of model files based on an input directory and a regex file pattern to search for.
See README.md for an example.
Settings:
:param self._file_pattern
:GUI "File name pattern"
:default r'.*.stl' (all .stl files)
RegEx that defines the files that should be searched for. Only files matching this pattern
are added to the list of files.
:param self._input_path
:GUI "Root directory"
:default None
The directory that will be walked when searching for files.
:param self._output_path
:GUI "Output directory"
:default None
The directory that .gcode files will be written to.
:param self._follow_dirs
:GUI "Follow directories
:default False
Whether or not to walk through directories found in self._input_path as well.
:param self._follow_depth
:GUI "Max depth"
:default 0
The maximum depth to walk when following directories. The root directory is treated as
depth 0.
:param self._preserve_dirs
:GUI: "Preserve directories in output
:default False
Whether or not to produce the same folder structure in the output directory as found in
the root directory.
"""
def __init__(self, parent=None):
QObject.__init__(self, parent)
Extension.__init__(self)
# add menu items in extensions menu
self.setMenuName(catalog.i18nc("@item:inmenu", "Multi slicing"))
self.addMenuItem(catalog.i18nc("@item:inmenu", "Configure and run"),
self._show_popup)
self._view = None # type: Optional[QObject]
# user options
self._file_pattern = r'.*.stl'
self._input_path = '' # type: Union[Path, str]
self._output_path = '' # type: Union[Path, str]
self._follow_dirs = False
self._follow_depth = 0 # type: Optional[int]
self._preserve_dirs = False
self._files = []
self._current_model = '' # type: Union[Path, str]
self._current_model_suffix = ''
self._current_model_name = ''
self._current_model_url = None # type: Optional[QUrl]
# gcode writer signal
self._write_done = Signal()
# event loop that allows us to wait for a signal
self._loop = QEventLoop()
# signal to handle output log messages
log = pyqtSignal(str, name='log')
def _log_msg(self, msg: str) -> None:
"""
Emits a message to the logger signal
"""
self.log.emit(msg)
# signal to handle error messages
error = pyqtSignal(str, name='error')
def _send_error(self, msg: str) -> None:
"""
Emits an error message to display in an error popup
"""
self.error.emit(msg)
# signal to send when processing is done
processingDone = pyqtSignal(name='processingDone')
def _signal_done(self) -> None:
"""
Signals to the frontend that the current session is finished
"""
self.processingDone.emit()
def _create_view(self) -> None:
"""
Create plugin view dialog
"""
path = Path(PluginRegistry.getInstance().getPluginPath(
'MultiSlice')) / 'MultiSliceView.qml'
self._view = CuraApplication.getInstance().createQmlComponent(
str(path), {'manager': self})
def _show_popup(self) -> None:
"""
Show plugin view dialog
"""
if self._view is None:
self._create_view()
if self._view is None:
Logger.log('e', 'Could not create QML')
self._view.show()
def _get_files(self, abs_paths: bool = False) -> List[Union[Path, str]]:
"""
Recursively collect files from input dir relative to follow depth
:param abs_paths: whether or not to collect absolute paths
"""
files = []
def _files(pattern: str, path: Path, depth: int):
# skip if we exceeded recursion depth
if depth > self._follow_depth:
return
try:
for d in Path_(path).iterdir():
# if we reached a directory, do recursive call
if d.is_dir():
_files(pattern, d, depth + 1)
# if we reached a file, check if it matches file pattern and add to list if so
elif d.is_file() and re.match(pattern, d.name):
nonlocal files
files.append(d if abs_paths else d.name)
except PermissionError:
# if we can't read the current step, notify and skip
self._log_msg(
'Could not access directory {0}, reason: permission denied. '
'Skipping.'.format(str(path)))
return
_files(self._file_pattern, self._input_path, 0)
return files
@pyqtProperty(list)
def files_names(self) -> List[str]:
"""
Retrieve names of all files matching settings
"""
return self._get_files() or []
@pyqtProperty(list)
def files_paths(self) -> List[Path]:
"""
Retrieve paths of all files matching settings
"""
return self._get_files(abs_paths=True) or []
@pyqtSlot(str)
def set_input_path(self, path: str) -> None:
"""
Set input path if valid
"""
if path and os.path.isdir(path):
self._input_path = Path(path)
@pyqtSlot(str)
def set_output_path(self, path: str) -> None:
"""
Set output path if valid
"""
if path and os.path.isdir(path):
self._output_path = Path(path)
@pyqtSlot(bool)
def set_follow_dirs(self, follow: bool) -> None:
"""
Set follow directories option
"""
self._follow_dirs = follow
@pyqtSlot(bool)
def set_preserve_dirs(self, preserve: bool) -> None:
"""
Set preserve directories option
"""
self._preserve_dirs = preserve
@pyqtSlot(str)
def set_file_pattern(self, regex: str) -> None:
"""
Set regex file pattern option if present, otherwise preserve default
"""
if regex:
self._file_pattern = regex
@pyqtSlot(str)
def set_follow_depth(self, depth: str) -> None:
"""
Set follow depth option if present, otherwise preserve default
"""
if depth:
self._follow_depth = depth
@pyqtProperty(bool)
def validate_input(self) -> bool:
"""
Try and validate applicable(bool options obviously don't need to be validated) options.
Emit error and return false if any fail.
"""
# file pattern should be valid regex
try:
re.compile(self._file_pattern)
except re.error:
self._send_error('Regex string \"{0}\" is not a valid regex. '
'Please try again.'.format(self._file_pattern))
return False
# input path should be a valid path
if type(self._input_path) is str or not Path_(
self._input_path).is_dir():
self._send_error('Input path \"{0}\" is not a valid path. '
'Please try again.'.format(self._input_path))
return False
# output path should be a valid path
if type(self._output_path) is str or not Path_(
self._output_path).is_dir():
self._send_error('Output path \"{0}\" is not a valid path. '
'Please try again.'.format(self._output_path))
return False
# follow depth should be an int
try:
self._follow_depth = int(self._follow_depth)
except ValueError:
self._send_error('Depth value \"{0}\" is not a valid integer. '
'Please try again.'.format(self._follow_depth))
return False
return True
@pyqtSlot()
def prepare_and_run(self) -> None:
"""
Do initial setup for running and start
"""
self._files = self.files_paths
# don't proceed if no files are found
if len(self._files) is 0:
self._log_msg('Found 0 files, please try again')
self._log_msg('-----')
self._signal_done()
return
self._log_msg('Found {0} files'.format(len(self._files)))
self._prepare_model()
# slice when model is loaded
CuraApplication.getInstance().fileCompleted.connect(self._slice)
# write gcode to file when slicing is done
Backend.Backend.backendStateChange.connect(self._write_gcode)
# run next model when file is written
self._write_done.connect(self._run_next)
self._clear_models()
self._run()
def _prepare_next(self) -> None:
"""
Prepare next model. If we don't have any models left to process, just set current to None.
"""
# if we don't have any files left, set current model to none
# current model is checked in function _run_next()
if len(self._files) == 0:
self._current_model = None
else:
self._log_msg('{0} file(s) to go'.format(len(self._files)))
self._prepare_model()
def _prepare_model(self) -> None:
"""
Perform necessary actions and field assignments for a given model
"""
self._current_model = self._files.pop()
self._current_model_suffix = self._current_model.suffix
self._current_model_name = self._current_model.name
self._current_model_url = QUrl().fromLocalFile(str(
self._current_model))
def _run(self) -> None:
"""
Run first iteration
"""
self._load_model_and_slice()
def _run_next(self) -> None:
"""
Run subsequent iterations
"""
self._log_msg('Clearing build plate and preparing next model')
self._clear_models()
self._prepare_next()
if not self._current_model:
self._log_msg('Found no more models. Done!')
# reset signal connectors once all models are done
self.__reset()
self._signal_done()
return
self._load_model_and_slice()
def _load_model_and_slice(self) -> None:
"""
Read .stl file into Cura and wait for fileCompleted signal
"""
self._log_msg('Loading model {0}'.format(self._current_model_name))
CuraApplication.getInstance().readLocalFile(self._current_model_url)
# wait for Cura to signal that it completed loading the file
self._loop.exec()
@staticmethod
def _clear_models() -> None:
"""
Clear all models on build plate
"""
CuraApplication.getInstance().deleteAll()
def _slice(self) -> None:
"""
Begin slicing models on build plate and wait for backendStateChange to signal state 3,
i.e. processing done
"""
self._log_msg('Slicing...')
CuraApplication.getInstance().backend.forceSlice()
# wait for CuraEngine to signal that slicing is done
self._loop.exec()
def _write_gcode(self, state) -> None:
"""
Write sliced model to file in output dir and emit signal once done
"""
# state = 3 = process is done
if state == 3:
# ensure proper file suffix
file_name = self._current_model_name.replace(
self._current_model_suffix, '.gcode')
# construct path relative to output directory using input path structure if we are
# following directories
# otherwise just dump it into the output directory
if self._preserve_dirs:
rel_path = self._current_model.relative_to(self._input_path)
path = (self._output_path / rel_path).parent / file_name
Path_(path.parent).mkdir(parents=True, exist_ok=True)
else:
path = self._output_path / file_name
self._log_msg('Writing gcode to file {0}'.format(file_name))
self._log_msg('Saving to directory: {0}'.format(str(path)))
with Path_(path).open(mode='w') as stream:
res = PluginRegistry.getInstance().getPluginObject(
"GCodeWriter").write(stream, [])
# GCodeWriter notifies success state with bool
if res:
self._write_done.emit()
def __reset(self) -> None:
"""
Reset all signal connectors to allow running subsequent processes without several
connector calls
"""
CuraApplication.getInstance().fileCompleted.disconnect(self._slice)
Backend.Backend.backendStateChange.disconnect(self._write_gcode)
self._write_done.disconnect(self._run_next)
@pyqtSlot()
def stop_multi_slice(self) -> None:
"""
Stop the session currently running
"""
self._log_msg("Cancel signal emitted, stopping Multislice")
self.__reset()
self._loop.exit()
@pyqtSlot(str)
def trim(self, path: str) -> str:
"""
Trims a file object from the frontend. What needs to be trimmed differes for
Linux and Windows.
"""
if platform.system() == "Windows":
return path.replace('file:///', '')
else:
return path.replace('file://', '')