class ParmerPanel(QWidget):
"""参数面板类"""
send_message_signal = Signal(str, str)
def __init__(self, username):
super(ParmerPanel, self).__init__()
self.username = username
self.id = None
self.setObjectName("parmPanel")
self.setupUI()
dayu_theme.apply(self)
# 设置UI界面
def setupUI(self):
self.setMaximumWidth(Data.getWindowWidth() / 3)
self.ui = loadUi(file_path + "\\res\\UI\\ParameterWindow.ui")
self.ui.setParent(self)
self.setLayout(QVBoxLayout())
self.layout().addWidget(self.ui)
self.widget_1 = self.ui.findChild(QWidget, "widget")
self.widget_2 = self.ui.findChild(QWidget, "widget_2")
self.widget_3 = self.ui.findChild(QWidget, "widget_3")
self.widget_4 = self.ui.findChild(QWidget, "widget_4")
self.widget_1.setLayout(QVBoxLayout())
self.widget_2.setLayout(QVBoxLayout())
self.widget_4.setLayout(QVBoxLayout())
self.widget_2.layout().setSpacing(8)
#设置布局
tab_card = MTabWidget()
self.label_filePic = MLabel("")
self.widget_1.setMinimumSize(Data.getWindowHeight() / 2.8,
Data.getWindowHeight() / 2.8)
tab_card.addTab(self.label_filePic, u'预览图')
# Todo 加载3d视口
# self.model_widget = QWidget()
self.model_widget = CefBrowser(self, url="editor")
# self.model_widget.setLayout(QVBoxLayout())
tab_card.addTab(self.model_widget, u'3D视口')
self.widget_1.layout().addWidget(tab_card)
self.widget_1.layout().setContentsMargins(0, 0, 0, 0)
self.widget_2.layout().addWidget(MDivider(u'操作面板'))
self.let_filename = MLineEdit(text='filename')
tool_button = MLabel(text=u'文件名').mark().secondary()
tool_button.setAlignment(Qt.AlignCenter)
tool_button.setFixedWidth(80)
self.let_filename.set_prefix_widget(tool_button)
self.widget_2.layout().addWidget(self.let_filename)
self.let_path = MLineEdit(text='filepath')
tool_button_2 = MLabel(text=u'文件地址').mark().secondary()
tool_button_2.setAlignment(Qt.AlignCenter)
tool_button_2.setFixedWidth(80)
self.let_path.set_prefix_widget(tool_button_2)
self.widget_2.layout().addWidget(self.let_path)
self.widget_2.layout().addWidget(MLabel(u'标签'))
self.let_tag = MLineEdit(text='tag')
self.btn_reviseTag = MPushButton(text=u'修改').primary()
self.btn_reviseTag.setFixedWidth(80)
self.let_tag.set_suffix_widget(self.btn_reviseTag)
self.widget_2.layout().addWidget(self.let_tag)
self.btn_export = MPushButton(u'导出到houdini').primary()
self.widget_2.layout().addWidget(self.btn_export)
self.btn_exportToMaya = MPushButton(u'导出到Maya').primary()
self.widget_2.layout().addWidget(self.btn_exportToMaya)
self.timer = QTimer()
self.timer.setInterval(0.1)
self.timer.timeout.connect(self.slot_timeout)
self.auto_color_progress = MProgressBar().auto_color()
self.widget_2.layout().addWidget(self.auto_color_progress)
self.widget_4.layout().addWidget(MDivider(u'操作记录'))
self.setWindowTitle(u"参数面板")
# #获取控件
self.tableWidget_operationNote = self.ui.findChild(
QTableWidget, "tableWidget_operationNote")
self.tableWidget_operationNote.setStyleSheet(Data.getQSS())
#设置默认值
self.let_filename.setReadOnly(True) #只读
self.let_path.setReadOnly(True) #只读
self.let_tag.setReadOnly(True) #只读
self.tableWidget_operationNote.setHorizontalHeaderLabels(
[u'用户', u'操作', u'时间'])
# #连接信号与槽
self.btn_export.clicked.connect(self.slot_run)
self.btn_reviseTag.clicked.connect(lambda: self.reviseTag())
# self.tableWidget_operationNote.setColumnCount(3)
setSectionResizeMode(self.tableWidget_operationNote.horizontalHeader(),
QHeaderView.Stretch) # 自适应
def setParam(self, type, name, path):
self.type = type
self.filename = name
self.filepath = path
# 根据图片类型设置图片
if (type == "jpg" or type == "jpeg" or type == "png"):
self.setPic(path)
elif (type == "obj" or type == "fbx"):
self.setObjPic()
filepath = os.path.dirname(path)
filename = filepath.split("/")[-1]
self.let_filename.setText(filename)
self.let_path.setText(filepath)
tags = ""
assetdb = client[type]
assetcol = assetdb[name]
for tagdic in assetcol.find({}, {"Tag": 1}):
if "Tag" in tagdic:
tag = tagdic["Tag"]
tags += tag + ","
self.let_tag.setText(tags)
#设置资产操作记录表：
assetdb = client[type]
assetcol = assetdb[name]
assetlist = assetcol.find({}, {
"UserName": 1,
"Operation": 1,
"Time": 1
})
#rowcount = len(assetlist) #assetlist并不是列表类型
i = 0
# for x in assetlist:
# i += 1 #求得行数减一
for xdir in assetlist:
if "UserName" in xdir:
str1 = xdir["UserName"]
newItem1 = QTableWidgetItem(str1)
self.tableWidget_operationNote.setItem(i, 0, newItem1)
if "Time" in xdir:
str3 = xdir["Time"]
newItem3 = QTableWidgetItem(str3)
self.tableWidget_operationNote.setItem(i, 2, newItem3)
if "Operation" in xdir:
str2 = xdir["Operation"]
newItem2 = QTableWidgetItem(str2)
self.tableWidget_operationNote.setItem(i, 1, newItem2)
i += 1
self.saveBrowseNode(self.username, name, type) #保存浏览信息到库
# 链接信号与槽函数
self.btn_export.clicked.connect(
lambda: self.exportModelToHoudini(name, type, path))
self.btn_exportToMaya.clicked.connect(
lambda: self.exportModelToMaya(name, type, path))
# self.saveBrowseNode(self.username, name, type)
# 进度条
def slot_run(self):
self.timer.start()
self.auto_color_progress.setValue(0)
def slot_timeout(self):
if self.auto_color_progress.value() > 99:
self.timer.stop()
else:
self.auto_color_progress.setValue(
self.auto_color_progress.value() + 1)
def setPic(self, path):
pixmap = QPixmap(path)
self.label_filePic.setPixmap(pixmap)
self.label_filePic.setScaledContents(True)
def setObjPic(self):
"""加载的是obj模型，设置obj的图片"""
pixmap = QPixmap(file_path + r"\res\image\objimg.jpg")
self.label_filePic.setPixmap(pixmap)
self.label_filePic.setScaledContents(True)
def saveBrowseNode(self, username, filename, type):
RTime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))
#将浏览信息存入用户数据库
col = userdb[username]
dict = {"Operation": "Browse", "Time": RTime, "FileName": filename}
col.insert_one(dict)
#将浏览信息存入资产数据库
assetdb = client[type]
assetcol = assetdb[filename]
adict = {"UserName": username, "Time": RTime, "Operation": "Browse"}
assetcol.insert_one(adict)
def exportModelToHoudini(self, filename, type, path):
try:
import hrpyc
connection, hou = hrpyc.import_remote_module()
name = filename.split(".")[0]
if (type == "obj" or type == "fbx"):
print("exportModelHoudiniOBJ")
geo = hou.node('/obj').createNode('geo', name)
fileNode = geo.createNode('file', name)
fileNode.parm('file').set(path)
prinShaderNode = hou.node('/mat').createNode(
'principledshader', name)
prinShaderNode.parm('basecolor_useTexture').set(1)
materialNode = geo.createNode('material', name + "_material")
materialNode.parm('shop_materialpath1').set("/mat/" + name)
materialNode.setInput(0, fileNode)
materialNode.moveToGoodPosition()
materialNode.setDisplayFlag(1)
if (type == "jpg" or type == "jpeg"):
try:
print("exportModelHoudiniJPG")
# 路径不能有中文
imgNode = hou.node('/img').createNode('img', "comp1")
imgNode = hou.node('/img/comp1').createNode('file', name)
# fileNode = imgNode.createNode('file', name)
imgNode.parm('filename1').set(path)
except:
print("ExportPictureFail")
#保存导出记录到资产数据库
RTime = time.strftime('%Y-%m-%d %H:%M:%S',
time.localtime(time.time()))
assetdb = client[type]
assetcol = assetdb[filename]
adict = {
"UserName": self.username,
"Time": RTime,
"Operation": "Export"
}
assetcol.insert_one(adict)
# 保存导出记录到用户数据库
usercol = userdb[self.username]
adict = {
"FileName": filename,
"Time": RTime,
"Operation": "Export"
}
usercol.insert_one(adict)
except:
self.slot_show_message(MMessage.info,
(u'导出失败！请确认Houdini是否配置成功或启动。'))
print(path)
print(type)
def exportModelToMaya(self, filename, type, path):
if type == "obj" or type == "fbx":
print("ExportModelToMaya")
import socket
try:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1', 7001))
#s.send('print("HelloWord!!!!!!!!!!!!");')
name = filename.split(".")[0]
command0 = "import maya.cmds as mc;"
command1 = "imported_objects = mc.file(r'" + path + "\', ns='ns', i=True, rnn=True);"
command2 = "transforms = mc.ls(imported_objects, type='transform');"
command = command0 + command1 + command2
s.send(command)
except:
self.slot_show_message(MMessage.info,
(u'导出失败！请确认Maya是否配置成功或启动。'))
print(path)
print(type)
else:
print("This can't export to Maya")
def saveExportModelNode(self, username, filename, type):
RTime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))
# 将浏览信息存入用户数据库
col = userdb[username]
dict = {"Operation": "Export", "Time": RTime, "FileName": filename}
col.insert_one(dict)
# 将浏览信息存入资产数据库
assetdb = client[type]
assetcol = assetdb[filename]
adict = {"UserName": username, "Time": RTime, "Operation": "Export"}
assetcol.insert_one(adict)
def reviseTag(self):
if self.username != None:
usercol = userdb[self.username]
idlist = usercol.find({"_id": "UserID"}, {"UserID": 1})
for iddir in idlist:
self.id = iddir["UserID"]
if self.id != "管理员":
self.send_message_signal.emit("warnning", u"只有管理员才可以修改标签")
return 0
self.let_tag.setReadOnly(False) #可写入
self.btn_reviseTag.setText(u"确认修改")
self.btn_reviseTag.clicked.disconnect()
self.btn_reviseTag.clicked.connect(lambda: self.confirmReviseTag())
def confirmReviseTag(self):
self.let_tag.setReadOnly(True) # 只读
#连接数据库
assetdb = client[self.type]
assetcol = assetdb[self.filename]
assetdb = client[self.type]
assetcol = assetdb[self.filename]
newTags = self.let_tag.text().split(",")
self.oldTags = []
taglist = assetcol.find({}, {"Tag": 1})
for tag in taglist:
if "Tag" in tag:
self.oldTags.append(tag["Tag"])
#是否有删除标签
for oldTag in self.oldTags:
delTag = True
for newTag in newTags:
if newTag == oldTag:
delTag = False
if delTag: #存在
# 删除资产库里的标签
deldir = {"Tag": oldTag}
assetcol.delete_one(deldir)
#删除标签文件库里对应的文件
tagfilecol = tagfiledb[oldTag]
deldir = {"FileName": self.filename}
tagfilecol.delete_one(deldir)
#是否有新标签
for newTag in newTags:
if newTag == "":
continue
# 是否为从未有过的标签
new = True
for x in tagcol.find({}, {"Tag": 1}): # 把数据库里的标签取出
if newTag == x["Tag"]:
new = False
if new:
newTagdir = {"Tag": newTag}
tagcol.insert_one(newTagdir) # 添加到标签数据库
addTag = True
for oldTag in self.oldTags:
if newTag == oldTag:
addTag = False
if addTag:
#资产库里添加新标签
adict = {"Tag": newTag}
assetcol.insert_one(adict)
# 标签文件库里添加对应的文件
tagfilecol = tagfiledb[newTag]
deldir = {"FileName": self.filename}
tagfilecol.insert_one(deldir)
self.btn_reviseTag.setText(u"修改")
self.send_message_signal.emit("info", u"已成功修改标签")
#保存操作记录到用户数据库
RTime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))
usercol = userdb[self.username]
adict = {
"FileName": self.filename,
"Time": RTime,
"Operation": "EditTag"
}
usercol.insert_one(adict)
#保存操作记录到资产数据库
RTime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))
adict = {
"UserName": self.username,
"Time": RTime,
"Operation": "EditTag"
}
assetcol.insert_one(adict)
self.btn_reviseTag.clicked.disconnect()
self.btn_reviseTag.clicked.connect(lambda: self.reviseTag())
# Todo: 传入预览图路径，加载3d视口资源到3d视口
def setModelWidget(self, path):
file = os.path.dirname(path)
tex_type = [
'albedo', 'bump', 'roughness', 'specular', 'opacity', 'normal',
'displacement'
]
data = {} #存储贴图数据
for name in os.listdir(file):
if name.split(".")[-1] == "obj" or name.split(".")[-1] == "fbx":
modelpath = os.path.join(file, name).replace("\\", "/")
data["model"] = modelpath
elif name.split(".")[-1] == "jpg":
for typ in tex_type:
_typ = "_%s." % typ
if _typ in name.lower():
data[typ] = os.path.join(file, name).replace("\\", "/")
break
import json
# self.model_widget.layout().addWidget(MLabel(modelpath))
data = json.dumps(data)
print data
self.model_widget.loadAsset(data)
# 弹出信息提示窗口
def slot_show_message(self, func, config):
func(config, parent=self.parent())
class editUserWinodw(QWidget):
"""
编辑用户信息窗口类
"""
editUser_signal = QtCore.Signal() #确认头像信号
def __init__(self, username):
super(editUserWinodw, self).__init__()
self.setWindowModality(Qt.ApplicationModal)
self.setMinimumSize(Data.getWindowHeight() / 1.5,
Data.getWindowHeight() / 3)
# self.setMinimumSize(Data.getWindowWidth() / 3, Data.getWindowHeight() / 3)
# self.setMaximumSize(Data.getWindowWidth() / 3, Data.getWindowHeight() / 3)
self.username = username
self.key = None
self.wchp = False
self.wcun = False
self.wcpw = False
self.wcid = False
# 设置窗口名称
self.setWindowTitle(u"用户窗口")
self.tab = MLineTabWidget()
widget = QWidget()
widget_child = QWidget()
widget_child_2 = QWidget()
widget.setLayout(QHBoxLayout())
widget_child.setLayout(QVBoxLayout())
widget_child_2.setLayout(QVBoxLayout())
self.label_headProfile = MLabel()
self.label_headProfile.setAlignment(Qt.AlignHCenter)
self.btn_changeHead = MPushButton(u'选择新头像')
widget_child.layout().addWidget(self.label_headProfile)
widget_child.layout().addWidget(self.btn_changeHead)
# widget_child.layout().addStretch()
self.let_username = MLineEdit(text='username')
tool_button = MLabel(text=u'用户名').mark().secondary()
tool_button.setAlignment(Qt.AlignCenter)
tool_button.setFixedWidth(80)
self.let_username.set_prefix_widget(tool_button)
widget_child_2.layout().addWidget(self.let_username)
self.let_ID = MLineEdit(text='identity')
self.btn_changeID = MPushButton(text=u'修改身份').primary()
self.btn_changeID.setFixedWidth(80)
self.let_ID.set_suffix_widget(self.btn_changeID)
widget_child_2.layout().addWidget(self.let_ID)
self.let_key = MLineEdit(text='')
tool_button = MLabel(text=u'密钥').mark().secondary()
tool_button.setAlignment(Qt.AlignCenter)
tool_button.setFixedWidth(80)
self.let_key.set_prefix_widget(tool_button)
widget_child_2.layout().addWidget(self.let_key)
self.let_password = MLineEdit(text='***********')
self.btn_changePassword = MPushButton(text=u'修改密码').primary()
self.btn_changePassword.setFixedWidth(80)
self.let_password.set_suffix_widget(self.btn_changePassword)
widget_child_2.layout().addWidget(self.let_password)
self.btn_ok = MPushButton(u'确定').large().primary()
self.btn_cancel = MPushButton(u'取消').large().primary()
layout = QHBoxLayout()
layout.addWidget(self.btn_ok)
layout.addWidget(self.btn_cancel)
widget_child_2.layout().addLayout(layout)
widget.layout().addWidget(widget_child)
widget.layout().addWidget(widget_child_2)
self.tab.add_tab(widget, u'用户信息')
widget2 = QWidget()
self.ui = loadUi(file_path + r"\res\UI\EditUserWindow.ui")
self.ui.setParent(widget2)
widget2.setLayout(QVBoxLayout())
widget2.layout().addWidget(self.ui)
self.tableWidget_operationNode = self.ui.findChild(
QTableWidget, "tableWidget_operationNode")
self.tableWidget_operationNode.setStyleSheet(Data.getQSS())
setSectionResizeMode(self.tableWidget_operationNode.horizontalHeader(),
QHeaderView.Stretch) # 自适应
# widget2.layout().addSpacing(100)
self.tab.add_tab(widget2, u'操作记录')
btn_layout = QHBoxLayout()
main_lay = QVBoxLayout()
main_lay.addSpacing(20)
main_lay.addWidget(self.tab)
main_lay.addWidget(MDivider(u''))
main_lay.addLayout(btn_layout)
main_lay.addSpacing(20)
self.setLayout(main_lay)
dayu_theme.background_color = "#262626"
dayu_theme.apply(self)
# 设置默认值
self.tableWidget_operationNode.setHorizontalHeaderLabels(
[u'操作', u'文件名', u'时间'])
# 设置资产操作记录表：
colUser = userdb[self.username]
userlist = colUser.find({}, {"FileName": 1, "Operation": 1, "Time": 1})
i = 0
for xdir in userlist:
if "Operation" in xdir:
str1 = xdir["Operation"]
newItem1 = QTableWidgetItem(str1)
self.tableWidget_operationNode.setItem(i, 0, newItem1)
if "FileName" in xdir:
str2 = xdir["FileName"]
newItem2 = QTableWidgetItem(str2)
self.tableWidget_operationNode.setItem(i, 1, newItem2)
if "Time" in xdir:
str3 = xdir["Time"]
newItem3 = QTableWidgetItem(str3)
self.tableWidget_operationNode.setItem(i, 2, newItem3)
i += 1
#设置默认值
pixmap = QtGui.QPixmap(hpPath + "\\" + self.username + ".jpg")
self.label_headProfile.setPixmap(pixmap)
self.let_username.setText(self.username)
self.let_username.setReadOnly(True) #只读
self.let_ID.setReadOnly(True) # 只读
self.let_password.setReadOnly(True) # 只读
self.let_password.setEchoMode(QLineEdit.Password) #输入密码形式
# 从数据库提取ID
for x in colUser.find({"_id": "UserID"}, {"UserID": 1}):
ID = x["UserID"]
self.let_ID.setText(ID) # 显示身份
self.let_password.setText("**********")
#链接信号与槽
self.btn_changeHead.clicked.connect(lambda: self.editHeadProfile())
self.btn_ok.clicked.connect(lambda: self.ok())
self.btn_cancel.clicked.connect(lambda: self.cancel())
self.btn_changePassword.clicked.connect(lambda: self.changePassword())
self.btn_changeID.clicked.connect(lambda: self.setID())
#修改头像
def editHeadProfile(self):
dialog = QFileDialog()
dialog.setStyleSheet(Data.getQSS())
#根据当前所在文件目录，设置默认打开文件格式
dialog.setNameFilter(u"图片文件(*.jpg *.png *.jpeg);;")
#加载对应的文件
dialog.setFileMode(QFileDialog.ExistingFiles)
dialog.setViewMode(QFileDialog.Detail)
if dialog.exec_():
imgPaths = dialog.selectedFiles()
imgPath = imgPaths[0]
self.editHeadProfileWin = headProfieWindow(imgPath)
self.editHeadProfileWin.show()
self.editHeadProfileWin.hPOK_signal.connect(
lambda: self.setPicture()) #选中图片，更新头像
self.editHeadProfileWin.hPOK_signal.connect(lambda: self.setWCHP())
#设置头像图片
def setPicture(self):
pixmap = QtGui.QPixmap(file_path + r"\res\headPortrial\buffer.jpg")
self.label_headProfile.setPixmap(pixmap)
def ok(self):
# 保存头像
if self.wchp:
saveHeadPorfile(self.username,
file_path + r"\res\headPortrial\buffer.jpg")
if self.wcpw:
self.setPassword()
self.saveChangePasswordNode(self.username)
if self.wcid:
colUser = userdb[self.username]
# 从数据库提取ID
for x in colUser.find({"_id": "UserID"}, {"UserID": 1}):
ID = x["UserID"]
oldD = {"UserID": ID}
newD = {"$set": {"UserID": self.let_ID.text()}}
x = colUser.update_many(oldD, newD)
self.editUser_signal.emit() #发射确认修改信号
self.slot_show_message(MMessage.success, (u'成功修改用户信息'))
self.close()
def cancel(self):
self.close()
def setWCHP(self):
self.wchp = True
def changePassword(self):
self.let_password.setReadOnly(False)
self.wcpw = True
def setPassword(self):
newPassword = self.let_password.text()
if not newPassword:
msgBox = QMessageBox()
msgBox.setText(u'请输入密码！')
msgBox.exec_()
return 0
colUser = userdb[self.username]
# 从数据库提取密码
for x in colUser.find({"_id": "Password"}, {"Password": 1}):
if "Password" in x:
DBpassword = x["Password"]
oldD = {"Password": DBpassword}
newD = {"$set": {"Password": newPassword}}
x = colUser.update_many(oldD, newD)
def setID(self):
self.key = self.let_key.text()
if self.key == "IDO":
if self.let_ID.text() == u"普通用户":
self.let_ID.setText(u"管理员")
elif self.let_ID.text() == u"管理员":
self.let_ID.setText(u"普通用户")
self.wcid = True
def saveChangePasswordNode(self, username):
RTime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))
# 将浏览信息存入用户数据库
col = userdb[username]
dict = {"Operation": "ChangePassword", "Time": RTime}
col.insert_one(dict)
# def changeUsername(self):
# self.let_username.setReadOnly(False)
# self.wcun = True
# def setUsername(self):
# newname = self.let_username.text()
# collist = userdb.list_collection_names()
# if newname in collist:
# print("The username already exists")
# return 0
# 弹出信息提示窗口
def slot_show_message(self, func, config):
func(config, parent=self)
class MSectionItem(QWidget):
sig_context_menu = Signal(object)
def __init__(self,
title='',
expand=False,
widget=None,
closeable=False,
parent=None):
super(MSectionItem, self).__init__(parent)
self._central_widget = None
self.setAttribute(Qt.WA_StyledBackground)
self.title_label = MLabel(parent=self)
self.expand_icon = MLabel(parent=self)
self.expand_icon.setSizePolicy(QSizePolicy.Minimum,
QSizePolicy.Minimum)
self._close_button = MToolButton().icon_only().tiny().svg(
'close_line.svg')
self._close_button.clicked.connect(self.close)
header_lay = QHBoxLayout()
header_lay.addWidget(self.expand_icon)
header_lay.addWidget(self.title_label)
header_lay.addStretch()
header_lay.addWidget(self._close_button)
self.header_widget = QWidget(parent=self)
self.header_widget.setAttribute(Qt.WA_StyledBackground)
self.header_widget.setObjectName('title')
self.header_widget.setLayout(header_lay)
self.header_widget.setSizePolicy(QSizePolicy.Minimum,
QSizePolicy.Minimum)
self.header_widget.setCursor(Qt.PointingHandCursor)
self.title_label.setCursor(Qt.PointingHandCursor)
self.header_widget.installEventFilter(self)
self.title_label.installEventFilter(self)
self.content_widget = QWidget(parent=self)
self.content_layout = QHBoxLayout()
self.content_widget.setLayout(self.content_layout)
self.main_lay = QVBoxLayout()
self.main_lay.setContentsMargins(0, 0, 0, 0)
self.main_lay.setSpacing(0)
self.main_lay.addWidget(self.header_widget)
self.main_lay.addWidget(self.content_widget)
self.setLayout(self.main_lay)
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Minimum)
self.setMouseTracking(True)
self.set_title(title)
self.set_closeable(closeable)
if widget:
self.set_content(widget)
self.set_expand(expand)
def set_content(self, widget):
if self._central_widget:
self.content_layout.removeWidget(self._central_widget)
self._central_widget.close()
self.content_layout.addWidget(widget)
self._central_widget = widget
def get_content(self):
return self._central_widget
def set_closeable(self, value):
self.setProperty('closeable', value)
def _set_closeable(self, value):
self.content_widget.setVisible(value)
self._close_button.setVisible(value)
def set_expand(self, value):
self.setProperty('expand', value)
def _set_expand(self, value):
self.content_widget.setVisible(value)
self.expand_icon.setPixmap(
MPixmap('down_line.svg' if value else 'right_line.svg').
scaledToHeight(12))
def set_title(self, value):
self.setProperty('title', value)
def _set_title(self, value):
self.title_label.setText(value)
def eventFilter(self, widget, event):
if widget in [self.header_widget, self.title_label]:
if event.type() == QEvent.MouseButtonRelease:
self.set_expand(not self.property('expand'))
return super(QWidget, self).eventFilter(widget, event)
class headProfieWindow(QWidget):
"""
头像编辑窗口
"""
hPOK_signal = QtCore.Signal() #确认头像信号
def __init__(self, imgPath):
super(headProfieWindow, self).__init__()
# 加载ui,并设置ui界面
#self.ui = loadUi(file_path + r"\res\UI\HeadProfileWindow.ui")
self.setWindowModality(Qt.ApplicationModal)
self.setWindowTitle(u"头像编辑器")
# self.ui = loadUi(file_path + r"\res\UI\HeadProfileWindow.ui")
# self.ui.setParent(self)
self.imgPath = imgPath
self.setLayout(QVBoxLayout())
self.label_view = MLabel()
self.label_view.setAlignment(Qt.AlignCenter)
self.layout().addWidget(self.label_view)
self.layout().addWidget(MDivider(u'缩放'))
self.hSlider_scale = MSlider(Qt.Horizontal)
self.layout().addWidget(self.hSlider_scale)
self.layout().addWidget(MDivider(u'垂直位移'))
self.hSlider_moveH = MSlider(Qt.Horizontal)
self.layout().addWidget(self.hSlider_moveH)
self.layout().addWidget(MDivider(u'水平位移'))
self.hSlider_moveV = MSlider(Qt.Horizontal)
self.layout().addWidget(self.hSlider_moveV)
self.layout().addStretch()
self.btn_save_image = MPushButton(u'保存图片').large().primary()
self.layout().addWidget(self.btn_save_image)
dayu_theme.apply(self)
self.setMinimumSize(Data.getWindowWidth() / 4.5,
Data.getWindowHeight() / 1.7)
self.setMaximumSize(Data.getWindowWidth() / 4.5,
Data.getWindowHeight() / 1.7)
self.layout().setContentsMargins(20, 40, 20, 40)
Data.setWindowCenter(self)
self.hSlider_scale.setRange(0, 100)
self.hSlider_moveH.setRange(0, 100)
self.hSlider_moveV.setRange(0, 100)
self.hSlider_scale.setValue(0) #初始值
self.hSlider_moveH.setValue(50)
self.hSlider_moveV.setValue(50)
#原图大小
img = cv2.imread(self.imgPath)
self.oiX = img.shape[0]
self.oiY = img.shape[1]
#获取初始值
self.scale = self.hSlider_scale.value()
self.mh = self.hSlider_moveH.value()
self.mv = self.hSlider_moveV.value()
#设置初始头像
self.editPicture(self.scale, self.mh, self.mv)
self.setPicteruView()
# #链接信号与槽
self.hSlider_scale.valueChanged.connect(lambda: self.scaleEdit())
self.hSlider_moveH.valueChanged.connect(lambda: self.moveHEdit())
self.hSlider_moveV.valueChanged.connect(lambda: self.moveVEdit())
self.btn_save_image.clicked.connect(lambda: self.ok())
def editPicture(self, sn, mhn, mvn):
img = cv2.imread(self.imgPath) #每一次都是用原图编辑
flag = min(img.shape[0], img.shape[1])
sx = (flag - 150) * 0.01 * sn #缩放的变化量
hn = (img.shape[0] - 150) * 0.01 * (mhn - 50) * 0.01 * sn #水平移动变化量
vn = (img.shape[1] - 150) * 0.01 * (mvn - 50) * 0.01 * sn #垂直移动变化量
if (((img.shape[0] - flag) + sx + hn) // 2) < 0 or ((
(img.shape[1] - flag) + sx) // 2) < 0:
newimg = img[(int((img.shape[0] - flag) + sx) //
2):(int((img.shape[0] + flag) - sx) // 2),
(int((img.shape[1] - flag) + sx) //
2):(int((img.shape[1] + flag) - sx) // 2)]
else:
newimg = img[(int((img.shape[0] - flag) + sx + hn) //
2):(int((img.shape[0] + flag) - sx + hn) // 2),
(int((img.shape[1] - flag) + sx + vn) //
2):(int((img.shape[1] + flag) - sx + vn) // 2)]
# newimg = img[(int((img.shape[0] - flag) + sx) // 2): (int((img.shape[0] + flag) - sx) // 2),
# (int((img.shape[1] - flag) + sx) // 2):(int((img.shape[1] + flag) - sx) // 2)]
newimg = cv2.resize(newimg, (150, 150), interpolation=cv2.INTER_AREA)
cv2.imwrite(hpPath + "\\" + "buffer.jpg", newimg) # 将图片作为缓存存入指定文件内
def setPicteruView(self):
imgPath = hpPath + "\\" + "buffer.jpg"
pixmap = QtGui.QPixmap(imgPath)
self.label_view.setPixmap(pixmap)
def scaleEdit(self):
self.scale = self.hSlider_scale.value()
self.editPicture(self.scale, self.mh, self.mv)
self.setPicteruView()
def moveHEdit(self):
self.mh = self.hSlider_moveH.value()
self.editPicture(self.scale, self.mh, self.mv)
self.setPicteruView()
def moveVEdit(self):
self.mv = self.hSlider_moveV.value()
self.editPicture(self.scale, self.mh, self.mv)
self.setPicteruView()
def ok(self):
self.hPOK_signal.emit()
self.close()
