ModelFileChecker.h

Go to the documentation of this file.

#ifndef MODEL_FILE_CHECKER_H
#define MODEL_FILE_CHECKER_H
 
#include <string>
#include <fstream>
#include <iostream>
#include <regex>
#include <optional>
#include <vector>
 
struct ModelClassInfo {
    std::string class_name;
    bool is_template = false;
};
 
class ModelFileChecker {
public:
    ModelFileChecker(const std::string& filePath);
 
    bool isAnyModelTemplate() const;
 
    ModelClassInfo resolveModelClass(const std::string& model) const;
 
    static std::vector<std::string> modelClassCandidates(const std::string& model);
 
private:
    std::string readContents() const;
    static bool hasClassDefinition(const std::string& contents,
                                   const std::string& class_name,
                                   bool& is_template);
    static std::string regexEscape(const std::string& value);
 
    std::string filePath;
};
 
#endif