BDFaceAdjustParamsModel.m 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308
  1. //
  2. // BDFaceAdjustParamsModel.m
  3. // FaceSDKSample_IOS
  4. //
  5. // Created by Zhang,Jian(MBD) on 2020/12/3.
  6. // Copyright © 2020 Baidu. All rights reserved.
  7. //
  8. #import "BDFaceAdjustParamsModel.h"
  9. #import "BDFaceAdjustParamsItem.h"
  10. static NSString *const minLightIntensityString = @"最小光照值";
  11. static NSString *const maxLightIntensityString = @"最大光照值";
  12. static NSString *const ambiguityString = @"模糊度";
  13. static NSString *const leftEyeOcclusionString = @"左眼";
  14. static NSString *const rightEyeOcclusionString = @"右眼";
  15. static NSString *const noseOcclusionString = @"鼻子";
  16. static NSString *const mouthOcclusionString = @"嘴巴";
  17. static NSString *const leftFaceOcclusionString = @"左脸颊";
  18. static NSString *const rightFaceOcclusionString = @"右脸颊";
  19. static NSString *const lowerJawOcclusionString = @"下巴";
  20. static NSString *const upAndDownAngleString = @"俯仰角";
  21. static NSString *const leftAndRightAngleString = @"左右角";
  22. static NSString *const spinAngleString = @"旋转角";
  23. static NSString *const BDFaceSelectTypeNormalString = @"正常";
  24. static NSString *const BDFaceSelectTypeStrictString = @"严格";
  25. static NSString *const BDFaceSelectTypeLooseString = @"宽松";
  26. static NSString *const BDFaceSelectTypeCustomString = @"自定义";
  27. @implementation BDFaceAdjustParamsModel
  28. + (NSMutableArray *)loadItemsArray:(BDFaceAdjustParams *)config
  29. recorverText:(NSString *)text
  30. selectType:(BDFaceSelectType)type {
  31. NSMutableArray *array = [NSMutableArray array];
  32. [array addObject:[self lightArray:config]];
  33. [array addObject:[self occlusionArray:config]];
  34. [array addObject:[self positionArray:config]];
  35. if (type != BDFaceSelectTypeCustom) {
  36. [array addObject:[self recoverArray:text]];
  37. }
  38. return array;
  39. }
  40. /// 光照相关
  41. + (NSMutableArray *)lightArray:(BDFaceAdjustParams *)config {
  42. NSMutableArray *section1 = [NSMutableArray array];
  43. {
  44. /// 最小光照值
  45. BDFaceAdjustParamsItem *item = [[BDFaceAdjustParamsItem alloc] init];
  46. item.itemTitle = minLightIntensityString;
  47. item.minValue = 20.0f;
  48. item.maxValue = 60.0f;
  49. item.interval = 1.0f;
  50. item.currentValue = config.minLightIntensity;
  51. item.configDetailType = BDFaceAdjustParamsTypeMinLightIntensity;
  52. [section1 addObject:item];
  53. }
  54. {
  55. /// 最大光照值
  56. BDFaceAdjustParamsItem *item = [[BDFaceAdjustParamsItem alloc] init];
  57. item.itemTitle = maxLightIntensityString;
  58. item.minValue = 200.0f;
  59. item.maxValue = 240.0f;
  60. item.interval = 1.0f;
  61. item.currentValue = config.maxLightIntensity;
  62. item.configDetailType = BDFaceAdjustParamsTypeMaxLightIntensity;
  63. [section1 addObject:item];
  64. }
  65. {
  66. /// 模糊度
  67. BDFaceAdjustParamsItem *item = [[BDFaceAdjustParamsItem alloc] init];
  68. item.itemTitle = ambiguityString;
  69. item.minValue = 0.1f;
  70. item.maxValue = 0.9f;
  71. item.interval = 0.05f;
  72. item.currentValue = config.ambiguity;
  73. item.configDetailType = BDFaceAdjustParamsTypeAmbiguity;
  74. [section1 addObject:item];
  75. }
  76. return section1;
  77. }
  78. + (NSMutableArray *)occlusionArray:(BDFaceAdjustParams *)config {
  79. NSMutableArray *section2 = [NSMutableArray array];
  80. {
  81. /// 左眼
  82. BDFaceAdjustParamsItem *item1 = [[BDFaceAdjustParamsItem alloc] init];
  83. item1.itemTitle = leftEyeOcclusionString;
  84. item1.currentValue = config.leftEyeOcclusion;
  85. item1.configDetailType = BDFaceAdjustParamsTypeLeftEyeOcclusion;
  86. [section2 addObject:item1];
  87. }
  88. {
  89. /// 右眼
  90. BDFaceAdjustParamsItem *item2 = [[BDFaceAdjustParamsItem alloc] init];
  91. item2.itemTitle = rightEyeOcclusionString;
  92. item2.currentValue = config.rightEyeOcclusion;
  93. item2.configDetailType = BDFaceAdjustParamsTypeRightEyeOcclusion;
  94. [section2 addObject:item2];
  95. }
  96. {
  97. /// 鼻子
  98. BDFaceAdjustParamsItem *item3 = [[BDFaceAdjustParamsItem alloc] init];
  99. item3.itemTitle = noseOcclusionString;
  100. item3.currentValue = config.noseOcclusion;
  101. item3.configDetailType = BDFaceAdjustParamsTypeNoseOcclusion;
  102. [section2 addObject:item3];
  103. }
  104. {
  105. /// 嘴巴
  106. BDFaceAdjustParamsItem *item4 = [[BDFaceAdjustParamsItem alloc] init];
  107. item4.itemTitle = mouthOcclusionString;
  108. item4.currentValue = config.mouthOcclusion;
  109. item4.configDetailType = BDFaceAdjustParamsTypeMouthOcclusion;
  110. [section2 addObject:item4];}
  111. {
  112. BDFaceAdjustParamsItem *item5 = [[BDFaceAdjustParamsItem alloc] init];
  113. item5.itemTitle = leftFaceOcclusionString;
  114. item5.currentValue = config.leftFaceOcclusion;
  115. item5.configDetailType = BDFaceAdjustParamsTypeLeftFaceOcclusion;
  116. [section2 addObject:item5];
  117. }
  118. {
  119. BDFaceAdjustParamsItem *item6 = [[BDFaceAdjustParamsItem alloc] init];
  120. item6.itemTitle = rightFaceOcclusionString;
  121. item6.currentValue = config.rightFaceOcclusion;
  122. item6.configDetailType = BDFaceAdjustParamsTypeRightFaceOcclusion;
  123. [section2 addObject:item6];
  124. }
  125. {
  126. BDFaceAdjustParamsItem *item7 = [[BDFaceAdjustParamsItem alloc] init];
  127. item7.itemTitle = lowerJawOcclusionString;
  128. item7.currentValue = config.lowerJawOcclusion;
  129. item7.configDetailType = BDFaceAdjustParamsTypeLowerJawOcclusion;
  130. [section2 addObject:item7];
  131. }
  132. /// 设置最大最小值和间隔
  133. for (BDFaceAdjustParamsItem *each in section2) {
  134. each.minValue = 0.3f;
  135. each.maxValue = 1.0f;
  136. each.interval = 0.05f;
  137. }
  138. return section2;
  139. }
  140. + (NSMutableArray *)positionArray:(BDFaceAdjustParams *)config {
  141. NSMutableArray *section3 = [NSMutableArray array];
  142. {
  143. // 俯仰角
  144. BDFaceAdjustParamsItem *item1 = [[BDFaceAdjustParamsItem alloc] init];
  145. item1.itemTitle = upAndDownAngleString;
  146. item1.currentValue = config.upAndDownAngle;
  147. item1.configDetailType = BDFaceAdjustParamsTypeUpAndDownAngle;
  148. [section3 addObject:item1];
  149. }
  150. {
  151. /// 左右角
  152. BDFaceAdjustParamsItem *item2 = [[BDFaceAdjustParamsItem alloc] init];
  153. item2.itemTitle = leftAndRightAngleString;
  154. item2.currentValue = config.leftAndRightAngle;
  155. item2.configDetailType = BDFaceAdjustParamsTypeLeftAndRightAngle;
  156. [section3 addObject:item2];
  157. }
  158. {
  159. /// 旋转角
  160. BDFaceAdjustParamsItem *item3 = [[BDFaceAdjustParamsItem alloc] init];
  161. item3.itemTitle = spinAngleString;
  162. item3.currentValue = config.spinAngle;
  163. item3.configDetailType = BDFaceAdjustParamsTypeSpinAngle;
  164. [section3 addObject:item3];
  165. }
  166. for (BDFaceAdjustParamsItem *each in section3) {
  167. each.minValue = 10.0f;
  168. each.maxValue = 50.0f;
  169. each.interval = 1.0f;
  170. }
  171. return section3;
  172. }
  173. + (NSMutableArray *)recoverArray:(NSString *)text {
  174. NSMutableArray *section = [NSMutableArray array];
  175. BDFaceAdjustParamsItem *item = [[BDFaceAdjustParamsItem alloc] init];
  176. item.itemTitle = text;
  177. item.contentType = BDFaceAdjustParamsContentTypeRecoverToNormal;
  178. [section addObject:item];
  179. return section;
  180. }
  181. //+ (BDFaceAdjustParams *)getConfig:(NSArray *)array {
  182. // BDFaceAdjustParams *config = [BDFaceAdjustParams new];
  183. // for (NSArray *subArray in array) {
  184. // if (subArray && [subArray isKindOfClass:[NSArray class]]) {
  185. // for (BDFaceAdjustParamsItem *item in subArray) {
  186. // switch (item.configDetailType) {
  187. // /// 光照
  188. // case BDFaceAdjustParamsTypeMinLightIntensity:
  189. // {
  190. // config.minLightIntensity = item.currentValue;
  191. // }
  192. // break;
  193. // case BDFaceAdjustParamsTypeMaxLightIntensity:
  194. // {
  195. // config.maxLightIntensity = item.currentValue;
  196. // }
  197. // break;
  198. // case BDFaceAdjustParamsTypeAmbiguity:
  199. // {
  200. // config.ambiguity = item.currentValue;
  201. // }
  202. // break;
  203. // /// 遮挡
  204. // case BDFaceAdjustParamsTypeLeftEyeOcclusion:
  205. // {
  206. // config.leftEyeOcclusion = item.currentValue;
  207. // }
  208. // break;
  209. // case BDFaceAdjustParamsTypeRightEyeOcclusion:
  210. // {
  211. // config.rightFaceOcclusion = item.currentValue;
  212. // }
  213. // break;
  214. // case BDFaceAdjustParamsTypeNoseOcclusion:
  215. // {
  216. // config.noseOcclusion = item.currentValue;
  217. // }
  218. // break;
  219. // case BDFaceAdjustParamsTypeMouthOcclusion:
  220. // {
  221. // config.mouthOcclusion = item.currentValue;
  222. // }
  223. // break;
  224. // case BDFaceAdjustParamsTypeLeftFaceOcclusion:
  225. // {
  226. // config.leftFaceOcclusion = item.currentValue;
  227. // }
  228. // break;
  229. // case BDFaceAdjustParamsTypeRightFaceOcclusion:
  230. // {
  231. // config.rightFaceOcclusion = item.currentValue;
  232. // }
  233. // break;
  234. // case BDFaceAdjustParamsTypeLowerJawOcclusion:
  235. // {
  236. // config.lowerJawOcclusion = item.currentValue;
  237. // }
  238. // break;
  239. // /// 姿势
  240. // case BDFaceAdjustParamsTypeUpAndDownAngle:
  241. // {
  242. // config.upAndDownAngle = item.currentValue;
  243. // }
  244. // break;
  245. // case BDFaceAdjustParamsTypeLeftAndRightAngle:
  246. // {
  247. // config.leftAndRightAngle = item.currentValue;
  248. // }
  249. // break;
  250. // case BDFaceAdjustParamsTypeSpinAngle:
  251. // {
  252. // config.spinAngle = item.currentValue;
  253. // }
  254. // break;
  255. // default:
  256. // break;
  257. // }
  258. // }
  259. // }
  260. // }
  261. // return config;
  262. //}
  263. + (NSString *)getSelectTypeString:(BDFaceSelectType)type {
  264. NSString *value = BDFaceSelectTypeNormalString;
  265. switch (type) {
  266. case BDFaceSelectTypeStrict:
  267. {
  268. value = BDFaceSelectTypeStrictString;
  269. }
  270. break;
  271. case BDFaceSelectTypeLoose:
  272. {
  273. value = BDFaceSelectTypeLooseString;
  274. }
  275. break;
  276. case BDFaceSelectTypeCustom:
  277. {
  278. value = BDFaceSelectTypeCustomString;
  279. }
  280. break;
  281. default:
  282. break;
  283. }
  284. return value;
  285. }
  286. @end