matcher.h

См. документацию.

 
#ifndef _MATCHER_H_
#define _MATCHER_H_
 
#pragma once
 
#include "config.h"
#include "extractor.h"
#include "export.h"
 
 
namespace Diametrix {
 
    class TemplateDB {
    public:
        virtual void add(const std::shared_ptr<Diametrix::Template>& template0, const long long id) = 0;
        
        virtual void clear() = 0;                                   
        virtual size_t size() = 0;                                  
        virtual size_t maxSize() = 0;                               
 
        virtual void getLoadedIDs(std::vector<long long>& ids) = 0; 
        virtual bool isLoaded(const long long id) = 0;              
        
        virtual void save(const std::string& path) = 0;             
        virtual void load(const std::string& path) = 0;             
 
        virtual ~TemplateDB() = default;
    };
 
    std::shared_ptr<Diametrix::TemplateDB> DMXSDK_API constructTemplateDB(size_t maxSize, int batchSize = 1);
 
 
    class Matcher {
    public:
        virtual float match(const std::shared_ptr<Diametrix::Template>& template0,
                            const std::shared_ptr<Diametrix::Template>& template1) = 0;
 
        virtual void matchN(const std::vector<std::shared_ptr<Diametrix::Template>>& templates,
                            std::shared_ptr<Diametrix::TemplateDB>& db,
                            std::vector<float*>& results, long long * ids = nullptr) = 0;
 
        virtual void matchK(const std::vector<std::shared_ptr<Diametrix::Template>>& templates,
                            std::shared_ptr<Diametrix::TemplateDB>& db,
                            unsigned int K, float threshold,
                            std::vector<std::vector<std::pair<long long,float>>>& results) = 0;
 
        virtual ~Matcher() = default;
 
    };
 
    std::unique_ptr<Diametrix::Matcher> DMXSDK_API createMatcher(const std::shared_ptr<Diametrix::Config>& cfg);
}
 
 
#endif /* _MATCHER_H_ */