mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-15 07:08:02 +08:00
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
from UM.Logger import Logger
|
|
from cura.Snapshot import Snapshot
|
|
from PyQt5.QtCore import QBuffer
|
|
|
|
from ..Script import Script
|
|
|
|
|
|
class X40Snapshot(Script):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def _createSnapshot(self, width, height):
|
|
Logger.log("d", "Creating thumbnail image...")
|
|
try:
|
|
return Snapshot.snapshot(width, height)
|
|
except Exception:
|
|
Logger.logException("w", "Failed to create snapshot image")
|
|
|
|
def _convertSnapshotToGcode(self, snapshot):
|
|
Logger.log("d", "Encoding thumbnail image...")
|
|
try:
|
|
thumbnail_buffer = QBuffer()
|
|
thumbnail_buffer.open(QBuffer.ReadWrite)
|
|
thumbnail_image = snapshot
|
|
thumbnail_image.save(thumbnail_buffer, "JPG")
|
|
|
|
thumbnail_buffer.seek(0)
|
|
gcode = []
|
|
newline = "W221"
|
|
byte = thumbnail_buffer.read(1)
|
|
counter = 0
|
|
while byte != b"":
|
|
if counter == 0:
|
|
gcode.append(newline)
|
|
newline = "W220 "
|
|
counter = 40
|
|
newline += byte.hex()
|
|
counter -= 1
|
|
|
|
byte = thumbnail_buffer.read(1)
|
|
gcode.append(newline)
|
|
gcode.append("W222")
|
|
|
|
thumbnail_buffer.close()
|
|
return gcode
|
|
except Exception:
|
|
Logger.logException("w", "Failed to encode snapshot image")
|
|
|
|
def getSettingDataString(self):
|
|
return """{
|
|
"name": "Create Thumbnail for Weedo X40",
|
|
"key": "X40Snapshot",
|
|
"metadata": {},
|
|
"version": 2,
|
|
"settings":
|
|
{}
|
|
}"""
|
|
|
|
def execute(self, data):
|
|
width = 180
|
|
height = 180
|
|
|
|
snapshot = self._createSnapshot(width, height)
|
|
if snapshot:
|
|
snapshot_gcode = self._convertSnapshotToGcode(snapshot)
|
|
|
|
for layer in data:
|
|
layer_index = data.index(layer)
|
|
lines = data[layer_index].split("\n")
|
|
for line in lines:
|
|
if line.startswith(";Generated with Cura"):
|
|
line_index = lines.index(line)
|
|
insert_index = line_index + 1
|
|
lines[insert_index:insert_index] = snapshot_gcode
|
|
break
|
|
|
|
final_lines = "\n".join(lines)
|
|
data[layer_index] = final_lines
|
|
|
|
return data
|