Merge pull request #11 from mourezwell/master

feat : Add Ability to distinct Abstract classes with (A) icon and italic
gboersma · Apr 10, 2020 · 79af2db · 79af2db
2 parents c546da5 + 0ec9eac
commit 79af2db
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/main/java/info/leadinglight/umljavadoclet/model/ModelClass.java b/src/main/java/info/leadinglight/umljavadoclet/model/ModelClass.java
@@ -28,7 +28,7 @@ public enum Visibility {
     }
 
     public enum ClassType {
-        INTERFACE, ENUM, CLASS
+        INTERFACE, ENUM, CLASS, ABSTRACT
     }
 
     public static class VisibilityItem {
@@ -168,6 +168,8 @@ public ClassType type() {
             return ClassType.INTERFACE;
         } else if (_classDoc.isEnum()) {
             return ClassType.ENUM;
+        } else if (_classDoc.isAbstract()) {
+            return ClassType.ABSTRACT;
         } else {
             return ClassType.CLASS;
         }

diff --git a/src/main/java/info/leadinglight/umljavadoclet/printer/PumlDiagramPrinter.java b/src/main/java/info/leadinglight/umljavadoclet/printer/PumlDiagramPrinter.java
@@ -180,6 +180,9 @@ public void classType(ModelClass modelClass) {
             case ENUM:
                 print("enum");
                 break;
+            case ABSTRACT:
+                print("abstract");
+                break;
             default:
                 print("class");
         }