1352 lines
58 KiB
Swift
1352 lines
58 KiB
Swift
//
|
||
// TravelAlbumPhotoPreviewViewController.swift
|
||
// suixinkan
|
||
//
|
||
|
||
import CoreImage
|
||
import Kingfisher
|
||
import SnapKit
|
||
import UIKit
|
||
|
||
/// 相册管理中的单张照片预览页，提供 AI 修图、删除和刷新操作。
|
||
final class TravelAlbumPhotoPreviewViewController: UIViewController {
|
||
/// 删除成功后通知相册管理页刷新当前素材列表。
|
||
var onDeleted: (() -> Void)?
|
||
|
||
private var previewState: TravelAlbumPhotoPreviewState
|
||
private let api: any TravelAlbumServing
|
||
private let imageView = UIImageView()
|
||
private let bottomBar = UIView()
|
||
private let bottomContentStack = UIStackView()
|
||
private let comparisonBar = UIStackView()
|
||
private let versionControl = UISegmentedControl(items: ["原图", "精修后"])
|
||
private var visibleDisplayModes: [TravelAlbumAIEditResultState.DisplayMode] = [.original, .edited]
|
||
private let highResolutionButton = UIButton(type: .system)
|
||
private let actionStack = UIStackView()
|
||
private let aiEditButton = TravelAlbumPreviewActionButton(
|
||
title: "AI修图",
|
||
icon: UIImage(named: "travel_album_preview_ai_edit") ?? UIImage(systemName: "wand.and.stars")
|
||
)
|
||
private let aiEditResultStore = TravelAlbumAIEditResultStore.shared
|
||
private let previewHeaderView = TravelAlbumPreviewHeaderView()
|
||
private var originalImage: UIImage?
|
||
private var originalImagesByMaterialID: [Int: UIImage] = [:]
|
||
|
||
init(materials: [TravelAlbumMaterial], initialIndex: Int, api: any TravelAlbumServing) {
|
||
precondition(!materials.isEmpty, "照片预览至少需要一张素材")
|
||
previewState = TravelAlbumPhotoPreviewState(materials: materials, initialIndex: initialIndex)
|
||
self.api = api
|
||
super.init(nibName: nil, bundle: nil)
|
||
// 与修图样式、封面模板预览保持同一种全屏容器，避免系统导航栏叠加。
|
||
modalPresentationStyle = .fullScreen
|
||
modalTransitionStyle = .crossDissolve
|
||
}
|
||
|
||
/// 兼容单张照片入口，并统一复用连续预览实现。
|
||
convenience init(material: TravelAlbumMaterial, api: any TravelAlbumServing) {
|
||
self.init(materials: [material], initialIndex: 0, api: api)
|
||
}
|
||
|
||
@available(*, unavailable)
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
|
||
override func viewDidLoad() {
|
||
super.viewDidLoad()
|
||
view.backgroundColor = .black
|
||
setupNavigationBar()
|
||
setupUI()
|
||
NotificationCenter.default.addObserver(
|
||
self,
|
||
selector: #selector(aiEditResultDidChange(_:)),
|
||
name: TravelAlbumAIEditResultStore.didChangeNotification,
|
||
object: aiEditResultStore
|
||
)
|
||
updateAIEditControls()
|
||
loadImage()
|
||
}
|
||
|
||
override var preferredStatusBarStyle: UIStatusBarStyle {
|
||
.lightContent
|
||
}
|
||
|
||
/// 使用沉浸式自定义头部展示返回入口、文件信息和当前照片位置。
|
||
private func setupNavigationBar() {
|
||
previewHeaderView.onBack = { [weak self] in
|
||
self?.backTapped()
|
||
}
|
||
view.addSubview(previewHeaderView)
|
||
previewHeaderView.snp.makeConstraints { make in
|
||
make.top.equalTo(view.safeAreaLayoutGuide)
|
||
make.leading.trailing.equalToSuperview()
|
||
make.height.equalTo(68)
|
||
}
|
||
updateNavigationTitle()
|
||
}
|
||
|
||
private func setupUI() {
|
||
imageView.contentMode = .scaleAspectFit
|
||
imageView.clipsToBounds = true
|
||
imageView.image = UIImage(systemName: "photo")
|
||
imageView.tintColor = UIColor.white.withAlphaComponent(0.5)
|
||
imageView.isUserInteractionEnabled = true
|
||
let previousGesture = UISwipeGestureRecognizer(target: self, action: #selector(photoSwiped(_:)))
|
||
previousGesture.direction = .right
|
||
let nextGesture = UISwipeGestureRecognizer(target: self, action: #selector(photoSwiped(_:)))
|
||
nextGesture.direction = .left
|
||
imageView.addGestureRecognizer(previousGesture)
|
||
imageView.addGestureRecognizer(nextGesture)
|
||
|
||
bottomBar.backgroundColor = .black
|
||
bottomBar.layer.borderColor = UIColor.white.withAlphaComponent(0.12).cgColor
|
||
bottomBar.layer.borderWidth = 1 / UIScreen.main.scale
|
||
bottomContentStack.axis = .vertical
|
||
bottomContentStack.spacing = 10
|
||
comparisonBar.axis = .horizontal
|
||
comparisonBar.alignment = .center
|
||
comparisonBar.distribution = .equalSpacing
|
||
comparisonBar.isHidden = true
|
||
|
||
versionControl.selectedSegmentIndex = 1
|
||
versionControl.selectedSegmentTintColor = UIColor.white.withAlphaComponent(0.2)
|
||
versionControl.setTitleTextAttributes([.foregroundColor: UIColor.white.withAlphaComponent(0.7)], for: .normal)
|
||
versionControl.setTitleTextAttributes([.foregroundColor: UIColor.white], for: .selected)
|
||
versionControl.addTarget(self, action: #selector(versionChanged), for: .valueChanged)
|
||
|
||
highResolutionButton.setTitle("查看高清图", for: .normal)
|
||
highResolutionButton.setTitleColor(.white, for: .normal)
|
||
highResolutionButton.titleLabel?.font = .systemFont(ofSize: 13, weight: .medium)
|
||
highResolutionButton.backgroundColor = UIColor.white.withAlphaComponent(0.18)
|
||
highResolutionButton.layer.cornerRadius = 16
|
||
highResolutionButton.addTarget(self, action: #selector(highResolutionTapped), for: .touchUpInside)
|
||
|
||
actionStack.axis = .horizontal
|
||
actionStack.distribution = .fillEqually
|
||
|
||
aiEditButton.addTarget(self, action: #selector(aiEditTapped), for: .touchUpInside)
|
||
actionStack.addArrangedSubview(aiEditButton)
|
||
[
|
||
("删除", "travel_album_preview_delete", "trash", #selector(deleteTapped)),
|
||
("刷新", "travel_album_preview_refresh", "arrow.clockwise", #selector(refreshTapped)),
|
||
].forEach { title, assetName, fallbackIcon, action in
|
||
let button = TravelAlbumPreviewActionButton(
|
||
title: title,
|
||
icon: UIImage(named: assetName) ?? UIImage(systemName: fallbackIcon)
|
||
)
|
||
button.addTarget(self, action: action, for: .touchUpInside)
|
||
actionStack.addArrangedSubview(button)
|
||
}
|
||
|
||
view.addSubview(imageView)
|
||
view.addSubview(bottomBar)
|
||
bottomBar.addSubview(bottomContentStack)
|
||
bottomContentStack.addArrangedSubview(comparisonBar)
|
||
bottomContentStack.addArrangedSubview(actionStack)
|
||
comparisonBar.addArrangedSubview(versionControl)
|
||
comparisonBar.addArrangedSubview(highResolutionButton)
|
||
bottomBar.snp.makeConstraints { make in
|
||
make.leading.trailing.bottom.equalToSuperview()
|
||
}
|
||
bottomContentStack.snp.makeConstraints { make in
|
||
make.top.equalToSuperview().offset(12)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.bottom.equalTo(view.safeAreaLayoutGuide).offset(-10)
|
||
}
|
||
comparisonBar.snp.makeConstraints { make in
|
||
make.height.equalTo(32).priority(.high)
|
||
}
|
||
versionControl.snp.makeConstraints { make in
|
||
make.width.equalTo(124)
|
||
make.height.equalTo(32)
|
||
}
|
||
highResolutionButton.snp.makeConstraints { make in
|
||
make.width.equalTo(112)
|
||
make.height.equalTo(32)
|
||
}
|
||
actionStack.snp.makeConstraints { make in
|
||
make.height.equalTo(58)
|
||
}
|
||
imageView.snp.makeConstraints { make in
|
||
make.top.equalTo(previewHeaderView.snp.bottom)
|
||
make.leading.trailing.equalToSuperview()
|
||
make.bottom.equalTo(bottomBar.snp.top).offset(-20)
|
||
}
|
||
}
|
||
|
||
/// 返回相册管理页；照片预览统一使用全屏模态展示。
|
||
@objc private func backTapped() {
|
||
closePreview()
|
||
}
|
||
|
||
/// 关闭当前预览；保留导航栈兼容分支，避免旧入口无法返回。
|
||
private func closePreview() {
|
||
if let navigationController,
|
||
navigationController.viewControllers.first !== self {
|
||
navigationController.popViewController(animated: true)
|
||
} else {
|
||
dismiss(animated: true)
|
||
}
|
||
}
|
||
|
||
/// 加载当前素材的原图，刷新时始终绕过 Kingfisher 内存缓存。
|
||
private func loadImage(forceRefresh: Bool = false) {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
if let assetName = material.bundledImageAssetName,
|
||
let bundledImage = UIImage(named: assetName) {
|
||
originalImagesByMaterialID[material.id] = bundledImage
|
||
originalImage = bundledImage
|
||
renderCurrentPhoto()
|
||
return
|
||
}
|
||
let imageURL = material.fileUrl.isEmpty ? material.coverUrl : material.fileUrl
|
||
guard let url = URL(string: imageURL), !imageURL.isEmpty else { return }
|
||
let options: KingfisherOptionsInfo = forceRefresh ? [.forceRefresh] : []
|
||
let materialID = material.id
|
||
KingfisherManager.shared.retrieveImage(with: url, options: options) { [weak self] result in
|
||
guard let self, case let .success(value) = result else { return }
|
||
Task { @MainActor in
|
||
self.originalImagesByMaterialID[materialID] = value.image
|
||
guard self.previewState.currentMaterial?.id == materialID else { return }
|
||
self.originalImage = value.image
|
||
self.renderCurrentPhoto()
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 响应左右滑动并切换至相邻照片。
|
||
@objc private func photoSwiped(_ gesture: UISwipeGestureRecognizer) {
|
||
let didMove: Bool
|
||
switch gesture.direction {
|
||
case .left:
|
||
didMove = previewState.moveNext()
|
||
case .right:
|
||
didMove = previewState.movePrevious()
|
||
default:
|
||
didMove = false
|
||
}
|
||
guard didMove else { return }
|
||
displayCurrentPhoto()
|
||
}
|
||
|
||
/// 更新当前照片的标题、原图和预览内容。
|
||
private func displayCurrentPhoto() {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
originalImage = originalImagesByMaterialID[material.id]
|
||
updateNavigationTitle()
|
||
renderCurrentPhoto()
|
||
updateAIEditControls()
|
||
if originalImage == nil {
|
||
loadImage()
|
||
}
|
||
}
|
||
|
||
/// 根据当前照片同步导航栏文件名与大小。
|
||
private func updateNavigationTitle() {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
previewHeaderView.configure(
|
||
title: material.fileName.isEmpty ? "照片预览" : material.fileName,
|
||
subtitle: TravelAlbumDisplayFormatter.fileSizeText(material.fileSize),
|
||
trailingText: "\(previewState.currentIndex + 1)/\(previewState.materials.count)"
|
||
)
|
||
}
|
||
|
||
@objc private func aiEditTapped() {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
let state = aiEditResultStore.state(for: material.id)
|
||
guard state.canStartEditing else { return }
|
||
guard let originalImage else {
|
||
showToast("照片加载中，请稍后再试")
|
||
return
|
||
}
|
||
let controller = TravelAlbumAIEditSheetViewController(sourceImage: originalImage, selectedPhotoCount: 1)
|
||
controller.onConfirm = { [weak self] selection in
|
||
guard let self else { return }
|
||
self.startAIEditing(
|
||
materialID: material.id,
|
||
sourceImage: originalImage,
|
||
selection: selection
|
||
)
|
||
}
|
||
present(controller, animated: true)
|
||
}
|
||
|
||
/// 启动本地演示修图任务；处理期间禁用入口，完成后用新结果覆盖旧结果。
|
||
private func startAIEditing(
|
||
materialID: Int,
|
||
sourceImage: UIImage,
|
||
selection: TravelAlbumAIEditSelection
|
||
) {
|
||
let state = aiEditResultStore.state(for: materialID)
|
||
guard state.canStartEditing else { return }
|
||
let existingCoverTemplate = state.appliedCoverTemplateID.flatMap { templateID in
|
||
TravelAlbumCoverTemplate.defaultOptions.first(where: { $0.id == templateID })
|
||
}
|
||
aiEditResultStore.startProcessing(materialIDs: [materialID])
|
||
updateAIEditControls()
|
||
|
||
let refinedImage = TravelAlbumAIEditImageProcessor.render(
|
||
effect: selection.refinedPreset.effect,
|
||
source: sourceImage
|
||
)
|
||
let atmosphereImage = selection.atmosphereOption.map { option in
|
||
TravelAlbumAIEditImageProcessor.render(effect: option.effect, source: sourceImage)
|
||
}
|
||
let coverImage = existingCoverTemplate.map { template in
|
||
TravelAlbumCoverTemplateImageProcessor.render(template: template, source: refinedImage)
|
||
}
|
||
let resultStore = aiEditResultStore
|
||
DispatchQueue.main.asyncAfter(deadline: .now() + 1.5) { [weak self] in
|
||
// 即使用户在处理中离开预览页，也要完成任务并让相册缩略图恢复可操作状态。
|
||
if selection.refinedPreset.effect == .original,
|
||
selection.atmosphereOption == nil,
|
||
existingCoverTemplate == nil {
|
||
resultStore.restoreOriginal(materialID: materialID)
|
||
} else {
|
||
resultStore.complete(
|
||
materialID: materialID,
|
||
presetID: selection.refinedPreset.id,
|
||
atmosphereOptionID: selection.atmosphereOption?.id,
|
||
coverTemplateID: existingCoverTemplate?.id,
|
||
editedImage: refinedImage,
|
||
atmosphereImage: atmosphereImage,
|
||
coverImage: coverImage
|
||
)
|
||
}
|
||
guard let self else { return }
|
||
if self.previewState.currentMaterial?.id == materialID {
|
||
let message: String
|
||
if selection.refinedPreset.effect == .original,
|
||
selection.atmosphereOption == nil,
|
||
existingCoverTemplate == nil {
|
||
message = "已还原为原图"
|
||
} else if selection.atmosphereOption != nil {
|
||
message = "AI修图完成，已生成 2 个结果"
|
||
} else {
|
||
message = "AI修图完成"
|
||
}
|
||
self.showToast(message)
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 根据当前照片的对比选项展示原图或最新修图结果。
|
||
private func renderCurrentPhoto() {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
let state = aiEditResultStore.state(for: material.id)
|
||
let displayImage: UIImage?
|
||
switch state.displayMode {
|
||
case .original:
|
||
displayImage = originalImagesByMaterialID[material.id]
|
||
case .edited:
|
||
displayImage = aiEditResultStore.editedImage(for: material.id) ?? originalImagesByMaterialID[material.id]
|
||
case .atmosphere:
|
||
displayImage = aiEditResultStore.atmosphereImage(for: material.id)
|
||
?? aiEditResultStore.editedImage(for: material.id)
|
||
?? originalImagesByMaterialID[material.id]
|
||
case .cover:
|
||
displayImage = aiEditResultStore.coverImage(for: material.id)
|
||
?? aiEditResultStore.editedImage(for: material.id)
|
||
?? originalImagesByMaterialID[material.id]
|
||
}
|
||
UIView.transition(with: imageView, duration: 0.2, options: .transitionCrossDissolve) {
|
||
self.imageView.image = displayImage ?? UIImage(systemName: "photo")
|
||
self.imageView.tintColor = UIColor.white.withAlphaComponent(0.5)
|
||
}
|
||
}
|
||
|
||
/// 刷新修图对比栏和 AI 按钮的可用状态。
|
||
private func updateAIEditControls() {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
let state = aiEditResultStore.state(for: material.id)
|
||
comparisonBar.isHidden = !state.hasEditedResult
|
||
updateVersionControl(for: state)
|
||
aiEditButton.isEnabled = state.canStartEditing
|
||
aiEditButton.alpha = state.canStartEditing ? 1 : 0.35
|
||
UIView.animate(withDuration: 0.2) {
|
||
self.view.layoutIfNeeded()
|
||
}
|
||
}
|
||
|
||
@objc private func versionChanged() {
|
||
guard let material = previewState.currentMaterial,
|
||
visibleDisplayModes.indices.contains(versionControl.selectedSegmentIndex) else { return }
|
||
aiEditResultStore.selectDisplayMode(
|
||
visibleDisplayModes[versionControl.selectedSegmentIndex],
|
||
materialID: material.id
|
||
)
|
||
renderCurrentPhoto()
|
||
}
|
||
|
||
/// 根据当前实际结果动态展示原图、精修后、氛围感和封面切换项。
|
||
private func updateVersionControl(for state: TravelAlbumAIEditResultState) {
|
||
var modes: [TravelAlbumAIEditResultState.DisplayMode] = [.original, .edited]
|
||
if state.hasAtmosphereResult {
|
||
modes.append(.atmosphere)
|
||
}
|
||
if state.isCover {
|
||
modes.append(.cover)
|
||
}
|
||
if visibleDisplayModes != modes {
|
||
visibleDisplayModes = modes
|
||
versionControl.removeAllSegments()
|
||
modes.enumerated().forEach { index, mode in
|
||
let title: String
|
||
switch mode {
|
||
case .original: title = "原图"
|
||
case .edited: title = "精修后"
|
||
case .atmosphere: title = "氛围感"
|
||
case .cover: title = "封面"
|
||
}
|
||
versionControl.insertSegment(withTitle: title, at: index, animated: false)
|
||
}
|
||
let width: CGFloat
|
||
switch modes.count {
|
||
case 4: width = 220
|
||
case 3: width = 180
|
||
default: width = 124
|
||
}
|
||
versionControl.snp.updateConstraints { make in
|
||
make.width.equalTo(width)
|
||
}
|
||
}
|
||
versionControl.selectedSegmentIndex = visibleDisplayModes.firstIndex(of: state.displayMode) ?? 1
|
||
}
|
||
|
||
@objc private func aiEditResultDidChange(_ notification: Notification) {
|
||
guard let materialID = notification.userInfo?[TravelAlbumAIEditResultStore.materialIDUserInfoKey] as? Int,
|
||
previewState.currentMaterial?.id == materialID else { return }
|
||
renderCurrentPhoto()
|
||
updateAIEditControls()
|
||
}
|
||
|
||
@objc private func highResolutionTapped() {
|
||
guard let image = imageView.image else { return }
|
||
let controller = TravelAlbumImagePreviewViewController(image: image, title: "高清图")
|
||
controller.modalPresentationStyle = .fullScreen
|
||
controller.modalTransitionStyle = .crossDissolve
|
||
present(controller, animated: true)
|
||
}
|
||
|
||
@objc private func deleteTapped() {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
guard !material.isPurchased else {
|
||
showToast("已购照片不可删除")
|
||
return
|
||
}
|
||
let alert = UIAlertController(title: "删除照片", message: "删除后无法恢复，确定删除这张照片吗？", preferredStyle: .alert)
|
||
alert.addAction(UIAlertAction(title: "取消", style: .cancel))
|
||
alert.addAction(UIAlertAction(title: "删除", style: .destructive) { [weak self] _ in
|
||
guard let self else { return }
|
||
Task {
|
||
do {
|
||
try await self.api.deleteMaterial(id: material.id)
|
||
await MainActor.run {
|
||
self.originalImagesByMaterialID.removeValue(forKey: material.id)
|
||
self.aiEditResultStore.remove(materialID: material.id)
|
||
self.onDeleted?()
|
||
if self.previewState.removeCurrent() {
|
||
self.displayCurrentPhoto()
|
||
} else {
|
||
self.closePreview()
|
||
}
|
||
}
|
||
} catch {
|
||
await MainActor.run {
|
||
self.showToast(error.localizedDescription.isEmpty ? "删除失败" : error.localizedDescription)
|
||
}
|
||
}
|
||
}
|
||
})
|
||
present(alert, animated: true)
|
||
}
|
||
|
||
@objc private func refreshTapped() {
|
||
guard let material = previewState.currentMaterial else { return }
|
||
imageView.image = UIImage(systemName: "arrow.triangle.2.circlepath")
|
||
imageView.tintColor = UIColor.white.withAlphaComponent(0.5)
|
||
originalImage = nil
|
||
originalImagesByMaterialID.removeValue(forKey: material.id)
|
||
loadImage(forceRefresh: true)
|
||
}
|
||
|
||
private func showToast(_ message: String) {
|
||
let alert = UIAlertController(title: nil, message: message, preferredStyle: .alert)
|
||
present(alert, animated: true)
|
||
DispatchQueue.main.asyncAfter(deadline: .now() + 1.2) { [weak alert] in
|
||
alert?.dismiss(animated: true)
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 全屏预览页底部的图文操作按钮。
|
||
private final class TravelAlbumPreviewActionButton: UIButton {
|
||
init(title: String, icon: UIImage?) {
|
||
super.init(frame: .zero)
|
||
var configuration = UIButton.Configuration.plain()
|
||
configuration.image = icon
|
||
configuration.title = title
|
||
configuration.imagePlacement = .top
|
||
configuration.imagePadding = 6
|
||
configuration.baseForegroundColor = .white
|
||
configuration.titleTextAttributesTransformer = UIConfigurationTextAttributesTransformer { attributes in
|
||
var attributes = attributes
|
||
attributes.font = .systemFont(ofSize: 13)
|
||
return attributes
|
||
}
|
||
self.configuration = configuration
|
||
}
|
||
|
||
@available(*, unavailable)
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
}
|
||
|
||
/// 相册预览统一头部，提供单一返回入口、两行标题和可选的右侧状态信息。
|
||
private final class TravelAlbumPreviewHeaderView: UIView {
|
||
var onBack: (() -> Void)?
|
||
|
||
private let backButton = UIButton(type: .system)
|
||
private let titleLabel = UILabel()
|
||
private let subtitleLabel = UILabel()
|
||
private let titleStack = UIStackView()
|
||
private let trailingLabel = UILabel()
|
||
|
||
override init(frame: CGRect) {
|
||
super.init(frame: frame)
|
||
setupUI()
|
||
}
|
||
|
||
@available(*, unavailable)
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
|
||
/// 刷新头部标题、辅助说明和右侧状态。
|
||
func configure(title: String, subtitle: String?, trailingText: String?) {
|
||
titleLabel.text = title
|
||
subtitleLabel.text = subtitle
|
||
subtitleLabel.isHidden = subtitle?.isEmpty != false
|
||
trailingLabel.text = trailingText
|
||
trailingLabel.isHidden = trailingText?.isEmpty != false
|
||
}
|
||
|
||
private func setupUI() {
|
||
backgroundColor = .black
|
||
|
||
var backConfiguration = UIButton.Configuration.plain()
|
||
backConfiguration.image = UIImage(
|
||
systemName: "chevron.left",
|
||
withConfiguration: UIImage.SymbolConfiguration(pointSize: 17, weight: .semibold)
|
||
)
|
||
backConfiguration.baseForegroundColor = .white
|
||
backConfiguration.contentInsets = .zero
|
||
backButton.configuration = backConfiguration
|
||
backButton.backgroundColor = UIColor.white.withAlphaComponent(0.10)
|
||
backButton.layer.cornerRadius = 20
|
||
backButton.accessibilityLabel = "返回"
|
||
backButton.addTarget(self, action: #selector(backTapped), for: .touchUpInside)
|
||
|
||
titleLabel.textColor = .white
|
||
titleLabel.font = .systemFont(ofSize: 17, weight: .semibold)
|
||
titleLabel.textAlignment = .center
|
||
titleLabel.lineBreakMode = .byTruncatingMiddle
|
||
|
||
subtitleLabel.textColor = UIColor.white.withAlphaComponent(0.55)
|
||
subtitleLabel.font = .systemFont(ofSize: 12, weight: .regular)
|
||
subtitleLabel.textAlignment = .center
|
||
|
||
titleStack.axis = .vertical
|
||
titleStack.alignment = .fill
|
||
titleStack.spacing = 3
|
||
titleStack.addArrangedSubview(titleLabel)
|
||
titleStack.addArrangedSubview(subtitleLabel)
|
||
|
||
trailingLabel.textColor = UIColor.white.withAlphaComponent(0.78)
|
||
trailingLabel.font = .monospacedDigitSystemFont(ofSize: 12, weight: .medium)
|
||
trailingLabel.textAlignment = .center
|
||
trailingLabel.backgroundColor = UIColor.white.withAlphaComponent(0.10)
|
||
trailingLabel.layer.cornerRadius = 14
|
||
trailingLabel.clipsToBounds = true
|
||
|
||
addSubview(backButton)
|
||
addSubview(titleStack)
|
||
addSubview(trailingLabel)
|
||
backButton.snp.makeConstraints { make in
|
||
make.leading.equalToSuperview().offset(16)
|
||
make.centerY.equalToSuperview()
|
||
make.width.height.equalTo(40)
|
||
}
|
||
trailingLabel.snp.makeConstraints { make in
|
||
make.trailing.equalToSuperview().offset(-16)
|
||
make.centerY.equalToSuperview()
|
||
make.width.greaterThanOrEqualTo(44)
|
||
make.height.equalTo(28)
|
||
}
|
||
titleStack.snp.makeConstraints { make in
|
||
make.center.equalToSuperview()
|
||
make.leading.greaterThanOrEqualTo(backButton.snp.trailing).offset(12)
|
||
make.trailing.lessThanOrEqualTo(trailingLabel.snp.leading).offset(-12)
|
||
make.width.lessThanOrEqualTo(230)
|
||
}
|
||
}
|
||
|
||
@objc private func backTapped() {
|
||
onBack?()
|
||
}
|
||
}
|
||
|
||
/// 图片统一大图预览页，支持双指缩放并沿用相册预览的沉浸式头部。
|
||
private final class TravelAlbumImagePreviewViewController: UIViewController, UIScrollViewDelegate {
|
||
private let image: UIImage
|
||
private let previewTitle: String
|
||
private let previewHeaderView = TravelAlbumPreviewHeaderView()
|
||
private let scrollView = UIScrollView()
|
||
private let imageView = UIImageView()
|
||
|
||
init(image: UIImage, title: String = "高清图") {
|
||
self.image = image
|
||
previewTitle = title
|
||
super.init(nibName: nil, bundle: nil)
|
||
}
|
||
|
||
@available(*, unavailable)
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
|
||
override func viewDidLoad() {
|
||
super.viewDidLoad()
|
||
view.backgroundColor = .black
|
||
previewHeaderView.configure(
|
||
title: previewTitle,
|
||
subtitle: "双指缩放查看细节",
|
||
trailingText: nil
|
||
)
|
||
previewHeaderView.onBack = { [weak self] in
|
||
self?.dismiss(animated: true)
|
||
}
|
||
|
||
scrollView.delegate = self
|
||
scrollView.minimumZoomScale = 1
|
||
scrollView.maximumZoomScale = 4
|
||
scrollView.showsHorizontalScrollIndicator = false
|
||
scrollView.showsVerticalScrollIndicator = false
|
||
imageView.image = image
|
||
imageView.contentMode = .scaleAspectFit
|
||
|
||
view.addSubview(previewHeaderView)
|
||
view.addSubview(scrollView)
|
||
scrollView.addSubview(imageView)
|
||
previewHeaderView.snp.makeConstraints { make in
|
||
make.top.equalTo(view.safeAreaLayoutGuide)
|
||
make.leading.trailing.equalToSuperview()
|
||
make.height.equalTo(68)
|
||
}
|
||
scrollView.snp.makeConstraints { make in
|
||
make.top.equalTo(previewHeaderView.snp.bottom)
|
||
make.leading.trailing.bottom.equalToSuperview()
|
||
}
|
||
imageView.snp.makeConstraints { make in
|
||
make.edges.equalTo(scrollView.contentLayoutGuide)
|
||
make.width.height.equalTo(scrollView.frameLayoutGuide)
|
||
}
|
||
}
|
||
|
||
func viewForZooming(in scrollView: UIScrollView) -> UIView? {
|
||
imageView
|
||
}
|
||
|
||
override var preferredStatusBarStyle: UIStatusBarStyle {
|
||
.lightContent
|
||
}
|
||
}
|
||
|
||
/// AI 修图底部弹窗，统一选择原图精修、可选氛围感修图和封面模板。
|
||
final class TravelAlbumAIEditSheetViewController: UIViewController {
|
||
var onConfirm: ((TravelAlbumAIEditSelection) -> Void)?
|
||
|
||
private let sourceImage: UIImage?
|
||
private let selectedPhotoCount: Int
|
||
private let presets = TravelAlbumEditPreset.defaultOptions
|
||
private let atmosphereOptions = TravelAlbumAtmosphereEditOption.defaultOptions
|
||
private let coverTemplates = TravelAlbumCoverTemplate.defaultOptions
|
||
private let showsCoverTemplates: Bool
|
||
private var selectedPresetID = TravelAlbumEditPreset.defaultOptions[0].id
|
||
private var selectedAtmosphereOptionID: String?
|
||
private var selectedCoverTemplateID: String?
|
||
private let scrollView = UIScrollView()
|
||
private let contentView = UIView()
|
||
private let tipsContainerView = UIView()
|
||
private let tipsLabel = UILabel()
|
||
private let quotaInfoView = UIView()
|
||
private var collectionView: UICollectionView!
|
||
private var dataSource: UICollectionViewDiffableDataSource<Int, TravelAlbumEditPreset>!
|
||
private var atmosphereCollectionView: UICollectionView!
|
||
private var atmosphereDataSource: UICollectionViewDiffableDataSource<Int, TravelAlbumAtmosphereEditOption>!
|
||
private var coverCollectionView: UICollectionView!
|
||
private var coverDataSource: UICollectionViewDiffableDataSource<Int, TravelAlbumCoverTemplate>!
|
||
|
||
init(sourceImage: UIImage?, selectedPhotoCount: Int) {
|
||
self.sourceImage = sourceImage
|
||
self.selectedPhotoCount = max(selectedPhotoCount, 1)
|
||
showsCoverTemplates = TravelAlbumCoverTemplate.shouldShow(for: selectedPhotoCount)
|
||
selectedCoverTemplateID = showsCoverTemplates ? TravelAlbumCoverTemplate.defaultOptions.first?.id : nil
|
||
super.init(nibName: nil, bundle: nil)
|
||
modalPresentationStyle = .pageSheet
|
||
if let sheetPresentationController {
|
||
let contentIdentifier = UISheetPresentationController.Detent.Identifier("aiEditContent")
|
||
let desiredHeight: CGFloat = showsCoverTemplates ? 920 : 770
|
||
let contentDetent = UISheetPresentationController.Detent.custom(identifier: contentIdentifier) { context in
|
||
return min(desiredHeight, context.maximumDetentValue)
|
||
}
|
||
sheetPresentationController.detents = [contentDetent, .large()]
|
||
sheetPresentationController.selectedDetentIdentifier = contentIdentifier
|
||
sheetPresentationController.prefersGrabberVisible = true
|
||
}
|
||
}
|
||
|
||
@available(*, unavailable)
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
|
||
override func viewDidLoad() {
|
||
super.viewDidLoad()
|
||
view.backgroundColor = AppColor.pageBackground
|
||
setupUI()
|
||
applySnapshot()
|
||
}
|
||
|
||
private func setupUI() {
|
||
let titleLabel = UILabel()
|
||
titleLabel.text = "AI修图"
|
||
titleLabel.font = .systemFont(ofSize: 18, weight: .semibold)
|
||
titleLabel.textColor = AppColor.textPrimary
|
||
titleLabel.textAlignment = .center
|
||
|
||
let countLabel = UILabel()
|
||
countLabel.text = "已选择 \(selectedPhotoCount) 张照片"
|
||
countLabel.font = .systemFont(ofSize: 15)
|
||
countLabel.textColor = AppColor.textSecondary
|
||
countLabel.textAlignment = .center
|
||
|
||
tipsContainerView.backgroundColor = UIColor(hex: 0xEAF2FF)
|
||
tipsContainerView.layer.cornerRadius = 10
|
||
tipsLabel.font = .systemFont(ofSize: 13)
|
||
tipsLabel.textColor = UIColor(hex: 0x43617F)
|
||
tipsLabel.numberOfLines = 0
|
||
tipsContainerView.addSubview(tipsLabel)
|
||
tipsLabel.snp.makeConstraints { make in
|
||
make.edges.equalToSuperview().inset(12)
|
||
}
|
||
|
||
let presetTitleLabel = UILabel()
|
||
presetTitleLabel.text = "原图精修"
|
||
presetTitleLabel.font = .systemFont(ofSize: 15, weight: .semibold)
|
||
presetTitleLabel.textColor = AppColor.textPrimary
|
||
|
||
let layout = UICollectionViewFlowLayout()
|
||
layout.scrollDirection = .horizontal
|
||
layout.minimumLineSpacing = 12
|
||
layout.itemSize = CGSize(width: 116, height: 142)
|
||
collectionView = UICollectionView(frame: .zero, collectionViewLayout: layout)
|
||
collectionView.backgroundColor = .clear
|
||
collectionView.showsHorizontalScrollIndicator = false
|
||
collectionView.delegate = self
|
||
collectionView.register(TravelAlbumEditPresetCell.self, forCellWithReuseIdentifier: TravelAlbumEditPresetCell.reuseIdentifier)
|
||
dataSource = UICollectionViewDiffableDataSource<Int, TravelAlbumEditPreset>(collectionView: collectionView) { [weak self] collectionView, indexPath, preset in
|
||
let cell = collectionView.dequeueReusableCell(withReuseIdentifier: TravelAlbumEditPresetCell.reuseIdentifier, for: indexPath) as! TravelAlbumEditPresetCell
|
||
cell.apply(preset: preset, sourceImage: self?.sourceImage, selected: self?.selectedPresetID == preset.id)
|
||
cell.onPreview = { [weak self] in
|
||
guard let self else { return }
|
||
let previewImage = self.sourceImage.map {
|
||
TravelAlbumAIEditImageProcessor.render(effect: preset.effect, source: $0)
|
||
}
|
||
self.presentStylePreview(title: preset.title, image: previewImage)
|
||
}
|
||
return cell
|
||
}
|
||
|
||
let atmosphereHeaderView = UIView()
|
||
let atmosphereTitleLabel = UILabel()
|
||
atmosphereTitleLabel.text = "氛围感修图"
|
||
atmosphereTitleLabel.font = .systemFont(ofSize: 15, weight: .semibold)
|
||
atmosphereTitleLabel.textColor = AppColor.textPrimary
|
||
let optionalLabel = UILabel()
|
||
optionalLabel.text = "选填"
|
||
optionalLabel.font = .systemFont(ofSize: 12, weight: .medium)
|
||
optionalLabel.textColor = AppColor.primary
|
||
optionalLabel.backgroundColor = UIColor(hex: 0xEAF2FF)
|
||
optionalLabel.textAlignment = .center
|
||
optionalLabel.layer.cornerRadius = 13
|
||
optionalLabel.clipsToBounds = true
|
||
atmosphereHeaderView.addSubview(atmosphereTitleLabel)
|
||
atmosphereHeaderView.addSubview(optionalLabel)
|
||
atmosphereTitleLabel.snp.makeConstraints { make in
|
||
make.leading.centerY.equalToSuperview()
|
||
}
|
||
optionalLabel.snp.makeConstraints { make in
|
||
make.leading.equalTo(atmosphereTitleLabel.snp.trailing).offset(8)
|
||
make.centerY.equalToSuperview()
|
||
make.width.equalTo(48)
|
||
make.height.equalTo(26)
|
||
}
|
||
|
||
let atmosphereLayout = UICollectionViewFlowLayout()
|
||
atmosphereLayout.scrollDirection = .horizontal
|
||
atmosphereLayout.minimumLineSpacing = 12
|
||
atmosphereLayout.itemSize = CGSize(width: 116, height: 142)
|
||
atmosphereCollectionView = UICollectionView(frame: .zero, collectionViewLayout: atmosphereLayout)
|
||
atmosphereCollectionView.backgroundColor = .clear
|
||
atmosphereCollectionView.showsHorizontalScrollIndicator = false
|
||
atmosphereCollectionView.delegate = self
|
||
atmosphereCollectionView.register(
|
||
TravelAlbumEditPresetCell.self,
|
||
forCellWithReuseIdentifier: TravelAlbumEditPresetCell.reuseIdentifier
|
||
)
|
||
atmosphereDataSource = UICollectionViewDiffableDataSource<Int, TravelAlbumAtmosphereEditOption>(
|
||
collectionView: atmosphereCollectionView
|
||
) { [weak self] collectionView, indexPath, option in
|
||
let cell = collectionView.dequeueReusableCell(
|
||
withReuseIdentifier: TravelAlbumEditPresetCell.reuseIdentifier,
|
||
for: indexPath
|
||
) as! TravelAlbumEditPresetCell
|
||
cell.apply(
|
||
atmosphereOption: option,
|
||
sourceImage: self?.sourceImage,
|
||
selected: self?.selectedAtmosphereOptionID == option.id
|
||
)
|
||
cell.onPreview = { [weak self] in
|
||
guard let self else { return }
|
||
let previewImage = self.sourceImage.map {
|
||
TravelAlbumAIEditImageProcessor.render(effect: option.effect, source: $0)
|
||
}
|
||
self.presentStylePreview(title: option.title, image: previewImage)
|
||
}
|
||
return cell
|
||
}
|
||
updateAtmosphereSelection()
|
||
|
||
let coverHeaderView = UIView()
|
||
coverHeaderView.isHidden = !showsCoverTemplates
|
||
let coverTitleLabel = UILabel()
|
||
coverTitleLabel.text = "封面风格模板"
|
||
coverTitleLabel.font = .systemFont(ofSize: 15, weight: .semibold)
|
||
coverTitleLabel.textColor = AppColor.textPrimary
|
||
let freeGiftLabel = UILabel()
|
||
freeGiftLabel.text = "赠送 · 不占精修张数"
|
||
freeGiftLabel.font = .systemFont(ofSize: 12, weight: .medium)
|
||
freeGiftLabel.textColor = UIColor(hex: 0xF29A38)
|
||
freeGiftLabel.backgroundColor = UIColor(hex: 0xFFF3DE)
|
||
freeGiftLabel.textAlignment = .center
|
||
freeGiftLabel.layer.cornerRadius = 13
|
||
freeGiftLabel.clipsToBounds = true
|
||
coverHeaderView.addSubview(coverTitleLabel)
|
||
coverHeaderView.addSubview(freeGiftLabel)
|
||
coverTitleLabel.snp.makeConstraints { make in
|
||
make.leading.centerY.equalToSuperview()
|
||
}
|
||
freeGiftLabel.snp.makeConstraints { make in
|
||
make.trailing.centerY.equalToSuperview()
|
||
make.height.equalTo(26)
|
||
make.width.equalTo(148)
|
||
}
|
||
|
||
let coverLayout = UICollectionViewFlowLayout()
|
||
coverLayout.scrollDirection = .horizontal
|
||
coverLayout.minimumLineSpacing = 12
|
||
coverLayout.itemSize = CGSize(width: 116, height: 142)
|
||
coverCollectionView = UICollectionView(frame: .zero, collectionViewLayout: coverLayout)
|
||
coverCollectionView.backgroundColor = .clear
|
||
coverCollectionView.showsHorizontalScrollIndicator = false
|
||
coverCollectionView.isHidden = !showsCoverTemplates
|
||
coverCollectionView.delegate = self
|
||
coverCollectionView.register(
|
||
TravelAlbumCoverTemplateCell.self,
|
||
forCellWithReuseIdentifier: TravelAlbumCoverTemplateCell.reuseIdentifier
|
||
)
|
||
coverDataSource = UICollectionViewDiffableDataSource<Int, TravelAlbumCoverTemplate>(
|
||
collectionView: coverCollectionView
|
||
) { [weak self] collectionView, indexPath, template in
|
||
let cell = collectionView.dequeueReusableCell(
|
||
withReuseIdentifier: TravelAlbumCoverTemplateCell.reuseIdentifier,
|
||
for: indexPath
|
||
) as! TravelAlbumCoverTemplateCell
|
||
cell.apply(
|
||
template: template,
|
||
sourceImage: self?.sourceImage,
|
||
selected: self?.selectedCoverTemplateID == template.id
|
||
)
|
||
cell.onPreview = { [weak self] in
|
||
guard let self else { return }
|
||
let previewImage = TravelAlbumCoverTemplateImageProcessor.render(
|
||
template: template,
|
||
source: self.sourceImage
|
||
)
|
||
self.presentStylePreview(title: template.title, image: previewImage)
|
||
}
|
||
return cell
|
||
}
|
||
|
||
let modeTitleLabel = UILabel()
|
||
modeTitleLabel.text = "修图模式"
|
||
modeTitleLabel.font = .systemFont(ofSize: 15)
|
||
modeTitleLabel.textColor = AppColor.textSecondary
|
||
|
||
quotaInfoView.backgroundColor = .white
|
||
quotaInfoView.layer.cornerRadius = 12
|
||
quotaInfoView.accessibilityLabel = "AI精修按张收费，剩余9张"
|
||
quotaInfoView.accessibilityTraits = .staticText
|
||
let modeLabel = UILabel()
|
||
modeLabel.text = "AI精修 · 按张收费"
|
||
modeLabel.font = .systemFont(ofSize: 16, weight: .medium)
|
||
modeLabel.textColor = AppColor.textPrimary
|
||
let quotaLabel = UILabel()
|
||
quotaLabel.text = "剩余 9 张"
|
||
quotaLabel.font = .systemFont(ofSize: 14, weight: .medium)
|
||
quotaLabel.textColor = AppColor.primary
|
||
quotaLabel.backgroundColor = UIColor(hex: 0xEAF2FF)
|
||
quotaLabel.textAlignment = .center
|
||
quotaLabel.layer.cornerRadius = 15
|
||
quotaLabel.clipsToBounds = true
|
||
quotaInfoView.addSubview(modeLabel)
|
||
quotaInfoView.addSubview(quotaLabel)
|
||
modeLabel.snp.makeConstraints { make in
|
||
make.leading.equalToSuperview().offset(16)
|
||
make.centerY.equalToSuperview()
|
||
}
|
||
quotaLabel.snp.makeConstraints { make in
|
||
make.trailing.equalToSuperview().offset(-16)
|
||
make.centerY.equalToSuperview()
|
||
make.width.equalTo(82)
|
||
make.height.equalTo(30)
|
||
}
|
||
|
||
let cancelButton = makeBottomButton(title: "取消", isPrimary: false)
|
||
cancelButton.addTarget(self, action: #selector(cancelTapped), for: .touchUpInside)
|
||
let confirmButton = makeBottomButton(title: "确定", isPrimary: true)
|
||
confirmButton.addTarget(self, action: #selector(confirmTapped), for: .touchUpInside)
|
||
let buttonStack = UIStackView(arrangedSubviews: [cancelButton, confirmButton])
|
||
buttonStack.axis = .horizontal
|
||
buttonStack.spacing = 12
|
||
buttonStack.distribution = .fillEqually
|
||
|
||
view.addSubview(scrollView)
|
||
view.addSubview(buttonStack)
|
||
scrollView.addSubview(contentView)
|
||
[
|
||
titleLabel,
|
||
countLabel,
|
||
tipsContainerView,
|
||
presetTitleLabel,
|
||
collectionView,
|
||
atmosphereHeaderView,
|
||
atmosphereCollectionView,
|
||
coverHeaderView,
|
||
coverCollectionView,
|
||
modeTitleLabel,
|
||
quotaInfoView,
|
||
].forEach(contentView.addSubview)
|
||
scrollView.showsVerticalScrollIndicator = false
|
||
scrollView.alwaysBounceVertical = showsCoverTemplates
|
||
scrollView.snp.makeConstraints { make in
|
||
make.top.leading.trailing.equalToSuperview()
|
||
make.bottom.equalTo(buttonStack.snp.top).offset(-12)
|
||
}
|
||
contentView.snp.makeConstraints { make in
|
||
make.edges.equalTo(scrollView.contentLayoutGuide)
|
||
make.width.equalTo(scrollView.frameLayoutGuide)
|
||
}
|
||
buttonStack.snp.makeConstraints { make in
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.bottom.equalTo(view.safeAreaLayoutGuide).offset(-16)
|
||
make.height.equalTo(48)
|
||
}
|
||
titleLabel.snp.makeConstraints { make in
|
||
make.top.equalToSuperview().offset(28)
|
||
make.leading.trailing.equalToSuperview().inset(24)
|
||
}
|
||
countLabel.snp.makeConstraints { make in
|
||
make.top.equalTo(titleLabel.snp.bottom).offset(12)
|
||
make.leading.trailing.equalTo(titleLabel)
|
||
}
|
||
tipsContainerView.snp.makeConstraints { make in
|
||
make.top.equalTo(countLabel.snp.bottom).offset(14)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
}
|
||
presetTitleLabel.snp.makeConstraints { make in
|
||
make.top.equalTo(tipsContainerView.snp.bottom).offset(20)
|
||
make.leading.equalToSuperview().offset(20)
|
||
}
|
||
collectionView.snp.makeConstraints { make in
|
||
make.top.equalTo(presetTitleLabel.snp.bottom).offset(12)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.height.equalTo(142)
|
||
}
|
||
atmosphereHeaderView.snp.makeConstraints { make in
|
||
make.top.equalTo(collectionView.snp.bottom).offset(20)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.height.equalTo(26)
|
||
}
|
||
atmosphereCollectionView.snp.makeConstraints { make in
|
||
make.top.equalTo(atmosphereHeaderView.snp.bottom).offset(12)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.height.equalTo(142)
|
||
}
|
||
coverHeaderView.snp.makeConstraints { make in
|
||
make.top.equalTo(atmosphereCollectionView.snp.bottom).offset(showsCoverTemplates ? 20 : 0)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.height.equalTo(showsCoverTemplates ? 26 : 0)
|
||
}
|
||
coverCollectionView.snp.makeConstraints { make in
|
||
make.top.equalTo(coverHeaderView.snp.bottom).offset(showsCoverTemplates ? 12 : 0)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.height.equalTo(showsCoverTemplates ? 142 : 0)
|
||
}
|
||
modeTitleLabel.snp.makeConstraints { make in
|
||
make.top.equalTo(coverCollectionView.snp.bottom).offset(20)
|
||
make.leading.equalToSuperview().offset(20)
|
||
}
|
||
quotaInfoView.snp.makeConstraints { make in
|
||
make.top.equalTo(modeTitleLabel.snp.bottom).offset(12)
|
||
make.leading.trailing.equalToSuperview().inset(20)
|
||
make.height.equalTo(62)
|
||
make.bottom.equalToSuperview().offset(-20)
|
||
}
|
||
}
|
||
|
||
private func makeBottomButton(title: String, isPrimary: Bool) -> UIButton {
|
||
let button = UIButton(type: .system)
|
||
button.setTitle(title, for: .normal)
|
||
button.setTitleColor(isPrimary ? .white : AppColor.textPrimary, for: .normal)
|
||
button.titleLabel?.font = .systemFont(ofSize: 16, weight: .semibold)
|
||
button.backgroundColor = isPrimary ? AppColor.primary : .white
|
||
button.layer.cornerRadius = 10
|
||
button.layer.borderWidth = isPrimary ? 0 : 1
|
||
button.layer.borderColor = UIColor(hex: 0xD5D5D5).cgColor
|
||
return button
|
||
}
|
||
|
||
private func applySnapshot() {
|
||
var snapshot = NSDiffableDataSourceSnapshot<Int, TravelAlbumEditPreset>()
|
||
snapshot.appendSections([0])
|
||
snapshot.appendItems(presets)
|
||
dataSource.apply(snapshot, animatingDifferences: false)
|
||
|
||
var atmosphereSnapshot = NSDiffableDataSourceSnapshot<Int, TravelAlbumAtmosphereEditOption>()
|
||
atmosphereSnapshot.appendSections([0])
|
||
atmosphereSnapshot.appendItems(atmosphereOptions)
|
||
atmosphereDataSource.apply(atmosphereSnapshot, animatingDifferences: false)
|
||
|
||
guard showsCoverTemplates else { return }
|
||
var coverSnapshot = NSDiffableDataSourceSnapshot<Int, TravelAlbumCoverTemplate>()
|
||
coverSnapshot.appendSections([0])
|
||
coverSnapshot.appendItems(coverTemplates)
|
||
coverDataSource.apply(coverSnapshot, animatingDifferences: false)
|
||
}
|
||
|
||
@objc private func cancelTapped() {
|
||
dismiss(animated: true)
|
||
}
|
||
|
||
/// 刷新氛围感修图的可选状态，并用 Tips 明确不同组合会生成几个结果。
|
||
private func updateAtmosphereSelection() {
|
||
let isAtmosphereSelected = selectedAtmosphereOptionID != nil
|
||
if showsCoverTemplates {
|
||
tipsLabel.text = isAtmosphereSelected
|
||
? "Tips：本次显示 3 类风格。每张照片生成「精修后 + 氛围感」2 个结果，第一张照片另生成所选封面结果。"
|
||
: "Tips：氛围感修图为选填，可横向选择一种样式；选中后每张照片会额外生成 1 个独立结果，第一张照片仍另生成封面。"
|
||
} else {
|
||
tipsLabel.text = isAtmosphereSelected
|
||
? "Tips：已同时选择原图精修和氛围感修图，每张照片会生成「精修后 + 氛围感」2 个独立结果。"
|
||
: "Tips：原图精修、氛围感修图为两类独立模板；同时选择这两种模板时，同一张原图将并行生成两份不同的修图成品。"
|
||
}
|
||
}
|
||
|
||
/// 展示当前样式的大图预览；原图尚未加载时给出明确提示，不改变当前选择。
|
||
private func presentStylePreview(title: String, image: UIImage?) {
|
||
guard let image else {
|
||
let alert = UIAlertController(
|
||
title: "暂时无法预览",
|
||
message: "当前照片仍在加载，请稍后再试。",
|
||
preferredStyle: .alert
|
||
)
|
||
alert.addAction(UIAlertAction(title: "知道了", style: .default))
|
||
present(alert, animated: true)
|
||
return
|
||
}
|
||
let controller = TravelAlbumImagePreviewViewController(image: image, title: title)
|
||
controller.modalPresentationStyle = .fullScreen
|
||
controller.modalTransitionStyle = .crossDissolve
|
||
present(controller, animated: true)
|
||
}
|
||
|
||
@objc private func confirmTapped() {
|
||
guard let preset = presets.first(where: { $0.id == selectedPresetID }) else { return }
|
||
let coverTemplate = selectedCoverTemplateID.flatMap { selectedID in
|
||
coverTemplates.first(where: { $0.id == selectedID })
|
||
}
|
||
let atmosphereOption = selectedAtmosphereOptionID.flatMap { selectedID in
|
||
atmosphereOptions.first(where: { $0.id == selectedID })
|
||
}
|
||
let selection = TravelAlbumAIEditSelection(
|
||
refinedPreset: preset,
|
||
atmosphereOption: atmosphereOption,
|
||
coverTemplate: coverTemplate
|
||
)
|
||
onConfirm?(selection)
|
||
dismiss(animated: true)
|
||
}
|
||
}
|
||
|
||
extension TravelAlbumAIEditSheetViewController: UICollectionViewDelegate {
|
||
func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
|
||
if collectionView === coverCollectionView {
|
||
guard let template = coverDataSource.itemIdentifier(for: indexPath) else { return }
|
||
selectedCoverTemplateID = template.id
|
||
collectionView.visibleCells.compactMap { $0 as? TravelAlbumCoverTemplateCell }.forEach { cell in
|
||
guard let visiblePath = collectionView.indexPath(for: cell),
|
||
let visibleTemplate = coverDataSource.itemIdentifier(for: visiblePath) else { return }
|
||
cell.setSelected(visibleTemplate.id == selectedCoverTemplateID)
|
||
}
|
||
return
|
||
}
|
||
|
||
if collectionView === atmosphereCollectionView {
|
||
guard let option = atmosphereDataSource.itemIdentifier(for: indexPath) else { return }
|
||
// 氛围感为非必选项，再次点击当前样式即可取消本次氛围感结果。
|
||
selectedAtmosphereOptionID = selectedAtmosphereOptionID == option.id ? nil : option.id
|
||
collectionView.visibleCells.compactMap { $0 as? TravelAlbumEditPresetCell }.forEach { cell in
|
||
guard let visiblePath = collectionView.indexPath(for: cell),
|
||
let visibleOption = atmosphereDataSource.itemIdentifier(for: visiblePath) else { return }
|
||
cell.setSelected(visibleOption.id == selectedAtmosphereOptionID)
|
||
}
|
||
updateAtmosphereSelection()
|
||
return
|
||
}
|
||
|
||
guard let preset = dataSource.itemIdentifier(for: indexPath) else { return }
|
||
selectedPresetID = preset.id
|
||
collectionView.visibleCells.compactMap { $0 as? TravelAlbumEditPresetCell }.forEach { cell in
|
||
guard let visiblePath = collectionView.indexPath(for: cell), let visiblePreset = dataSource.itemIdentifier(for: visiblePath) else { return }
|
||
cell.setSelected(visiblePreset.id == selectedPresetID)
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 带效果缩略图的修图预设单元格。
|
||
private final class TravelAlbumEditPresetCell: UICollectionViewCell {
|
||
static let reuseIdentifier = "TravelAlbumEditPresetCell"
|
||
|
||
var onPreview: (() -> Void)?
|
||
|
||
private let imageView = UIImageView()
|
||
private let titleLabel = UILabel()
|
||
private let selectionView = UIImageView(image: UIImage(systemName: "checkmark.circle.fill"))
|
||
private let previewButton = UIButton(type: .system)
|
||
|
||
override init(frame: CGRect) {
|
||
super.init(frame: frame)
|
||
contentView.backgroundColor = .white
|
||
contentView.layer.cornerRadius = 10
|
||
contentView.clipsToBounds = true
|
||
imageView.contentMode = .scaleAspectFill
|
||
imageView.clipsToBounds = true
|
||
imageView.backgroundColor = UIColor(hex: 0xEAF2FF)
|
||
titleLabel.font = .systemFont(ofSize: 12, weight: .medium)
|
||
titleLabel.textColor = AppColor.textPrimary
|
||
titleLabel.textAlignment = .center
|
||
titleLabel.numberOfLines = 2
|
||
selectionView.tintColor = AppColor.primary
|
||
selectionView.backgroundColor = .white
|
||
selectionView.layer.cornerRadius = 10
|
||
selectionView.clipsToBounds = true
|
||
previewButton.setTitle("预览", for: .normal)
|
||
previewButton.setTitleColor(.white, for: .normal)
|
||
previewButton.titleLabel?.font = .systemFont(ofSize: 11, weight: .medium)
|
||
previewButton.backgroundColor = UIColor.black.withAlphaComponent(0.58)
|
||
previewButton.layer.cornerRadius = 12
|
||
previewButton.addTarget(self, action: #selector(previewTapped), for: .touchUpInside)
|
||
contentView.addSubview(imageView)
|
||
contentView.addSubview(titleLabel)
|
||
contentView.addSubview(selectionView)
|
||
contentView.addSubview(previewButton)
|
||
imageView.snp.makeConstraints { make in
|
||
make.top.leading.trailing.equalToSuperview()
|
||
make.height.equalTo(96)
|
||
}
|
||
titleLabel.snp.makeConstraints { make in
|
||
make.top.equalTo(imageView.snp.bottom).offset(6)
|
||
make.leading.trailing.equalToSuperview().inset(5)
|
||
make.bottom.lessThanOrEqualToSuperview().offset(-5)
|
||
}
|
||
selectionView.snp.makeConstraints { make in
|
||
make.top.trailing.equalToSuperview().inset(6)
|
||
make.size.equalTo(20)
|
||
}
|
||
previewButton.snp.makeConstraints { make in
|
||
make.trailing.bottom.equalTo(imageView).inset(6)
|
||
make.width.equalTo(46)
|
||
make.height.equalTo(24)
|
||
}
|
||
}
|
||
|
||
@available(*, unavailable)
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
|
||
func apply(preset: TravelAlbumEditPreset, sourceImage: UIImage?, selected: Bool) {
|
||
apply(
|
||
title: preset.title,
|
||
effect: preset.effect,
|
||
sourceImage: sourceImage,
|
||
selected: selected
|
||
)
|
||
}
|
||
|
||
/// 更新氛围感样式的效果图、名称和单选状态。
|
||
func apply(
|
||
atmosphereOption: TravelAlbumAtmosphereEditOption,
|
||
sourceImage: UIImage?,
|
||
selected: Bool
|
||
) {
|
||
apply(
|
||
title: atmosphereOption.title,
|
||
effect: atmosphereOption.effect,
|
||
sourceImage: sourceImage,
|
||
selected: selected
|
||
)
|
||
accessibilityHint = "再次点击已选样式可取消，点击预览按钮查看大图"
|
||
}
|
||
|
||
private func apply(
|
||
title: String,
|
||
effect: TravelAlbumEditPreset.Effect,
|
||
sourceImage: UIImage?,
|
||
selected: Bool
|
||
) {
|
||
titleLabel.text = title
|
||
imageView.image = sourceImage.map {
|
||
TravelAlbumAIEditImageProcessor.render(effect: effect, source: $0)
|
||
} ?? UIImage(systemName: "photo")
|
||
setSelected(selected)
|
||
accessibilityLabel = title
|
||
accessibilityValue = selected ? "已选中" : "未选中"
|
||
}
|
||
|
||
func setSelected(_ selected: Bool) {
|
||
contentView.layer.borderWidth = selected ? 2 : 0
|
||
contentView.layer.borderColor = selected ? AppColor.primary.cgColor : UIColor.clear.cgColor
|
||
selectionView.isHidden = !selected
|
||
accessibilityValue = selected ? "已选中" : "未选中"
|
||
}
|
||
|
||
override func prepareForReuse() {
|
||
super.prepareForReuse()
|
||
onPreview = nil
|
||
accessibilityHint = nil
|
||
}
|
||
|
||
@objc private func previewTapped() {
|
||
onPreview?()
|
||
}
|
||
}
|
||
|
||
/// 带本地排版效果图的封面模板单元格。
|
||
private final class TravelAlbumCoverTemplateCell: UICollectionViewCell {
|
||
static let reuseIdentifier = "TravelAlbumCoverTemplateCell"
|
||
|
||
var onPreview: (() -> Void)?
|
||
|
||
private let imageView = UIImageView()
|
||
private let titleLabel = UILabel()
|
||
private let selectionView = UIImageView(image: UIImage(systemName: "checkmark.circle.fill"))
|
||
private let previewButton = UIButton(type: .system)
|
||
|
||
override init(frame: CGRect) {
|
||
super.init(frame: frame)
|
||
contentView.backgroundColor = .white
|
||
contentView.layer.cornerRadius = 10
|
||
contentView.clipsToBounds = true
|
||
|
||
imageView.contentMode = .scaleAspectFill
|
||
imageView.clipsToBounds = true
|
||
imageView.backgroundColor = UIColor(hex: 0xEAF2FF)
|
||
titleLabel.font = .systemFont(ofSize: 12, weight: .medium)
|
||
titleLabel.textColor = AppColor.textPrimary
|
||
titleLabel.textAlignment = .center
|
||
selectionView.tintColor = AppColor.primary
|
||
selectionView.backgroundColor = .white
|
||
selectionView.layer.cornerRadius = 10
|
||
selectionView.clipsToBounds = true
|
||
previewButton.setTitle("预览", for: .normal)
|
||
previewButton.setTitleColor(.white, for: .normal)
|
||
previewButton.titleLabel?.font = .systemFont(ofSize: 11, weight: .medium)
|
||
previewButton.backgroundColor = UIColor.black.withAlphaComponent(0.58)
|
||
previewButton.layer.cornerRadius = 12
|
||
previewButton.addTarget(self, action: #selector(previewTapped), for: .touchUpInside)
|
||
|
||
contentView.addSubview(imageView)
|
||
contentView.addSubview(titleLabel)
|
||
contentView.addSubview(selectionView)
|
||
contentView.addSubview(previewButton)
|
||
imageView.snp.makeConstraints { make in
|
||
make.top.leading.trailing.equalToSuperview()
|
||
make.height.equalTo(104)
|
||
}
|
||
titleLabel.snp.makeConstraints { make in
|
||
make.top.equalTo(imageView.snp.bottom).offset(6)
|
||
make.leading.trailing.equalToSuperview().inset(5)
|
||
make.bottom.lessThanOrEqualToSuperview().offset(-5)
|
||
}
|
||
selectionView.snp.makeConstraints { make in
|
||
make.top.trailing.equalToSuperview().inset(6)
|
||
make.size.equalTo(20)
|
||
}
|
||
previewButton.snp.makeConstraints { make in
|
||
make.trailing.bottom.equalTo(imageView).inset(6)
|
||
make.width.equalTo(46)
|
||
make.height.equalTo(24)
|
||
}
|
||
}
|
||
|
||
@available(*, unavailable)
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
|
||
/// 更新模板效果图与单选状态。
|
||
func apply(template: TravelAlbumCoverTemplate, sourceImage: UIImage?, selected: Bool) {
|
||
titleLabel.text = template.title
|
||
imageView.image = TravelAlbumCoverTemplateImageProcessor.render(
|
||
template: template,
|
||
source: sourceImage,
|
||
targetSize: CGSize(width: 232, height: 208)
|
||
)
|
||
setSelected(selected)
|
||
accessibilityLabel = "\(template.title)封面模板"
|
||
accessibilityValue = selected ? "已选中" : "未选中"
|
||
accessibilityHint = "点击预览按钮查看完整模板样式"
|
||
}
|
||
|
||
/// 刷新当前模板的选中描边和勾选标识。
|
||
func setSelected(_ selected: Bool) {
|
||
contentView.layer.borderWidth = selected ? 2 : 0
|
||
contentView.layer.borderColor = selected ? AppColor.primary.cgColor : UIColor.clear.cgColor
|
||
selectionView.isHidden = !selected
|
||
accessibilityValue = selected ? "已选中" : "未选中"
|
||
}
|
||
|
||
override func prepareForReuse() {
|
||
super.prepareForReuse()
|
||
onPreview = nil
|
||
}
|
||
|
||
@objc private func previewTapped() {
|
||
onPreview?()
|
||
}
|
||
}
|