TSAIDiyVideoPTVVC.swift 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. //
  2. // TSAIDiyVideoPTVVC.swift
  3. // AIEmoji
  4. //
  5. // Created by 100Years on 2025/7/28.
  6. //
  7. class TSAIDiyVideoPTVVC: TSAIDiyVideoTTBaseVC, UIPopoverPresentationControllerDelegate {
  8. lazy var aiModelView = TSAIModelElementView(frame: .zero)
  9. lazy var promptTextView: TSDiyVideoPromptElementView = {
  10. let promptTextView = TSDiyVideoPromptElementView(randomTextArray: kRandomImageToVideoArray) { [weak self] _ in
  11. guard let self = self else { return }
  12. }
  13. promptTextView.uploadClickAction = { [weak self] in
  14. guard let self = self else { return }
  15. TSDiscoverVC.enterSelectPhotos(target: self, style: .pictureToVideo, hintType: 0) { [weak self] image in
  16. guard let self = self else { return }
  17. self.viewModel.uploadImage = image
  18. }
  19. }
  20. promptTextView.deleteClickAction = { [weak self] in
  21. guard let self = self else { return }
  22. self.viewModel.uploadImage = nil
  23. }
  24. promptTextView.textChangedBlock = { [weak self] text in
  25. guard let self = self else { return }
  26. self.viewModel.prompt = text
  27. }
  28. promptTextView.sendBolck = { [weak self] text in
  29. guard let self = self else { return }
  30. self.viewModel.prompt = text
  31. }
  32. promptTextView.AIBolck = { [weak self] in
  33. guard let self = self else { return }
  34. let vc = TSChatViewController()
  35. vc.viewModel.uiStyle = .perfectHint
  36. vc.textBlock = { [weak self] text, _ in
  37. guard let self = self else { return }
  38. promptTextView.setText(text: text)
  39. }
  40. kPushVC(target: self, modelVC: vc)
  41. }
  42. return promptTextView
  43. }()
  44. lazy var resolution = TSResolutionElementView(aiType: viewModel.selectedAiModelType)
  45. lazy var length = TSVideoLengthElementView(aiType: viewModel.selectedAiModelType)
  46. lazy var movement = TSMovementElementView(aiType: viewModel.selectedAiModelType)
  47. var viewModel: TSAIDiyVideoPTVViewModel = .init()
  48. override var videoGenerateModel: TSAIDiyVideoGenerateModel {
  49. let model = TSAIDiyVideoGenerateModel()
  50. model.model = aiModelView.param
  51. model.duration = Int(length.param) ?? 0
  52. model.movement = movement.param
  53. model.resolution = resolution.param
  54. model.prompt = viewModel.prompt
  55. return model
  56. }
  57. override var elements: [any TSDiyVideoElement] {
  58. [
  59. aiModelView,
  60. promptTextView,
  61. resolution,
  62. length,
  63. movement,
  64. ]
  65. }
  66. override func createView() {
  67. super.createView()
  68. addObservers()
  69. addActions()
  70. }
  71. func addObservers() {
  72. viewModel.$selectedAiModelType.receive(on: DispatchQueue.main).sink { [weak self] type in
  73. self?.aiModelView.updateType(type: type)
  74. self?.length.updateAiModelType(aitype: type)
  75. self?.resolution.updateAiModelType(aitype: type)
  76. }.store(in: &cancellable)
  77. viewModel.$uploadImage.receive(on: DispatchQueue.main).dropFirst().sink { [weak self] image in
  78. self?.promptTextView.uploadView.upLoadImage = image
  79. self?.viewModel.judgeWasFinishedEdit()
  80. }.store(in: &cancellable)
  81. }
  82. func addActions() {
  83. aiModelView.addTapAction { [weak self] in
  84. guard let self = self else { return }
  85. // 初始化内容控制器
  86. let contentVC = TSDiyVideoPopoverViewController()
  87. contentVC.currentType = self.viewModel.selectedAiModelType
  88. contentVC.onSelected = { [weak self] type in
  89. guard let self = self else { return }
  90. viewModel.selectedAiModelType = type
  91. }
  92. // 设置模态呈现样式为Popover
  93. contentVC.modalPresentationStyle = .popover
  94. // 配置Popover属性
  95. if let popoverPresentationController = contentVC.popoverPresentationController {
  96. popoverPresentationController.delegate = self
  97. popoverPresentationController.sourceView = self.aiModelView
  98. popoverPresentationController.sourceRect = .init(origin: .init(x: self.aiModelView.bounds.origin.x, y: self.aiModelView.bounds.origin.y + 100), size: self.aiModelView.bounds.size)
  99. popoverPresentationController.permittedArrowDirections = []
  100. popoverPresentationController.backgroundColor = .fromHex("#222222")
  101. }
  102. // 呈现Popover
  103. self.present(contentVC, animated: true, completion: nil)
  104. }
  105. }
  106. // 关键: 确保在iPhone上以Popover形式显示而非全屏
  107. func adaptivePresentationStyle(for controller: UIPresentationController) -> UIModalPresentationStyle {
  108. return .none
  109. }
  110. }