318 lines
17 KiB
Python
318 lines
17 KiB
Python
import sys, os, subprocess
|
|
from PySide6.QtWidgets import QApplication, QDialog, QButtonGroup, QFileDialog, QMessageBox
|
|
from PySide6.QtCore import QCoreApplication, Slot
|
|
from PySide6.QtGui import QIntValidator
|
|
# Import the generated UI class from the ui_form.py file
|
|
from ui_read_disk import Ui_ReadDialog
|
|
|
|
class ReadDiskWindow(QDialog):
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
|
|
self.ui = Ui_ReadDialog()
|
|
self.ui.setupUi(self)
|
|
|
|
# --- Set up Logical Button Groups ---
|
|
self.flippy_type_group = QButtonGroup(self)
|
|
self.flippy_type_group.addButton(self.ui.rb_panasonic, 1)
|
|
self.flippy_type_group.addButton(self.ui.rb_teac, 2)
|
|
self.pin2_setting_group = QButtonGroup(self)
|
|
self.pin2_setting_group.addButton(self.ui.rb_pin2_high, 1)
|
|
self.pin2_setting_group.addButton(self.ui.rb_pin2_low, 2)
|
|
|
|
# --- Connect Signals to Slots ---
|
|
self.ui.cb_bitrate.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_double_step.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_revs.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_drive_select.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_pllspec.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_retries.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_head_sets.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_head_swap.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_ss_legacy.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_cylinder_sets.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_rev_track_data.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_hard_sectors.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_no_clobber.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_raw.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_pin2.toggled.connect(self.on_update_settings)
|
|
self.ui.rb_pin2_high.toggled.connect(self.on_update_settings)
|
|
self.ui.rb_pin2_low.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_flippy.toggled.connect(self.on_update_settings)
|
|
self.ui.rb_panasonic.toggled.connect(self.on_update_settings)
|
|
self.ui.rb_teac.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_adjust_speed.toggled.connect(self.on_update_settings)
|
|
self.ui.combo_drive_select.currentIndexChanged.connect(self.on_update_settings)
|
|
self.ui.combo_fake_index.currentIndexChanged.connect(self.on_update_settings)
|
|
self.ui.combo_adjust_speed.currentIndexChanged.connect(self.on_update_settings)
|
|
self.ui.combo_disktype.currentIndexChanged.connect(self.on_update_settings)
|
|
self.ui.combo_format.currentIndexChanged.connect(self.on_update_settings)
|
|
self.ui.le_fake_index.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_lowpass.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_period.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_phase.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_bitrate.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_revs.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_double_step.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_retries.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_head_sets.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_cylinder_sets.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_adjust_speed.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_suffix.textChanged.connect(self.on_update_settings)
|
|
self.ui.le_fileprefix.textChanged.connect(self.on_update_settings)
|
|
self.ui.btn_file_select.clicked.connect(self.on_btn_file_select_clicked)
|
|
self.ui.cb_format.toggled.connect(self.on_update_settings)
|
|
self.ui.cb_inc.toggled.connect(self.on_update_settings)
|
|
self.ui.btn_launch.clicked.connect(self.on_launch)
|
|
self.ui.btn_back.clicked.connect(self.reject)
|
|
self.ui.btn_path_select.clicked.connect(self.on_btn_path_select_clicked)
|
|
|
|
# Initialize command attribute
|
|
self.command = ""
|
|
self.device = ""
|
|
|
|
@Slot()
|
|
def on_btn_abort_clicked(self) -> None:
|
|
self.abort = True
|
|
self.ui.btn_abort.setEnabled(False)
|
|
|
|
|
|
def calc_suffix(self, inc: int) -> None:
|
|
"""Increment or decrement the numeric value in the suffix text field."""
|
|
try:
|
|
current_value = int(self.ui.le_suffix.text())
|
|
current_value = current_value + inc
|
|
except ValueError:
|
|
current_value = 0
|
|
self.ui.le_suffix.setText(str(current_value))
|
|
|
|
@Slot()
|
|
def on_btn_suffix_inc_clicked(self) -> None:
|
|
self.calc_suffix(1)
|
|
|
|
@Slot()
|
|
def on_btn_suffix_dec_clicked(self) -> None:
|
|
self.calc_suffix(-1)
|
|
|
|
@Slot()
|
|
def on_btn_file_select_clicked(self) -> None:
|
|
file, _ = QFileDialog.getOpenFileName(self, "Select File", self.ui.le_filepath.text())
|
|
if not file:
|
|
return
|
|
file_path, file_name = os.path.split(file)
|
|
name, extension = os.path.splitext(file_name)
|
|
self.ui.le_filepath.setText(file_path)
|
|
self.ui.le_fileprefix.setText(name)
|
|
self.ui.le_suffix.setText("")
|
|
self.ui.combo_disktype.setCurrentText(extension)
|
|
self.on_update_settings()
|
|
|
|
@Slot()
|
|
def on_btn_path_select_clicked(self) -> None:
|
|
path = QFileDialog.getExistingDirectory(self, "Select Folder")
|
|
if path:
|
|
self.ui.le_filepath.setText(path)
|
|
self.on_update_settings()
|
|
|
|
@Slot()
|
|
def on_cb_fake_index_toggled(self, checked: bool) -> None:
|
|
"""Enables or disables the Fake Index widgets."""
|
|
self.ui.le_fake_index.setEnabled(checked)
|
|
self.ui.combo_fake_index.setEnabled(checked)
|
|
self.on_update_settings()
|
|
|
|
def get_settings(self) -> dict:
|
|
"""Gather all settings from the UI into a dictionary."""
|
|
return {
|
|
"fileprefix": self.ui.le_fileprefix.text(),
|
|
"filepath": self.ui.le_filepath.text(),
|
|
"double_step_enabled": self.ui.cb_double_step.isChecked(),
|
|
"double_step_value": self.ui.le_double_step.text(),
|
|
"fake_index_enabled": self.ui.cb_fake_index.isChecked(),
|
|
"fake_index_value": self.ui.le_fake_index.text(),
|
|
"fake_index_unit": self.ui.combo_fake_index.currentText(),
|
|
"bitrate_enabled": self.ui.cb_bitrate.isChecked(),
|
|
"bitrate_value": self.ui.le_bitrate.text(),
|
|
"revs_enabled": self.ui.cb_revs.isChecked(),
|
|
"revs_value": self.ui.le_revs.text(),
|
|
"drive_select_enabled": self.ui.cb_drive_select.isChecked(),
|
|
"drive_select_value": self.ui.combo_drive_select.currentText(),
|
|
"retries_enabled": self.ui.cb_retries.isChecked(),
|
|
"retries_value": self.ui.le_retries.text(),
|
|
"pllspec_enabled": self.ui.cb_pllspec.isChecked(),
|
|
"pll_period": self.ui.le_period.text(),
|
|
"pll_phase": self.ui.le_phase.text(),
|
|
"pll_lowpass": self.ui.le_lowpass.text(),
|
|
"head_sets_enabled": self.ui.cb_head_sets.isChecked(),
|
|
"head_sets_value": self.ui.le_head_sets.text(),
|
|
"head_swap_enabled": self.ui.cb_head_swap.isChecked(),
|
|
"ss_legacy_enabled": self.ui.cb_ss_legacy.isChecked(),
|
|
"cylinder_sets_enabled": self.ui.cb_cylinder_sets.isChecked(),
|
|
"cylinder_sets_value": self.ui.le_cylinder_sets.text(),
|
|
"rev_track_data_enabled": self.ui.cb_rev_track_data.isChecked(),
|
|
"hard_sectors_enabled": self.ui.cb_hard_sectors.isChecked(),
|
|
"no_clobber_enabled": self.ui.cb_no_clobber.isChecked(),
|
|
"raw_enabled": self.ui.cb_raw.isChecked(),
|
|
"pin2_enabled": self.ui.cb_pin2.isChecked(),
|
|
"pin2_high": self.ui.rb_pin2_high.isChecked(),
|
|
"pin2_low": self.ui.rb_pin2_low.isChecked(),
|
|
"flippy_enabled": self.ui.cb_flippy.isChecked(),
|
|
"flippy_panasonic": self.ui.rb_panasonic.isChecked(),
|
|
"flippy_teac": self.ui.rb_teac.isChecked(),
|
|
"adjust_speed_enabled": self.ui.cb_adjust_speed.isChecked(),
|
|
"adjust_speed_value": self.ui.le_adjust_speed.text(),
|
|
"adjust_speed_unit": self.ui.combo_adjust_speed.currentText(),
|
|
"suffix_value": self.ui.le_suffix.text(),
|
|
"inc_enabled": self.ui.cb_inc.isChecked(),
|
|
"format_enabled": self.ui.cb_format.isChecked(),
|
|
"format_value": self.ui.combo_format.currentText(),
|
|
"disktype_value": self.ui.combo_disktype.currentText(),
|
|
}
|
|
|
|
def set_settings(self, settings: dict) -> None:
|
|
"""Apply a settings dictionary to the UI."""
|
|
self.ui.le_fileprefix.setText(settings.get("fileprefix", ""))
|
|
self.ui.le_filepath.setText(settings.get("filepath", ""))
|
|
self.ui.cb_double_step.setChecked(settings.get("double_step_enabled", False))
|
|
self.ui.le_double_step.setText(settings.get("double_step_value", ""))
|
|
self.ui.cb_fake_index.setChecked(settings.get("fake_index_enabled", False))
|
|
self.ui.le_fake_index.setText(settings.get("fake_index_value", ""))
|
|
self.ui.combo_fake_index.setCurrentText(settings.get("fake_index_unit", ""))
|
|
self.ui.cb_bitrate.setChecked(settings.get("bitrate_enabled", False))
|
|
self.ui.le_bitrate.setText(settings.get("bitrate_value", ""))
|
|
self.ui.cb_revs.setChecked(settings.get("revs_enabled", False))
|
|
self.ui.le_revs.setText(settings.get("revs_value", ""))
|
|
self.ui.cb_drive_select.setChecked(settings.get("drive_select_enabled", False))
|
|
self.ui.combo_drive_select.setCurrentText(settings.get("drive_select_value", ""))
|
|
self.ui.cb_retries.setChecked(settings.get("retries_enabled", False))
|
|
self.ui.le_retries.setText(settings.get("retries_value", ""))
|
|
self.ui.cb_pllspec.setChecked(settings.get("pllspec_enabled", False))
|
|
self.ui.le_period.setText(settings.get("pll_period", ""))
|
|
self.ui.le_phase.setText(settings.get("pll_phase", ""))
|
|
self.ui.le_lowpass.setText(settings.get("pll_lowpass", ""))
|
|
self.ui.cb_head_sets.setChecked(settings.get("head_sets_enabled", False))
|
|
self.ui.le_head_sets.setText(settings.get("head_sets_value", ""))
|
|
self.ui.cb_head_swap.setChecked(settings.get("head_swap_enabled", False))
|
|
self.ui.cb_ss_legacy.setChecked(settings.get("ss_legacy_enabled", False))
|
|
self.ui.cb_cylinder_sets.setChecked(settings.get("cylinder_sets_enabled", False))
|
|
self.ui.le_cylinder_sets.setText(settings.get("cylinder_sets_value", ""))
|
|
self.ui.cb_rev_track_data.setChecked(settings.get("rev_track_data_enabled", False))
|
|
self.ui.cb_hard_sectors.setChecked(settings.get("hard_sectors_enabled", False))
|
|
self.ui.cb_no_clobber.setChecked(settings.get("no_clobber_enabled", False))
|
|
self.ui.cb_raw.setChecked(settings.get("raw_enabled", False))
|
|
self.ui.cb_pin2.setChecked(settings.get("pin2_enabled", False))
|
|
self.ui.rb_pin2_high.setChecked(settings.get("pin2_high", False))
|
|
self.ui.rb_pin2_low.setChecked(settings.get("pin2_low", False))
|
|
self.ui.cb_flippy.setChecked(settings.get("flippy_enabled", False))
|
|
self.ui.rb_panasonic.setChecked(settings.get("flippy_panasonic", False))
|
|
self.ui.rb_teac.setChecked(settings.get("flippy_teac", False))
|
|
self.ui.cb_adjust_speed.setChecked(settings.get("adjust_speed_enabled", False))
|
|
self.ui.le_adjust_speed.setText(settings.get("adjust_speed_value", ""))
|
|
self.ui.combo_adjust_speed.setCurrentText(settings.get("adjust_speed_unit", ""))
|
|
self.ui.le_suffix.setText(settings.get("suffix_value", ""))
|
|
self.ui.cb_inc.setChecked(settings.get("inc_enabled", False))
|
|
self.ui.cb_format.setChecked(settings.get("format_enabled", False))
|
|
self.ui.combo_format.setCurrentText(settings.get("format_value", ""))
|
|
self.ui.combo_disktype.setCurrentText(settings.get("disktype_value", ""))
|
|
# Note: command is not set here, as it is generated from UI state
|
|
self.device = settings.get("device", "")
|
|
self.on_update_settings()
|
|
|
|
def build_command(self) -> tuple[list, str]:
|
|
"""Build the command list based on the current UI state."""
|
|
fileprefix = self.ui.le_fileprefix.text()
|
|
drive = f"--drive={self.ui.combo_drive_select.currentText()}" if self.ui.cb_drive_select.isChecked() else ""
|
|
bitrate = f"::bitrate={self.ui.le_bitrate.text()}" if self.ui.cb_bitrate.isChecked() else ""
|
|
legacy_ss = "::legacy_ss" if self.ui.cb_ss_legacy.isChecked() else ""
|
|
revs = f"--revs={self.ui.le_revs.text()}" if self.ui.cb_revs.isChecked() else ""
|
|
retries = f"--retries={self.ui.le_retries.text()}" if self.ui.cb_retries.isChecked() else ""
|
|
fake_index = f"--fake-index={self.ui.le_fake_index.text()}{self.ui.combo_fake_index.currentText()}" if self.ui.cb_fake_index.isChecked() else ""
|
|
pll = ""
|
|
if self.ui.cb_pllspec.isChecked():
|
|
pll = f"--pll=period={self.ui.le_period.text()}:phase={self.ui.le_phase.text()}"
|
|
if self.ui.le_lowpass.text():
|
|
pll += f":lowpass={self.ui.le_lowpass.text()}"
|
|
tracks = self._build_tracks()
|
|
reverse = "--reverse" if self.ui.cb_rev_track_data.isChecked() else ""
|
|
hard_sectors = "--hard-sectors" if self.ui.cb_hard_sectors.isChecked() else ""
|
|
no_clobber = "--no-clobber" if self.ui.cb_no_clobber.isChecked() else ""
|
|
densel = ""
|
|
if self.ui.cb_pin2.isChecked():
|
|
densel = "--densel H" if self.ui.rb_pin2_high.isChecked() else "--densel L"
|
|
raw = "--raw" if self.ui.cb_raw.isChecked() else ""
|
|
adjust_speed = f"--adjust-speed={self.ui.le_adjust_speed.text()}{self.ui.combo_adjust_speed.currentText()}" if self.ui.cb_adjust_speed.isChecked() else ""
|
|
suffix = f"-{self.ui.le_suffix.text()}" if self.ui.le_suffix.text() else ""
|
|
filepath = self.ui.le_filepath.text()
|
|
disktype = self.ui.combo_disktype.currentText()
|
|
format = f"--format={self.ui.combo_format.currentText()}" if self.ui.combo_format.currentText() else ""
|
|
generated_filename = f"{fileprefix}{suffix}{disktype}"
|
|
filename = os.path.join(filepath, generated_filename) + legacy_ss + bitrate
|
|
device = f"--device={self.device}" if self.device and self.device != "Auto" else ""
|
|
command = [
|
|
item for item in [
|
|
"./gw.sh", "read", tracks, revs, device, drive, densel,
|
|
format, adjust_speed, no_clobber, raw, reverse,
|
|
hard_sectors, fake_index, pll, retries, filename
|
|
] if item
|
|
]
|
|
return command, generated_filename
|
|
|
|
def _build_tracks(self) -> str:
|
|
"""Build the --tracks argument from relevant UI fields."""
|
|
tracks = []
|
|
if self.ui.cb_double_step.isChecked():
|
|
tracks.append(f"step={self.ui.le_double_step.text()}")
|
|
if self.ui.cb_cylinder_sets.isChecked():
|
|
tracks.append(f"c={self.ui.le_cylinder_sets.text()}")
|
|
if self.ui.cb_head_sets.isChecked():
|
|
tracks.append(f"h={self.ui.le_head_sets.text()}")
|
|
if self.ui.cb_head_swap.isChecked():
|
|
tracks.append("hswap")
|
|
if self.ui.cb_flippy.isChecked():
|
|
if self.ui.rb_panasonic.isChecked():
|
|
tracks.append("h1.off=-8")
|
|
else:
|
|
tracks.append("h0.off=+8")
|
|
return f"--tracks={':'.join(tracks)}" if tracks else ""
|
|
|
|
def on_update_settings(self, *args, **kwargs) -> None:
|
|
"""Update the command and filename fields based on current UI state."""
|
|
self.command, generated_filename = self.build_command()
|
|
self.ui.le_filename.setText(generated_filename)
|
|
self.ui.te_command_line.setPlainText(" ".join(self.command))
|
|
|
|
@Slot()
|
|
def on_launch(self) -> None:
|
|
"""Run the constructed command using subprocess.Popen and write output to the console widget."""
|
|
self.ui.btn_abort.setEnabled(True)
|
|
self.ui.te_console.clear()
|
|
self.abort = False
|
|
|
|
try:
|
|
process = subprocess.Popen(
|
|
self.command,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
text=True,
|
|
bufsize=1
|
|
)
|
|
if process.stdout:
|
|
for line in iter(process.stdout.readline, ''):
|
|
self.ui.te_console.append(line.rstrip())
|
|
QCoreApplication.processEvents()
|
|
if self.abort:
|
|
process.terminate()
|
|
self.ui.te_console.append("*** Aborted")
|
|
break
|
|
|
|
process.wait()
|
|
|
|
if process.returncode == 0 and self.ui.cb_inc.isChecked():
|
|
self.calc_suffix(1)
|
|
except FileNotFoundError as e:
|
|
QMessageBox.critical(self, "Error", f"Command not found: {e}")
|
|
except Exception as e:
|
|
QMessageBox.critical(self, "Error", f"Failed to launch command: {e}")
|
|
|
|
self.ui.btn_abort.setEnabled(False)
|