//
//  FUManager.m
//  FULiveDemo
//
//  Created by 刘洋 on 2017/8/18.
//  Copyright © 2017年 刘洋. All rights reserved.
//

#import "FUManager.h"
#import "FUBeautyComponentManager.h"
#import "authpack.h"
#import <FURenderKit/FUGLDisplayView.h>

static FUManager *shareManager = NULL;

@interface FUManager ()

@property (nonatomic, assign) FUDevicePerformanceLevel devicePerformanceLevel;

@end

@implementation FUManager

+ (FUManager *)shareManager
{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        shareManager = [[FUManager alloc] init];
    });
    return shareManager;
}

- (instancetype)init
{
    if (self = [super init]) {
        
        CFAbsoluteTime startTime = CFAbsoluteTimeGetCurrent();

        NSString *controllerPath = [[NSBundle mainBundle] pathForResource:@"controller_cpp" ofType:@"bundle"];
        NSString *controllerConfigPath = [[NSBundle mainBundle] pathForResource:@"controller_config" ofType:@"bundle"];
        FUSetupConfig *setupConfig = [[FUSetupConfig alloc] init];
        setupConfig.authPack = FUAuthPackMake(g_auth_package, sizeof(g_auth_package));
        setupConfig.controllerPath = controllerPath;
        setupConfig.controllerConfigPath = controllerConfigPath;
        
        // 初始化 FURenderKit
        [FURenderKit setupWithSetupConfig:setupConfig];
        
        [FURenderKit setLogLevel:FU_LOG_LEVEL_INFO];
        
        self.devicePerformanceLevel = [FURenderKit devicePerformanceLevel];
        
        dispatch_async(dispatch_get_global_queue(0, 0), ^{
            // 加载人脸 AI 模型
            NSString *faceAIPath = [[NSBundle mainBundle] pathForResource:@"ai_face_processor" ofType:@"bundle"];
            [FUAIKit loadAIModeWithAIType:FUAITYPE_FACEPROCESSOR dataPath:faceAIPath];
            
            // 加载身体 AI 模型，注意：高性能机型加载ai_human_processor_gpu.bundle
            NSString *humanBundleName = self.devicePerformanceLevel == FUDevicePerformanceLevelHigh ? @"ai_human_processor_gpu" : @"ai_human_processor";
            NSString *bodyAIPath = [[NSBundle mainBundle] pathForResource:humanBundleName ofType:@"bundle"];
            [FUAIKit loadAIModeWithAIType:FUAITYPE_HUMAN_PROCESSOR dataPath:bodyAIPath];
            
            CFAbsoluteTime endTime = (CFAbsoluteTimeGetCurrent() - startTime);
            
            NSString *path = [[NSBundle mainBundle] pathForResource:@"tongue" ofType:@"bundle"];
            [FUAIKit loadTongueMode:path];
            
            //TODO: todo 是否需要用？？？？？
            /* 设置嘴巴灵活度 默认= 0*/ //
            float flexible = 0.5;
            [FUAIKit setFaceTrackParam:@"mouth_expression_more_flexible" value:flexible];
            NSLog(@"---%lf",endTime);
            
            // 设置人脸算法质量
            [FUAIKit shareKit].faceProcessorFaceLandmarkQuality = self.devicePerformanceLevel == FUDevicePerformanceLevelHigh ? FUFaceProcessorFaceLandmarkQualityHigh : FUFaceProcessorFaceLandmarkQualityMedium;
            
            // 设置小脸检测是否打开
            [FUAIKit shareKit].faceProcessorDetectSmallFace = self.devicePerformanceLevel == FUDevicePerformanceLevelHigh;
        });
        [FURenderKit shareRenderKit].beauty = [self defaultBeauty];
        [FUBeautyComponentManager.sharedManager beautyStyleSetAllDefaultValues];
        [FUAIKit shareKit].maxTrackFaces = 4;
        
        FUBeauty *beauty = [FURenderKit shareRenderKit].beauty;
        beauty.colorLevel = 1;
        beauty.redLevel = 1;
        beauty.cheekThinning = 1;
    }
    
    return self;
}


- (void)destoryItems {
    [FURenderKit shareRenderKit].beauty = nil;
    [FURenderKit shareRenderKit].bodyBeauty = nil;
    [FURenderKit shareRenderKit].makeup = nil;
    [[FURenderKit shareRenderKit].stickerContainer removeAllSticks];
}


- (void)onCameraChange {
    [FUAIKit resetTrackedResult];
}

- (FUBeauty *)defaultBeauty {
    NSString *path = [[NSBundle mainBundle] pathForResource:@"face_beautification" ofType:@"bundle"];
    FUBeauty *beauty = [[FUBeauty alloc] initWithPath:path name:@"FUBeauty"];
    beauty.colorLevel = 1;
    beauty.redLevel = 1;
    beauty.cheekThinning = 1;
    return beauty;
}


#pragma mark - VideoFilterDelegate

- (CVPixelBufferRef)renderItemsToPixelBuffer:(CVPixelBufferRef)frame {
    if (UIApplication.sharedApplication.applicationState == UIApplicationStateBackground) {
        return frame;
    } else {
        // 前台才调用 FURenderKit 渲染
        FURenderInput *input = [[FURenderInput alloc] init];
        input.pixelBuffer = frame;
        //默认图片内部的人脸始终是朝上，旋转屏幕也无需修改该属性。
        input.renderConfig.imageOrientation = FUImageOrientationUP;
        //开启重力感应，内部会自动计算正确方向，设置fuSetDefaultRotationMode，无须外面设置
        input.renderConfig.gravityEnable = YES;
        //如果来源相机捕获的图片一定要设置，否则将会导致内部检测异常
        input.renderConfig.isFromFrontCamera = YES;
        //该属性是指系统相机是否做了镜像: 一般情况前置摄像头出来的帧都是设置过镜像，所以默认需要设置下。如果相机属性未设置镜像，改属性不用设置。
        input.renderConfig.isFromMirroredCamera = YES;
        FURenderOutput *output = [[FURenderKit shareRenderKit] renderWithInput:input];
        return output.pixelBuffer;
    }
}

@end