NSMapTable wrapper

ZMapTable.swift

//
//  ZMapTable.swift
//  ZKit
//
//  Created by Kaz Yoshikawa on 3/26/16.
//  Copyright © 2016 Electricwoods LLC. All rights reserved.
//

import Foundation



class ZStronToWeakMapTable<K: AnyObject, V: AnyObject>: CustomStringConvertible {

	let mapTable: NSMapTable

	init() {
		self.mapTable = NSMapTable.strongToWeakObjectsMapTable()
	}

	subscript(key: K) -> V? {
		get {
			return self.mapTable.objectForKey(key) as? V
		}
		set {
			self.mapTable.setObject(newValue, forKey: key)
		}
	}
	
	var description: String {
		return mapTable.description
	}
}


class ZWeakToStrongMapTable<K: AnyObject, V: AnyObject>: CustomStringConvertible {

	let mapTable: NSMapTable

	init() {
		self.mapTable = NSMapTable.weakToStrongObjectsMapTable()
	}

	subscript(key: K) -> V? {
		get {
			return self.mapTable.objectForKey(key) as? V
		}
		set {
			self.mapTable.setObject(newValue, forKey: key)
		}
	}

	var description: String {
		return mapTable.description
	}
}