Swift Fractal Tree

Created By : Debasis Das (May 2016)

Swift Fractal Tree 1

Swift Fractal Tree 2

Swift Fractal Tree 3

Swift Fractal Tree 4

Swift Fractal Tree 5

Swift Fractal Tree 6

Swift Fractal Tree 7

Swift Fractal Tree 8

Swift Fractal Tree 9

Swift Fractal Tree 10

Swift Fractal Tree 11

Swift Fractal Tree 12

Source Code

//  Swift Fractal Tree
//  Created by Debasis Das on 15/05/16.
//  Copyright © 2016 Knowstack. All rights reserved.

import Cocoa
import Foundation

class Tree: NSView {

    let PI = 3.14156
    var leftAngle:Double = 10
    var rightAngle:Double = 10
    var treeDepth:Float = 10
    var backgroundColor:NSColor = NSColor.whiteColor()
    var leafColor:NSColor = NSColor(calibratedRed: 0, green: 30, blue: 0, alpha: 1.0)
    var trunkColor:NSColor = NSColor.darkGrayColor()
    
    @IBAction func changeDepth(sender:AnyObject){
        self.treeDepth = Float(sender.intValue)
        Swift.print(self.treeDepth)
        self.needsDisplay = true
    }
    
    @IBAction func changeLeftAngle(sender:AnyObject){
        self.leftAngle = sender.doubleValue
        self.needsDisplay = true
    }
    
    @IBAction func changeRightAngle(sender:AnyObject){
        self.rightAngle = sender.doubleValue
        self.needsDisplay = true
    }
    
    @IBAction func changeBackgroundColor(sender:AnyObject){
        self.backgroundColor = ((sender as? NSColorWell)?.color)!
        self.needsDisplay = true
    }
    
    @IBAction func changeLeafColor(sender:AnyObject){
        self.leafColor = ((sender as? NSColorWell)?.color)!
        self.needsDisplay = true
    }
    @IBAction func changeTrunkColor(sender:AnyObject){
        self.trunkColor = ((sender as? NSColorWell)?.color)!
        self.needsDisplay = true
    }
    
    override func drawRect(dirtyRect: NSRect) {
        super.drawRect(dirtyRect)
        self.backgroundColor.setFill()
        NSRectFill(dirtyRect)
        self.drawBranch(Float(self.frame.size.width/2), y1: 100.0, angle: 90.0, depth: self.treeDepth)
    }
    
    func drawBranch(x1:Float, y1:Float, angle:Double,depth:Float){
        var branchArmLength = depth
        if depth > 0 && depth < 3{             branchArmLength = depth * 0.7         }         else if depth > 3 && depth < 7{
            branchArmLength = depth * 0.7
        }
        else{
            branchArmLength = depth*0.7
        }
        if depth != 0{
            let x2 = x1 + (Float(cos(angle * PI/180)) * depth * branchArmLength)
            let y2 = y1 + (Float(sin(angle * PI/180)) * depth * branchArmLength)
            drawLine(depth, fromPoint: NSMakePoint(CGFloat(x1), CGFloat(y1)), toPoint: NSMakePoint(CGFloat(x2), CGFloat(y2)))
            drawBranch(x2, y1: y2, angle: angle - self.leftAngle, depth: depth-1)
            drawBranch(x2, y1: y2, angle: angle + self.rightAngle, depth: depth-1)
        }
    }
    
    func drawLine(lineWidth:Float, fromPoint:NSPoint, toPoint:NSPoint){
        let path = NSBezierPath()
        if (lineWidth < 5){
            self.leafColor.set()
            path.lineWidth = 1.0//CGFloat(lineWidth*0.2)
        }
        else{
            self.trunkColor.set()
            path.lineWidth = 2.0//CGFloat(lineWidth*0.5)
        }
        path.moveToPoint(fromPoint)
        path.lineToPoint(toPoint)
        path.stroke()
    }
}

The Source Code can be downloaded hereSwiftTree