X-Git-Url: http://www2.svjatoslav.eu/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Feu%2Fsvjatoslav%2Finspector%2Fjava%2Fstructure%2FClassGraph.java;fp=src%2Fmain%2Fjava%2Feu%2Fsvjatoslav%2Finspector%2Fjava%2Fstructure%2FClassGraph.java;h=62e091157d56de91a9feb04416fbf7ce2c632a9b;hb=55258e8ea4f092ae1f1fe6b4385bf7920c20acc8;hp=f3afd808371cb05b025475fc14847edb13e74fdc;hpb=dda66ce5668b5e27e696759ef5c895d044a572bc;p=javainspect.git

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 f3afd80..62e0911 100755
--- a/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java
+++ b/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java
@@ -119,10 +119,9 @@ public class ClassGraph {
 			} finally {
 			}
 
-			if (!keepDotFile) {
+			if (!keepDotFile)
 				// delete dot file
 				new File(dotFilePath).delete();
-			}
 		} catch (final IOException e) {
 			System.err.println(e);
 		}
@@ -173,12 +172,13 @@ public class ClassGraph {
 	/**
 	 * Hide orphaned class that have no references
 	 */
-	public void hideOrphanedClasses() {
+	public ClassGraph hideOrphanedClasses() {
 
 		for (final ClassDescriptor classDescriptor : fullyQualifiedNameToClassMap
 				.values())
 			classDescriptor.hideClassIfNoReferences();
 
+		return this;
 	}
 
 	protected boolean isClassShown(final String className) {