Swift Set Type

gistfile1.swift

struct Set<T: AnyObject> : Printable {
    typealias ElementType = (T: AnyObject)

    var set: NSMutableSet

    init() {
        self.init(array: [] )
    }

    init(array: ElementType[]) {
        self.set = NSMutableSet(set: NSSet(objects: array))
    }

    init(object: ElementType) {
        self.init(array: [object])
    }

    var objects: ElementType[] {
        get { return self.set.allObjects }
    }

    var count: Int {
        get { return self.set.count }
    }

    func setByAdding(object: ElementType) -> Set<T> {
        var objects = self.objects
        objects.append(object)

        return Set(array: objects)
    }
}

extension Set {
    mutating func add(object: ElementType) {
        self.set.addObject(object)
    }

    mutating func remove(object: ElementType) {
        self.set.removeObject(object)
    }

    mutating func removeAll() {
        self.set.removeAllObjects()
    }
}

extension Set : Printable {
    var description: String {
        get {
            return self.objects.description
        }
    }
}