Fix scale and translation for different content modes

Author: fruitcoder

Sources/PlayButton/PlayButton.swift

@@ -179,8 +179,25 @@ public class PlayButton: UIButton {
       yScale = scale
     }
 
-    let xTranslate = (xScale - 1) * scaleLayer.bounds.width / 2
-    let yTranslate = (yScale - 1) * scaleLayer.bounds.height / 2
+    var xTranslate = (xScale - 1.0) * scaleLayer.bounds.height / 2.0
+    var yTranslate = (yScale - 1.0) * scaleLayer.bounds.width / 2.0
+
+    switch contentMode {
+    case .scaleToFill:
+      break
+    case .scaleAspectFill:
+      if bounds.width < bounds.height {
+        xTranslate += (bounds.width - xScale * scaleLayer.bounds.width) / 2.0
+      } else if bounds.width > bounds.height {
+        yTranslate += (bounds.height - yScale * scaleLayer.bounds.height) / 2.0
+      }
+    default:
+      if bounds.width > bounds.height {
+        xTranslate += (bounds.width - xScale * scaleLayer.bounds.width) / 2.0
+      } else if bounds.width < bounds.height {
+        yTranslate += (bounds.height - yScale * scaleLayer.bounds.height) / 2.0
+      }
+    }
 
     scaleLayer.transform = CATransform3DConcat(
       CATransform3DMakeScale(xScale, yScale, 1),

Tests/PlayButtonTests/PlayButtonTests.swift

@@ -21,7 +21,7 @@ final class PlayButtonTests: XCTestCase {
     windowView = vc.view
 
     SnapshotTesting.diffTool = "ksdiff"
-    //		isRecording = true
+//    isRecording = true
   }
 
   func testPlayToPause() {
@@ -225,6 +225,60 @@ final class PlayButtonTests: XCTestCase {
     button.setStop(true, animated: false)
     XCTAssertEqual(button.isStop, true)
   }
+
+  func testScaleAspectFillWidthGreaterThanHeight() {
+    let button = PlayButton(frame: CGRect(x: 0, y: 0, width: 60, height: 30))
+    button.backgroundColor = .blue
+    button.contentMode = .scaleAspectFill
+
+    windowView.addSubview(button)
+    assertSnapshot(matching: button, as: .image)
+  }
+
+  func testScaleAspectFillHeightGreaterThanWidth() {
+    let button = PlayButton(frame: CGRect(x: 0, y: 0, width: 30, height: 60))
+    button.backgroundColor = .blue
+    button.contentMode = .scaleAspectFill
+
+    windowView.addSubview(button)
+    assertSnapshot(matching: button, as: .image)
+  }
+
+  func testScaleAspectFitWidthGreaterThanHeight() {
+    let button = PlayButton(frame: CGRect(x: 0, y: 0, width: 60, height: 30))
+    button.backgroundColor = .blue
+    button.contentMode = .scaleAspectFit
+
+    windowView.addSubview(button)
+    assertSnapshot(matching: button, as: .image)
+  }
+
+  func testScaleAspectFitHeightGreaterThanWidth() {
+    let button = PlayButton(frame: CGRect(x: 0, y: 0, width: 30, height: 60))
+    button.backgroundColor = .blue
+    button.contentMode = .scaleAspectFit
+
+    windowView.addSubview(button)
+    assertSnapshot(matching: button, as: .image)
+  }
+
+  func testScaleToFillWidthGreaterThanHeight() {
+    let button = PlayButton(frame: CGRect(x: 0, y: 0, width: 60, height: 30))
+    button.backgroundColor = .blue
+    button.contentMode = .scaleToFill
+
+    windowView.addSubview(button)
+    assertSnapshot(matching: button, as: .image)
+  }
+
+  func testScaleToFillHeightGreaterThanWidth() {
+    let button = PlayButton(frame: CGRect(x: 0, y: 0, width: 30, height: 60))
+    button.backgroundColor = .blue
+    button.contentMode = .scaleToFill
+
+    windowView.addSubview(button)
+    assertSnapshot(matching: button, as: .image)
+  }
 }
 
 private enum CompletionCondition {