Carl Philipp Klemm
d155f1d92a
Some checks are pending
Build eismuliplexer for linux / Build (push) Waiting to run
61 lines
1.3 KiB
C++
61 lines
1.3 KiB
C++
/*
|
|
* Copyright Carl Philipp Klemm 2026
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
|
*/
|
|
|
|
#ifndef ACESTEPWORKER_H
|
|
#define ACESTEPWORKER_H
|
|
|
|
#include <QObject>
|
|
#include <QString>
|
|
#include <QProcess>
|
|
#include <QStandardPaths>
|
|
|
|
#include "SongItem.h"
|
|
|
|
class AceStep : public QObject
|
|
{
|
|
Q_OBJECT
|
|
QProcess qwenProcess;
|
|
QProcess ditVaeProcess;
|
|
|
|
bool busy = false;
|
|
|
|
struct Request
|
|
{
|
|
SongItem song;
|
|
uint64_t uid;
|
|
QString aceStepPath;
|
|
QString textEncoderModelPath;
|
|
QString ditModelPath;
|
|
QString vaeModelPath;
|
|
QString requestFilePath;
|
|
QString requestLlmFilePath;
|
|
};
|
|
|
|
Request request;
|
|
|
|
const QString tempDir = QStandardPaths::writableLocation(QStandardPaths::TempLocation);
|
|
|
|
signals:
|
|
void songGenerated(SongItem song);
|
|
void generationCancled(SongItem song);
|
|
void generationError(QString error);
|
|
void progressUpdate(int progress);
|
|
|
|
public slots:
|
|
bool requestGeneration(SongItem song, QString requestTemplate, QString aceStepPath,
|
|
QString qwen3ModelPath, QString textEncoderModelPath, QString ditModelPath,
|
|
QString vaeModelPath);
|
|
|
|
public:
|
|
AceStep(QObject* parent = nullptr);
|
|
bool isGenerateing(SongItem* song = nullptr);
|
|
void cancleGenerateion();
|
|
|
|
private slots:
|
|
void qwenProcFinished(int code, QProcess::ExitStatus status);
|
|
void ditProcFinished(int code, QProcess::ExitStatus status);
|
|
};
|
|
|
|
#endif // ACESTEPWORKER_H
|