wx_idphoto_detect.h

Go to the documentation of this file.

 
#ifndef WX_IDPHOTO_DETECT_H_
#define WX_IDPHOTO_DETECT_H_
 
#include <memory>
#include <string>
#include <vector>
#include "wx_api.h"
#include "wx_exception.h"
#include "wx_image.h"
#include "wx_idphoto.h"
 
namespace wonxai {
 
class WXIDPhotoDetectorImpl;
 
class WX_API WXIDPhotoDetectResult {
 
public:
    virtual ~WXIDPhotoDetectResult();
 
    virtual int getInt(const std::string &key) = 0;
 
    virtual float getFloat(const std::string &key) = 0;
 
    virtual bool getBool(const std::string &key) = 0;
 
    virtual const std::string& getString(const std::string &key) = 0;
 
    virtual const std::vector<int> &getIntArray(const std::string &key) = 0;
 
    virtual const std::vector<float> &getFloatArray(const std::string &key) = 0;
 
    virtual bool exists(const std::string &key) = 0;
 
    virtual std::vector<std::string> keys() = 0;
 
    virtual std::string dumps() = 0;
 
protected:
    WXIDPhotoDetectResult();
};
 
class WX_API WXIDPhotoDetector final {
public:
    WXIDPhotoDetector(WXDeviceType deviceType=kDeviceCPU);
 
    ~WXIDPhotoDetector();
 
    const std::shared_ptr<WXIDPhotoDetectResult>
    detect(std::shared_ptr<WXImage>inImg);
 
    WXIDPhotoDetectResult *
    detect(const WXImage *inImg);
 
    void setSettings(const std::shared_ptr<WXIDPhotoSettings> &settings);
 
    void setSettings(const WXIDPhotoSettings *settings);
 
private:
    std::unique_ptr<WXIDPhotoDetectorImpl> _impl;
};
 
} // end of namespace wonxai
 
#endif //WX_IDPHOTO_DETECT_H_