Add the TypeKind name to the IR graphviz output

1 files changed, 31 insertions, 0 deletions

diff --git a/src/ir/ty.rs b/src/ir/ty.rs
index b588dbb4..8cfbde10 100644
--- a/src/ir/ty.rs
+++ b/src/ir/ty.rs
@@ -443,6 +443,8 @@ impl DotAttributes for TypeKind {
     where
         W: io::Write,
     {
+        writeln!(out, "<tr><td>type kind</td><td>{}</td></tr>", self.kind_name())?;
+
         if let TypeKind::Comp(ref comp) = *self {
             comp.dot_attributes(ctx, out)?;
         }
@@ -451,6 +453,35 @@ impl DotAttributes for TypeKind {
     }
 }
 
+impl TypeKind {
+    fn kind_name(&self) -> &'static str {
+        match *self {
+            TypeKind::Void => "Void",
+            TypeKind::NullPtr => "NullPtr",
+            TypeKind::Comp(..) => "Comp",
+            TypeKind::Opaque => "Opaque",
+            TypeKind::Int(..) => "Int",
+            TypeKind::Float(..) => "Float",
+            TypeKind::Complex(..) => "Complex",
+            TypeKind::Alias(..) => "Alias",
+            TypeKind::TemplateAlias(..) => "TemplateAlias",
+            TypeKind::Array(..) => "Array",
+            TypeKind::Function(..) => "Function",
+            TypeKind::Enum(..) => "Enum",
+            TypeKind::Pointer(..) => "Pointer",
+            TypeKind::BlockPointer => "BlockPointer",
+            TypeKind::Reference(..) => "Reference",
+            TypeKind::TemplateInstantiation(..) => "TemplateInstantiation",
+            TypeKind::UnresolvedTypeRef(..) => "UnresolvedTypeRef",
+            TypeKind::ResolvedTypeRef(..) => "ResolvedTypeRef",
+            TypeKind::TypeParam => "TypeParam",
+            TypeKind::ObjCInterface(..) => "ObjCInterface",
+            TypeKind::ObjCId => "ObjCId",
+            TypeKind::ObjCSel => "ObjCSel",
+        }
+    }
+}
+
 #[test]
 fn is_invalid_type_param_valid() {
     let ty = Type::new(Some("foo".into()), None, TypeKind::TypeParam, false);