""" Player Panel — UI for playing files with EQ applied. """ import os from PyQt6.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QFileDialog, QProgressBar, QListWidget, QSplitter, QMessageBox, QCheckBox) from PyQt6.QtCore import Qt, QTimer, QSettings from core.file_loader import get_audio_duration from core.audio_player import AudioPlayer from ui.waveform_widget import WaveformWidget from ui.spectrum_widget import SpectrumWidget class VUMeter(QProgressBar): """Custom progress bar for VU metering.""" def __init__(self): super().__init__() self.setOrientation(Qt.Orientation.Vertical) self.setRange(-60, 0) # dB range self.setValue(-60) self.setTextVisible(False) self.setFixedWidth(15) # Styling handled in global QSS, but we can dynamic color here if needed self.setStyleSheet(""" QProgressBar { border: 1px solid #30304a; background-color: #161625; } QProgressBar::chunk { background-color: qlineargradient(x1: 0, y1: 1, x2: 0, y2: 0, stop: 0 #4CAF50, stop: 0.7 #FFEB3B, stop: 1 #F44336); } """) class PlayerPanel(QWidget): def __init__(self, player: AudioPlayer): super().__init__() self.player = player self.playlist = [] self.current_track_index = -1 self.settings = QSettings("TapeRecorder", "Player") self._init_ui() # Connect player callbacks # Note: These are called from the audio thread. # For thread safety, we use a QTimer to poll state instead of direct signals. self.poll_timer = QTimer() self.poll_timer.timeout.connect(self._poll_player_state) self.poll_timer.start(50) # 20Hz update def _init_ui(self): # --- Left Side: Player Controls --- left_widget = QWidget() left_layout = QVBoxLayout(left_widget) # --- Top: File Info & Load --- top_row = QHBoxLayout() self.file_lbl = QLabel("No file loaded") self.file_lbl.setStyleSheet("font-weight: bold; font-size: 12pt;") load_btn = QPushButton("Load Files...") load_btn.clicked.connect(self.load_files) load_folder_btn = QPushButton("Load Folder...") load_folder_btn.clicked.connect(self.load_folder) top_row.addWidget(self.file_lbl, 1) top_row.addWidget(load_btn) top_row.addWidget(load_folder_btn) left_layout.addLayout(top_row) # --- Middle: Waveform --- self.waveform = WaveformWidget() self.waveform.setFixedHeight(120) self.waveform.seek_requested.connect(self.player.seek) left_layout.addWidget(self.waveform) # --- Time display --- self.time_lbl = QLabel("00:00.0 / 00:00.0") self.time_lbl.setAlignment(Qt.AlignmentFlag.AlignCenter) self.time_lbl.setStyleSheet("font-family: monospace; font-size: 14pt; color: #ff7b00;") left_layout.addWidget(self.time_lbl) # --- Controls & VU --- control_row = QHBoxLayout() # Buttons btn_layout = QHBoxLayout() self.play_btn = QPushButton("▶ Play") self.play_btn.clicked.connect(self.toggle_play) self.stop_btn = QPushButton("■ Stop") self.stop_btn.clicked.connect(self._on_stop_clicked) self.norm_cb = QCheckBox("Normalize Volume") self.norm_cb.setChecked(self.settings.value("normalize_volume", True, type=bool)) self.norm_cb.clicked.connect(self._on_norm_changed) btn_layout.addWidget(self.play_btn) btn_layout.addWidget(self.stop_btn) btn_layout.addWidget(self.norm_cb) btn_layout.addStretch() control_row.addLayout(btn_layout, 1) # VU Meters vu_layout = QHBoxLayout() self.vu_l = VUMeter() self.vu_r = VUMeter() vu_layout.addWidget(QLabel("L")) vu_layout.addWidget(self.vu_l) vu_layout.addWidget(self.vu_r) vu_layout.addWidget(QLabel("R")) control_row.addLayout(vu_layout) left_layout.addLayout(control_row) # --- Bottom: Spectrum Analyzer --- self.spectrum = SpectrumWidget() self.spectrum.setFixedHeight(200) left_layout.addWidget(self.spectrum) # --- Right Side: Playlist --- right_widget = QWidget() right_layout = QVBoxLayout(right_widget) self.playlist_lbl = QLabel("Playlist") self.playlist_lbl.setStyleSheet("font-weight: bold;") right_layout.addWidget(self.playlist_lbl) self.playlist_widget = QListWidget() self.playlist_widget.itemDoubleClicked.connect(self._on_playlist_double_click) right_layout.addWidget(self.playlist_widget) # Splitter splitter = QSplitter(Qt.Orientation.Horizontal) splitter.addWidget(left_widget) splitter.addWidget(right_widget) splitter.setSizes([700, 300]) # Default proportions layout = QVBoxLayout(self) layout.addWidget(splitter) def load_files(self): files, _ = QFileDialog.getOpenFileNames( self, "Open Audio Files", "", "Audio Files (*.wav *.mp3 *.opus *.flac *.ogg)" ) if files: self.playlist = files self.current_track_index = 0 self._update_playlist_ui() self._load_track(self.playlist[0]) def load_folder(self): last_folder = self.settings.value("last_folder", "") folder_path = QFileDialog.getExistingDirectory(self, "Select Folder", last_folder) if folder_path: self.settings.setValue("last_folder", folder_path) supported = ('.wav', '.mp3', '.opus', '.flac', '.ogg', '.aiff') files = [] for f in os.listdir(folder_path): if f.lower().endswith(supported): files.append(os.path.join(folder_path, f)) if not files: QMessageBox.information(self, "Info", "No supported audio files found in folder.") return self.playlist = sorted(files) self.current_track_index = 0 self._update_playlist_ui() self._load_track(self.playlist[self.current_track_index], auto_play=True) def _update_playlist_ui(self): self.playlist_widget.clear() total_duration = 0.0 for f in self.playlist: self.playlist_widget.addItem(os.path.basename(f)) total_duration += get_audio_duration(f) dur_m = int(total_duration // 60) dur_s = int(total_duration % 60) self.playlist_lbl.setText(f"Playlist (Total: {dur_m:02d}:{dur_s:02d})") if self.playlist and self.current_track_index >= 0: self.playlist_widget.setCurrentRow(self.current_track_index) def _on_playlist_double_click(self, item): idx = self.playlist_widget.row(item) if 0 <= idx < len(self.playlist): self.current_track_index = idx self._load_track(self.playlist[self.current_track_index], auto_play=True) def _on_norm_changed(self): self.settings.setValue("normalize_volume", self.norm_cb.isChecked()) def _load_track(self, file_path, auto_play=False): try: should_normalize = self.norm_cb.isChecked() self.player.load_file(file_path, normalize=should_normalize) filename = os.path.basename(file_path) prefix = f"[{self.current_track_index + 1}/{len(self.playlist)}] " if self.playlist else "" self.file_lbl.setText(f"Loaded: {prefix}{filename}") self.waveform.load_audio(self.player.audio_data) self._update_time_display(0) self.spectrum.set_sample_rate(self.player.device_manager.sample_rate) if auto_play: try: self.player.play() self.play_btn.setText("⏸ Pause") except Exception as e: QMessageBox.warning(self, "Playback Error", f"Could not start playback. Is your output device set?\n\n{e}") except Exception as e: self.file_lbl.setText(f"Error loading file: {e}") def play_next_track(self): if self.playlist and self.current_track_index < len(self.playlist) - 1: self.current_track_index += 1 self.playlist_widget.setCurrentRow(self.current_track_index) self._load_track(self.playlist[self.current_track_index], auto_play=True) else: self.play_btn.setText("▶ Play") def _on_stop_clicked(self): self.play_btn.setText("▶ Play") self.player.stop() def toggle_play(self): if self.player.is_playing: self.player.pause() self.play_btn.setText("▶ Play") else: if self.player.audio_data is not None: try: self.player.play() self.play_btn.setText("⏸ Pause") except Exception as e: QMessageBox.warning(self, "Playback Error", f"Could not start playback. Is your output device set?\n\n{e}") def _poll_player_state(self): """Poll the player state to update UI (thread-safe).""" # Update play button state if stopped automatically (reached end of file) if not self.player.is_playing and self.play_btn.text() == "⏸ Pause": self.play_next_track() # Update playhead if self.player.audio_data: pos_sec = self.player.current_frame / self.player.audio_data.sample_rate self.waveform.set_position(pos_sec) self._update_time_display(pos_sec) # If playing, we can pull the latest chunk from the player for the spectrum. # In a real app we'd pass this via queue, but this is a simple approximation. if self.player.is_playing and self.player.current_frame > 1024: idx = self.player.current_frame chunk = self.player.audio_data.samples[max(0, idx-1024):idx] self.spectrum.add_audio_data(chunk) # Update VU meters l_db, r_db = self.player.vu_levels # Map dB to progress bar (-60 to 0) self.vu_l.setValue(int(max(-60, l_db))) self.vu_r.setValue(int(max(-60, r_db))) # Decay VU meters when stopped if not self.player.is_playing: self.player.vu_levels = (max(-100, l_db - 2), max(-100, r_db - 2)) def _update_time_display(self, pos_sec: float): if not self.player.audio_data: return dur_sec = self.player.audio_data.duration pos_m = int(pos_sec // 60) pos_s = pos_sec % 60 dur_m = int(dur_sec // 60) dur_s = dur_sec % 60 self.time_lbl.setText(f"{pos_m:02d}:{pos_s:04.1f} / {dur_m:02d}:{dur_s:04.1f}")