FURenderViewModel.m 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324
  1. //
  2. // FURenderViewModel.m
  3. // FULiveDemo
  4. //
  5. // Created by 项林平 on 2022/7/15.
  6. // Copyright © 2022 FaceUnity. All rights reserved.
  7. //
  8. #import "FURenderViewModel.h"
  9. @interface FURenderViewModel ()<FURenderKitDelegate, FUCaptureCameraDataSource>
  10. /// 缓存渲染线程的人脸中心点
  11. @property (nonatomic, assign) CGPoint faceCenter;
  12. @property (nonatomic, assign) BOOL trackedCompleteFace;
  13. @property (nonatomic, assign) CGFloat inputBufferWidth;
  14. @property (nonatomic, assign) CGFloat inputBufferHeight;
  15. @end
  16. @implementation FURenderViewModel {
  17. // 计算帧率相关变量
  18. CFAbsoluteTime startTime, lastCalculateTime;
  19. int rate;
  20. NSTimeInterval currentCalculateTime;
  21. }
  22. - (instancetype)init {
  23. self = [super init];
  24. if (self) {
  25. self.rendering = YES;
  26. self.maxFaceNumber = 4;
  27. self.maxHumanNumber = 4;
  28. self.detectingParts = FUDetectingPartsFace;
  29. [FURenderKit shareRenderKit].delegate = self;
  30. [FURenderKit shareRenderKit].captureCamera.dataSource = self;
  31. if ((self.necessaryAIModelTypes & FUAIModelTypeFace) && self.loadAIModelAutomatically) {
  32. // 加载人脸AI模型
  33. [FURenderKitManager loadFaceAIModel];
  34. }
  35. if ((self.necessaryAIModelTypes & FUAIModelTypeHuman) && self.loadAIModelAutomatically) {
  36. // 加载人体AI模型
  37. [FURenderKitManager loadHumanAIModel:FUHumanSegmentationModeCPUCommon];
  38. }
  39. if ((self.necessaryAIModelTypes & FUAIModelTypeHand) && self.loadAIModelAutomatically) {
  40. // 加载手势AI模型
  41. [FURenderKitManager loadHandAIModel];
  42. // 设置未跟踪到手势时每次检测的间隔帧数为3
  43. [FUAIKit setHandDetectEveryFramesWhenNoHand:3];
  44. }
  45. // 设置设备性能相关细项
  46. [[FURenderKitManager sharedManager] setDevicePerformanceDetails];
  47. }
  48. return self;
  49. }
  50. - (void)setRendering:(BOOL)rendering{
  51. _rendering = rendering;
  52. }
  53. #pragma mark - Instance methods
  54. - (void)startCameraWithRenderView:(FUGLDisplayView *)renderView {
  55. // 需要音频轨道
  56. [FURenderKit shareRenderKit].internalCameraSetting.needsAudioTrack = YES;
  57. [[FURenderKit shareRenderKit] startInternalCamera];
  58. [FURenderKit shareRenderKit].glDisplayView = renderView;
  59. }
  60. - (void)stopCamera {
  61. [[FURenderKit shareRenderKit] stopInternalCamera];
  62. [FURenderKit shareRenderKit].glDisplayView = nil;
  63. }
  64. - (void)switchCameraOutputFormat {
  65. OSType format = [FURenderKit shareRenderKit].internalCameraSetting.format == kCVPixelFormatType_32BGRA ? kCVPixelFormatType_420YpCbCr8BiPlanarFullRange : kCVPixelFormatType_32BGRA;
  66. [FURenderKit shareRenderKit].internalCameraSetting.format = format;
  67. }
  68. - (void)switchCapturePreset:(AVCaptureSessionPreset)preset unsupportedPresetHandler:(void (^)(void))handler {
  69. if (![FURenderKit shareRenderKit].captureCamera) {
  70. return;
  71. }
  72. FUCaptureCamera *camera = [FURenderKit shareRenderKit].captureCamera;
  73. if ([camera changeSessionPreset:preset]) {
  74. [FURenderKit shareRenderKit].internalCameraSetting.sessionPreset = preset;
  75. } else {
  76. // 硬件不支持该分辨率
  77. if (handler) {
  78. handler();
  79. }
  80. }
  81. }
  82. - (void)switchCameraBetweenFrontAndRear:(BOOL)front unsupportedPresetHandler:(void (^)(void))handler {
  83. if (![FURenderKit shareRenderKit].captureCamera) {
  84. return;
  85. }
  86. FUCaptureCamera *camera = [FURenderKit shareRenderKit].captureCamera;
  87. if (![camera supportsAVCaptureSessionPreset:front]) {
  88. // 硬件不支持分辨率
  89. if (handler) {
  90. handler();
  91. }
  92. } else {
  93. [FURenderKit shareRenderKit].internalCameraSetting.position = front ? AVCaptureDevicePositionFront : AVCaptureDevicePositionBack;
  94. }
  95. }
  96. - (void)switchCameraFocusMode:(FUCaptureCameraFocusMode)mode {
  97. if (![FURenderKit shareRenderKit].captureCamera) {
  98. return;
  99. }
  100. [[FURenderKit shareRenderKit].captureCamera cameraChangeMode:mode];
  101. }
  102. - (void)setCameraFocusPoint:(CGPoint)point {
  103. if (![FURenderKit shareRenderKit].captureCamera) {
  104. return;
  105. }
  106. [[FURenderKit shareRenderKit].captureCamera focusWithMode:AVCaptureFocusModeAutoFocus exposeWithMode:AVCaptureExposureModeAutoExpose atDevicePoint:point monitorSubjectAreaChange:YES];
  107. }
  108. - (void)setCameraExposureValue:(CGFloat)value {
  109. if (![FURenderKit shareRenderKit].captureCamera) {
  110. return;
  111. }
  112. [[FURenderKit shareRenderKit].captureCamera setExposureValue:value];
  113. }
  114. - (void)resetCameraFocusAndExposureMode {
  115. if (![FURenderKit shareRenderKit].captureCamera) {
  116. return;
  117. }
  118. [[FURenderKit shareRenderKit].captureCamera resetFocusAndExposureModes];
  119. }
  120. - (void)resetCameraSettings {
  121. [FURenderKit shareRenderKit].internalCameraSetting.sessionPreset = AVCaptureSessionPreset1920x1080;
  122. [FURenderKit shareRenderKit].internalCameraSetting.position = AVCaptureDevicePositionFront;
  123. [FURenderKit shareRenderKit].internalCameraSetting.format = kCVPixelFormatType_32BGRA;
  124. }
  125. #pragma mark - Private methods
  126. - (void)updateFaceCenterPoint {
  127. CGPoint center = CGPointMake(0.5, 0.5);
  128. if ([FURenderKitManager faceTracked] && [FURenderKit shareRenderKit].captureCamera) {
  129. center = [self trackedFaceCenter];
  130. if ([FURenderKit shareRenderKit].captureCamera.isFrontCamera) {
  131. center = CGPointMake(center.y, center.x);
  132. } else {
  133. center = CGPointMake(center.y, 1 - center.x);
  134. }
  135. }
  136. self.faceCenter = center;
  137. }
  138. - (CGPoint)trackedFaceCenter {
  139. // 获取人脸信息
  140. float rect[4];
  141. [FUAIKit getFaceInfo:0 name:@"face_rect" pret:rect number:4];
  142. CGFloat minX = rect[0];
  143. CGFloat minY = rect[1];
  144. CGFloat maxX = rect[2];
  145. CGFloat maxY = rect[3];
  146. // 判断人脸是否完整
  147. if (minX < 0 || minY < 0 || maxX > self.inputBufferWidth || maxY > self.inputBufferHeight) {
  148. self.trackedCompleteFace = NO;
  149. } else {
  150. self.trackedCompleteFace = YES;
  151. }
  152. // 计算中心点的坐标
  153. CGFloat centerX = (minX + maxX) * 0.5;
  154. CGFloat centerY = (minY + maxY) * 0.5;
  155. // 转换坐标
  156. centerX = centerX / self.inputBufferWidth;
  157. centerY = centerY / self.inputBufferHeight;
  158. return CGPointMake(centerX, centerY);
  159. }
  160. #pragma mark - FURenderKitDelegate
  161. - (BOOL)renderKitShouldDoRender {
  162. return self.rendering;
  163. }
  164. - (void)renderKitWillRenderFromRenderInput:(FURenderInput *)renderInput {
  165. if (renderInput.pixelBuffer != NULL) {
  166. CVPixelBufferLockBaseAddress(renderInput.pixelBuffer, 0);
  167. self.inputBufferWidth = CVPixelBufferGetWidth(renderInput.pixelBuffer);
  168. self.inputBufferHeight = CVPixelBufferGetHeight(renderInput.pixelBuffer);
  169. CVPixelBufferUnlockBaseAddress(renderInput.pixelBuffer, 0);
  170. }
  171. [FURenderKitManager updateBeautyBlurEffect];
  172. if (self.delegate && [self.delegate respondsToSelector:@selector(renderWillInputPixelBuffer:)]) {
  173. [self.delegate renderWillInputPixelBuffer:renderInput.pixelBuffer];
  174. }
  175. startTime = CFAbsoluteTimeGetCurrent();
  176. }
  177. - (void)renderKitDidRenderToOutput:(FURenderOutput *)renderOutput {
  178. CFAbsoluteTime endTime = CFAbsoluteTimeGetCurrent();
  179. // 加一帧占用时间
  180. currentCalculateTime += (endTime - startTime);
  181. // 加帧数
  182. rate += 1;
  183. if (endTime - lastCalculateTime >= 1) {
  184. // 一秒钟计算一次
  185. int width = (int)CVPixelBufferGetWidth(renderOutput.pixelBuffer);
  186. int height = (int)CVPixelBufferGetHeight(renderOutput.pixelBuffer);
  187. NSString *debugString = [NSString stringWithFormat:@"resolution:\n%dx%d\nfps: %d\nrender time:\n%.0fms", width, height, rate, currentCalculateTime * 1000 / rate];
  188. if (self.delegate && [self.delegate respondsToSelector:@selector(renderDidOutputDebugInformations:)]) {
  189. [self.delegate renderDidOutputDebugInformations:debugString];
  190. }
  191. // 恢复计算数据
  192. lastCalculateTime = endTime;
  193. currentCalculateTime = 0;
  194. rate = 0;
  195. }
  196. // 更新人脸中心点
  197. [self updateFaceCenterPoint];
  198. if (self.detectingParts != FUDetectingPartsNone) {
  199. // 需要检查人脸/人体/手势检测状态
  200. if (self.delegate && [self.delegate respondsToSelector:@selector(renderShouldCheckDetectingStatus:)]) {
  201. [self.delegate renderShouldCheckDetectingStatus:self.detectingParts];
  202. }
  203. }
  204. }
  205. #pragma mark - FUCaptureCameraDataSource
  206. - (CGPoint)fuCaptureFaceCenterInImage:(FUCaptureCamera *)camera {
  207. return self.faceCenter;
  208. }
  209. #pragma mark - Setters
  210. - (void)setMaxFaceNumber:(NSInteger)maxFaceNumber {
  211. if (_maxFaceNumber == maxFaceNumber) {
  212. return;
  213. }
  214. _maxFaceNumber = maxFaceNumber;
  215. [FURenderKitManager setMaxFaceNumber:maxFaceNumber];
  216. }
  217. - (void)setMaxHumanNumber:(NSInteger)maxHumanNumber {
  218. if (_maxHumanNumber == maxHumanNumber) {
  219. return;
  220. }
  221. _maxHumanNumber = maxHumanNumber;
  222. [FURenderKitManager setMaxHumanNumber:maxHumanNumber];
  223. }
  224. #pragma mark - Overriding attributes
  225. - (FUAIModelType)necessaryAIModelTypes {
  226. return FUAIModelTypeFace;
  227. }
  228. - (BOOL)loadAIModelAutomatically {
  229. return YES;
  230. }
  231. - (BOOL)supportCaptureAndRecording {
  232. return YES;
  233. }
  234. - (BOOL)supportMediaRendering {
  235. return NO;
  236. }
  237. - (BOOL)supportPresetSelection {
  238. return NO;
  239. }
  240. - (BOOL)supportSwitchingOutputFormat {
  241. return YES;
  242. }
  243. - (BOOL)supportVideoRecording {
  244. return YES;
  245. }
  246. - (NSArray<AVCaptureSessionPreset> *)presets {
  247. return @[AVCaptureSessionPreset640x480, AVCaptureSessionPreset1280x720, AVCaptureSessionPreset1920x1080];
  248. }
  249. - (NSArray<NSString *> *)presetTitles {
  250. NSMutableArray *titles = [[NSMutableArray alloc] init];
  251. for (AVCaptureSessionPreset preset in self.presets) {
  252. if (preset == AVCaptureSessionPreset640x480) {
  253. [titles addObject:@"480x640"];
  254. } else if (preset == AVCaptureSessionPreset1280x720) {
  255. [titles addObject:@"720x1280"];
  256. } else if (preset == AVCaptureSessionPreset1920x1080) {
  257. [titles addObject:@"1080x1920"];
  258. }
  259. }
  260. return [titles copy];
  261. }
  262. - (BOOL)needsLoadingBeauty {
  263. return YES;
  264. }
  265. - (CGFloat)captureButtonBottomConstant {
  266. return FUHeightIncludeBottomSafeArea(10);
  267. }
  268. #pragma mark - Getters
  269. - (AVCaptureSessionPreset)capturePreset {
  270. return [FURenderKit shareRenderKit].internalCameraSetting.sessionPreset;
  271. }
  272. - (AVCaptureDevicePosition)captureDevicePostion {
  273. return [FURenderKit shareRenderKit].internalCameraSetting.position;
  274. }
  275. @end