MiMO_iOS/ByteEffectLib/effect-sdk.framework/Headers/bef_effect_ai_pet_face.h

#ifndef ANDROIDDEMO_BEF_EFFECT_AI_PET_FACE_H
#define ANDROIDDEMO_BEF_EFFECT_AI_PET_FACE_H

#if defined(__ANDROID__) || defined(TARGET_OS_ANDROID)
#if BEF_EFFECT_ANDROID_WITH_JNI
#include<jni.h>
#endif
#endif

#include "bef_effect_ai_public_define.h"

#define AI_PET_MAX_POINT_NUM 90                     // 宠物脸的关键点的最大点数
#define AI_CAT_POINT_NUM 82                         // 猫：82点
#define AI_DOG_POINT_NUM 76                         // 狗：76点（不加耳朵）
#define AI_OTHER_POINT_NUM 4                        // 其它动物：4点(目前不支持)
#define AI_MAX_PET_NUM 10

#define AI_PET_OPEN_LEFT_EYE          0x00000001    // 左眼睛是否睁开
#define AI_PET_OPEN_RIGHT_EYE         0x00000002    // 右眼睛是否睁开
#define AI_PET_OPEN_MOUTH             0x00000004    // 嘴巴是否张开

typedef void *PetFaceHandle;                        ///< 关键点检测句柄

typedef enum {
    BEF_CAT                 =                       1,   // 猫
    BEF_DOG                 =                       2,   // 狗
    BEF_HUMAN               =                       3,   // 人（目前不支持）
    BEF_OTHERS              =                      99,   // 其它宠物类型（目前不支持）
}bef_ai_pet_face_type;


typedef enum {
    BEF_DetCat              =             0x00000001,  // 开启猫脸检测
    BEF_DetDog              =             0x00000002,  // 开启狗脸检测
    BEF_QuickMode           =             0x00000004,  // 开启快速版本
}bef_ai_pet_face_config_type;

typedef struct bef_ai_pet_info_st {
    bef_ai_pet_face_type type;                          /// < 宠物类型
    bef_ai_rect rect;                            /// < 代表面部的矩形区域
    float score;                            /// < 宠物脸检测的置信度
    bef_ai_fpoint points_array[AI_PET_MAX_POINT_NUM]; /// < 宠物脸关键点的数组
    float yaw;                              /// < 水平转角,真实度量的左负右正
    float pitch;                            /// < 俯仰角,真实度量的上负下正
    float roll;                             /// < 旋转角,真实度量的左负右正
    int id;                                 /// < faceID: 每个检测到的宠物脸拥有唯一id，跟踪丢失以后重新被检测到,会有一个新的id
    unsigned int action;                    /// < 脸部动作，目前只包括：左眼睛睁闭，右眼睛睁闭，嘴巴睁闭，
    /// < action 的第1，2，3位分别编码： 左眼睛睁闭，右眼睛睁闭，嘴巴睁闭，其余位数预留
    int ear_type;                                    ///< 判断是竖耳还是垂耳，竖耳为0，垂耳为1
} bef_ai_pet_face_info, *p_bef_ai_pet_face_info;

/// @brief 检测结果
typedef struct bef_ai_pet_face_result_st {
    bef_ai_pet_face_info p_faces[AI_MAX_PET_NUM];       ///< 检测到的宠物脸信息
    int face_count;                        ///< 检测到的宠物脸数目，p_faces 数组中，只有face_count个结果是有效的；
} bef_ai_pet_face_result, *p_bef_ai_pet_face_result;



/**
 * @brief 创建宠物脸检测的句柄
 *
 * @param [in] strModelPath 模型文件所在路径
 *
 * @param [in] config Config of pet face detect algorithm 宠物脸检测算法的配置
 *  可以配置只检测猫，只检测狗，或者同时检测猫狗
 *  例如：只检测狗 detect_config = bef_ai_pet_face_config_type::BEF_DetDog
 *       同时检测猫狗： 只检测狗 detect_config = bef_ai_pet_face_config_type::BEF_DetDog|bef_ai_pet_face_config_type::BEF_DetCat
 *
 * @param: max_face_num：指定最多能够检测到的宠物脸数目；
 *
 * @param [out] handle Created face detect handle
 *                     创建的宠物脸检测句柄
 * @return If succeed return BEF_RESULT_SUC, other value please see bef_effect_ai_public_define.h
 *         成功返回 BEF_RESULT_SUC, 失败返回相应错误码, 具体请参考 bef_effect_ai_public_define.h
 */
BEF_SDK_API
bef_effect_result_t
bef_effect_ai_pet_face_create(
        const char * strModelPath,
        long long config,
        unsigned int maxNum,
        bef_effect_handle_t *handle
);



/**
 * @brief 宠物脸检测
 * @param [in] handle Created face detect handle
 *                    已创建的宠物脸检测句柄
 * @param [in] image Image base address
 *                   输入图片的数据指针
 * @param [in] pixel_format Pixel format of input image
 *                          输入图片的格式
 * @param [in] image_width  Image width
 *                          输入图像的宽度 (以像素为单位)
 * @param [in] image_height Image height
 *                          输入图像的高度 (以像素为单位)
 * @param [in] image_stride Image stride in each row
 *                          输入图像每一行的步长 (以像素为单位)
 * @param [in] orientation Image orientation
 *                         输入图像的转向，具体请参考 bef_effect_ai_public_define.h 中的 bef_rotate_type
 * @param [out] p_pet_face_result  存放结果信息，需外部分配好内存，需保证空间大于等于设置的最大检测宠物脸数
 * @return If succeed return BEF_RESULT_SUC, other value please see bef_effect_ai_public_define.h
 *         成功返回 BEF_RESULT_SUC, 失败返回相应错误码, 具体请参考 bef_effect_ai_public_define.h
 */
BEF_SDK_API
bef_effect_result_t bef_effect_ai_pet_face_detect(
        bef_effect_handle_t handle,
        const unsigned char *image,
        bef_ai_pixel_format pixel_format,
        int image_width,
        int image_height,
        int image_stride,
        bef_ai_rotate_type orientation,
        bef_ai_pet_face_result *p_pet_face_result
);



/**
 * @param [in] handle Destroy the created pet face detect handle
 *                    销毁创建的宠物脸检测句柄
 */
BEF_SDK_API
bef_effect_result_t bef_effect_ai_pet_face_release(bef_effect_handle_t handle);



/**
 * @brief 宠物脸授权
 * @param [in] handle Created pet face handle
 *                    已创建的宠物脸检测句柄
 * @param [in] license 授权文件字符串
 * @param [in] length  授权文件字符串长度
 * @return If succeed return BEF_RESULT_SUC, other value please refer bef_effect_ai_public_define.h
 *         成功返回 BEF_RESULT_SUC, 授权码非法返回 BEF_RESULT_INVALID_LICENSE ，其它失败返回相应错误码, 具体请参考 bef_effect_ai_public_define.h
 */
#if defined(__ANDROID__) || defined(TARGET_OS_ANDROID)
#if BEF_EFFECT_ANDROID_WITH_JNI
BEF_SDK_API bef_effect_result_t bef_effect_ai_pet_face_check_license(JNIEnv *env, jobject context,
                                                                 bef_effect_handle_t handle,
                                                                 const char *licensePath);
#else
BEF_SDK_API bef_effect_result_t bef_effect_ai_pet_face_check_license(bef_effect_handle_t handle,
                                                                 const char *licensePath);
#endif
#else
#ifdef __APPLE__
BEF_SDK_API bef_effect_result_t bef_effect_ai_pet_face_check_license(bef_effect_handle_t handle, const char *licensePath);
#endif
#endif

BEF_SDK_API bef_effect_result_t
bef_effect_ai_pet_face_check_online_license(bef_effect_handle_t handle, const char *licensePath);
#endif //ANDROIDDEMO_BEF_EFFECT_AI_PET_FACE_H