wx_human_matting.h

Go to the documentation of this file.

 
#ifndef WX_HUMAN_MATTING_H_
#define WX_HUMAN_MATTING_H_
 
#include <cstdint>
#include <memory>
#include <string>
#include "wx_api.h"
#include "wx_exception.h"
#include "wx_image.h"
#ifdef ANDROID
#include <jni.h>
#endif
namespace wonxai {
class WXHumanMattingImpl;
 
 
#ifdef ANDROID
WX_API void WXSDKInitWithLicense(
        JNIEnv *jniEnv, jobject applicationContext,
        const char *appKey,
        const char *licenseFilePath);
 
WX_API void WXSDKInitWithLicense(
        JNIEnv *jniEnv, jobject applicationContext,
        const char *appKey,
        uint8_t *licData, size_t licSize);
#else
WX_API void WXSDKInitWithLicense(const char *appKey,
                              const char* licenseFilePath);
 
WX_API void WXSDKInitWithLicense(const char *appKey,
                              uint8_t *licData, size_t licSize);
 
#endif
 
WX_API void WXSDKTerminate();
 
WX_API const char *WXSDKVersion();
 
class WX_API WXSDKLicense {
public:
 
    static void setLicense(const std::string &licensePath);
 
    static void setLicense(uint8_t *buffer, size_t size);
 
    static void setLicense(const char *licensePath, const char *appKey);
 
    static void setLicense(uint8_t *buffer, size_t size, const char *appKey);
 
    static uint64_t validFrom();
 
    static uint64_t expiredAt();
 
    static const char *licenseId();
private:
    WXSDKLicense(){};
};
 
class WX_API WXHumanMatting final {
public:
    static void initialize();
 
    static void terminate();
 
    static const char *version();
 
    WXHumanMatting(WXDeviceType deviceType=kDeviceCPU);
 
    ~WXHumanMatting();
 
    void matting(const uint8_t *inRGB,
                 int width,
                 int height,
                 float *outAlpha);
 
    std::shared_ptr<WXImage> removeBackground(const std::shared_ptr<WXImage> &inImage,
                                              const std::shared_ptr<WXImage> &bkImage);
 
    WXImage* removeBackground(const WXImage *inImage,
                              const WXImage *bkImage);
 
private:
    std::unique_ptr<WXHumanMattingImpl> _impl;
};
} //namespace wonxai
 
#endif //WX_HUMAN_MATTING_H_