CscDetectorResources.h

Go to the documentation of this file.

#ifndef CscDetectorResources_H
#define CscDetectorResources_H
 
#include <string>
#include <filesystem>
#include <memory>
 
#include "CoreObjects/Resources/CscResourceSpec.h"
#include "CoreObjects/Resources/CscLocalFileResourceSpec.h"
#include "CoreObjects/Resources/CscResourceFromServiceSpec.h"
 
using std::string;
using namespace conscience_core::core_objects::resources;
namespace fs = std::filesystem;
template<class T>
using ptr = std::shared_ptr<T>;
 
namespace conscience_core::detector_engine {
    class CscDetectorResources {
    private:
        inline static ptr<CscResourceSpec> getServiceResourceSpecWithBackwardCompatibility(
            string resourceId,
            fs::path obsoleteFilePathCoreRelative
        ) {
            fs::path coreRelativePath = fs::path(__FILE__).parent_path().parent_path();
            try {
                if (fs::exists(coreRelativePath)) {
                    
                    fs::path corePath = fs::canonical(coreRelativePath);
                    fs::path obsoleteFullPath = corePath / obsoleteFilePathCoreRelative;
                    if (fs::exists(obsoleteFullPath)) {
                        return std::make_shared<CscLocalFileResourceSpec>(obsoleteFullPath);
                    }
                }
            } catch(std::filesystem::filesystem_error const& ex) {
                // legacy location cannot be resolved - whatever, we choose the standard way!
            }
            return std::make_shared<CscResourceFromServiceSpec>(resourceId);
        }
 
    public:
        inline static ptr<CscResourceSpec> mireAstral = getServiceResourceSpecWithBackwardCompatibility(
                "mire_astral.png", "DetectorEngine/mire_astral_v0.2.png"
        );
        inline static ptr<CscResourceSpec> haarCascadeFrontalFace = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_frontalface.xml", "DetectorEngineFiles/haarcascade_frontalface_default.xml"
        );
        inline static ptr<CscResourceSpec> haarCascadeMireA = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_mireA_classifier.xml", "DetectorEngineFiles/mireA/mireA_classifier.xml"
        );
        inline static ptr<CscResourceSpec> haarCascadeMireB = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_mireB_classifier.xml", "DetectorEngineFiles/mireB/mireB_classifier.xml"
        );
        inline static ptr<CscResourceSpec> haarCascadePedestrian = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_pedestrian.xml", "DetectorEngineFiles/pedestrian/pedestrian.xml"
        );
        inline static ptr<CscResourceSpec> haarCascadeHandDefault = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_hand_default.xml", "conscience_test_data/haarcascade_hand_default.xml"
        );
        inline static ptr<CscResourceSpec> haarCascadeHandFist = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_hand_fist.xml", "conscience_test_data/haarcascade_hand_fist.xml"
        );
        inline static ptr<CscResourceSpec> haarCascadeHandPalm = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_hand_palm.xml", "conscience_test_data/haarcascade_hand_palm.xml"
        );
        inline static ptr<CscResourceSpec> haarCascadePerson = getServiceResourceSpecWithBackwardCompatibility(
                "haarcascade_person.xml", "DetectorEngineFiles/haarcascade_person.xml"
        );
 
        static ptr<CscResourceSpec> DNNConfigurationYoloV4(const string & dataSet) {
            return std::make_shared<CscResourceFromServiceSpec>("dnn_yolov4_" + dataSet + ".cfg");
        }
        
        static ptr<CscResourceSpec> DNNWeightsYoloV4(const string & dataSet) {
            return std::make_shared<CscResourceFromServiceSpec>("dnn_yolov4_" + dataSet + ".weights");
        }
        
        static ptr<CscResourceSpec> DNNNamesYoloV4(const string & dataSet) {
            return std::make_shared<CscResourceFromServiceSpec>("dnn_yolov4_" + dataSet + ".names.json");
        }
 
        static ptr<CscResourceSpec> DNNWeightsYoloV8() {
            return std::make_shared<CscResourceFromServiceSpec>("dnn_yolov8x.onnx");
        }
        
    };
}
 
#endif