bugfixes

[javainspect.git] / src / main / java / eu / svjatoslav / inspector / java / structure / MethodDescriptor.java

diff --git a/src/main/java/eu/svjatoslav/inspector/java/structure/MethodDescriptor.java b/src/main/java/eu/svjatoslav/inspector/java/structure/MethodDescriptor.java
index efe6992..1929348 100755 (executable)
--- a/src/main/java/eu/svjatoslav/inspector/java/structure/MethodDescriptor.java
+++ b/src/main/java/eu/svjatoslav/inspector/java/structure/MethodDescriptor.java
@@ -98,8 +98,8 @@ public class MethodDescriptor implements GraphElement,
                             + classDescriptor.getColor()
                             + "\", style=\"dotted, bold\"];\n");
 
-        if (!returnType.isVisible())
-            return result.toString();
+        if (returnType == null) return result.toString();
+        if (!returnType.isVisible()) return result.toString();
 
         // main type
         if (returnType.areReferencesShown())
@@ -121,7 +121,7 @@ public class MethodDescriptor implements GraphElement,
         result.append("        // " + methodName + "\n");
 
         result.append("        <TR><td ALIGN=\"right\">"
-                + "<FONT POINT-SIZE=\"8.0\">" + returnType.getClassName(true)
+                + "<FONT POINT-SIZE=\"8.0\">" + describeReturnType()
                 + "</FONT>" + "</td><TD PORT=\"" + getMethodLabel()
                 + "\" ALIGN=\"left\"><FONT COLOR =\"red\" POINT-SIZE=\"11.0\">"
                 + getMethodLabel() + "</FONT></TD></TR>\n");
@@ -129,6 +129,12 @@ public class MethodDescriptor implements GraphElement,
         return result.toString();
     }
 
+    private String describeReturnType() {
+        if (returnType == null) return "-null-";
+
+        return returnType.getClassName(true);
+    }
+
     @Override
     public String getGraphId() {
         return parentClass.getGraphId() + ":" + methodName;