ModelManager.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <optional>
 
#include <QObject>
#include <QHash>
#include <QMetaType>
#include <QString>
#include <QStringList>
#include <QVector>
#include <QJsonObject>
 
#include <tl/expected.hpp>
 
#include <ModelConfig.h>
 
namespace MCFG {
 
class ModelManager : public QObject {
    Q_OBJECT
 
public:
    enum class ModelState {
        Ready = 0,
        MissingConfig = 1,
        MissingModel = 2,
        InvalidConfig = 3,
        Incompatible = 4
    };
 
    struct ModelEntry {
        QString name;
        QString jsonPath;
        QString onnxPath;
        ModelState state = ModelState::MissingConfig;
        QString error;
        std::shared_ptr<ModelConfig> config;
    };
 
    explicit ModelManager(QString modelDir = defaultModelDirectory());
 
    ~ModelManager();
 
    static QString defaultModelDirectory();
 
    void setNameFilter(const QString& pattern);
 
    QString nameFilter() const noexcept;
 
    void refresh();
 
    const QVector<ModelEntry>& models() const noexcept;
 
    QStringList availableModelNames() const;
 
    ModelState modelState(const QString& name) const noexcept;
 
    QString modelError(const QString& name) const noexcept;
 
    std::optional<ModelEntry> activateModel(const QString& name);
 
    std::optional<ModelEntry> activeModel() const noexcept;
 
    const QString& activeModelName() const noexcept;
 
    bool setClassSelected(const QString& modelName,
                          ModelConfig::ClassId id, bool selected);
 
    void markModelIncompatible(const QString& name, const QString& reason);
 
    QString modelToString(const QString& name) const noexcept;
 
    QJsonObject modelToJson(const QString& name) const noexcept;
 
    std::optional<ModelEntry> modelFromJson(const QJsonObject& obj) const noexcept;
 
signals:
    void modelActivated(const QString& modelName, ModelManager::ModelState state, const QString& error);
 
    void classListUpdated(const QVector<ModelConfig::ClassInfo>& classes);
 
    void modelsListUpdated(const QVector<ModelEntry>& models);
 
public slots:
    void onModelActivationRequested(const QString& modelName);
 
    void onClassSelectionRequested(const QString& modelName, uint classId, bool selected);
 
    void onModelsRefreshRequested();
 
    void onApplyMeanStdRequested(const QString& modelName, bool apply);
 
    void onNormalizeTo01Requested(const QString& modelName, bool normalize);
 
    void onInputAlignmentRequested(const QString& modelName, uint alignment);
 
private:
    const ModelEntry* findEntry(const QString& name) const noexcept;
 
    QString modelDir_;
    QString nameFilter_;  // Wildcard pattern for filtering models
    QVector<ModelEntry> models_;
    QHash<QString, int> indexByName_;
 
    QString activeModelName_;
};
 
} // namespace MCFG
 
// Register custom types for use in signals/slots
Q_DECLARE_METATYPE(MCFG::ModelManager::ModelState)
Q_DECLARE_METATYPE(MCFG::ModelManager::ModelEntry)
Q_DECLARE_METATYPE(QVector<MCFG::ModelManager::ModelEntry>)