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.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) # 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)