Impressia/Vernissage/Widgets/ImageViewer.swift

259 lines
10 KiB
Swift

//
// https://mczachurski.dev
// Copyright © 2023 Marcin Czachurski and the repository contributors.
// Licensed under the Apache License 2.0.
//
import SwiftUI
struct ImageViewer: View {
@Environment(\.dismiss) private var dismiss
private let attachmentModel: AttachmentModel
private let image: Image
private let closeDragDistance = UIScreen.main.bounds.height / 1.8
private let imageHeight: Double
private let imageWidth: Double
private let imagePosition: Double?
// Magnification.
@State private var currentMagnification = 0.0
@State private var finalMagnification = 1.0
// Rotation.
@State private var rotationAngle = Angle.zero
// Draging.
@State private var currentOffset = CGSize.zero
@State private var accumulatedOffset = CGSize.zero
init(attachmentModel: AttachmentModel, imagePosition: Double) {
self.attachmentModel = attachmentModel
self.imagePosition = imagePosition
if let data = attachmentModel.data, let uiImage = UIImage(data: data) {
self.image = Image(uiImage: uiImage)
self.imageHeight = uiImage.size.height
self.imageWidth = uiImage.size.width
} else {
self.image = Image(systemName: "photo")
self.imageHeight = 200
self.imageWidth = 200
}
}
var body: some View {
image
.resizable()
.aspectRatio(contentMode: .fit)
.tag(attachmentModel.id)
.offset(currentOffset)
.rotationEffect(rotationAngle)
.scaleEffect(finalMagnification + currentMagnification)
.gesture(dragGesture)
.gesture(magnificationGesture)
.gesture(doubleTapGesture)
.gesture(tapGesture)
.onAppear {
self.currentOffset = self.calculateStartingOffset()
withAnimation {
self.currentOffset = CGSize.zero
}
}
}
@MainActor
var magnificationGesture: some Gesture {
MagnificationGesture()
.onChanged { amount in
self.currentMagnification = (amount - 1) * self.finalMagnification
}
.onEnded { _ in
self.revertToPrecalculatedMagnification()
}
}
var doubleTapGesture: some Gesture {
TapGesture(count: 2)
.onEnded { _ in
withAnimation {
if self.finalMagnification == 1.0 {
self.currentOffset = CGSize.zero
self.accumulatedOffset = CGSize.zero
self.currentMagnification = 0
self.finalMagnification = 2.0
} else {
self.currentOffset = CGSize.zero
self.accumulatedOffset = CGSize.zero
self.currentMagnification = 0
self.finalMagnification = 1.0
}
}
}
}
@MainActor
var dragGesture: some Gesture {
DragGesture()
.onChanged { amount in
// Opacity and rotation is working only when we have small image size.
if self.finalMagnification == 1.0 {
// We can move image whatever we want.
self.currentOffset = CGSize(width: amount.translation.width + self.accumulatedOffset.width,
height: amount.translation.height + self.accumulatedOffset.height)
// Changing angle.
self.rotationAngle = Angle(degrees: Double(self.currentOffset.width / 30))
} else {
// Bigger images we can move only horizontally (we have to include magnifications).
let offsetWidth = (amount.predictedEndTranslation.width / self.finalMagnification) + self.accumulatedOffset.width
withAnimation(.spring()) {
self.currentOffset = CGSize(width: offsetWidth, height: 0)
}
}
} .onEnded { amount in
self.accumulatedOffset = CGSize(width: (amount.predictedEndTranslation.width / self.finalMagnification) + self.accumulatedOffset.width,
height: (amount.predictedEndTranslation.height / self.finalMagnification) + self.accumulatedOffset.height)
// Animations only for small images sizes,
if self.finalMagnification == 1.0 {
// When we still are in range visible image then we have to only revert back image to starting position..
if self.accumulatedOffset.height > -closeDragDistance && self.accumulatedOffset.height < closeDragDistance {
withAnimation(.linear(duration: 0.1)) {
self.currentOffset = self.accumulatedOffset
}
// Revert back image offset.
withAnimation(.linear(duration: 0.3).delay(0.1)) {
self.currentOffset = CGSize.zero
self.accumulatedOffset = CGSize.zero
self.rotationAngle = Angle.zero
}
} else {
// Close the screen.
withAnimation(.linear(duration: 0.4)) {
// We have to set end translations for sure outside the screen.
self.currentOffset = CGSize(width: amount.predictedEndTranslation.width * 2, height: amount.predictedEndTranslation.height * 2)
self.rotationAngle = Angle(degrees: Double(amount.predictedEndTranslation.width / 30))
self.accumulatedOffset = CGSize.zero
}
DispatchQueue.main.asyncAfter(deadline: .now() + 0.35) {
withoutAnimation {
self.dismiss()
}
}
}
} else {
self.moveToEdge()
}
}
}
var tapGesture: some Gesture {
TapGesture().onEnded({ _ in
withAnimation {
self.currentOffset = self.calculateStartingOffset()
}
DispatchQueue.main.asyncAfter(deadline: .now() + 0.35) {
withoutAnimation {
self.dismiss()
}
}
})
}
@MainActor
private func revertToPrecalculatedMagnification() {
let magnification = self.finalMagnification + self.currentMagnification
if magnification < 1.0 {
// When image is small we are returning to starting point.
withAnimation {
self.finalMagnification = 1.0
self.currentMagnification = 0
// Also we have to move image to orginal position.
self.currentOffset = CGSize.zero
}
HapticService.shared.fireHaptic(of: .animation)
} else if magnification > 3.0 {
// When image is magnified to much we are rturning to 1.5 maginification.
withAnimation {
self.finalMagnification = 3.0
self.currentMagnification = 0
}
HapticService.shared.fireHaptic(of: .animation)
} else {
self.finalMagnification = magnification
self.currentMagnification = 0
// Verify if we have to move image to nearest edge.
self.moveToEdge()
}
}
@MainActor
private func moveToEdge() {
let maxEdgeDistance = ((UIScreen.main.bounds.width * self.finalMagnification) - UIScreen.main.bounds.width) / (2 * self.finalMagnification)
if self.currentOffset.width > maxEdgeDistance {
withAnimation(.linear(duration: 0.15)) {
self.currentOffset = CGSize(width: maxEdgeDistance, height: 0)
self.accumulatedOffset = self.currentOffset
}
HapticService.shared.fireHaptic(of: .animation)
} else if self.currentOffset.width < -maxEdgeDistance {
withAnimation(.linear(duration: 0.15)) {
self.currentOffset = CGSize(width: -maxEdgeDistance, height: 0)
self.accumulatedOffset = self.currentOffset
}
HapticService.shared.fireHaptic(of: .animation)
}
}
private func calculateStartingOffset() -> CGSize {
// Image size on the screen.
let imageOnScreenHeight = self.calculateHeight(width: self.imageWidth, height: self.imageHeight)
// Calculate full space for image.
let safeAreaInsetsTop = UIApplication.shared.keyWindow?.safeAreaInsets.top ?? 20.0
let safeAreaInsetsBottom = UIApplication.shared.keyWindow?.safeAreaInsets.bottom ?? 20.0
let spaceForNavigationBar = self.spaceForNavigationBar()
let spaceForImage = UIScreen.main.bounds.height - safeAreaInsetsTop - safeAreaInsetsBottom - spaceForNavigationBar
// Calculate empty space.
let emptySpace = spaceForImage - imageOnScreenHeight
// Shift of image when image is exactly at the top of the screen.
let imageShiftToTopScreen = -(emptySpace / 2)
// Shidt of image also when image has been scrolled top/bottom.
let imageShiftToImagePosition = imageShiftToTopScreen - (self.imagePosition ?? 0.0)
// Calculate image shift.
return CGSize(width: 0, height: imageShiftToImagePosition)
}
private func spaceForNavigationBar() -> CGFloat {
if UIScreen.main.bounds.height == 852.0 || UIScreen.main.bounds.height == 932.0 {
// iPhone 14 Pro, iPhone 14 Pro Max
return 78.0
} else {
// iPhone SE, iPhone 12 Pro, iPhone 12 Pro Max, iPhone 13 Pro, iPhone 13 Pro Max
return 88.0
}
}
private func calculateHeight(width: Double, height: Double) -> CGFloat {
let divider = width / UIScreen.main.bounds.size.width
return height / divider
}
}