145 lines
6.3 KiB
Objective-C
Executable File
145 lines
6.3 KiB
Objective-C
Executable File
//
|
||
// FURenderInput.h
|
||
// FURenderKit
|
||
//
|
||
// Created by liuyang on 2021/1/4.
|
||
//
|
||
|
||
#import <Foundation/Foundation.h>
|
||
#import <AVFoundation/AVFoundation.h>
|
||
#import "CNamaSDK.h"
|
||
|
||
typedef enum : NSUInteger {
|
||
FUImageBufferFormatRGBA,
|
||
FUImageBufferFormatBGRA,
|
||
FUImageBufferFormatYUV420V,
|
||
FUImageBufferFormatYUV420F,
|
||
FUImageBufferFormatYUVI420,
|
||
} FUImageBufferFormat;
|
||
|
||
typedef enum : NSUInteger {
|
||
FUImageOrientationUP,
|
||
FUImageOrientationRight,
|
||
FUImageOrientationDown,
|
||
FUImageOrientationLeft
|
||
} FUImageOrientation;
|
||
|
||
/// CPU image 数据
|
||
struct FUImageBuffer {
|
||
Byte *buffer0; // 格式与传入 buffer 类型对应关系：RGBA: rgba_buffer; BGRA: bgra_buffer; YUV420V/YUV420F/YUVI420: y_buffer
|
||
size_t stride0; // 格式与传入 stride 类型对应关系：RGBA: rgba_stride; BGRA: bgra_stride; YUV420V/YUV420F/YUVI420: y_stride
|
||
|
||
Byte *buffer1; // 格式与传入 buffer 类型对应关系：RGBA/BGRA: NULL; YUV420V/YUV420F: uv_buffer; YUVI420: u_buffer
|
||
size_t stride1; // 格式与传入 stride 类型对应关系：RGBA/BGRA: 0; YUV420V/YUV420F: uv_stride; YUVI420: u_stride
|
||
|
||
Byte *buffer2; // 格式与传入 buffer 类型对应关系：RGBA/BGRA: NULL; YUV420V/YUV420F: NULL; YUVI420: v_buffer
|
||
size_t stride2; // 格式与传入 stride 类型对应关系：RGBA/BGRA: 0; YUV420V/YUV420F: 0; YUVI420: v_stride
|
||
|
||
CGSize size; // 图像宽高
|
||
FUImageBufferFormat format; // 图像格式
|
||
|
||
};
|
||
typedef struct FUImageBuffer FUImageBuffer;
|
||
|
||
FUImageBuffer FUImageBufferMakeRGBA(Byte *buffer0, size_t width, size_t height, size_t stride0);
|
||
FUImageBuffer FUImageBufferMakeBGRA(Byte *buffer0, size_t width, size_t height, size_t stride0);
|
||
FUImageBuffer FUImageBufferMakeYUV420F(Byte *buffer0, Byte *buffer1, size_t width, size_t height, size_t stride0, size_t stride1);
|
||
FUImageBuffer FUImageBufferMakeYUV420V(Byte *buffer0, Byte *buffer1, size_t width, size_t height, size_t stride0, size_t stride1);
|
||
FUImageBuffer FUImageBufferMakeI420(Byte *buffer0, Byte *buffer1, Byte *buffer2, size_t width, size_t height, size_t stride0, size_t stride1, size_t stride2);
|
||
FUImageBuffer FUImageBufferMake(Byte *buffer0, Byte *buffer1, Byte *buffer2, size_t width, size_t height, size_t stride0, size_t stride1, size_t stride2, FUImageBufferFormat format);
|
||
|
||
|
||
/// 纹理数据
|
||
struct FUTexture {
|
||
GLuint ID; // 纹理ID
|
||
CGSize size; // 纹理宽高
|
||
};
|
||
typedef struct FUTexture FUTexture;
|
||
|
||
#pragma -mark FURenderConfig
|
||
@interface FURenderConfig : NSObject
|
||
|
||
/// 自定义输出结果的大小，当前只会对输出的纹理及pixelBuffer有效
|
||
@property (nonatomic, assign) CGSize customOutputSize;
|
||
|
||
/// 当前图片是否来源于前置摄像头，默认为 NO
|
||
@property (nonatomic, assign) BOOL isFromFrontCamera;
|
||
|
||
/// 当前图片是否来源于镜像摄像头，默认为 NO
|
||
@property (nonatomic, assign) BOOL isFromMirroredCamera;
|
||
|
||
/// 原始图像的朝向
|
||
@property (nonatomic, assign) FUImageOrientation imageOrientation;
|
||
|
||
/// 贴纸水平镜像
|
||
@property (nonatomic, assign) BOOL stickerFlipH;
|
||
|
||
/// 贴纸垂直镜像
|
||
@property (nonatomic, assign) BOOL stickerFlipV;
|
||
|
||
/// 重力开关，开启此功能可以根据已设置的 imageRotation 自动适配AI检测的方向。
|
||
@property (nonatomic, assign) BOOL gravityEnable;
|
||
|
||
/// 设置为YES 只会生效美颜结果
|
||
@property (nonatomic, assign) BOOL onlyRenderBeauty;
|
||
|
||
/// 设置输入pixelBuffer/imageBuffer的旋转方向，以使buffer数据与textureTransform作用后纹理的方向一致，该参数仅用于AI算法检测，不会改变buffer的方向或镜像属性
|
||
@property (nonatomic, assign) TRANSFORM_MATRIX bufferTransform;
|
||
|
||
/// 设置输入纹理的旋转及镜像信息，设置该属性会影响输出纹理的方向。默认基于 CPU 图像创建的纹理与CPU 图像成上下镜像关系，此时 textureTransform 对应的值为 DEFAULT，以此类推，如果对默认生成的纹理做了其他变换，则将该参数设置为对应的变换即可。
|
||
@property (nonatomic, assign) TRANSFORM_MATRIX textureTransform;
|
||
|
||
/// 设置输出pixelBuffer/imageBuffer的旋转方向，以使buffer数据与textureTransform作用后纹理的方向一致，该参数仅用于AI算法检测，不会改变buffer的方向或镜像属性
|
||
@property (nonatomic, assign) TRANSFORM_MATRIX outputTransform;
|
||
|
||
/// 是否渲染到当前的FBO，设置为YES时，返回的 FURenderOutput 内的所有数据均为空值。
|
||
@property (nonatomic, assign) BOOL renderToCurrentFBO;
|
||
|
||
//是否读回到输入的buffer, 如果手机在后台状态也不回写数据，返回的output对象内部buffer为空。
|
||
@property (nonatomic, assign) BOOL readBackToPixelBuffer;
|
||
|
||
/// 设置为YES 且 renderToCurrentFBO 为 NO 时，只会输出纹理，不会输出CPU层的图像。
|
||
@property (nonatomic, assign) BOOL onlyOutputTexture;
|
||
|
||
/// 控制输出图像是否保留透明信息，默认值为NO，可能会输出不透明的图像，开启该参数可以保持图像中透明信息。
|
||
@property (nonatomic, assign) BOOL keepAlpha;
|
||
|
||
/// 如果 renderKit 有 glDisplayView ，控制绘制结果将直接渲染到 glDisplayView
|
||
@property (nonatomic, assign) BOOL autoDisplay;
|
||
|
||
@end
|
||
|
||
#pragma -mark FURenderInput
|
||
@interface FURenderInput : NSObject
|
||
|
||
/// 输入的纹理
|
||
@property (nonatomic, assign) FUTexture texture;
|
||
|
||
/// 输入的 pixelBuffer
|
||
@property (nonatomic, assign) CVPixelBufferRef pixelBuffer;
|
||
|
||
/// 输入的 imageBuffer，如果同时传入了 pixelBuffer，将优先使用 pixelBuffer
|
||
/// 输入 imageBuffer，在 renderConfig 的 onlyOutputTexture 为 NO 时，render 结果会直接读会输入的 imageBuffer，大小格式与输入均保持一致。
|
||
@property (nonatomic, assign) FUImageBuffer imageBuffer;
|
||
|
||
/// 设置render相关的输入输出配置，详细参数请查看 FURenderConfig 类的接口注释。
|
||
@property (nonatomic, strong, readonly) FURenderConfig *renderConfig;
|
||
|
||
@end
|
||
|
||
|
||
#pragma -mark FURenderOutput
|
||
@interface FURenderOutput : NSObject
|
||
|
||
/// 输出的纹理
|
||
@property (nonatomic, assign) FUTexture texture;
|
||
|
||
/// 输出的 pixelBuffer
|
||
@property (nonatomic, assign) CVPixelBufferRef pixelBuffer;
|
||
|
||
/// 输出的 imageBuffer，内部数据与输入的 imageBuffer 一致。
|
||
@property (nonatomic, assign) FUImageBuffer imageBuffer;
|
||
|
||
@end
|
||
|