class MainWindow(QWidget):
def __init__(self):
QMainWindow.__init__(self)
self.resize(500, 300)
self.mainLayout = QHBoxLayout()
self.chooseLayout = QHBoxLayout()
self.layout = QVBoxLayout()
self.pixLayout = QHBoxLayout()
self.thresholdLayout = QHBoxLayout()
self.timeLayout = QHBoxLayout()
self.contoursLayout = QHBoxLayout()
self.setLayout(self.mainLayout)
self.setWindowTitle(
"Image To Gcode V1.0 ----- build By yizheneng [email protected]")
self.imageLabel = QLabel("image")
self.mainLayout.addWidget(self.imageLabel)
self.mainLayout.addLayout(self.layout)
self.mainLayout.setStretchFactor(self.layout, 1)
self.mainLayout.setStretchFactor(self.imageLabel, 3)
self.pixLengthLabel = QLabel(u"像素大小(mm):")
self.pixDoubleSpinBox = QDoubleSpinBox()
self.pixDoubleSpinBox.setValue(1)
self.pixDoubleSpinBox.setDecimals(6)
self.pixLayout.addWidget(self.pixLengthLabel)
self.pixLayout.addWidget(self.pixDoubleSpinBox)
self.thresholdLabel = QLabel(u"阈值:")
self.thresholdSpinBox = QSpinBox()
self.thresholdSpinBox.valueChanged.connect(self.ThresholdValChange)
self.thresholdSpinBox.setMaximum(255)
self.thresholdSpinBox.setValue(120)
self.thresholdLayout.addWidget(self.thresholdLabel)
self.thresholdLayout.addWidget(self.thresholdSpinBox)
self.timeLabel = QLabel(u"灼烧时间:")
self.timeDoubleSpinBox = QDoubleSpinBox()
self.timeDoubleSpinBox.setValue(0.3)
self.timeLayout.addWidget(self.timeLabel)
self.timeLayout.addWidget(self.timeDoubleSpinBox)
self.chooseLabel = QLabel(u"只雕刻轮廓:")
self.chooseBox = QCheckBox()
self.chooseLayout.addWidget(self.chooseLabel)
self.chooseLayout.addWidget(self.chooseBox)
self.chooseBox.stateChanged.connect(self.ChooseValChanged)
self.contoursWidthLabel = QLabel(u"边框宽度")
self.ContoursWidthSpinBox = QSpinBox()
self.ContoursWidthSpinBox.setEnabled(False)
self.ContoursWidthSpinBox.setValue(1)
self.contoursLayout.addWidget(self.contoursWidthLabel)
self.contoursLayout.addWidget(self.ContoursWidthSpinBox)
self.loadImageButton = QPushButton(u"加载图片")
self.loadImageButton.clicked.connect(self.LoadImageButtonClicked)
self.previewButton = QPushButton(u"预览")
self.previewButton.clicked.connect(self.ThresholdValChange)
self.makeCodeButton = QPushButton(u"生成G代码")
self.makeCodeButton.clicked.connect(self.MakeGcode)
self.layout.addLayout(self.pixLayout)
self.layout.addLayout(self.thresholdLayout)
self.layout.addLayout(self.timeLayout)
self.layout.addLayout(self.chooseLayout)
self.layout.addLayout(self.contoursLayout)
self.layout.addWidget(self.loadImageButton)
self.layout.addWidget(self.previewButton)
self.layout.addWidget(self.makeCodeButton)
def LoadImageButtonClicked(self):
self.filePath = QFileDialog.getOpenFileName(self, u"选择图片文件", "",
"Images (*.bmp)")
if self.filePath == "":
QMessageBox.warning(self, u"发生错误", u"没有选择可以识别的文件！！")
return
self.srcImage = QImage(self.filePath)
self.grayImage = QImage(self.srcImage.size(), QImage.Format_Indexed8)
for i in range(256):
self.grayImage.setColor(i, qRgb(i, i, i))
for i in range(self.srcImage.width()):
for j in range(self.srcImage.height()):
temp = qGray(self.srcImage.pixel(i, j))
self.grayImage.setPixel(i, j, temp)
self.srcImage = QImage(self.grayImage)
self.resultImage = QImage(self.grayImage)
self.imageLabel.setPixmap(QPixmap(self.srcImage))
def ChooseValChanged(self):
self.ContoursWidthSpinBox.setEnabled(self.chooseBox.isChecked())
def ThresholdValChange(self):
for i in range(self.srcImage.width()):
for j in range(self.srcImage.height()):
temp = self.srcImage.pixelIndex(i, j)
if (temp >= self.thresholdSpinBox.value()):
self.grayImage.setPixel(i, j, 255)
else:
self.grayImage.setPixel(i, j, 0)
self.resultImage = QImage(self.grayImage)
#如果选中了只雕刻轮廓
if self.chooseBox.isChecked():
img = np.zeros(
(self.grayImage.height(), self.grayImage.width(), 1), np.uint8)
for i in range(self.grayImage.width()):
for j in range(self.grayImage.height()):
img[j, i] = self.grayImage.pixelIndex(i, j)
#提取轮廓
contours = cv.findContours(img, cv.RETR_LIST,
cv.CHAIN_APPROX_SIMPLE)
img = np.zeros(
(self.grayImage.height(), self.grayImage.width(), 1), np.uint8)
cv.drawContours(img, contours[1][:-1], -1, (255, 255, 255),
self.ContoursWidthSpinBox.value())
#转换轮廓到显示界面
for i in range(self.resultImage.width()):
for j in range(self.resultImage.height()):
if img[j, i] == 0:
self.resultImage.setPixel(i, j, 255)
else:
self.resultImage.setPixel(i, j, 0)
self.imageLabel.setPixmap(QPixmap(self.resultImage))
def MakeGcode(self):
path = QFileDialog.getSaveFileName(self, u"选择保存路径", "", " (*.nc)")
if path == "":
QMessageBox.warning(self, u"发生错误", u"路径错误！！")
return
f = open(path, 'w')
f.write("M5\n")
for i in range(self.resultImage.width()):
flag = False
#检测这一行是否有点
for j in range(self.resultImage.height()):
if self.resultImage.pixelIndex(i, j) < 128:
flag = True
break
#如果这一行都没有点则跳过这一行
if flag:
f.write("G0 Y%f\n" % (i * self.pixDoubleSpinBox.value()))
else:
continue
if (i % 2) > 0:
for j in range(self.resultImage.height()):
if self.resultImage.pixelIndex(i, j) < 128:
f.write("G0 X%f\n" %
(j * self.pixDoubleSpinBox.value()))
f.write("M3\n")
f.write("G4 P%f\n" % self.timeDoubleSpinBox.value())
f.write("M5\n")
else:
for j in range(self.resultImage.height())[::-1]:
if self.resultImage.pixelIndex(i, j) < 128:
f.write("G0 X%f\n" %
(j * self.pixDoubleSpinBox.value()))
f.write("M3\n")
f.write("G4 P%f\n" % self.timeDoubleSpinBox.value())
f.write("M5\n")
f.write("M5\n")
f.write("G0 X0 Y0\n")
f.close()
QMessageBox.information(self, u"成功", u"生成G代码文件成功！！")
def run(self):
self.exit = False
global psi1, psi2, psi3, psi4, M, N, x0, X, y0, Y, t0, T, n, f, phi
global Max, Min
U0 = []
global K
for k in xrange(K+1):
U0.append([phi(x0+m*h1,y0+k*h2) for m in xrange(M+1)])
#self.MU.append(U0)
img = QImage(M+1,K+1,QImage.Format_RGB32)
c = QColor()
U = U0
for i in xrange(len(U)):
for j in xrange(len(U[0])):
c.setHsvF(0.833333333*(1.0-(U[i][j]-Min)/(Max-Min)),1,1)
img.setPixel(i,j,c.rgb())
self.Images.append(img)
global u,tau
for n in xrange(N+1):
U = vardir(f, U0, psi1, psi2, psi3, psi4, M, K, N, x0, X, y0, Y, t0, T, n)
#U=[]
#for k in xrange(K+1):
# U.append([u(t0+n*tau,x0+m*h1,y0+k*h2) for m in xrange(M+1)])
U0 = []
for k in xrange(K+1):
U0.append(U[k])
#self.MU.append(U0)
#print (n/float(N)*100)
img = QImage(M+1,K+1,QImage.Format_RGB32)
c = QColor()
for i in xrange(len(U)):
for j in xrange(len(U[0])):
#print U[i][j]
fl = min(max(0, 0.833333333*(1.0-(U[i][j]-Min)/(Max-Min))), 1)
c.setHsvF(fl,1,1)
img.setPixel(i,j,c.rgb())
self.Images.append(img)
self.progr = (int(n/float(N)*100))
if(self.exit):
break
