def setFont(self, font):
"""Set the font selector to the given font.
Arguments:
font -- the QFont to set.
"""
fontInfo = QFontInfo(font)
family = fontInfo.family()
matches = self.familyList.findItems(family, Qt.MatchExactly)
if matches:
self.familyList.setCurrentItem(matches[0])
self.familyList.scrollToItem(matches[0],
QAbstractItemView.PositionAtTop)
style = QFontDatabase().styleString(fontInfo)
matches = self.styleList.findItems(style, Qt.MatchExactly)
if matches:
self.styleList.setCurrentItem(matches[0])
self.styleList.scrollToItem(matches[0])
else:
self.styleList.setCurrentRow(0)
self.styleList.scrollToItem(self.styleList.currentItem())
size = repr(fontInfo.pointSize())
matches = self.sizeList.findItems(size, Qt.MatchExactly)
if matches:
self.sizeList.setCurrentItem(matches[0])
self.sizeList.scrollToItem(matches[0])
def __init__(self, di, prefix, sortcus, sortdies):
QAbstractItemModel.__init__(self)
self.prefix = prefix
self.top_dies = [
decorate_die(CU.get_top_DIE(), i) for (i, CU) in enumerate(di._CUs)
]
self.highlight_condition = None
fi = QFontInfo(QApplication.font())
self.bold_font = QFont(fi.family(), fi.pointSize(), QFont.Bold)
self.blue_brush = QBrush(Qt.GlobalColor.blue)
self.sortcus = sortcus
self.sortdies = sortdies
class Ui_MarkWindow(QMainWindow):
# 构造方法
def __init__(self):
super(Ui_MarkWindow, self).__init__()
self.setWindowFlags(QtCore.Qt.WindowCloseButtonHint) # 只显示关闭按钮
self.setupUi(self) # 初始化窗体设置
# 自动生成的代码，用来对窗体进行设置
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(578, 418)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
# 设置图片显示列表
self.listWidget = QtWidgets.QListWidget(self.centralwidget)
self.listWidget.setGeometry(QtCore.QRect(0, 0, 141, 391))
self.listWidget.setObjectName("listWidget")
# 设置加载图片按钮
self.pushButton = QtWidgets.QPushButton(self.centralwidget)
self.pushButton.setGeometry(QtCore.QRect(150, 0, 75, 23))
self.pushButton.setObjectName("pushButton")
self.groupBox = QtWidgets.QGroupBox(self.centralwidget)
self.groupBox.setGeometry(QtCore.QRect(150, 30, 421, 151))
self.groupBox.setObjectName("groupBox")
# 设置文字水印单选按钮
self.radioButton = QtWidgets.QRadioButton(self.groupBox)
self.radioButton.setGeometry(QtCore.QRect(20, 20, 101, 16))
self.radioButton.setChecked(True) # 默认选中
self.radioButton.setObjectName("radioButton")
self.label = QtWidgets.QLabel(self.groupBox)
self.label.setGeometry(QtCore.QRect(30, 50, 54, 16))
self.label.setObjectName("label")
# 设置要输入水印文字的文本框
self.lineEdit = QtWidgets.QLineEdit(self.groupBox)
self.lineEdit.setGeometry(QtCore.QRect(90, 50, 241, 20))
self.lineEdit.setObjectName("lineEdit")
# 设置“字体设置”按钮
self.pushButton_2 = QtWidgets.QPushButton(self.groupBox)
self.pushButton_2.setGeometry(QtCore.QRect(340, 50, 75, 23))
self.pushButton_2.setObjectName("pushButton_2")
# 设置图片水印单选按钮
self.radioButton_2 = QtWidgets.QRadioButton(self.groupBox)
self.radioButton_2.setGeometry(QtCore.QRect(20, 80, 91, 16))
self.radioButton_2.setChecked(False) # 默认不选中
self.radioButton_2.setObjectName("radioButton_2")
# 设置选择水印图片按钮
self.pushButton_3 = QtWidgets.QPushButton(self.groupBox)
self.pushButton_3.setGeometry(QtCore.QRect(340, 110, 75, 23))
self.pushButton_3.setObjectName("pushButton_3")
self.label_2 = QtWidgets.QLabel(self.groupBox)
self.label_2.setGeometry(QtCore.QRect(30, 110, 54, 16))
self.label_2.setObjectName("label_2")
# 设置显示水印图片路径的文本框
self.lineEdit_2 = QtWidgets.QLineEdit(self.groupBox)
self.lineEdit_2.setGeometry(QtCore.QRect(90, 110, 241, 20))
self.lineEdit_2.setObjectName("lineEdit_2")
self.groupBox_2 = QtWidgets.QGroupBox(self.centralwidget)
self.groupBox_2.setGeometry(QtCore.QRect(150, 190, 421, 71))
self.groupBox_2.setObjectName("groupBox_2")
self.label_3 = QtWidgets.QLabel(self.groupBox_2)
self.label_3.setGeometry(QtCore.QRect(270, 31, 54, 21))
self.label_3.setObjectName("label_3")
# 设置水印位置选择框
self.comboBox = QtWidgets.QComboBox(self.groupBox_2)
self.comboBox.setGeometry(QtCore.QRect(330, 30, 71, 22))
self.comboBox.setObjectName("comboBox")
self.comboBox.addItem('左上角')
self.comboBox.addItem('右上角')
self.comboBox.addItem('左下角')
self.comboBox.addItem('右下角')
self.comboBox.addItem('居中位置')
self.comboBox.setCurrentIndex(0) # 设置默认选择第一项
self.label_4 = QtWidgets.QLabel(self.groupBox_2)
self.label_4.setGeometry(QtCore.QRect(20, 30, 51, 21))
self.label_4.setObjectName("label_4")
# 设置水印透明度的滑动条
self.horizontalSlider = QtWidgets.QSlider(self.groupBox_2)
self.horizontalSlider.setGeometry(QtCore.QRect(70, 30, 181, 22))
self.horizontalSlider.setMinimum(1)
self.horizontalSlider.setMaximum(10)
self.horizontalSlider.setOrientation(QtCore.Qt.Horizontal)
self.horizontalSlider.setObjectName("horizontalSlider")
self.groupBox_3 = QtWidgets.QGroupBox(self.centralwidget)
self.groupBox_3.setGeometry(QtCore.QRect(150, 270, 421, 71))
self.groupBox_3.setObjectName("groupBox_3")
self.label_6 = QtWidgets.QLabel(self.groupBox_3)
self.label_6.setGeometry(QtCore.QRect(20, 30, 61, 21))
self.label_6.setObjectName("label_6")
# 设置显示保存路径的文本框
self.lineEdit_3 = QtWidgets.QLineEdit(self.groupBox_3)
self.lineEdit_3.setGeometry(QtCore.QRect(80, 30, 241, 20))
self.lineEdit_3.setObjectName("lineEdit_3")
# 设置选择图片保存路径的按钮
self.pushButton_4 = QtWidgets.QPushButton(self.groupBox_3)
self.pushButton_4.setGeometry(QtCore.QRect(330, 30, 75, 23))
self.pushButton_4.setObjectName("pushButton_4")
#设置执行按钮
self.pushButton_5 = QtWidgets.QPushButton(self.centralwidget)
self.pushButton_5.setGeometry(QtCore.QRect(480, 350, 75, 23))
self.pushButton_5.setObjectName("pushButton_5")
MainWindow.setCentralWidget(self.centralwidget)
#设置状态栏
self.statusBar = QtWidgets.QStatusBar(MainWindow)
self.statusBar.setObjectName("statusBar")
self.statusBar.showMessage('准备就绪…… ') # 设置状态栏默认值
MainWindow.setStatusBar(self.statusBar)
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
# 自动生成的代码，用来设置窗体中控件的默认值
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "图片批量添加水印"))
self.pushButton.setText(_translate("MainWindow", "加载图片"))
self.groupBox.setTitle(_translate("MainWindow", "水印设置"))
self.radioButton.setText(_translate("MainWindow", "添加文字水印"))
self.label.setText(_translate("MainWindow", "水印文字："))
self.pushButton_2.setText(_translate("MainWindow", "字体设置"))
self.radioButton_2.setText(_translate("MainWindow", "添加图片水印"))
self.pushButton_3.setText(_translate("MainWindow", "浏览"))
self.label_2.setText(_translate("MainWindow", "水印图片："))
self.groupBox_2.setTitle(_translate("MainWindow", "透明度及位置设置"))
self.label_3.setText(_translate("MainWindow", "水印位置："))
self.label_4.setText(_translate("MainWindow", "透明度："))
self.groupBox_3.setTitle(_translate("MainWindow", "路径设置"))
self.label_6.setText(_translate("MainWindow", "保存位置："))
self.pushButton_4.setText(_translate("MainWindow", "浏览"))
self.pushButton_5.setText(_translate("MainWindow", "执行"))
# 关联“加载图片”按钮的方法
self.pushButton.clicked.connect(self.getFiles)
# 关联“字体设置”按钮的方法
self.pushButton_2.clicked.connect(self.setFont)
# 关联“选择图片”按钮的方法
self.pushButton_3.clicked.connect(self.setImg)
# 关联“选择保存路径”按钮的方法
self.pushButton_4.clicked.connect(self.msg)
# 关联“执行”按钮的方法
self.pushButton_5.clicked.connect(self.addMark)
# 关联列表单击方法，用来预览选中的图片
self.listWidget.itemClicked.connect(self.itemClick)
# 设置字体
def setFont(self):
self.waterfont, ok = QFontDialog.getFont() # 显示字体对话框
if ok: # 判断是否选择了字体
self.lineEdit.setFont(self.waterfont) # 设置水印文字的字体
self.fontSize = QFontMetrics(self.waterfont) # 获取字体尺寸
self.fontInfo = QFontInfo(self.waterfont) # 获取字体信息
# 是否为图片
def isImg(self, file):
file = file.lower()
if file == '.jpg':
return True
elif file == '.png':
return True
elif file == '.jpeg':
return True
elif file == '.bmp':
return True
else:
return False
# 获取所有文件
def getFiles(self):
try:
# 选择图片文件夹路径
self.img_path = QFileDialog.getExistingDirectory(
None, "选择图片文件夹路径", os.getcwd())
self.list = os.listdir(self.img_path) # 遍历选择的文件夹
num = 0 # 记录图片数量
self.listWidget.clear() # 清空列表项
for i in range(0, len(self.list)): # 遍历图片列表
filepath = os.path.join(self.img_path,
self.list[i]) # 记录遍历到的文件名
if os.path.isfile(filepath): # 判断是否为文件
imgType = os.path.splitext(filepath)[1] # 获取扩展名
if self.isImg(imgType): # 判断是否为图片
num += 1 # 数量加1
self.item = QtWidgets.QListWidgetItem(
self.listWidget) # 创建列表项
self.item.setText(self.list[i]) # 显示图片列表
self.statusBar.showMessage('共有图片 ' + str(num) + ' 张') # 显示图片总数
except Exception:
QMessageBox.warning(None, '警告', '请选择一个有效路径……', QMessageBox.Ok)
# 预览图片
def itemClick(self, item):
os.startfile(self.img_path + '\\' + item.text())
# 选择水印图片
def setImg(self):
try:
# waterimg即为选择的水印图片，第二形参为对话框标题，第三个为对话框打开后默认的路径
self.waterimg = QFileDialog.getOpenFileName(
None, '选择水印图片', 'C:\\', "图片文件(*.jpeg;*.png;*.jpg;*.bmp)")
self.lineEdit_2.setText(self.waterimg[0]) # 显示选择的水印图片
except Exception as e:
print(e)
# 选择保存路径
def msg(self):
try:
# dir_path即为选择的文件夹的绝对路径，第二形参为对话框标题，第三个为对话框打开后默认的路径
self.dir_path = QFileDialog.getExistingDirectory(
None, "选择路径", os.getcwd())
self.lineEdit_3.setText(self.dir_path) # 显示选择的保存路径
except Exception as e:
print(e)
# 文字水印
def textMark(self, img, newImgPath):
try:
im = Image.open(img).convert('RGBA') # 打开原始图片，并转换为RGBA
newImg = Image.new('RGBA', im.size,
(255, 255, 255, 0)) # 存储添加水印后的图片
# 创建字体,说明：默认使用楷体，如果需要使用其他字体，需要将字体文件复制到当前目录中
# 然后对下面第一个参数进行修改，可以使用self.fontInfo.family()动态获取字体名称，后面加扩展名即可
font = ImageFont.truetype('simkai.ttf', self.fontInfo.pointSize())
imagedraw = ImageDraw.Draw(newImg) # 创建绘制对象
imgwidth, imgheight = im.size # 记录图片大小
txtwidth = self.fontSize.maxWidth() * len(
self.lineEdit.text()) # 获取字体宽度
txtheight = self.fontSize.height() # 获取字体高度
# 设置水印文字位置
if self.comboBox.currentText() == '左上角':
position = (0, 0)
elif self.comboBox.currentText() == '左下角':
position = (0, imgheight - txtheight)
elif self.comboBox.currentText() == '右上角':
position = (imgwidth - txtwidth, 0)
elif self.comboBox.currentText() == '右下角':
position = (imgwidth - txtwidth, imgheight - txtheight)
elif self.comboBox.currentText() == '居中位置':
position = (imgwidth / 2, imgheight / 2)
# 设置文本颜色
imagedraw.text(position,
self.lineEdit.text(),
font=font,
fill="#FCA454")
# 设置透明度
alpha = newImg.split()[3]
alpha = ImageEnhance.Brightness(alpha).enhance(
int(self.horizontalSlider.value()) / 10.0)
newImg.putalpha(alpha)
Image.alpha_composite(im, newImg).save(newImgPath) # 保存图片
except Exception:
QMessageBox.warning(None, '错误', '图片格式有误，请重新选择……', QMessageBox.Ok)
# 图片水印
def imgMark(self, img, newImgPath):
im = Image.open(img) # 打开原始图片
mark = Image.open(self.lineEdit_2.text()) # 打开水印图片
rgbaim = im.convert('RGBA') # 将原始图片转换为RGBA
rgbamark = mark.convert('RGBA') # 将水印图片转换为RGBA
imgwidth, imgheight = rgbaim.size # 获取原始图片尺寸
nimgwidth, nimgheight = rgbamark.size # 获取水印图片尺寸
# 缩放水印图片
scale = 10
markscale = max(imgwidth / (scale * nimgwidth),
imgheight / (scale * nimgheight))
newsize = (int(nimgwidth * markscale), int(nimgheight * markscale)
) # 计算新的尺寸大小
rgbamark = rgbamark.resize(newsize,
resample=Image.ANTIALIAS) # 重新设置水印图片大小
nimgwidth, nimgheight = rgbamark.size # 获取水印图片缩放后的尺寸
# 计算水印位置
if self.comboBox.currentText() == '左上角':
position = (0, 0)
elif self.comboBox.currentText() == '左下角':
position = (0, imgheight - nimgheight)
elif self.comboBox.currentText() == '右上角':
position = (imgwidth - nimgwidth, 0)
elif self.comboBox.currentText() == '右下角':
position = (imgwidth - nimgwidth, imgheight - nimgheight)
elif self.comboBox.currentText() == '居中位置':
position = (int(imgwidth / 2), int(imgheight / 2))
# 设置透明度：img.point(function)接受一个参数，且对图片中的每一个点执行这个函数，这个函数是一个匿名函数，使用lambda表达式来完成
# convert()函数，用于不同模式图像之间的转换，模式“L”为灰色图像，它的每个像素用8个bit表示，0表示黑，255表示白，其他数字表示不同的灰度。
# 在PIL中，从模式“RGB”转换为“L”模式是按照下面的公式转换的：L = R * 299/1000 + G * 587/1000+ B * 114/1000
rgbamarkpha = rgbamark.convert("L").point(
lambda x: x / int(self.horizontalSlider.value()))
rgbamark.putalpha(rgbamarkpha)
# 水印位置
rgbaim.paste(rgbamark, position, rgbamarkpha)
try:
rgbaim.save(newImgPath) # 保存水印图片
except Exception:
QMessageBox.warning(None, '错误', '请选择其他路径……', QMessageBox.Ok)
# 添加水印
def addMark(self):
if self.lineEdit_3.text() == '': # 判断是否选择了保存路径
QMessageBox.warning(None, '警告', '请选择保存路径', QMessageBox.Ok)
return
else:
num = 0 # 记录处理图片数量
for i in range(0, self.listWidget.count()): # 遍历图片列表
# 设置原始图片路径（包括文件名）
filepath = os.path.join(self.img_path,
self.listWidget.item(i).text())
# 设置水印图片保存路径（包括文件名）
newfilepath = os.path.join(self.lineEdit_3.text(),
self.listWidget.item(i).text())
if self.radioButton.isChecked(): # 判断是否选择文字水印单选按钮
if self.lineEdit.text() == '': # 判断是否输入了水印文字
QMessageBox.warning(None, '警告', '请输入水印文字',
QMessageBox.Ok)
return
else:
self.textMark(filepath,
newfilepath) # 调用textMark方法添加文字水印
num += 1 # 处理图片数量加1
else:
if self.lineEdit_2.text() != '': # 判断水印图片不为空
self.imgMark(filepath,
newfilepath) # 调用imgMark方法添加图片水印
num += 1 # 处理图片数量加1
else:
QMessageBox.warning(None, '警告', '请选择水印图片',
QMessageBox.Ok)
self.statusBar.showMessage('任务完成，此次共处理 ' + str(num) +
' 张图片') # 显示处理图片总数
def reportCodeViewerProperties(self, font):
wposition = self.pos()
font_info = QFontInfo(font)
print("---------- Code Viewer Report ----------")
print("Size: width: %d, height: %d" % (self.width, self.height))
print("Top-Left position in screen: (%d, %d)" %
(wposition.x(), wposition.y()))
print("Top-Right position in screen: (%d, %d)" %
(wposition.x() + self.width, wposition.y()))
print("Bottom-Left position in screen: (%d, %d)" %
(wposition.x(), wposition.y() + self.height))
print("Bottom-Right position in screen: (%d, %d)" %
(wposition.x() + self.width, wposition.y() + self.height))
print("---------- Side Bar Report ----------")
print("Size: width: %d, height: %d" %
(self.listWidth, self.editorHeight))
print("Top-Left position in screen: (%d, %d)" %
(wposition.x(), wposition.y()))
print("Top-Right position in screen: (%d, %d)" %
(wposition.x() + self.listWidth, wposition.y()))
print("Bottom-Left position in screen: (%d, %d)" %
(wposition.x(), wposition.y() + self.editorHeight))
print("Bottom-Right position in screen: (%d, %d)" %
(wposition.x() + self.listWidth,
wposition.y() + self.editorHeight))
print("---------- Code Editor Report ----------")
print("Size: width: %d, height: %d" %
(self.editorWidth, self.editorHeight))
print("Margins: %f pixels" % self.editor.document().documentMargin())
print("Top-Left position in screen: (%d, %d)" %
(wposition.x() + self.listWidth, wposition.y()))
print(
"Top-Right position in screen: (%d, %d)" %
(wposition.x() + self.listWidth + self.editorWidth, wposition.y()))
print("Bottom-Left position in screen: (%d, %d)" %
(wposition.x() + self.listWidth,
wposition.y() + self.editorHeight))
print("Bottom-Right position in screen: (%d, %d)" %
(wposition.x() + self.listWidth + self.editorWidth,
wposition.y() + self.editorHeight))
print("Padding: left: %d, top: %d, bottom: %d, right: %d" %
(self.padding_left, self.padding_top, self.padding_bottom,
self.padding_right))
print("---------- Font report (font) ----------")
print("font family: %s" % font.family())
print("font pixelSize: %f" % font.pixelSize())
print("font pointSize: %f" % font.pointSize())
print("font pointSizeF: %f" % font.pointSizeF())
print("QFont.pixelSize(): %f" % font.pixelSize())
print("---------- QFontInfo (font) ----------")
print("family(): %s" % font_info.family())
print("pixelSize(): %f" % font_info.pixelSize())
print("pointSize(): %f" % font_info.pointSize())
print("pointSizeF(): %f" % font_info.pointSizeF())
print("QFontInfo.pixelSize(): %d" % font_info.pixelSize())
print("---------- QFontMetrics (font) ---------")
print("QFontMetrics.lineSpacing(): %f" %
QFontMetrics(font).lineSpacing())
print("QFontMetrics.leading(): %f" % QFontMetrics(font).leading())
print("QFontMetrics.height(): %f" % QFontMetrics(font).height())
print("---------- Eye tracking area ----------")
x1 = wposition.x() + self.listWidth + self.padding_left
y1 = wposition.y() + self.padding_top
x2 = wposition.x(
) + self.listWidth + self.editorWidth - self.padding_right
y2 = wposition.y() + self.editorHeight - self.padding_bottom
self.editor.x_offset = x1 # For the eye tracker, as it uses the entire screen
self.editor.y_offset = y1 # For the eye tracker, as it uses the entire screen
self.editor.most_right_x = x2
self.editor.most_bottom_y = y2
print("Size: width: %d, height: %d" % (x2 - x1, y2 - y1))
print("Top-Left position in screen: (%d, %d)" % (x1, y1))
print("Top-Right position in screen: (%d, %d)" % (x2, y1))
print("Bottom-Left position in screen: (%d, %d)" % (x1, y2))
print("Bottom-Right position in screen: (%d, %d)" % (x2, y2))
print("x_offset = %d" % x1)
print("y_offset = %d" % y1)
print("---------- Status Bar Report ----------")
print("Size: width: %d, height: %d" %
(self.width, self.status_bar_height))
print("Top-Left position in screen: (%d, %d)" %
(wposition.x(), wposition.y() + self.editorHeight))
print("Top-Right position in screen: (%d, %d)" %
(wposition.x() + self.width, wposition.y() + self.editorHeight))
print("Bottom-Left position in screen: (%d, %d)" %
(wposition.x(),
wposition.y() + self.editorHeight + self.status_bar_height))
print("Bottom-Right position in screen: (%d, %d)" %
(wposition.x() + self.width,
wposition.y() + self.editorHeight + self.status_bar_height))
print("----------------------------------------")