def build_menu_by_recursive(root: QTreeWidgetItem, menu_dict: dict):
for key in menu_dict:
item = QTreeWidgetItem()
item_font = QFont()
item_font.setPointSize(
self.config.getint('master', 'master_item_font_size'))
try:
text = str(menu_dict[key]['text'])
item.setText(0, text)
item.setFont(0, item_font)
except KeyError:
pass
try:
descript = str(menu_dict[key]['descript'])
item.setToolTip(0, descript)
except KeyError:
pass
try:
scl = str(menu_dict[key]['scl'])
item.setText(2, scl)
except KeyError:
pass
try:
children = menu_dict[key]['children']
build_menu_by_recursive(item, children)
except KeyError:
pass
try:
expanded = menu_dict[key]['expanded']
item.setExpanded(expanded)
except KeyError:
pass
root.addChild(item)
def update_tree(self, value):
if value != None:
self.tree_sem.acquire()
current_addr_space = ''
for region in value:
parent_node = self.tree
parent = region['parent']
if parent != '':
root = self.tree.invisibleRootItem()
for i in range(root.childCount()):
if root.child(i).text(0) == current_addr_space:
root = root.child(i)
break
parent_node = self.find(parent, root)
else:
current_addr_space = region['name']
node = QTreeWidgetItem(parent_node)
node.setText(0, region['name'])
start = region['start']
end = region['end']
if start < 0:
start = start + (1 << constants['bits'])
if end < 0:
end = end + (1 << constants['bits'])
node.setText(1, f'{start:016x}')
node.setText(2, f'{end:016x}')
node.setFont(0, QFont('Courier New'))
node.setFont(1, QFont('Courier New'))
node.setFont(2, QFont('Courier New'))
self.tree_sem.release()
def _add_table_item(self, tablename, item, parent):
for detail in self.db.item_detail(tablename):
if detail.pk:
color = Qt.darkGreen
elif detail.notnull:
color = Qt.black
else:
color = Qt.darkGray
child = QTreeWidgetItem((detail.name, detail.type.upper()))
child.setForeground(0, QBrush(color))
font = child.font(1)
font.setPointSize(max(6, font.pointSize() - 1))
child.setFont(1, font)
item.addChild(child)
def handle_dict_to_tree(self, root):
""" change json dict to item """
if self.tree_top_item:
root_item = self.window.treeWidget.invisibleRootItem()
root_item.removeChild(self.tree_top_item)
self.tree_top_item = None
stack = []
item_stack = []
current_stack = 0
stack.append(root)
self.total_time = root["total_time"]
while stack:
current = stack.pop()
if current is None:
current_stack = current_stack - 1
continue
while len(item_stack) > current_stack:
item_stack.pop()
item = QTreeWidgetItem(None, [
current["function_name"],
str(current["count"]),
str(current["total_time"]),
str(current["self_time"]),
str(current["children_time"])
])
current_total_time = current["total_time"]
brush = get_brush(current_total_time, self.total_time)
item.setBackground(0, brush)
for column_index in [1, 2, 3, 4]:
item.setTextAlignment(column_index, Qt.AlignRight)
item.setBackground(column_index, brush)
item.setFont(column_index, self.mono_space_font)
if current_stack == 0:
self.window.treeWidget.addTopLevelItem(item)
self.tree_top_item = item
else:
item_stack[-1].addChild(item)
item_stack.append(item)
if "children" not in current:
continue
stack.append(None)
current_stack = current_stack + 1
for child in current["children"][::-1]:
stack.append(child)
def add_list_to_view(self):
""" add list to view """
self.window.listWidget.invisibleRootItem().takeChildren()
list_data = []
for value in self.list_dict.values():
list_data.append(value)
for data in list_data:
item_data = [
data["function_name"], (data["count"]), (data["total_time"]),
(data["self_time"]), (data["children_time"])
]
item = QTreeWidgetItem(None, item_data)
for column_index in [1, 2, 3, 4]:
item.setTextAlignment(column_index, Qt.AlignRight)
item.setBackground(
column_index,
get_brush(item_data[column_index], self.total_time))
item.setFont(column_index, self.mono_space_font)
item.setData(column_index, Qt.DisplayRole,
item_data[column_index])
self.window.listWidget.addTopLevelItem(item)
self.window.listWidget.sortItems(2, Qt.DescendingOrder)
def build_toplevel_menu(self, surpac_scl_cfg: str):
# Tree递归构建
def build_menu_by_recursive(root: QTreeWidgetItem, menu_dict: dict):
for key in menu_dict:
item = QTreeWidgetItem()
item_font = QFont()
item_font.setPointSize(
self.config.getint('master', 'master_item_font_size'))
try:
text = str(menu_dict[key]['text'])
item.setText(0, text)
item.setFont(0, item_font)
except KeyError:
pass
try:
descript = str(menu_dict[key]['descript'])
item.setToolTip(0, descript)
except KeyError:
pass
try:
scl = str(menu_dict[key]['scl'])
item.setText(2, scl)
except KeyError:
pass
try:
children = menu_dict[key]['children']
build_menu_by_recursive(item, children)
except KeyError:
pass
try:
expanded = menu_dict[key]['expanded']
item.setExpanded(expanded)
except KeyError:
pass
root.addChild(item)
surpac_scl_encoding = self.config.get('master', 'surpac_scl_encoding')
menus = []
with open(file=surpac_scl_cfg, encoding=surpac_scl_encoding) as _f:
menu_dict = yaml.load(_f, yaml.loader.FullLoader)
for key in menu_dict:
menu_toplevel_item = QTreeWidgetItem()
menu_toplevel_item_font = QFont()
menu_toplevel_item_font.setPointSize(
self.config.getint('master', 'master_root_font_size'))
try:
text = str(menu_dict[key]['text'])
menu_toplevel_item.setText(0, text)
menu_toplevel_item.setFont(0, menu_toplevel_item_font)
except KeyError:
pass
try:
descript = str(menu_dict[key]['descript'])
menu_toplevel_item.setToolTip(0, descript)
except KeyError:
pass
try:
scl = str(menu_dict[key]['scl'])
menu_toplevel_item.setText(2, scl)
except KeyError:
pass
try:
expanded = menu_dict[key]['expanded']
menu_toplevel_item.setExpanded(expanded)
except KeyError:
pass
try:
children = menu_dict[key]['children']
build_menu_by_recursive(menu_toplevel_item, children)
except KeyError:
pass
menus.append(menu_toplevel_item)
return menus
def __init__(self, parent=None):
super(ToolTree, self).__init__(parent)
group_names = []
tool_names = []
tool_infos = []
tool_progress = [
] # 0 = da fare, 1 = iniziato, 2 = funzionante, 3 = completo
# [0]
group_names.append(self.tr('[General]'))
tool_names.append([
self.tr('Original Image'),
self.tr('File Digest'),
self.tr('Hex Editor'),
self.tr('Similar Search')
])
tool_infos.append([
self.tr(
'Display the unaltered reference image for visual inspection'),
self.
tr('Retrieve physical file information, crypto and perceptual hashes'
),
self.
tr('Open an external hexadecimal editor to show and edit raw bytes'
),
self.
tr('Browse online search services to find visually similar images')
])
tool_progress.extend([3, 3, 2, 2])
# [1]
group_names.append(self.tr('[Metadata]'))
tool_names.append([
self.tr('Header Structure'),
self.tr('EXIF Full Dump'),
self.tr('Thumbnail Analysis'),
self.tr('Geolocation data')
])
tool_infos.append([
self.
tr('Dump the file header structure and display an interactive view'
),
self.
tr('Scan through file metadata and gather all available information'
),
self.tr(
'Extract optional embedded thumbnail and compare with original'
),
self.
tr('Retrieve optional geolocation data and show it on a world map')
])
tool_progress.extend([3, 3, 3, 2])
# [2]
group_names.append(self.tr('[Inspection]'))
tool_names.append([
self.tr('Enhancing Magnifier'),
self.tr('Channel Histogram'),
self.tr('Global Adjustments'),
self.tr('Reference Comparison')
])
tool_infos.append([
self.
tr('Magnifying glass with enhancements for better identifying forgeries'
),
self.
tr('Display single color channels or RGB composite interactive histogram'
),
self.
tr('Apply standard image adjustments (brightness, hue, saturation, ...)'
),
self.
tr('Open a synchronized double view for comparison with another picture'
)
])
tool_progress.extend([3, 3, 3, 3])
# [3]
group_names.append(self.tr('[Detail]'))
tool_names.append([
self.tr('Luminance Gradient'),
self.tr('Echo Edge Filter'),
self.tr('Wavelet Threshold'),
self.tr('Correlation Plot')
])
tool_infos.append([
self.
tr('Analyze horizontal/vertical brightness variations across the image'
),
self.
tr('Use derivative filters to reveal artificial out-of-focus regions'
),
self.
tr('Reconstruct image with different wavelet coefficient thresholds'
),
self.
tr('Exploit spatial correlation patterns among neighboring pixels')
])
tool_progress.extend([3, 3, 3, 0])
# [4]
group_names.append(self.tr('[Colors]'))
tool_names.append([
self.tr('RGB/HSV Plots'),
self.tr('Space Conversion'),
self.tr('PCA Projection'),
self.tr('Pixel Statistics')
])
tool_infos.append([
self.
tr('Display interactive 2D and 3D plots of RGB and HSV pixel values'
),
self.
tr('Convert RGB channels into HSV/YCbCr/Lab/Luv/CMYK/Gray spaces'),
self.tr(
'Use color PCA to project pixel onto most salient components'),
self.tr(
'Compute minimum/maximum/average RGB values for every pixel')
])
tool_progress.extend([3, 3, 3, 3])
# [5]
group_names.append(self.tr('[Noise]'))
tool_names.append([
self.tr('Noise Estimation'),
self.tr('Min/Max Deviation'),
self.tr('Frequency Split'),
self.tr('Bit Plane Values')
])
tool_infos.append([
self.tr(
'Estimate different kind of noise components of the image'),
self.
tr('Highlight pixels deviating from block-based min/max statistics'
),
self.
tr('Split image luminance into high and low frequency components'),
self.
tr('Show individual bit planes to find inconsistent noise patterns'
)
])
tool_progress.extend([3, 3, 3, 3])
# [6]
group_names.append(self.tr('[JPEG]'))
tool_names.append([
self.tr('Error Level Analysis'),
self.tr('Quality Estimation'),
self.tr('Multiple Compression'),
self.tr('DCT Dimples Map')
])
tool_infos.append([
self.
tr('Show pixel-level difference against fixed compression levels'),
self.
tr('Extract quantization tables and estimate last saved JPEG quality'
),
self.
tr('Use residuals to detect multiple compressions at different levels'
),
self.
tr('Analyze periodic quantization artifacts introduced by devices')
])
tool_progress.extend([3, 3, 1, 0])
# [7]
group_names.append(self.tr('[Tampering]'))
tool_names.append([
self.tr('Contrast Enhancement'),
self.tr('Copy-Move Forgery'),
self.tr('Image Resampling'),
self.tr('Composite Splicing')
])
tool_infos.append([
self.tr(
'Analyze color distributions to detect contrast enhancements'),
self.tr(
'Use invariant feature descriptors to detect cloned regions'),
self.
tr('Estimate 2D pixel interpolation for detecting resampling traces'
),
self.tr(
'Exploit DCT statistics for automatic splicing zone detection')
])
tool_progress.extend([3, 3, 0, 0])
# [8]
group_names.append(self.tr('[Various]'))
tool_names.append([
self.tr('Median Filtering'),
self.tr('Illuminant Map'),
self.tr('PRNU Identification'),
self.tr('Stereogram Decoder')
])
tool_infos.append([
self.tr(
'Detect processing traces left by nonlinear median filtering'),
self.tr(
'Estimate scene local light direction on estimated 3D surfaces'
),
self.
tr('Exploit sensor pattern noise introduced by different cameras'),
self.
tr('Decode 3D images concealed inside crossed-eye autostereograms')
])
tool_progress.extend([0, 0, 0, 3])
count = 0
for i, group in enumerate(group_names):
group_item = QTreeWidgetItem()
group_item.setText(0, group)
font = group_item.font(0)
font.setBold(True)
group_item.setFont(0, font)
group_item.setData(0, Qt.UserRole, False)
group_item.setIcon(0, QIcon('icons/{}.svg'.format(i)))
for j, tool in enumerate(tool_names[i]):
tool_item = QTreeWidgetItem(group_item)
tool_item.setText(0, tool)
tool_item.setData(0, Qt.UserRole, True)
tool_item.setData(0, Qt.UserRole + 1, i)
tool_item.setData(0, Qt.UserRole + 2, j)
tool_item.setToolTip(0, tool_infos[i][j])
if tool_progress[count] == 0:
modify_font(tool_item, italic=True)
count += 1
self.addTopLevelItem(group_item)
self.expandAll()
self.setColumnCount(1)
self.header().setVisible(False)
self.setMaximumWidth(300)
self.version = '{:.2f}a'.format(sum(tool_progress) / (count * 3))
def __init__(self, parent=None):
super(ToolTree, self).__init__(parent)
group_names = []
tool_names = []
tool_infos = []
tool_progress = [
] # 0 = da fare, 1 = iniziato, 2 = funzionante, 3 = completo
# [0]
group_names.append(self.tr('[General]'))
tool_names.append([
self.tr('Original Image'),
self.tr('File Digest'),
self.tr('Hex Editor'),
self.tr('Reverse Search')
])
tool_infos.append([
self.tr(
'Display the unaltered reference image for visual inspection'),
self.
tr('Retrieve file information and compute many hashes and ballistics'
),
self.
tr('Open an external hexadecimal editor to show and edit raw bytes'
),
self.tr(
'Use online search services to find visually similar images')
])
tool_progress.extend([3, 3, 2, 2])
# [1]
group_names.append(self.tr('[Metadata]'))
tool_names.append([
self.tr('Header Structure'),
self.tr('Metadata Extraction'),
self.tr('Thumbnail Analysis'),
self.tr('Geolocation data')
])
tool_infos.append([
self.
tr('Dump the physical EXIF structure and display an interactive view'
),
self.
tr('Scan through file metadata and gather all available information'
),
self.tr(
'Extract optional embedded thumbnail and compare with original'
),
self.
tr('Retrieve optional geo-location data and show it on a world map'
)
])
tool_progress.extend([3, 3, 3, 2])
# [2]
group_names.append(self.tr('[Inspection]'))
tool_names.append([
self.tr('Enhancing Magnifier'),
self.tr('Reference Comparison'),
self.tr('Global Adjustments'),
self.tr('Fourier Transform')
])
tool_infos.append([
self.
tr('Use a loupe with visual enhancement for better identifying forgeries'
),
self.
tr('Open a synchronized double view to compare two different pictures'
),
self.
tr('Apply standard adjustments (contrast, brightness, hue, saturation)'
),
self.
tr('Compute amplitude and phase components of the 2D Fourier Transform'
)
])
tool_progress.extend([2, 2, 2, 3])
# [3]
group_names.append(self.tr('[JPEG]'))
tool_names.append([
self.tr('Quality Estimation'),
self.tr('Error Level Analysis'),
self.tr('Multiple Compression'),
self.tr('DCT Dimples Map')
])
tool_infos.append([
self.
tr('Extract quantization tables and estimate last saved JPEG quality'
),
self.
tr('Show pixel-level difference against different compression levels'
),
self.
tr('Use residuals to detect multiple compressions at different levels'
),
self.
tr('Analyze periodic quantization artifacts to detect manipulations'
)
])
tool_progress.extend([2, 3, 1, 0])
# [4]
group_names.append(self.tr('[Colors]'))
tool_names.append([
self.tr('RGB/HSV Plots'),
self.tr('PCA Projection'),
self.tr('Pixel Statistics'),
self.tr('Space Conversion')
])
tool_infos.append([
self.tr(
'Display interactive 2D and 3D plots of RGB and HSV pixel data'
),
self.
tr('Use color PCA to project RGB values onto reduced vector spaces'
),
self.tr(
'Compute minimum/maximum/average RGB values for every pixel'),
self.
tr('Convert color channels into RGB/HSV/YCbCr/Lab/Luv/CMYK spaces')
])
tool_progress.extend([0, 2, 3, 2])
# [5]
group_names.append(self.tr('[Tonality]'))
tool_names.append([
self.tr('Luminance Gradient'),
self.tr('Echo Edge Filter'),
self.tr('Correlation Plot'),
self.tr('Wavelet Threshold')
])
tool_infos.append([
self.
tr('Analyze horizontal/vertical brightness variations across the image'
),
self.
tr('Use derivative filters to reveal artificial out-of-focus zones'
),
self.tr(
'Exploit spatial correlation patterns among neighboring pixels'
),
self.
tr('Reconstruct image with different wavelet coefficient thresholds'
)
])
tool_progress.extend([2, 3, 0, 0])
# [6]
group_names.append(self.tr('[Noise]'))
tool_names.append([
self.tr('Noise Estimation'),
self.tr('Min/Max Deviation'),
self.tr('Image Bit Planes'),
self.tr('Frequency Separation')
])
tool_infos.append([
self.tr(
'Estimate and visualize gaussian noise components of the image'
),
self.
tr('Highlight pixels deviating from block-based min/max statistics'
),
self.
tr('Visualize bit planes values to find different noise patterns'),
self.
tr('Estimate high/low frequency components of the luminance channel'
)
])
tool_progress.extend([3, 2, 3, 0])
# [7]
group_names.append(self.tr('[Tampering]'))
tool_names.append([
self.tr('Contrast Enhancement'),
self.tr('Region Cloning'),
self.tr('Image Resampling'),
self.tr('Composite Splicing')
])
tool_infos.append([
self.tr(
'Analyze color distribuions to detect contrast enhancements'),
self.
tr('Use feature descriptors for copy/rotate clone area detection'),
self.
tr('Analyze 2D pixel interpolation for detecting resampling traces'
),
self.tr(
'Exploit DCT statistics for automatic splicing zone detection')
])
tool_progress.extend([0, 0, 0, 0])
count = 0
for i, group in enumerate(group_names):
group_item = QTreeWidgetItem()
group_item.setText(0, group)
font = group_item.font(0)
font.setBold(True)
group_item.setFont(0, font)
group_item.setData(0, Qt.UserRole, False)
group_item.setIcon(0, QIcon('icons/{}.svg'.format(i)))
for j, tool in enumerate(tool_names[i]):
tool_item = QTreeWidgetItem(group_item)
tool_item.setText(0, tool)
tool_item.setData(0, Qt.UserRole, True)
tool_item.setData(0, Qt.UserRole + 1, i)
tool_item.setData(0, Qt.UserRole + 2, j)
tool_item.setToolTip(0, tool_infos[i][j])
if tool_progress[count] == 0:
modify_font(tool_item, italic=True)
count += 1
self.addTopLevelItem(group_item)
self.expandAll()
self.setColumnCount(1)
self.header().setVisible(False)
self.setMaximumWidth(300)
self.version = '{:.2f}a'.format(sum(tool_progress) / (count * 3))
def __init__(self, parent=None):
super(ToolTree, self).__init__(parent)
group_names = []
tool_names = []
tool_infos = []
tool_progress = [
] # 0 = da fare, 1 = debug, 2 = funzionante, 3 = completo
# [0]
group_names.append(self.tr("[General]"))
tool_names.append([
self.tr("Original Image"),
self.tr("File Digest"),
self.tr("Hex Editor"),
self.tr("Similar Search")
])
tool_infos.append([
self.tr(
"Display the unaltered reference image for visual inspection"),
self.
tr("Retrieve physical file information, crypto and perceptual hashes"
),
self.
tr("Open an external hexadecimal editor to show and edit raw bytes"
),
self.tr(
"Browse online search services to find visually similar images"
),
])
tool_progress.extend([3, 3, 2, 2])
# [1]
group_names.append(self.tr("[Metadata]"))
tool_names.append([
self.tr("Header Structure"),
self.tr("EXIF Full Dump"),
self.tr("Thumbnail Analysis"),
self.tr("Geolocation data"),
])
tool_infos.append([
self.
tr("Dump the file header structure and display an interactive view"
),
self.
tr("Scan through file metadata and gather all available information"
),
self.tr(
"Extract optional embedded thumbnail and compare with original"
),
self.tr(
"Retrieve optional geolocation data and show it on a world map"
),
])
tool_progress.extend([3, 3, 3, 2])
# [2]
group_names.append(self.tr("[Inspection]"))
tool_names.append([
self.tr("Enhancing Magnifier"),
self.tr("Channel Histogram"),
self.tr("Global Adjustments"),
self.tr("Reference Comparison"),
])
tool_infos.append([
self.
tr("Magnifying glass with enhancements for better identifying forgeries"
),
self.
tr("Display single color channels or RGB composite interactive histogram"
),
self.
tr("Apply standard image adjustments (brightness, hue, saturation, ...)"
),
self.
tr("Open a synchronized double view for comparison with another picture"
),
])
tool_progress.extend([3, 3, 3, 3])
# [3]
group_names.append(self.tr("[Detail]"))
tool_names.append([
self.tr("Luminance Gradient"),
self.tr("Echo Edge Filter"),
self.tr("Wavelet Threshold"),
self.tr("Frequency Split"),
])
tool_infos.append([
self.
tr("Analyze horizontal/vertical brightness variations across the image"
),
self.
tr("Use derivative filters to reveal artificial out-of-focus regions"
),
self.
tr("Reconstruct image with different wavelet coefficient thresholds"
),
self.tr(
"Divide image luminance into high and low frequency components"
),
])
tool_progress.extend([3, 3, 3, 3])
# [4]
group_names.append(self.tr("[Colors]"))
tool_names.append([
self.tr("RGB/HSV Plots"),
self.tr("Space Conversion"),
self.tr("PCA Projection"),
self.tr("Pixel Statistics"),
])
tool_infos.append([
self.
tr("Display interactive 2D and 3D plots of RGB and HSV pixel values"
),
self.
tr("Convert RGB channels into HSV/YCbCr/Lab/Luv/CMYK/Gray spaces"),
self.tr(
"Use color PCA to project pixel onto most salient components"),
self.tr(
"Compute minimum/maximum/average RGB values for every pixel"),
])
tool_progress.extend([3, 3, 3, 3])
# [5]
group_names.append(self.tr("[Noise]"))
tool_names.append([
self.tr("Noise Separation"),
self.tr("Min/Max Deviation"),
self.tr("Bit Plane Values"),
self.tr("PRNU Identification"),
])
tool_infos.append([
self.tr(
"Estimate and extract different kind of image noise components"
),
self.
tr("Highlight pixels deviating from block-based min/max statistics"
),
self.
tr("Show individual bit planes to find inconsistent noise patterns"
),
self.
tr("Exploit sensor pattern noise introduced by different cameras"),
])
tool_progress.extend([3, 3, 3, 0])
# [6]
group_names.append(self.tr("[JPEG]"))
tool_names.append([
self.tr("Quality Estimation"),
self.tr("Error Level Analysis"),
self.tr("Multiple Compression"),
self.tr("JPEG Ghost Maps"),
])
tool_infos.append([
self.
tr("Extract quantization tables and estimate last saved JPEG quality"
),
self.tr(
"Show pixel-wise differences against a fixed compression level"
),
self.tr(
"Use a machine learning model to detect multiple compression"),
self.
tr("Highlight traces of different compressions in difference images"
),
])
tool_progress.extend([3, 3, 0, 0])
# [7]
group_names.append(self.tr("[Tampering]"))
tool_names.append([
self.tr("Contrast Enhancement"),
self.tr("Copy-Move Forgery"),
self.tr("Composite Splicing"),
self.tr("Image Resampling"),
])
tool_infos.append([
self.tr(
"Analyze color distributions to detect contrast enhancements"),
self.tr(
"Use invariant feature descriptors to detect cloned regions"),
self.
tr("Exploit DCT statistics for automatic splicing zone detection"),
self.
tr("Estimate 2D pixel interpolation for detecting resampling traces"
),
])
tool_progress.extend([3, 2, 3, 0])
# [8]
group_names.append(self.tr("[Various]"))
tool_names.append([
self.tr("Median Filtering"),
self.tr("Illuminant Map"),
self.tr("Dead/Hot Pixels"),
self.tr("Stereogram Decoder"),
])
tool_infos.append([
self.tr(
"Detect nonlinear processing traces left by median filtering"),
self.tr(
"Estimate scene local light direction on estimated 3D surfaces"
),
self.tr(
"Detect and fix dead/hot pixels caused by sensor imperfections"
),
self.tr(
"Decode 3D images concealed inside crossed-eye autostereograms"
),
])
tool_progress.extend([2, 0, 0, 3])
count = 0
for i, group in enumerate(group_names):
group_item = QTreeWidgetItem()
group_item.setText(0, group)
font = group_item.font(0)
font.setBold(True)
group_item.setFont(0, font)
group_item.setData(0, Qt.UserRole, False)
group_item.setIcon(0, QIcon(f"icons/{i}.svg"))
for j, tool in enumerate(tool_names[i]):
tool_item = QTreeWidgetItem(group_item)
tool_item.setText(0, tool)
tool_item.setData(0, Qt.UserRole, True)
tool_item.setData(0, Qt.UserRole + 1, i)
tool_item.setData(0, Qt.UserRole + 2, j)
tool_item.setToolTip(0, tool_infos[i][j])
if tool_progress[count] == 0:
modify_font(tool_item, italic=True)
count += 1
self.addTopLevelItem(group_item)
self.expandAll()
self.setColumnCount(1)
self.header().setVisible(False)
self.setMaximumWidth(300)
self.version = f"{sum(tool_progress) / 100:.2f}a"
def __init__(self):
super(TreeWidget, self).__init__()
self.setColumnCount(3)
self.setColumnWidth(0, 80)
self.setColumnWidth(1, 80)
self.setColumnWidth(2, 80)
self.setIndentation(0)
self.setHeaderHidden(True)
self.foreground_font = QFont()
self.foreground_font.setBold(True)
self.normal_font = QFont()
self.normal_font.setBold(False)
self.setExpandsOnDoubleClick(False)
p = QTreeWidgetItem(self)
font = QFont()
font.setPointSize(10)
font.setBold(True)
p.setFont(0, font)
p.setText(0, "父级1父级1父级1父级1父级1父级1")
p.setIcon(0, QIcon("icons/arrow_right.png"))
p.setFirstColumnSpanned(True)
c = QTreeWidgetItem(p)
c.setText(0, "子级1")
c.setText(1, "子级2")
c.setText(2, "子级3")
c.setTextAlignment(0, Qt.AlignCenter)
c.setTextAlignment(1, Qt.AlignCenter)
c.setTextAlignment(2, Qt.AlignCenter)
c.setData(0, Qt.UserRole, "c_00")
c.setData(1, Qt.UserRole, "c_01")
c.setData(2, Qt.UserRole, "c_02")
c = QTreeWidgetItem(p)
c.setText(0, "子级1")
c.setText(1, "子级2")
c.setTextAlignment(0, Qt.AlignCenter)
c.setTextAlignment(1, Qt.AlignCenter)
c.setData(0, Qt.UserRole, "c_03")
c.setData(1, Qt.UserRole, "c_04")
p.addChild(c)
p = QTreeWidgetItem(self)
p.setText(0, "父级2父级2父级2父级2父级2父级2")
p.setIcon(0, QIcon("icons/arrow_right.png"))
p.setFirstColumnSpanned(True)
p.setFont(0, font)
c = QTreeWidgetItem(p)
c.setText(0, "子级2")
c.setText(1, "子级22")
c.setText(2, "子级23")
c.setTextAlignment(0, Qt.AlignCenter)
c.setTextAlignment(1, Qt.AlignCenter)
c.setTextAlignment(2, Qt.AlignCenter)
c.setData(0, Qt.UserRole, "c_10")
c.setData(1, Qt.UserRole, "c_11")
c.setData(2, Qt.UserRole, "c_12")
p.addChild(c)
self.itemClicked.connect(self.click)
self.setFocusPolicy(Qt.NoFocus)
self.setObjectName("tree")
self.setStyleSheet(
"#tree::item:selected{background-color:rgb(255,255,255);color:rgb(180,12,34);}"
"#tree::item:hover{background-color:rgb(255,255,255);color:rgb(10,160,10)}"
)
self.setAnimated(True)
self.setSelectionBehavior(QAbstractItemView.SelectItems)
self.setSelectionMode(QAbstractItemView.SingleSelection)