Getting and Comparing Type of Class

swift-compare-inherited-type

class BaseClass: NSObject {
    func test() {
        let aType = type(of: self)
        if aType == dict["My Base"] {
            print("this is base")
        } else if aType == dict["My Inherited"] {
            print("this is inherited")
        }
        print(aType)
    }
}

class InheritedClass: BaseClass {
    override func test() {
        print("from inherited...")
        super.test()
    }
}

let dict = [
    "My Base": BaseClass.self,
    "My Inherited": InheritedClass.self,
]

BaseClass().test()
InheritedClass().test()

Output:

this is base
BaseClass
from inherited...
this is inherited
InheritedClass