Functionality for launching fireworks

20-fireworks-night/Fireworks Night/Fireworks Night/GameScene.swift

@@ -12,11 +12,20 @@ import GameplayKit
 let sceneWidth = 1024.0
 let sceneHeight = 768.0
 
+enum NodeName: String {
+    case fireworkRocket
+}
+
+enum LaunchStyle: CaseIterable {
+    case straightUp, fanUp, leftToRight, rightToLeft
+}
+
 
 class GameScene: SKScene {
-    let fireworkInterval = 6.0
-    let leftEdge = -22
-    let bottomEdge = -22
+    let fireworkInterval = 4.0
+    let fireworkSpeed = 400.0
+    let leftEdge = -22.0
+    let bottomEdge = -22.0
     let rightEdge = sceneWidth + 22
     
     var gameTimer: Timer!
@@ -28,6 +37,10 @@ class GameScene: SKScene {
         }
     }
     
+    var randomFireworkColor: UIColor {
+        return [UIColor.cyan, UIColor.green, UIColor.red].randomElement()!
+    }
+    
     lazy var sceneCenterPoint = CGPoint(x: sceneWidth / 2.0, y: sceneHeight / 2.0)
     
     override func didMove(to view: SKView) {
@@ -36,7 +49,13 @@ class GameScene: SKScene {
     }
     
     func setupTimer() {
-        gameTimer = Timer.scheduledTimer(timeInterval: fireworkInterval, target: self, selector: #selector(launchFireworks), userInfo: nil, repeats: true)
+        gameTimer = Timer.scheduledTimer(
+            timeInterval: fireworkInterval,
+            target: self,
+            selector: #selector(launchFireworks),
+            userInfo: nil,
+            repeats: true
+        )
     }
     
     func setupBackground() {
@@ -49,8 +68,123 @@ class GameScene: SKScene {
         addChild(background)
     }
 
-
+    
+    /**
+     * Launch fireworks five at a time, in four different shapes
+     */
     @objc func launchFireworks() {
+        let burstQuantity = 5
         
+        switch LaunchStyle.allCases.randomElement()! {
+        case .straightUp:
+            launchStraightUp(burstQuantity)
+        case .fanUp:
+            launchFanUp(burstQuantity)
+        case .leftToRight:
+            launchLeftToRight(burstQuantity)
+        case .rightToLeft:
+            launchRightToLeft(burstQuantity)
+        }
     }
+    
+    
+    func createLaunch(xMovement: Double, xPos: Double, yPos: Double) {
+        let firework = makeFirework()
+        let motion = makeFireworkMotion(xMovement: xMovement, speed: fireworkSpeed)
+        
+        firework.position = CGPoint(x: xPos, y: yPos)
+        
+        fireworks.append(firework)
+        firework.run(motion)
+        addChild(firework)
+    }
+
+    
+    func makeFirework() -> SKNode {
+        let firework = SKNode()
+        let fireworkGlow = SKEmitterNode(fileNamed: "fuse")!
+        let fireworkRocket = SKSpriteNode(imageNamed: "rocket")
+        
+        fireworkRocket.colorBlendFactor = 1
+        fireworkRocket.name = NodeName.fireworkRocket.rawValue
+        fireworkRocket.color = randomFireworkColor
+        
+        fireworkGlow.position = CGPoint(x: 0, y: -22)
+        
+        firework.addChild(fireworkGlow)
+        firework.addChild(fireworkRocket)
+        
+        return firework
+    }
+    
+    
+    func makeFireworkMotion(xMovement: Double, speed: Double) -> SKAction {
+        let path = UIBezierPath()
+        
+        path.move(to: CGPoint(x: 0, y: 0))
+        path.addLine(to: CGPoint(x: xMovement, y: sceneHeight * 1.5))
+        
+        // `asOffset` makes the motion start relative to the node's position
+        let motionPath = SKAction.follow(path.cgPath, asOffset: true, orientToPath: true, speed: CGFloat(speed))
+        
+        return motionPath
+    }
+    
+    
+    func launchStraightUp(_ numFireworks: Int) {
+        let spacingIncrement = (sceneHeight / Double(numFireworks)) / 2.0
+        var currentSpacing = 0.0
+        
+        for n in 0 ..< numFireworks {
+            currentSpacing = (n % 2 == 0) ? currentSpacing : currentSpacing + spacingIncrement
+            let directionFromCenter = (n % 2 == 0) ? 1.0 : -1.0
+            let xPosition = Double(sceneCenterPoint.x) + (currentSpacing * directionFromCenter)
+                
+            createLaunch(xMovement: 0, xPos: xPosition, yPos: bottomEdge)
+        }
+    }
+    
+    func launchFanUp(_ numFireworks: Int) {
+        let spacingIncrement = (sceneHeight / Double(numFireworks)) / 2.0
+        var currentSpacing = 0.0
+        
+        for n in 0 ..< numFireworks {
+            currentSpacing = (n % 2 == 0) ? currentSpacing : currentSpacing + spacingIncrement
+            let directionFromCenter = (n % 2 == 0) ? 1.0 : -1.0
+            let xPosition = Double(sceneCenterPoint.x) + (currentSpacing * directionFromCenter)
+            let xMovement = xPosition
+            
+            print("X position: \(xPosition)")
+            createLaunch(xMovement: xMovement, xPos: xPosition, yPos: bottomEdge)
+        }
+    }
+    
+    func launchLeftToRight(_ numFireworks: Int) {
+        let spacingIncrement = (sceneHeight / Double(numFireworks)) / 2.0
+        var currentSpacing = 0.0
+        
+        for n in 0 ..< numFireworks {
+            currentSpacing = (n % 2 == 0) ? currentSpacing : currentSpacing + spacingIncrement
+            let directionFromCenter = (n % 2 == 0) ? 1.0 : -1.0
+            let yPosition = Double(sceneCenterPoint.y) + (currentSpacing * directionFromCenter)
+            let xMovement = sceneWidth * 1.5
+
+            createLaunch(xMovement: xMovement, xPos: leftEdge, yPos: yPosition)
+        }
+    }
+    
+    func launchRightToLeft(_ numFireworks: Int) {
+        let spacingIncrement = (sceneHeight / Double(numFireworks)) / 2.0
+        var currentSpacing = 0.0
+
+        for n in 0 ..< numFireworks {
+            currentSpacing = (n % 2 == 0) ? currentSpacing : currentSpacing + spacingIncrement
+            let directionFromCenter = (n % 2 == 0) ? 1.0 : -1.0
+            let yPosition = Double(sceneCenterPoint.y) + (currentSpacing * directionFromCenter)
+            let xMovement = sceneWidth * 1.5
+            
+            createLaunch(xMovement: -xMovement, xPos: rightEdge, yPos: yPosition)
+        }
+    }
+    
 }