wx_idphoto.h

Go to the documentation of this file.

 
#ifndef WX_IDPHOTO_H_
#define WX_IDPHOTO_H_
 
#include <memory>
#include <string>
#include <vector>
#include "wx_api.h"
#include "wx_exception.h"
#include "wx_image.h"
 
namespace wonxai {
 
class WXIDPhotoImpl;
 
class WX_API WXIDPhotoConfig {
public:
    int width;
 
    int height;
 
    float faceRatio;
 
    float eyesTop;
 
    float topEmpty[2];
 
    std::shared_ptr<WXImage> bgImage;
 
    unsigned short dpi;
 
    WXIDPhotoConfig(int w, int h):
        width(w),
        height(h),
        faceRatio(0.5f),
        eyesTop(0.4f),
        dpi(300),
        bgImage(nullptr) {
        topEmpty[0] = .0f;
        topEmpty[1] = .0f;
    }
 
    WXIDPhotoConfig():
        WXIDPhotoConfig(0, 0) {
    }
};
 
class WX_API WXIDPhotoSettings {
public:
    static std::shared_ptr<WXIDPhotoSettings> create();
 
    static WXIDPhotoSettings *create2();
 
    virtual ~WXIDPhotoSettings() = default;
 
    virtual void setValue(const std::string &key, int value) = 0;
 
    virtual void setValue(const std::string &key, float value) = 0;
 
    virtual void setValue(const std::string &key, bool value) = 0;
 
    virtual void setValue(const std::string &key, const std::string &data) = 0;
 
    virtual void setValue(const std::string &key, const std::vector<int> &arr) = 0;
 
    virtual void setValue(const std::string &key, const std::vector<float> &arr) = 0;
 
    virtual int getIntValue(const std::string &key, int defValue=0) = 0;
 
    virtual float getFloatValue(const std::string &key, float defValue=0) = 0;
 
    virtual bool getBoolValue(const std::string &key, bool defValue=false) = 0;
 
    virtual const std::string& getStringValue(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 hasKey(const std::string &key) = 0;
 
};
 
enum WXGlassesType{
    NoGlasses = 0,
    NormalGlasses = 1,
    DarkGlasses = 2
};
 
class WX_API WXIDPhotoPreDetectResult {
public:
    typedef enum WXGlassesType GlassesType;
 
public:
    virtual ~WXIDPhotoPreDetectResult();
 
    virtual int faceCount() = 0;
 
    virtual int upEmpty() = 0;
 
    virtual int rightEmpty() = 0;
 
    virtual int bottomEmpty() = 0;
 
    virtual int leftEmpty() = 0;
 
    virtual int sizeScore() = 0;
 
    virtual int darkillum() = 0;
 
    virtual int faceBlur() = 0;
 
    virtual int faceUnbalance() = 0;
 
    virtual GlassesType glasses() = 0;
 
    virtual int eyesClose() = 0;
 
    virtual int mouthClose() = 0;
 
    virtual int faceOcclusion() = 0;
 
    virtual const float* headPose() = 0;
 
    virtual int gaze() = 0;
 
protected:
    WXIDPhotoPreDetectResult();
};
 
class WX_API WXIDPhoto final {
public:
    WXIDPhoto(WXDeviceType deviceType=kDeviceCPU);
 
    ~WXIDPhoto();
 
    const std::shared_ptr<WXImage>
    makeIDPhoto(const std::shared_ptr<WXImage> &inImg,
                const WXIDPhotoConfig &config);
 
    WXImage* makeIDPhoto(const WXImage *inImg,
                         const WXIDPhotoConfig &config);
 
    const std::shared_ptr<WXImage>
    makeIDPhoto(const std::shared_ptr<WXImage> &inImg,
                const WXIDPhotoConfig &config,
                const std::shared_ptr<WXIDPhotoPreDetectResult> &preResult);
 
    WXImage *
    makeIDPhoto(const WXImage *inImg,
                const WXIDPhotoConfig &config,
                const WXIDPhotoPreDetectResult *preResult);
 
    std::shared_ptr<WXIDPhotoPreDetectResult>
    preDetect(const std::shared_ptr<WXImage> &inImg);
 
    WXIDPhotoPreDetectResult*
    preDetect(const WXImage *inImg);
 
    void setSettings(const std::shared_ptr<WXIDPhotoSettings> &settings);
 
    void setSettings(const WXIDPhotoSettings* settings);
 
private:
    std::unique_ptr<WXIDPhotoImpl> _impl;
};
 
} // end of namespace wonxai
 
 
#endif //WX_IDPHOTO_H_