uvos/Aceradio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

playlist handling

Browse source
This commit is contained in:
Carl Philipp Klemm 2026-03-04 22:16:20 +01:00
parent c4d2fa3ffa
commit 134e827053
5 changed files with 418 additions and 67 deletions
Download patch file Download diff file Expand all files Collapse all files

267
MainWindow.cpp
View file

@ -10,6 +10,8 @@
#include <QTextEdit>
#include <QDialogButtonBox>
#include <QLabel>
#include <QFile>
#include <QDir>
MainWindow::MainWindow(QWidget *parent)
: QMainWindow(parent),
@ -32,16 +34,22 @@ MainWindow::MainWindow(QWidget *parent)
// Load settings
loadSettings();
// Auto-load playlist from config location on startup
autoLoadPlaylist();
// Connect signals and slots
connect(ui->actionAdvancedSettings, &QAction::triggered, this, &MainWindow::on_advancedSettingsButton_clicked);
connect(ui->actionSavePlaylist, &QAction::triggered, this, &MainWindow::on_actionSavePlaylist);
connect(ui->actionLoadPlaylist, &QAction::triggered, this, &MainWindow::on_actionLoadPlaylist);
connect(ui->actionQuit, &QAction::triggered, this, [this](){close();});
connect(audioPlayer, &AudioPlayer::playbackFinished, this, &MainWindow::playNextSong);
connect(audioPlayer, &AudioPlayer::playbackStarted, this, &MainWindow::playbackStarted);
connect(aceStepWorker, &AceStepWorker::songGenerated, this, &MainWindow::songGenerated);
connect(aceStepWorker, &AceStepWorker::generationError, this, &MainWindow::generationError);
connect(aceStepWorker, &AceStepWorker::progressUpdate, ui->progressBar, &QProgressBar::setValue);
// Connect double-click on song list for editing
connect(ui->songListView, &QListView::doubleClicked, this, &MainWindow::on_songListView_doubleClicked);
// Connect double-click on song list for editing (works with QTableView too)
connect(ui->songListView, &QTableView::doubleClicked, this, &MainWindow::on_songListView_doubleClicked);
// Connect audio player error signal
connect(audioPlayer, &AudioPlayer::playbackError, [this](const QString &error) {
@ -51,6 +59,9 @@ MainWindow::MainWindow(QWidget *parent)
MainWindow::~MainWindow()
{
// Auto-save playlist before closing
autoSavePlaylist();
saveSettings();
delete ui;
}
@ -60,9 +71,20 @@ void MainWindow::setupUI()
// Setup song list view
ui->songListView->setModel(songModel);
// Make sure the list view is read-only (no inline editing)
// Make sure the table view is read-only (no inline editing)
ui->songListView->setEditTriggers(QAbstractItemView::NoEditTriggers);
// Hide headers for cleaner appearance
ui->songListView->horizontalHeader()->hide();
ui->songListView->verticalHeader()->hide();
// Configure column sizes
ui->songListView->setColumnWidth(0, 40); // Fixed width for play indicator column
ui->songListView->horizontalHeader()->setSectionResizeMode(1, QHeaderView::Stretch); // Expand caption column
// Enable row selection and disable column selection
ui->songListView->setSelectionBehavior(QAbstractItemView::SelectRows);
// Add some default songs
SongItem defaultSong1("Upbeat pop rock anthem with driving electric guitars", "");
SongItem defaultSong2("Chill electronic music with smooth synths and relaxing beats", "");
@ -84,8 +106,7 @@ void MainWindow::loadSettings()
QSettings settings("MusicGenerator", "AceStepGUI");
// Load JSON template (default to simple configuration)
jsonTemplate = settings.value("jsonTemplate",
"{\"inference_steps\": 8, \"shift\": 3.0, \"vocal_language\": \"en\"}").toString();
jsonTemplate = settings.value("jsonTemplate", "{\n\t\"inference_steps\": 8,\n\t\"shift\": 3.0,\n\t\"vocal_language\": \"en\"\n}").toString();
// Load shuffle mode
shuffleMode = settings.value("shuffleMode", false).toBool();
@ -217,7 +238,6 @@ void MainWindow::on_stopButton_clicked()
audioPlayer->stop();
isPlaying = false;
isPaused = false;
ui->statusLabel->setText("Stopped");
updateControls();
}
}
@ -251,9 +271,22 @@ void MainWindow::on_songListView_doubleClicked(const QModelIndex &index)
// Temporarily disconnect the signal to prevent multiple invocations
// This happens when the dialog closes and triggers another double-click event
disconnect(ui->songListView, &QListView::doubleClicked, this, &MainWindow::on_songListView_doubleClicked);
disconnect(ui->songListView, &QTableView::doubleClicked, this, &MainWindow::on_songListView_doubleClicked);
int row = index.row();
// Different behavior based on which column was clicked
if (index.column() == 0) {
// Column 0 (play indicator): Stop current playback and play this song
if (isPlaying) {
audioPlayer->stop();
}
// Set this as the current song and generate/play it
currentSongIndex = row;
generateAndPlayNext();
} else if (index.column() == 1) {
// Column 1 (caption): Edit the song
SongItem song = songModel->getSong(row);
SongDialog dialog(this, song.caption, song.lyrics);
@ -262,13 +295,14 @@ void MainWindow::on_songListView_doubleClicked(const QModelIndex &index)
QString caption = dialog.getCaption();
QString lyrics = dialog.getLyrics();
// Update the model
songModel->setData(songModel->index(row, 0), caption, SongListModel::CaptionRole);
songModel->setData(songModel->index(row, 0), lyrics, SongListModel::LyricsRole);
// Update the model - use column 1 for the song name
songModel->setData(songModel->index(row, 1), caption, SongListModel::CaptionRole);
songModel->setData(songModel->index(row, 1), lyrics, SongListModel::LyricsRole);
}
}
// Reconnect the signal after dialog is closed
connect(ui->songListView, &QListView::doubleClicked, this, &MainWindow::on_songListView_doubleClicked);
connect(ui->songListView, &QTableView::doubleClicked, this, &MainWindow::on_songListView_doubleClicked);
}
void MainWindow::on_removeSongButton_clicked()
@ -276,13 +310,9 @@ void MainWindow::on_removeSongButton_clicked()
QModelIndex currentIndex = ui->songListView->currentIndex();
if (!currentIndex.isValid()) return;
// Get the row from the current selection (works with table view)
int row = currentIndex.row();
QMessageBox::StandardButton reply;
reply = QMessageBox::question(this, "Remove Song", "Are you sure you want to remove this song?",
QMessageBox::Yes | QMessageBox::No);
if (reply == QMessageBox::Yes) {
songModel->removeSong(row);
// Select next item or previous if at end
@ -291,7 +321,6 @@ void MainWindow::on_removeSongButton_clicked()
QModelIndex newIndex = songModel->index(newRow, 0);
ui->songListView->setCurrentIndex(newIndex);
}
}
}
void MainWindow::on_advancedSettingsButton_clicked()
@ -375,6 +404,14 @@ void MainWindow::playbackStarted()
startNextSongGeneration();
}
void MainWindow::highlightCurrentSong()
{
if (currentSongIndex >= 0 && currentSongIndex < songModel->rowCount()) {
// Update the model to show play icon for current song
songModel->setPlayingIndex(currentSongIndex);
}
}
void MainWindow::songGenerated(const QString &filePath)
{
if (!QFile::exists(filePath)) {
@ -389,11 +426,14 @@ void MainWindow::songGenerated(const QString &filePath)
return;
}
ui->statusLabel->setText("Playing: " + QFileInfo(filePath).baseName());
ui->statusLabel->setText("");
// Play the generated song
audioPlayer->play(filePath);
// Highlight the current song in the list
highlightCurrentSong();
// Connect position and duration updates for the slider
connect(audioPlayer, &AudioPlayer::positionChanged, this, &MainWindow::updatePosition);
connect(audioPlayer, &AudioPlayer::durationChanged, this, &MainWindow::updateDuration);
@ -405,7 +445,6 @@ void MainWindow::playNextSong()
// Check if we have a pre-generated next song
if (!nextSongFilePath.isEmpty()) {
ui->statusLabel->setText("Playing: " + QFileInfo(nextSongFilePath).baseName());
audioPlayer->play(nextSongFilePath);
nextSongFilePath.clear();
@ -415,7 +454,8 @@ void MainWindow::playNextSong()
currentSongIndex = nextIndex;
}
// Start generating the song after this one
// Highlight the current song and start generating the next one
highlightCurrentSong();
startNextSongGeneration();
} else {
// Find next song index and generate it
@ -428,7 +468,6 @@ void MainWindow::playNextSong()
// No more songs
isPlaying = false;
isPaused = false;
ui->statusLabel->setText("Finished playback");
updateControls();
}
}
@ -488,3 +527,189 @@ void MainWindow::on_positionSlider_sliderMoved(int position)
audioPlayer->setPosition(position);
}
}
// Playlist save/load methods
void MainWindow::on_actionSavePlaylist()
{
QString filePath = QFileDialog::getSaveFileName(this, "Save Playlist",
QStandardPaths::writableLocation(QStandardPaths::DocumentsLocation) + "/playlist.json",
"JSON Files (*.json);;All Files (*)");
if (!filePath.isEmpty()) {
savePlaylist(filePath);
}
}
void MainWindow::on_actionLoadPlaylist()
{
QString filePath = QFileDialog::getOpenFileName(this, "Load Playlist",
QStandardPaths::writableLocation(QStandardPaths::DocumentsLocation),
"JSON Files (*.json);;All Files (*)");
if (!filePath.isEmpty()) {
loadPlaylist();
}
}
void MainWindow::savePlaylist(const QString &filePath)
{
// Get current songs from the model
QList<SongItem> songs;
for (int i = 0; i < songModel->rowCount(); ++i) {
songs.append(songModel->getSong(i));
}
savePlaylistToJson(filePath, songs);
}
void MainWindow::loadPlaylist()
{
QString filePath = QFileDialog::getOpenFileName(this, "Load Playlist",
QStandardPaths::writableLocation(QStandardPaths::DocumentsLocation),
"JSON Files (*.json);;All Files (*)");
if (!filePath.isEmpty()) {
QList<SongItem> songs;
if (loadPlaylistFromJson(filePath, songs)) {
// Clear current playlist
while (songModel->rowCount() > 0) {
songModel->removeSong(0);
}
// Add loaded songs
for (const SongItem &song : songs) {
songModel->addSong(song);
}
}
}
}
void MainWindow::autoSavePlaylist()
{
QString configPath = QStandardPaths::writableLocation(QStandardPaths::ConfigLocation);
QString appConfigPath = configPath + "/MusicGenerator/AceStepGUI";
// Create directory if it doesn't exist
QDir().mkpath(appConfigPath);
QString filePath = appConfigPath + "/playlist.json";
// Get current songs from the model
QList<SongItem> songs;
for (int i = 0; i < songModel->rowCount(); ++i) {
songs.append(songModel->getSong(i));
}
savePlaylistToJson(filePath, songs);
}
void MainWindow::autoLoadPlaylist()
{
QString configPath = QStandardPaths::writableLocation(QStandardPaths::ConfigLocation);
QString appConfigPath = configPath + "/MusicGenerator/AceStepGUI";
QString filePath = appConfigPath + "/playlist.json";
// Check if the auto-save file exists
if (QFile::exists(filePath)) {
QList<SongItem> songs;
if (loadPlaylistFromJson(filePath, songs)) {
// Clear default songs and add loaded ones
while (songModel->rowCount() > 0) {
songModel->removeSong(0);
}
for (const SongItem &song : songs) {
songModel->addSong(song);
}
}
}
}
bool MainWindow::savePlaylistToJson(const QString &filePath, const QList<SongItem> &songs)
{
QJsonArray songsArray;
for (const SongItem &song : songs) {
QJsonObject songObj;
songObj["caption"] = song.caption;
songObj["lyrics"] = song.lyrics;
songsArray.append(songObj);
}
QJsonObject rootObj;
rootObj["songs"] = songsArray;
rootObj["version"] = "1.0";
QJsonDocument doc(rootObj);
QByteArray jsonData = doc.toJson();
QFile file(filePath);
if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) {
qWarning() << "Could not open file for writing:" << filePath;
return false;
}
file.write(jsonData);
file.close();
return true;
}
bool MainWindow::loadPlaylistFromJson(const QString &filePath, QList<SongItem> &songs)
{
QFile file(filePath);
if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
qWarning() << "Could not open file for reading:" << filePath;
return false;
}
QByteArray jsonData = file.readAll();
file.close();
QJsonParseError parseError;
QJsonDocument doc = QJsonDocument::fromJson(jsonData, &parseError);
if (parseError.error != QJsonParseError::NoError) {
qWarning() << "JSON parse error:" << parseError.errorString();
return false;
}
if (!doc.isObject()) {
qWarning() << "JSON root is not an object";
return false;
}
QJsonObject rootObj = doc.object();
// Check for version compatibility
if (rootObj.contains("version") && rootObj["version"].toString() != "1.0") {
qWarning() << "Unsupported playlist version:" << rootObj["version"].toString();
return false;
}
if (!rootObj.contains("songs") || !rootObj["songs"].isArray()) {
qWarning() << "Invalid playlist format: missing songs array";
return false;
}
QJsonArray songsArray = rootObj["songs"].toArray();
for (const QJsonValue &value : songsArray) {
if (!value.isObject()) continue;
QJsonObject songObj = value.toObject();
SongItem song;
if (songObj.contains("caption")) {
song.caption = songObj["caption"].toString();
}
if (songObj.contains("lyrics")) {
song.lyrics = songObj["lyrics"].toString();
}
songs.append(song);
}
return true;
}

18
MainWindow.h
View file

@ -5,6 +5,10 @@
#include <QListWidgetItem>
#include <QStandardItemModel>
#include <QTimer>
#include <QStandardPaths>
#include <QJsonArray>
#include <QJsonObject>
#include <QJsonDocument>
#include "SongListModel.h"
#include "AudioPlayer.h"
#include "AceStepWorker.h"
@ -43,6 +47,9 @@ private slots:
void generationFinished();
void generationError(const QString &error);
void on_actionSavePlaylist();
void on_actionLoadPlaylist();
private:
void startNextSongGeneration();
@ -72,8 +79,19 @@ private:
// Pre-generated song file path
QString nextSongFilePath;
private:
void highlightCurrentSong();
void loadSettings();
void saveSettings();
void loadPlaylist();
void savePlaylist(const QString &filePath);
void autoSavePlaylist();
void autoLoadPlaylist();
bool savePlaylistToJson(const QString &filePath, const QList<SongItem> &songs);
bool loadPlaylistFromJson(const QString &filePath, QList<SongItem> &songs);
void setupUI();
void updateControls();
void generateAndPlayNext();

100
MainWindow.ui
View file

@ -11,7 +11,7 @@
</rect>
</property>
<property name="windowTitle">
<string>Music Generator GUI</string>
<string>Aceradio</string>
</property>
<widget class="QWidget" name="centralwidget">
<layout class="QVBoxLayout" name="verticalLayout">
@ -21,13 +21,12 @@
<string>Music Generator</string>
</property>
<property name="alignment">
<set>Qt::AlignCenter</set>
<set>Qt::AlignmentFlag::AlignCenter</set>
</property>
</widget>
</item>
<item>
<widget class="QListView" name="songListView">
<widget class="QTableView" name="songListView">
<property name="minimumSize">
<size>
<width>0</width>
@ -35,7 +34,10 @@
</size>
</property>
<property name="editTriggers">
<set>QAbstractItemView::NoEditTriggers</set>
<set>QAbstractItemView::EditTrigger::NoEditTriggers</set>
</property>
<property name="selectionBehavior">
<enum>QAbstractItemView::SelectionBehavior::SelectRows</enum>
</property>
</widget>
</item>
@ -58,7 +60,7 @@
<item>
<spacer name="horizontalSpacer">
<property name="orientation">
<enum>Qt::Horizontal</enum>
<enum>Qt::Orientation::Horizontal</enum>
</property>
<property name="sizeHint" stdset="0">
<size>
@ -81,8 +83,14 @@
</item>
<item>
<widget class="QSlider" name="positionSlider">
<property name="mouseTracking">
<bool>false</bool>
</property>
<property name="tabletTracking">
<bool>true</bool>
</property>
<property name="orientation">
<enum>Qt::Horizontal</enum>
<enum>Qt::Orientation::Horizontal</enum>
</property>
</widget>
</item>
@ -98,10 +106,10 @@
<item>
<widget class="QFrame" name="controlsFrame">
<property name="frameShape">
<enum>QFrame::StyledPanel</enum>
<enum>QFrame::Shape::StyledPanel</enum>
</property>
<property name="frameShadow">
<enum>QFrame::Raised</enum>
<enum>QFrame::Shadow::Raised</enum>
</property>
<layout class="QHBoxLayout" name="controlsLayout">
<item>
@ -110,9 +118,7 @@
<string>Play</string>
</property>
<property name="icon">
<iconset theme="media-playback-start">
<normaloff>.</normaloff>.
</iconset>
<iconset theme="media-playback-start"/>
</property>
</widget>
</item>
@ -122,9 +128,7 @@
<string>Pause</string>
</property>
<property name="icon">
<iconset theme="media-playback-pause">
<normaloff>.</normaloff>.
</iconset>
<iconset theme="media-playback-pause"/>
</property>
</widget>
</item>
@ -134,9 +138,7 @@
<string>Skip</string>
</property>
<property name="icon">
<iconset theme="media-skip-forward">
<normaloff>.</normaloff>.
</iconset>
<iconset theme="media-skip-forward"/>
</property>
</widget>
</item>
@ -146,9 +148,7 @@
<string>Stop</string>
</property>
<property name="icon">
<iconset theme="media-playback-stop">
<normaloff>.</normaloff>.
</iconset>
<iconset theme="media-playback-stop"/>
</property>
</widget>
</item>
@ -157,20 +157,18 @@
<property name="text">
<string>Shuffle</string>
</property>
<property name="icon">
<iconset theme="media-playlist-shuffle"/>
</property>
<property name="checkable">
<bool>true</bool>
</property>
<property name="icon">
<iconset theme="media-playlist-shuffle">
<normaloff>.</normaloff>.
</iconset>
</property>
</widget>
</item>
<item>
<spacer name="horizontalSpacer_2">
<property name="orientation">
<enum>Qt::Horizontal</enum>
<enum>Qt::Orientation::Horizontal</enum>
</property>
<property name="sizeHint" stdset="0">
<size>
@ -195,7 +193,7 @@
<item>
<widget class="QLabel" name="statusLabel">
<property name="text">
<string>Ready</string>
<string/>
</property>
</widget>
</item>
@ -209,21 +207,63 @@
<x>0</x>
<y>0</y>
<width>800</width>
<height>22</height>
<height>32</height>
</rect>
</property>
<widget class="QMenu" name="menuFile">
<property name="title">
<string>File</string>
</property>
<addaction name="actionSavePlaylist"/>
<addaction name="actionLoadPlaylist"/>
<addaction name="actionQuit"/>
</widget>
<widget class="QMenu" name="menuSettings">
<property name="title">
<string>Settings</string>
</property>
<addaction name="actionAdvancedSettings"/>
</widget>
<addaction name="menuFile"/>
<addaction name="menuSettings"/>
</widget>
<widget class="QStatusBar" name="statusbar"/>
<action name="actionSavePlaylist">
<property name="icon">
<iconset theme="QIcon::ThemeIcon::DocumentSaveAs"/>
</property>
<property name="text">
<string>Save Playlist</string>
</property>
<property name="shortcut">
<string>Ctrl+S</string>
</property>
</action>
<action name="actionLoadPlaylist">
<property name="icon">
<iconset theme="QIcon::ThemeIcon::DocumentOpen"/>
</property>
<property name="text">
<string>Load Playlist...</string>
</property>
<property name="shortcut">
<string>Ctrl+O</string>
</property>
</action>
<action name="actionAdvancedSettings">
<property name="text">
<string>Advanced Settings...</string>
<string>Ace Step</string>
</property>
</action>
<action name="actionQuit">
<property name="icon">
<iconset theme="QIcon::ThemeIcon::ApplicationExit"/>
</property>
<property name="text">
<string>Quit</string>
</property>
<property name="shortcut">
<string>Ctrl+Q</string>
</property>
</action>
</widget>

62
SongListModel.cpp
View file

@ -1,10 +1,13 @@
#include "SongListModel.h"
#include <QApplication>
#include <QTime>
#include <QRandomGenerator>
#include <QDebug>
#include <QFont>
SongListModel::SongListModel(QObject *parent)
: QAbstractListModel(parent)
: QAbstractTableModel(parent),
m_playingIndex(-1)
{
}
@ -15,6 +18,12 @@ int SongListModel::rowCount(const QModelIndex &parent) const
return songList.size();
}
int SongListModel::columnCount(const QModelIndex &parent) const
{
// We have 2 columns: play indicator and song name
return 2;
}
QVariant SongListModel::data(const QModelIndex &index, int role) const
{
if (!index.isValid() || index.row() >= songList.size())
@ -24,13 +33,40 @@ QVariant SongListModel::data(const QModelIndex &index, int role) const
switch (role) {
case Qt::DisplayRole:
// Column 0: Play indicator column
if (index.column() == 0) {
return index.row() == m_playingIndex ? "▶" : "";
}
// Column 1: Song name
else if (index.column() == 1) {
return song.caption;
}
break;
case Qt::FontRole:
// Make play indicator bold and larger
if (index.column() == 0 && index.row() == m_playingIndex) {
QFont font = QApplication::font();
font.setBold(true);
return font;
}
break;
case Qt::TextAlignmentRole:
// Center align the play indicator
if (index.column() == 0) {
return Qt::AlignCenter;
}
break;
case CaptionRole:
return song.caption;
case LyricsRole:
return song.lyrics;
case IsPlayingRole:
return index.row() == m_playingIndex;
default:
return QVariant();
}
return QVariant();
}
bool SongListModel::setData(const QModelIndex &index, const QVariant &value, int role)
@ -88,6 +124,30 @@ SongItem SongListModel::getSong(int index) const
return SongItem();
}
QVariant SongListModel::headerData(int section, Qt::Orientation orientation, int role) const
{
if (role == Qt::DisplayRole && orientation == Qt::Horizontal) {
// Hide headers since we don't need column titles
return QVariant();
}
return QAbstractTableModel::headerData(section, orientation, role);
}
void SongListModel::setPlayingIndex(int index)
{
int oldPlayingIndex = m_playingIndex;
m_playingIndex = index;
// Update both the old and new playing indices to trigger UI updates
if (oldPlayingIndex >= 0 && oldPlayingIndex < songList.size()) {
emit dataChanged(this->index(oldPlayingIndex, 0), this->index(oldPlayingIndex, 0));
}
if (index >= 0 && index < songList.size()) {
emit dataChanged(this->index(index, 0), this->index(index, 0));
}
}
int SongListModel::findNextIndex(int currentIndex, bool shuffle) const
{
if (songList.isEmpty())

12
SongListModel.h
View file

@ -14,21 +14,24 @@ public:
: caption(caption), lyrics(lyrics) {}
};
class SongListModel : public QAbstractListModel
class SongListModel : public QAbstractTableModel
{
Q_OBJECT
public:
enum Roles {
CaptionRole = Qt::UserRole + 1,
LyricsRole = Qt::UserRole + 2
LyricsRole = Qt::UserRole + 2,
IsPlayingRole = Qt::UserRole + 3
};
explicit SongListModel(QObject *parent = nullptr);
// Basic functionality:
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
int columnCount(const QModelIndex &parent = QModelIndex()) const override;
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
// Editable:
bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
@ -40,8 +43,13 @@ public:
SongItem getSong(int index) const;
int findNextIndex(int currentIndex, bool shuffle = false) const;
// Playing indicator
void setPlayingIndex(int index);
int playingIndex() const { return m_playingIndex; }
private:
QList<SongItem> songList;
int m_playingIndex;
};
#endif // SONGLISTMODEL_H