def add_material(self, material):
mat_path = BrresPath(material=material)
name = mat_path.get_path()
if name not in self.materials:
label = MaterialWidget(self, self, material, mat_path,
self.is_material_removable)
self.materials[name] = label
# label.setFixedWidth(120)
self.grid.addWidget(label, *self.increment_grid())
return True
return False
def dragMoveEvent(self, a0):
data = a0.mimeData()
if self.enable_mat_drag and self.poly and data.hasText() \
and self.handler.locate_material(BrresPath(data.text())):
a0.accept()
else:
a0.ignore()
def set_material(self, material, brres_path=None):
if self.material is not None:
ImageManager.get().unsubscribe(self, self.material.parent.parent)
self.material.unregister(self)
self.material = material
if brres_path is None:
self.brres_path = BrresPath(material=material)
else:
self.brres_path = brres_path
if material is not None:
self.setToolTip(self.brres_path.get_path())
self.text_label.setText(material.name)
self.img_label.setPixmap(QPixmap(None))
self.material.register_observer(self)
ImageManager.get().subscribe(self,
material.parent.parent, material.get_first_layer_name())
else:
self.text_label.setText('Null')
self.setToolTip('Null')
def dropEvent(self, a0):
data = a0.mimeData()
if self.enable_mat_drag and self.poly and data.hasText():
text = data.text()
mat = self.handler.locate_material(BrresPath(text))
# mat = get_material_by_url(text, trace_path=True)
if mat:
a0.accept()
self.poly.material.paste(mat)
# self.poly.set_material(mat)
return
a0.ignore()
def mouseMoveEvent(self, ev):
if not ev.buttons() & Qt.LeftButton:
return
drag = QDrag(self)
mimedata = QMimeData()
mimedata.setText(BrresPath(material=self.material).get_path())
drag.setMimeData(mimedata)
pixmap = QPixmap(self.size())
painter = QPainter(pixmap)
painter.drawPixmap(self.rect(), self.grab())
painter.end()
drag.setPixmap(pixmap)
drag.setHotSpot(ev.pos())
drag.exec_(Qt.CopyAction | Qt.MoveAction)
def remove_material(self, material):
b_path = BrresPath(material=material).get_path()
if b_path in self.materials:
self.brres.models[0].remove_material(material)
widget = self.materials.pop(b_path)
self.grid.removeWidget(widget)
def can_add_material(self, mat_path):
bpath = BrresPath(mat_path)
return mat_path not in self.materials and bpath.trace_path(
)[0] is not None
class MaterialWidget(QWidget, ClipableObserver, ImageObserver):
def on_image_update(self, dir):
self.layer_name = name = self.material.get_first_layer_name()
if name:
self.__update_image(dir, name)
def on_node_update(self, node):
self.on_child_update(node) # redirect
if node is self.material:
if self.handler.should_remove_unused_mats() and not node.is_used():
self.remove_material()
def on_rename_update(self, node, old_name):
self.text_label.setText(node.name)
self.get_brres_path()
def on_child_update(self, child):
name = self.material.get_first_layer_name()
if name != self.layer_name:
self.__update_image(ImageManager.get().get_image_dir(self.material.getBrres()), name)
def __update_image(self, dir, name):
if not update_image(self.img_label, dir, name, scale_width=self.w):
pass
def __init__(self, parent, handler=None, material=None, brres_path=None, removable=False, width=64):
super().__init__(parent)
self.handler = handler
self.__init_ui()
self.layer_name = None
self.material = None
self.removable = removable
self.w = width
if material is not None:
self.set_material(material, brres_path)
self.__init_context_menu()
def __init_ui(self):
self.main_layout = QVBoxLayout(self)
self.text_label = QLabel(self)
self.img_label = QLabel(self)
self.main_layout.addWidget(self.text_label)
self.main_layout.addWidget(self.img_label)
self.setLayout(self.main_layout)
def __del__(self):
if self.material is not None:
self.material.unregister(self)
def rename(self):
if self.material is not None:
current_name = self.material.name
text, ok = QInputDialog.getText(self, 'Rename Node', 'Rename to:', text=current_name)
if ok and text != current_name:
if self.material.parent.get_material_by_name(text) is not None:
AutoFix.get().error('Material with name {} already exists!'.format(text))
return
self.material.rename(text)
def __init_context_menu(self):
self.setContextMenuPolicy(Qt.ActionsContextMenu)
rename = QAction('Rename', self)
rename.triggered.connect(self.rename)
edit_action = QAction('Edit', self)
edit_action.triggered.connect(self.edit_material)
self.addAction(rename)
self.addAction(edit_action)
if self.removable:
remove_action = QAction('Remove', self)
remove_action.triggered.connect(self.remove_material)
self.addAction(remove_action)
def remove_material(self):
ImageManager.get().unsubscribe(self, self.material.parent.parent)
self.handler.on_material_remove(self.material)
self.setParent(None)
def get_brres_path(self):
self.brres_path = BrresPath(material=self.material).get_path()
self.setToolTip(self.brres_path)
return self.brres_path
def edit_material(self):
if self.material is not None:
self.handler.on_material_edit(self.material)
def set_material(self, material, brres_path=None):
if self.material is not None:
ImageManager.get().unsubscribe(self, self.material.parent.parent)
self.material.unregister(self)
self.material = material
if brres_path is None:
self.brres_path = BrresPath(material=material)
else:
self.brres_path = brres_path
if material is not None:
self.setToolTip(self.brres_path.get_path())
self.text_label.setText(material.name)
self.img_label.setPixmap(QPixmap(None))
self.material.register_observer(self)
ImageManager.get().subscribe(self,
material.parent.parent, material.get_first_layer_name())
else:
self.text_label.setText('Null')
self.setToolTip('Null')
def mousePressEvent(self, ev):
if self.handler:
self.handler.on_material_select(self.material)
if ev.button() == Qt.LeftButton:
self.drag_start_position = ev.pos()
def mouseDoubleClickEvent(self, a0):
if self.material is not None:
self.handler.on_material_edit(self.material)
def mouseMoveEvent(self, ev):
if not ev.buttons() & Qt.LeftButton:
return
drag = QDrag(self)
mimedata = QMimeData()
mimedata.setText(BrresPath(material=self.material).get_path())
drag.setMimeData(mimedata)
pixmap = QPixmap(self.size())
painter = QPainter(pixmap)
painter.drawPixmap(self.rect(), self.grab())
painter.end()
drag.setPixmap(pixmap)
drag.setHotSpot(ev.pos())
drag.exec_(Qt.CopyAction | Qt.MoveAction)
def get_brres_path(self):
self.brres_path = BrresPath(material=self.material).get_path()
self.setToolTip(self.brres_path)
return self.brres_path