mirror of
https://github.com/vosen/ZLUDA.git
synced 2025-04-19 16:04:44 +00:00
add a Ez to use GUI
This commit is contained in:
Mattscreative
parent
d704e92c97
commit
1d25762097
2 changed files with 901 additions and 0 deletions
2
zluda_gui/requirements.txt
Normal file
2
zluda_gui/requirements.txt
Normal file
|
|
@ -0,0 +1,2 @@
|
|||
PyQt6==6.6.1
|
||||
requests==2.31.0
|
||||
899
zluda_gui/zluda_gui.py
Normal file
899
zluda_gui/zluda_gui.py
Normal file
|
|
@ -0,0 +1,899 @@
|
|||
import sys
|
||||
import os
|
||||
import subprocess
|
||||
import requests
|
||||
import zipfile
|
||||
import io
|
||||
import platform
|
||||
import tarfile
|
||||
import psutil
|
||||
import time
|
||||
import threading
|
||||
from concurrent.futures import ThreadPoolExecutor
|
||||
from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
|
||||
QHBoxLayout, QPushButton, QLineEdit, QLabel,
|
||||
QFileDialog, QMessageBox, QTextEdit, QFrame,
|
||||
QTabWidget, QProgressBar, QGroupBox, QGridLayout,
|
||||
QCheckBox)
|
||||
from PyQt6.QtCore import Qt, QThread, pyqtSignal, QSize
|
||||
from PyQt6.QtGui import QPalette, QColor, QScreen
|
||||
|
||||
class DownloadThread(QThread):
|
||||
"""Thread for downloading ZLUDA with progress tracking"""
|
||||
progress_signal = pyqtSignal(int)
|
||||
log_signal = pyqtSignal(str)
|
||||
finished_signal = pyqtSignal(bool, str)
|
||||
|
||||
def __init__(self, url, chunk_size=1024*1024):
|
||||
super().__init__()
|
||||
self.url = url
|
||||
self.chunk_size = chunk_size
|
||||
self.stop_download = False
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
self.log_signal.emit(f"Starting download from {self.url}")
|
||||
response = requests.get(self.url, stream=True)
|
||||
response.raise_for_status()
|
||||
|
||||
total_size = int(response.headers.get('content-length', 0))
|
||||
downloaded = 0
|
||||
|
||||
# Create a temporary file to store the download
|
||||
temp_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
temp_file = os.path.join(temp_dir, "zluda_download.tmp")
|
||||
|
||||
with open(temp_file, 'wb') as f:
|
||||
for data in response.iter_content(chunk_size=self.chunk_size):
|
||||
if self.stop_download:
|
||||
self.log_signal.emit("Download cancelled")
|
||||
self.finished_signal.emit(False, "Download cancelled")
|
||||
return
|
||||
|
||||
downloaded += len(data)
|
||||
f.write(data)
|
||||
|
||||
if total_size:
|
||||
progress = int((downloaded / total_size) * 100)
|
||||
self.progress_signal.emit(progress)
|
||||
self.log_signal.emit(f"Download progress: {progress}%")
|
||||
|
||||
self.log_signal.emit("Download completed")
|
||||
self.finished_signal.emit(True, temp_file)
|
||||
|
||||
except Exception as e:
|
||||
self.log_signal.emit(f"Download error: {str(e)}")
|
||||
self.finished_signal.emit(False, str(e))
|
||||
|
||||
def stop(self):
|
||||
self.stop_download = True
|
||||
|
||||
class ProcessMonitor(QThread):
|
||||
"""Thread to monitor process output and GPU usage"""
|
||||
log_signal = pyqtSignal(str)
|
||||
process_ended = pyqtSignal()
|
||||
|
||||
def __init__(self, process, zluda_path, gpu_monitor):
|
||||
super().__init__()
|
||||
self.process = process
|
||||
self.zluda_path = zluda_path
|
||||
self.running = True
|
||||
self.gpu_monitor = gpu_monitor
|
||||
|
||||
def run(self):
|
||||
while self.running and self.process.poll() is None:
|
||||
try:
|
||||
# Check for ZLUDA debug output
|
||||
if platform.system() == "Windows":
|
||||
try:
|
||||
output = self.process.stdout.readline().decode('utf-8', errors='ignore')
|
||||
if output:
|
||||
self.log_signal.emit(f"Debug: {output.strip()}")
|
||||
except:
|
||||
pass
|
||||
else:
|
||||
try:
|
||||
output = self.process.stderr.readline().decode('utf-8', errors='ignore')
|
||||
if output:
|
||||
self.log_signal.emit(f"Debug: {output.strip()}")
|
||||
except:
|
||||
pass
|
||||
|
||||
# Check GPU usage
|
||||
if self.gpu_monitor:
|
||||
try:
|
||||
if platform.system() == "Windows":
|
||||
gpu_info = subprocess.check_output(['nvidia-smi', '--query-gpu=utilization.gpu,memory.used', '--format=csv,noheader,nounits'], stderr=subprocess.DEVNULL)
|
||||
else:
|
||||
gpu_info = subprocess.check_output(['nvidia-smi', '--query-gpu=utilization.gpu,memory.used', '--format=csv,noheader,nounits'], stderr=subprocess.DEVNULL)
|
||||
self.log_signal.emit(f"GPU Usage: {gpu_info.decode().strip()}")
|
||||
except:
|
||||
pass
|
||||
|
||||
time.sleep(1)
|
||||
except:
|
||||
pass
|
||||
|
||||
self.process_ended.emit()
|
||||
|
||||
def stop(self):
|
||||
self.running = False
|
||||
|
||||
class ZLUDA_GUI(QMainWindow):
|
||||
def __init__(self):
|
||||
# Enable high DPI scaling
|
||||
os.environ["QT_ENABLE_HIGHDPI_SCALING"] = "1"
|
||||
os.environ["QT_SCALE_FACTOR"] = "1"
|
||||
os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1"
|
||||
|
||||
super().__init__()
|
||||
self.setWindowTitle("ZLUDA GUI")
|
||||
|
||||
# Get the primary screen
|
||||
screen = QApplication.primaryScreen()
|
||||
if screen:
|
||||
# Get the screen's geometry and scale factor
|
||||
geometry = screen.geometry()
|
||||
scale_factor = screen.devicePixelRatio()
|
||||
|
||||
# Adjust scale factor to be more reasonable
|
||||
scale_factor = min(scale_factor, 1.5) # Cap the scaling
|
||||
|
||||
# Set minimum size scaled according to DPI
|
||||
self.setMinimumSize(900, 700)
|
||||
|
||||
# Calculate the window size based on screen size and scale factor
|
||||
width = int(geometry.width() * 0.7) # 70% of screen width
|
||||
height = int(geometry.height() * 0.7) # 70% of screen height
|
||||
|
||||
# Set the window size
|
||||
self.resize(width, height)
|
||||
|
||||
# Center the window on screen
|
||||
self.move(
|
||||
int((geometry.width() - width) / 2),
|
||||
int((geometry.height() - height) / 2)
|
||||
)
|
||||
|
||||
# Initialize widgets
|
||||
self.zluda_path = QLineEdit()
|
||||
self.app_path = QLineEdit()
|
||||
self.libs_path = QLineEdit()
|
||||
self.download_button = QPushButton("Download ZLUDA")
|
||||
self.run_button = QPushButton("Run Application")
|
||||
self.download_button.clicked.connect(self.download_zluda)
|
||||
self.run_button.clicked.connect(self.run_application)
|
||||
|
||||
# Adjust font sizes based on DPI
|
||||
base_font_size = int(11 * scale_factor)
|
||||
title_font_size = int(18 * scale_factor)
|
||||
button_font_size = int(12 * scale_factor)
|
||||
|
||||
# Update the style sheet with adjusted sizes
|
||||
self.setStyleSheet(f"""
|
||||
/* Main window and widget backgrounds */
|
||||
QMainWindow {{
|
||||
background-color: #1a1a1a;
|
||||
}}
|
||||
QWidget {{
|
||||
background-color: transparent;
|
||||
color: #ffffff;
|
||||
}}
|
||||
|
||||
/* Labels */
|
||||
QLabel {{
|
||||
color: #ffffff;
|
||||
font-size: {base_font_size}px;
|
||||
}}
|
||||
|
||||
/* Group boxes */
|
||||
QGroupBox {{
|
||||
background-color: #212121;
|
||||
border: 1px solid #2a2a2a;
|
||||
border-radius: 6px;
|
||||
margin-top: 12px;
|
||||
padding: 15px;
|
||||
font-size: {base_font_size}px;
|
||||
}}
|
||||
QGroupBox::title {{
|
||||
subcontrol-origin: margin;
|
||||
left: 10px;
|
||||
padding: 3px 6px;
|
||||
color: #00e5ff;
|
||||
font-weight: bold;
|
||||
background-color: #212121;
|
||||
}}
|
||||
|
||||
/* Frames */
|
||||
QFrame {{
|
||||
background-color: #212121;
|
||||
border: 1px solid #2a2a2a;
|
||||
border-radius: 6px;
|
||||
margin: 2px 0;
|
||||
}}
|
||||
|
||||
/* Line edits */
|
||||
QLineEdit {{
|
||||
background-color: #2a2a2a;
|
||||
color: #ffffff;
|
||||
border: 1px solid #333333;
|
||||
border-radius: 4px;
|
||||
padding: 6px 10px;
|
||||
font-size: {base_font_size}px;
|
||||
selection-background-color: #006064;
|
||||
min-height: 24px;
|
||||
}}
|
||||
QLineEdit:focus {{
|
||||
border: 2px solid #00e5ff;
|
||||
background-color: #2d2d2d;
|
||||
}}
|
||||
QLineEdit:hover {{
|
||||
background-color: #2d2d2d;
|
||||
border: 1px solid #404040;
|
||||
}}
|
||||
|
||||
/* Buttons */
|
||||
QPushButton {{
|
||||
border: none;
|
||||
border-radius: 4px;
|
||||
padding: 6px 16px;
|
||||
font-size: {button_font_size}px;
|
||||
font-weight: bold;
|
||||
color: white;
|
||||
background-color: #424242;
|
||||
min-height: 28px;
|
||||
}}
|
||||
QPushButton:hover {{
|
||||
background-color: #4a4a4a;
|
||||
}}
|
||||
QPushButton:pressed {{
|
||||
background-color: #383838;
|
||||
}}
|
||||
QPushButton[cssClass="browse"] {{
|
||||
background-color: #333333;
|
||||
padding: 6px 12px;
|
||||
min-width: 70px;
|
||||
}}
|
||||
QPushButton[cssClass="browse"]:hover {{
|
||||
background-color: #3d3d3d;
|
||||
}}
|
||||
QPushButton[cssClass="browse"]:pressed {{
|
||||
background-color: #2a2a2a;
|
||||
}}
|
||||
|
||||
/* Checkboxes */
|
||||
QCheckBox {{
|
||||
spacing: 6px;
|
||||
color: #ffffff;
|
||||
font-size: {base_font_size}px;
|
||||
min-height: 20px;
|
||||
padding: 2px;
|
||||
}}
|
||||
QCheckBox::indicator {{
|
||||
width: 16px;
|
||||
height: 16px;
|
||||
border-radius: 3px;
|
||||
border: 1px solid #404040;
|
||||
background-color: #2a2a2a;
|
||||
}}
|
||||
QCheckBox::indicator:hover {{
|
||||
border-color: #00e5ff;
|
||||
background-color: #2d2d2d;
|
||||
}}
|
||||
QCheckBox::indicator:checked {{
|
||||
background-color: #00e5ff;
|
||||
border-color: #00e5ff;
|
||||
}}
|
||||
|
||||
/* Tab widget */
|
||||
QTabWidget::pane {{
|
||||
border: 1px solid #2a2a2a;
|
||||
border-radius: 6px;
|
||||
background-color: #212121;
|
||||
top: -1px;
|
||||
}}
|
||||
QTabBar::tab {{
|
||||
background-color: #2a2a2a;
|
||||
color: #b0b0b0;
|
||||
padding: 8px 16px;
|
||||
border-top-left-radius: 4px;
|
||||
border-top-right-radius: 4px;
|
||||
margin-right: 2px;
|
||||
font-size: {base_font_size}px;
|
||||
min-height: 24px;
|
||||
min-width: 80px;
|
||||
}}
|
||||
QTabBar::tab:selected {{
|
||||
background-color: #212121;
|
||||
color: #00e5ff;
|
||||
border: 1px solid #2a2a2a;
|
||||
border-bottom: 2px solid #00e5ff;
|
||||
}}
|
||||
QTabBar::tab:hover:!selected {{
|
||||
background-color: #333333;
|
||||
color: #ffffff;
|
||||
}}
|
||||
|
||||
/* Progress bar */
|
||||
QProgressBar {{
|
||||
border: 1px solid #2a2a2a;
|
||||
border-radius: 4px;
|
||||
background-color: #2a2a2a;
|
||||
text-align: center;
|
||||
font-size: {int(10 * scale_factor)}px;
|
||||
color: white;
|
||||
min-height: 20px;
|
||||
}}
|
||||
QProgressBar::chunk {{
|
||||
background-color: #00e5ff;
|
||||
border-radius: 3px;
|
||||
}}
|
||||
|
||||
/* Text edit */
|
||||
QTextEdit {{
|
||||
background-color: #2a2a2a;
|
||||
color: #ffffff;
|
||||
border: 1px solid #333333;
|
||||
border-radius: 4px;
|
||||
padding: 8px;
|
||||
font-size: {base_font_size}px;
|
||||
line-height: 1.4;
|
||||
selection-background-color: #006064;
|
||||
}}
|
||||
QTextEdit:focus {{
|
||||
border: 2px solid #00e5ff;
|
||||
}}
|
||||
|
||||
/* Scrollbars */
|
||||
QScrollBar:vertical {{
|
||||
border: none;
|
||||
background: #2a2a2a;
|
||||
width: 10px;
|
||||
margin: 2px;
|
||||
border-radius: 5px;
|
||||
}}
|
||||
QScrollBar::handle:vertical {{
|
||||
background: #404040;
|
||||
min-height: 20px;
|
||||
border-radius: 5px;
|
||||
}}
|
||||
QScrollBar::handle:vertical:hover {{
|
||||
background: #4a4a4a;
|
||||
}}
|
||||
QScrollBar::add-line:vertical, QScrollBar::sub-line:vertical {{
|
||||
height: 0;
|
||||
}}
|
||||
QScrollBar::add-page:vertical, QScrollBar::sub-page:vertical {{
|
||||
background: none;
|
||||
}}
|
||||
""")
|
||||
|
||||
# Create main widget and layout
|
||||
main_widget = QWidget()
|
||||
self.setCentralWidget(main_widget)
|
||||
layout = QVBoxLayout(main_widget)
|
||||
layout.setSpacing(12)
|
||||
layout.setContentsMargins(15, 15, 15, 15)
|
||||
|
||||
# Header with title and version
|
||||
header = QHBoxLayout()
|
||||
header.setSpacing(6)
|
||||
|
||||
title = QLabel("ZLUDA GUI")
|
||||
title.setStyleSheet(f"""
|
||||
font-size: {title_font_size}px;
|
||||
font-weight: bold;
|
||||
color: #00e5ff;
|
||||
""")
|
||||
|
||||
version = QLabel("v4.0")
|
||||
version.setStyleSheet(f"""
|
||||
font-size: {button_font_size}px;
|
||||
color: #808080;
|
||||
padding-top: 4px;
|
||||
padding-left: 6px;
|
||||
""")
|
||||
|
||||
header.addWidget(title)
|
||||
header.addWidget(version)
|
||||
header.addStretch()
|
||||
layout.addLayout(header)
|
||||
|
||||
# Create tab widget
|
||||
tabs = QTabWidget()
|
||||
layout.addWidget(tabs)
|
||||
|
||||
# Main tab
|
||||
main_tab = QWidget()
|
||||
main_layout = QVBoxLayout(main_tab)
|
||||
main_layout.setSpacing(12)
|
||||
main_layout.setContentsMargins(15, 15, 15, 15)
|
||||
|
||||
# Debug settings group
|
||||
debug_group = QGroupBox("Debug Settings")
|
||||
debug_layout = QGridLayout(debug_group)
|
||||
debug_layout.setSpacing(10)
|
||||
debug_layout.setContentsMargins(12, 20, 12, 12)
|
||||
|
||||
# Debug checkboxes with tooltips
|
||||
self.zluda_debug = QCheckBox("Enable ZLUDA Debug")
|
||||
self.zluda_debug.setChecked(True)
|
||||
self.zluda_debug.setToolTip("Enable detailed ZLUDA debugging output")
|
||||
|
||||
self.lib_debug = QCheckBox("Enable Library Debug")
|
||||
self.lib_debug.setChecked(True)
|
||||
self.lib_debug.setToolTip("Enable library loading debug information")
|
||||
|
||||
self.gpu_monitor = QCheckBox("Enable GPU Monitoring")
|
||||
self.gpu_monitor.setChecked(True)
|
||||
self.gpu_monitor.setToolTip("Monitor GPU usage while application is running")
|
||||
|
||||
debug_layout.addWidget(self.zluda_debug, 0, 0)
|
||||
debug_layout.addWidget(self.lib_debug, 0, 1)
|
||||
debug_layout.addWidget(self.gpu_monitor, 1, 0)
|
||||
|
||||
# Debug file path
|
||||
debug_file_layout = QHBoxLayout()
|
||||
debug_file_layout.setSpacing(6)
|
||||
|
||||
debug_file_label = QLabel("Debug File:")
|
||||
debug_file_label.setStyleSheet("""
|
||||
color: #ffffff;
|
||||
font-weight: bold;
|
||||
""")
|
||||
|
||||
self.debug_file = QLineEdit("/tmp/zluda_debug.log")
|
||||
self.debug_file.setPlaceholderText("Debug log file path")
|
||||
|
||||
debug_file_layout.addWidget(debug_file_label)
|
||||
debug_file_layout.addWidget(self.debug_file)
|
||||
debug_layout.addLayout(debug_file_layout, 1, 1)
|
||||
|
||||
main_layout.addWidget(debug_group)
|
||||
|
||||
# Path input frames
|
||||
path_frames = [
|
||||
("ZLUDA Library Path", self.zluda_path, "Select ZLUDA library file", self.browse_zluda),
|
||||
("Application Path", self.app_path, "Select application executable", self.browse_application),
|
||||
("Additional Libraries Path", self.libs_path, "Select additional libraries directory (optional)", self.browse_libs)
|
||||
]
|
||||
|
||||
for label_text, line_edit, placeholder, browse_handler in path_frames:
|
||||
frame = QFrame()
|
||||
frame_layout = QVBoxLayout(frame)
|
||||
frame_layout.setSpacing(6)
|
||||
frame_layout.setContentsMargins(12, 12, 12, 12)
|
||||
|
||||
label = QLabel(label_text)
|
||||
label.setStyleSheet("""
|
||||
color: #00e5ff;
|
||||
font-weight: bold;
|
||||
font-size: 12px;
|
||||
margin-bottom: 3px;
|
||||
""")
|
||||
frame_layout.addWidget(label)
|
||||
|
||||
path_layout = QHBoxLayout()
|
||||
path_layout.setSpacing(6)
|
||||
|
||||
line_edit.setPlaceholderText(placeholder)
|
||||
path_layout.addWidget(line_edit)
|
||||
|
||||
browse_btn = QPushButton("Browse")
|
||||
browse_btn.setProperty("cssClass", "browse")
|
||||
browse_btn.clicked.connect(browse_handler)
|
||||
path_layout.addWidget(browse_btn)
|
||||
|
||||
frame_layout.addLayout(path_layout)
|
||||
main_layout.addWidget(frame)
|
||||
|
||||
# Action buttons
|
||||
buttons_layout = QHBoxLayout()
|
||||
buttons_layout.setSpacing(10)
|
||||
|
||||
self.download_button.setStyleSheet("""
|
||||
QPushButton {
|
||||
background-color: #00796b;
|
||||
min-width: 150px;
|
||||
}
|
||||
QPushButton:hover {
|
||||
background-color: #00897b;
|
||||
}
|
||||
QPushButton:pressed {
|
||||
background-color: #00695c;
|
||||
}
|
||||
QPushButton:disabled {
|
||||
background-color: #2d3436;
|
||||
color: #666666;
|
||||
}
|
||||
""")
|
||||
|
||||
self.run_button.setStyleSheet("""
|
||||
QPushButton {
|
||||
background-color: #006064;
|
||||
min-width: 150px;
|
||||
}
|
||||
QPushButton:hover {
|
||||
background-color: #00838f;
|
||||
}
|
||||
QPushButton:pressed {
|
||||
background-color: #005662;
|
||||
}
|
||||
QPushButton:disabled {
|
||||
background-color: #2d3436;
|
||||
color: #666666;
|
||||
}
|
||||
""")
|
||||
|
||||
buttons_layout.addWidget(self.download_button)
|
||||
buttons_layout.addWidget(self.run_button)
|
||||
main_layout.addLayout(buttons_layout)
|
||||
|
||||
# Progress bar
|
||||
self.progress_bar = QProgressBar()
|
||||
main_layout.addWidget(self.progress_bar)
|
||||
|
||||
# Log areas
|
||||
log_frame = QFrame()
|
||||
log_layout = QVBoxLayout(log_frame)
|
||||
log_layout.setSpacing(6)
|
||||
log_layout.setContentsMargins(12, 12, 12, 12)
|
||||
|
||||
log_header = QHBoxLayout()
|
||||
log_label = QLabel("Log Output")
|
||||
log_label.setStyleSheet("""
|
||||
color: #00e5ff;
|
||||
font-weight: bold;
|
||||
font-size: 12px;
|
||||
margin-bottom: 3px;
|
||||
""")
|
||||
log_header.addWidget(log_label)
|
||||
log_layout.addLayout(log_header)
|
||||
|
||||
self.log_area = QTextEdit()
|
||||
self.log_area.setReadOnly(True)
|
||||
self.log_area.setMinimumHeight(150)
|
||||
log_layout.addWidget(self.log_area)
|
||||
|
||||
main_layout.addWidget(log_frame)
|
||||
|
||||
# Debug tab
|
||||
debug_tab = QWidget()
|
||||
debug_layout = QVBoxLayout(debug_tab)
|
||||
debug_layout.setSpacing(12)
|
||||
debug_layout.setContentsMargins(15, 15, 15, 15)
|
||||
|
||||
debug_log_frame = QFrame()
|
||||
debug_log_layout = QVBoxLayout(debug_log_frame)
|
||||
debug_log_layout.setSpacing(6)
|
||||
debug_log_layout.setContentsMargins(12, 12, 12, 12)
|
||||
|
||||
debug_log_header = QHBoxLayout()
|
||||
debug_log_label = QLabel("Debug Output")
|
||||
debug_log_label.setStyleSheet("""
|
||||
color: #00e5ff;
|
||||
font-weight: bold;
|
||||
font-size: 12px;
|
||||
margin-bottom: 3px;
|
||||
""")
|
||||
debug_log_header.addWidget(debug_log_label)
|
||||
debug_log_layout.addLayout(debug_log_header)
|
||||
|
||||
self.debug_log_area = QTextEdit()
|
||||
self.debug_log_area.setReadOnly(True)
|
||||
self.debug_log_area.setMinimumHeight(300)
|
||||
debug_log_layout.addWidget(self.debug_log_area)
|
||||
|
||||
debug_layout.addWidget(debug_log_frame)
|
||||
|
||||
# Add tabs
|
||||
tabs.addTab(main_tab, "Main")
|
||||
tabs.addTab(debug_tab, "Debug")
|
||||
|
||||
# Set window flags for proper window controls
|
||||
self.setWindowFlags(
|
||||
Qt.WindowType.Window |
|
||||
Qt.WindowType.CustomizeWindowHint |
|
||||
Qt.WindowType.WindowCloseButtonHint |
|
||||
Qt.WindowType.WindowMinimizeButtonHint |
|
||||
Qt.WindowType.WindowMaximizeButtonHint
|
||||
)
|
||||
|
||||
# Maximize the window by default
|
||||
self.setWindowState(Qt.WindowState.WindowMaximized)
|
||||
|
||||
# Initialize variables
|
||||
self.current_process = None
|
||||
self.process_monitor = None
|
||||
self.download_thread = None
|
||||
self.download_progress = None
|
||||
|
||||
def log(self, message):
|
||||
"""Add a message to the main log area"""
|
||||
self.log_area.append(message)
|
||||
QApplication.processEvents()
|
||||
|
||||
def debug_log(self, message):
|
||||
"""Add a message to the debug log area"""
|
||||
self.debug_log_area.append(message)
|
||||
QApplication.processEvents()
|
||||
|
||||
def browse_zluda(self):
|
||||
if platform.system() == "Windows":
|
||||
file_filter = "DLL Files (*.dll)"
|
||||
else:
|
||||
file_filter = "Library Files (*.so)"
|
||||
|
||||
file_path, _ = QFileDialog.getOpenFileName(
|
||||
self, "Select ZLUDA Library", "", file_filter
|
||||
)
|
||||
if file_path:
|
||||
self.zluda_path.setText(file_path)
|
||||
|
||||
def browse_application(self):
|
||||
if platform.system() == "Windows":
|
||||
file_filter = "Executable Files (*.exe)"
|
||||
else:
|
||||
file_filter = "All Files (*)"
|
||||
|
||||
file_path, _ = QFileDialog.getOpenFileName(
|
||||
self, "Select Application", "", file_filter
|
||||
)
|
||||
if file_path:
|
||||
self.app_path.setText(file_path)
|
||||
|
||||
def browse_libs(self):
|
||||
dir_path = QFileDialog.getExistingDirectory(
|
||||
self, "Select Libs Folder"
|
||||
)
|
||||
if dir_path:
|
||||
self.libs_path.setText(dir_path)
|
||||
|
||||
def download_zluda(self):
|
||||
try:
|
||||
self.log("Starting ZLUDA download process...")
|
||||
self.log(f"Detected operating system: {platform.system()}")
|
||||
|
||||
# Determine the appropriate download URL based on OS
|
||||
if platform.system() == "Windows":
|
||||
download_url = "https://github.com/vosen/ZLUDA/releases/download/v4/zluda-4-windows.zip"
|
||||
archive_type = "zip"
|
||||
lib_extension = ".dll"
|
||||
else: # Linux
|
||||
download_url = "https://github.com/vosen/ZLUDA/releases/download/v4/zluda-4-linux.tar.gz"
|
||||
archive_type = "tar.gz"
|
||||
lib_extension = ".so"
|
||||
|
||||
# Start download thread
|
||||
self.download_thread = DownloadThread(download_url)
|
||||
self.download_thread.progress_signal.connect(self.progress_bar.setValue)
|
||||
self.download_thread.log_signal.connect(self.log)
|
||||
self.download_thread.finished_signal.connect(lambda success, result: self.handle_download_finished(success, result, archive_type, lib_extension))
|
||||
self.download_thread.start()
|
||||
|
||||
# Disable download button while downloading
|
||||
self.download_button.setEnabled(False)
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to start download: {str(e)}"
|
||||
self.log(f"Error: {error_msg}")
|
||||
QMessageBox.critical(self, "Error", error_msg)
|
||||
|
||||
def handle_download_finished(self, success, result, archive_type, lib_extension):
|
||||
self.download_button.setEnabled(True)
|
||||
|
||||
if not success:
|
||||
QMessageBox.critical(self, "Error", f"Download failed: {result}")
|
||||
return
|
||||
|
||||
try:
|
||||
self.log("Extracting files...")
|
||||
temp_file = result
|
||||
|
||||
# Extract the archive based on type
|
||||
if archive_type == "zip":
|
||||
with zipfile.ZipFile(temp_file) as zip_ref:
|
||||
file_list = zip_ref.namelist()
|
||||
self.log(f"Found {len(file_list)} files in the archive")
|
||||
self.log("Extracting to current directory...")
|
||||
zip_ref.extractall(os.path.dirname(os.path.abspath(__file__)))
|
||||
else: # tar.gz
|
||||
with tarfile.open(temp_file, mode='r:gz') as tar_ref:
|
||||
file_list = tar_ref.getnames()
|
||||
self.log(f"Found {len(file_list)} files in the archive")
|
||||
self.log("Extracting to current directory...")
|
||||
tar_ref.extractall(os.path.dirname(os.path.abspath(__file__)))
|
||||
|
||||
self.log("Extraction completed successfully!")
|
||||
self.log("ZLUDA has been downloaded and installed")
|
||||
|
||||
# Update the ZLUDA path if a library file was found
|
||||
for file in file_list:
|
||||
if file.endswith(lib_extension):
|
||||
zluda_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), file)
|
||||
self.zluda_path.setText(zluda_path)
|
||||
self.log(f"Automatically set ZLUDA path to: {zluda_path}")
|
||||
break
|
||||
|
||||
# Clean up the temporary file
|
||||
try:
|
||||
os.remove(temp_file)
|
||||
self.log("Cleaned up temporary download file")
|
||||
except:
|
||||
pass
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to extract files: {str(e)}"
|
||||
self.log(f"Error: {error_msg}")
|
||||
QMessageBox.critical(self, "Error", error_msg)
|
||||
|
||||
def check_zluda_loaded(self, pid):
|
||||
"""Check if ZLUDA is loaded in the process"""
|
||||
try:
|
||||
process = psutil.Process(pid)
|
||||
if platform.system() == "Windows":
|
||||
# On Windows, check loaded DLLs
|
||||
dlls = [dll.path.lower() for dll in process.memory_maps()]
|
||||
zluda_dll = self.zluda_path.text().lower()
|
||||
return any(zluda_dll in dll for dll in dlls)
|
||||
else:
|
||||
# On Linux, check LD_PRELOAD
|
||||
env = process.environ()
|
||||
return self.zluda_path.text() in env.get("LD_PRELOAD", "")
|
||||
except (psutil.NoSuchProcess, psutil.AccessDenied):
|
||||
return False
|
||||
|
||||
def run_application(self):
|
||||
zluda_path = self.zluda_path.text()
|
||||
app_path = self.app_path.text()
|
||||
libs_path = self.libs_path.text()
|
||||
|
||||
if not zluda_path or not app_path:
|
||||
QMessageBox.warning(self, "Warning", "Please specify both ZLUDA library and application paths")
|
||||
return
|
||||
|
||||
try:
|
||||
self.log("Preparing to run application...")
|
||||
self.log(f"ZLUDA path: {zluda_path}")
|
||||
self.log(f"Application path: {app_path}")
|
||||
if libs_path:
|
||||
self.log(f"Libs path: {libs_path}")
|
||||
|
||||
# Prepare environment variables
|
||||
env = os.environ.copy()
|
||||
if platform.system() == "Windows":
|
||||
zluda_dir = os.path.dirname(zluda_path)
|
||||
env["PATH"] = f"{zluda_dir}{os.pathsep}{env.get('PATH', '')}"
|
||||
self.log(f"Added ZLUDA directory to PATH: {zluda_dir}")
|
||||
else:
|
||||
# On Linux, use absolute paths
|
||||
zluda_path = os.path.abspath(zluda_path)
|
||||
zluda_dir = os.path.dirname(zluda_path)
|
||||
|
||||
# Set up environment variables
|
||||
env["LD_PRELOAD"] = zluda_path
|
||||
env["LD_LIBRARY_PATH"] = f"{zluda_dir}{os.pathsep}{env.get('LD_LIBRARY_PATH', '')}"
|
||||
|
||||
# Add CUDA specific variables
|
||||
env["CUDA_CACHE_PATH"] = os.path.expanduser("~/.nv/ComputeCache")
|
||||
env["CUDA_CACHE_DISABLE"] = "0"
|
||||
|
||||
self.log(f"Set LD_PRELOAD to: {zluda_path}")
|
||||
self.log(f"Set LD_LIBRARY_PATH to: {zluda_dir}")
|
||||
|
||||
if libs_path:
|
||||
libs_path = os.path.abspath(libs_path)
|
||||
if platform.system() == "Windows":
|
||||
env["PATH"] = f"{libs_path}{os.pathsep}{env.get('PATH', '')}"
|
||||
self.log(f"Added libs directory to PATH: {libs_path}")
|
||||
else:
|
||||
env["LD_LIBRARY_PATH"] = f"{libs_path}{os.pathsep}{env.get('LD_LIBRARY_PATH', '')}"
|
||||
self.log(f"Added libs directory to LD_LIBRARY_PATH: {libs_path}")
|
||||
|
||||
# Set debug flags based on GUI settings
|
||||
if self.zluda_debug.isChecked():
|
||||
env["ZLUDA_DEBUG"] = "1"
|
||||
self.log("Enabled ZLUDA debug output")
|
||||
|
||||
if self.lib_debug.isChecked() and platform.system() == "Linux":
|
||||
env["LD_DEBUG"] = "libs"
|
||||
debug_file = self.debug_file.text()
|
||||
env["LD_DEBUG_OUTPUT"] = debug_file
|
||||
self.log(f"Enabled library debug output to: {debug_file}")
|
||||
|
||||
# Run the application
|
||||
self.log("Starting application...")
|
||||
try:
|
||||
# Clean up any existing process
|
||||
if self.process_monitor:
|
||||
self.process_monitor.stop()
|
||||
self.process_monitor.wait()
|
||||
self.process_monitor = None
|
||||
|
||||
if self.current_process:
|
||||
try:
|
||||
self.current_process.terminate()
|
||||
self.current_process.wait(timeout=5)
|
||||
except:
|
||||
pass
|
||||
self.current_process = None
|
||||
|
||||
# On Linux, run the application directly
|
||||
if platform.system() == "Linux":
|
||||
app_path = os.path.abspath(app_path)
|
||||
# Make sure the application is executable
|
||||
os.chmod(app_path, 0o755)
|
||||
|
||||
self.log("Running application with the following environment:")
|
||||
for key, value in env.items():
|
||||
if key.startswith(("LD_", "CUDA_")):
|
||||
self.log(f"{key}={value}")
|
||||
|
||||
process = subprocess.Popen(
|
||||
[app_path],
|
||||
env=env,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
stdin=subprocess.PIPE,
|
||||
text=True,
|
||||
bufsize=1,
|
||||
universal_newlines=True
|
||||
)
|
||||
else:
|
||||
process = subprocess.Popen(
|
||||
[app_path],
|
||||
env=env,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
stdin=subprocess.PIPE,
|
||||
text=True,
|
||||
bufsize=1,
|
||||
universal_newlines=True
|
||||
)
|
||||
|
||||
self.current_process = process
|
||||
self.log(f"Application started with PID: {process.pid}")
|
||||
|
||||
# Start process monitor
|
||||
self.process_monitor = ProcessMonitor(process, zluda_path, self.gpu_monitor.isChecked())
|
||||
self.process_monitor.log_signal.connect(self.debug_log)
|
||||
self.process_monitor.process_ended.connect(self.on_process_ended)
|
||||
self.process_monitor.start()
|
||||
|
||||
# Check if ZLUDA is loaded
|
||||
if self.check_zluda_loaded(process.pid):
|
||||
self.log("✅ ZLUDA is successfully loaded in the application!")
|
||||
else:
|
||||
self.log("❌ ZLUDA is not loaded in the application. Please check the configuration.")
|
||||
|
||||
self.log("Application started successfully!")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to start application: {str(e)}"
|
||||
self.log(f"Error: {error_msg}")
|
||||
QMessageBox.critical(self, "Error", error_msg)
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"Failed to run application: {str(e)}"
|
||||
self.log(f"Error: {error_msg}")
|
||||
QMessageBox.critical(self, "Error", error_msg)
|
||||
|
||||
def on_process_ended(self):
|
||||
"""Handle when the application process ends"""
|
||||
self.log("Application has terminated")
|
||||
self.debug_log("Process monitor stopped")
|
||||
|
||||
# Clean up process monitor
|
||||
if self.process_monitor:
|
||||
self.process_monitor.stop()
|
||||
self.process_monitor.wait()
|
||||
self.process_monitor = None
|
||||
|
||||
# Reset process reference
|
||||
self.current_process = None
|
||||
|
||||
if __name__ == "__main__":
|
||||
app = QApplication(sys.argv)
|
||||
window = ZLUDA_GUI()
|
||||
window.show()
|
||||
sys.exit(app.exec())
|
||||
Loading…
Add table
Reference in a new issue