ModelConfig.h

Go to the documentation of this file.

#pragma once
 
#include <QColor>
#include <QDir>
#include <QMap>
#include <QMetaType>
#include <QString>
#include <QVector>
#include <cstdint>
#include <string>
#include <vector>
 
#include <tl/expected.hpp>
 
namespace MCFG {
 
class ModelConfig {
public:
    enum class ErrorCode {
        ConfigFileNotFound = 0,
        ModelFileNotFound = 1,
        IoError = 2,
        ConfigParseError = 3,
        DuplicateClassIds = 4,
    };
 
    struct Error {
        ErrorCode code;
        QString message;
    };
 
    using ClassId = uint32_t;
 
    struct ClassInfo {
        ClassId id;
        QString name;
        QColor color;
        bool selected = false;
    };
 
    static tl::expected<ModelConfig, ModelConfig::Error> loadFromFile(const QString& jsonPath);
 
    const std::vector<float>& getMean() const noexcept;
 
    const std::vector<float>& getStd() const noexcept;
 
    const std::string& getModelPath() const noexcept;
 
    uint getInputAlignment() const noexcept;
 
    bool getApplyMeanStd() const noexcept;
 
    bool getNormalizeTo01() const noexcept;
 
    const std::vector<ClassInfo>& getClasses() const noexcept;
    
    const ClassInfo* getClassById(ClassId id) const noexcept;
 
    bool setClassSelected(ClassId id, bool selected) noexcept;
 
    void setApplyMeanStd(bool apply) noexcept;
 
    void setNormalizeTo01(bool normalize) noexcept;
 
    void setInputAlignment(uint alignment) noexcept;
 
 
private:
    // Private members - only accessible via getters
    std::vector<float> mean_;
    std::vector<float> std_;
    std::vector<ClassInfo> classes_;
    std::string modelPath_;
    uint inputAlignment_ = 1;
    bool applyMeanStd_ = true;      // Whether to apply mean/std normalization
    bool normalizeTo01_ = true;     // Whether to normalize input to [0,1]
};
 
} // namespace MCFG
 
// Register custom types for use in signals/slots
Q_DECLARE_METATYPE(MCFG::ModelConfig::ClassInfo)
Q_DECLARE_METATYPE(QVector<MCFG::ModelConfig::ClassInfo>)