possibility to hide orphaned classes from graph

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

diff --git a/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java b/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java
index 4c1dcbc..6155876 100644 (file)
--- a/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java
+++ b/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java
@@ -21,6 +21,15 @@ import eu.svjatoslav.inspector.java.methods.ProjectScanner;
 
 public class ClassGraph {
 
+       public static void render(final String graphName, final Class... classes) {
+               final ClassGraph classGraph = new ClassGraph();
+
+               for (final Class clazz : classes)
+                       classGraph.addClass(clazz);
+
+               classGraph.generateGraph(graphName);
+       }
+
        /**
         * Maps class fully qualified names to class descriptors.
         */
@@ -124,4 +133,11 @@ public class ClassGraph {
                return resultStr;
        }
 
+       public void hideClassesWithoutReferences() {
+
+               for (final ClassDescriptor classDescriptor : nameToClassMap.values())
+                       classDescriptor.hideClassIfNoReferences();
+
+       }
+
 }