Added reset things

Author: BlindDev

SnakeGame/GameFieldView.swift

@@ -36,11 +36,9 @@ class GameFieldView: UIView {
 
             let parameters = SegmentParameters(segment: value)
 
-            var segmentView: SegmentView!
-            
             if index == segmentViews.count {
 
-                segmentView = SegmentView(parameters: parameters)
+                let segmentView = SegmentView(parameters: parameters)
 
                 self.addSubview(segmentView)
                 segmentViews.append(segmentView)
@@ -63,6 +61,14 @@ class GameFieldView: UIView {
         }
     }
 
+    func clearSubviews() {
+        
+        segmentViews = []
+        for view in subviews {
+            view.removeFromSuperview()
+        }
+    }
+    
     override func drawRect(rect: CGRect) {
 
         delegate?.viewDidDraw()

SnakeGame/GameViewController.swift

@@ -20,7 +20,9 @@ class GameViewController: UIViewController {
     }
 
     @IBAction func refreshAction(sender: UIBarButtonItem) {
-        resetBrain()
+        
+        gameFieldView.clearSubviews()
+        gameFieldView.setNeedsDisplay()
     }
 
     private var brain: GameBrain!
@@ -38,8 +40,11 @@ class GameViewController: UIViewController {
     }
 
     private func resetBrain() {
+        
         brain = GameBrain(viewSize: gameFieldView.bounds.size)
+        
         brain.delegate = self
+        
         gameFieldView.renderBorders(brain.borders)
 
         timer = NSTimer.scheduledTimerWithTimeInterval(0.10,  target: self, selector: #selector(movePoint), userInfo: nil, repeats: true)