X-Git-Url: http://www2.svjatoslav.eu/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Feu%2Fsvjatoslav%2Finspector%2Fjava%2Fstructure%2FClassGraph.java;h=31ddbede42d3749cb8a8a538a958993017134e78;hb=31e8e0b97c9f60f8820d708fa86c11d65f5445f3;hp=bda0882713bb0588d2f1835822fb1b9384ffaca5;hpb=d717e90f4c46e26f9f54ba5638aade6688527bf4;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
old mode 100644
new mode 100755
index bda0882..31ddbed
--- a/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java
+++ b/src/main/java/eu/svjatoslav/inspector/java/structure/ClassGraph.java
@@ -1,10 +1,10 @@
 /*
  * JavaInspect - Utility to visualize java software
- * Copyright (C) 2013, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
+ * Copyright (C) 2013-2014, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
  * 
  * This program is free software; you can redistribute it and/or
- * modify it under the terms of version 2 of the GNU General Public License
- * as published by the Free Software Foundation.
+ * modify it under the terms of version 3 of the GNU Lesser General Public License
+ * or later as published by the Free Software Foundation.
  */
 
 package eu.svjatoslav.inspector.java.structure;
@@ -16,25 +16,49 @@ import java.util.HashMap;
 import java.util.Map;
 
 import eu.svjatoslav.commons.file.CommonPathResolver;
+import eu.svjatoslav.inspector.java.methods.Clazz;
+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(classes);
+
+		classGraph.generateGraph(graphName);
+	}
+
 	/**
 	 * Maps class fully qualified names to class descriptors.
 	 */
 	Map<String, ClassDescriptor> nameToClassMap = new HashMap<String, ClassDescriptor>();
 
+	private Filter filter = new Filter();
+
 	public ClassGraph() {
 	}
 
-	public ClassGraph(final Class<? extends Object> clazz) {
-		addClass(clazz);
+	/**
+	 * @param classes
+	 *            classes that shall be added to graph
+	 */
+	public ClassGraph(final Class<? extends Object>... classes) {
+		for (final Class<? extends Object> clazz : classes)
+			addClass(clazz);
 	}
 
-	public ClassGraph(final Object root) {
-		addClass(root.getClass());
+	/**
+	 * @param objects
+	 *            objects that shall be added to graph
+	 */
+	public ClassGraph(final Object... objects) {
+		for (final Object object : objects)
+			addClass(object.getClass());
 	}
 
+	/**
+	 * @param clazz
+	 *            class that shall be added to graph
+	 */
 	public ClassDescriptor addClass(final Class<? extends Object> clazz) {
 
 		if (clazz == null)
@@ -48,21 +72,60 @@ public class ClassGraph {
 		return new ClassDescriptor(clazz, this);
 	}
 
+	/**
+	 * @param object
+	 *            object that shall be added to graph
+	 */
 	public ClassDescriptor addObject(final Object object) {
 		return addClass(object.getClass());
 	}
 
-	public void generateGraph(final String graphName) {
-		generateGraph(graphName, false);
+	/**
+	 * @param path
+	 *            path to recursively scan for java source code could be
+	 *            relative to current project or absolute
+	 */
+	public void addProject(final String path) {
+		final ProjectScanner projectScanner = new ProjectScanner(new File(path));
+		for (final Clazz clazz : projectScanner.getAllClasses())
+			try {
+				System.out.println("Class full name: " + clazz.getFullName());
+				final Class c = this.getClass().forName(clazz.getFullName());
+				addClass(c);
+			} catch (final Exception exception) {
+				System.out.println("cannot add class: "
+						+ exception.getMessage());
+			}
 	}
 
-	public void generateGraph(final String graphName, final boolean keepDotFile) {
+	/**
+	 * @param resultFileName
+	 *            file name for the generated graph. Existing file with the same
+	 *            name will be overwritten.
+	 */
+	public void generateGraph(final String resultFileName) {
+		generateGraph(resultFileName, false);
+	}
+
+	/**
+	 * @param resultFileName
+	 *            file name for the generated graph. File extension will be
+	 *            added automatically. Existing file with the same name will be
+	 *            overwritten.
+	 * 
+	 * @param keepDotFile
+	 *            if set to <code>true</code> then intermediary GraphViz DOT
+	 *            file will be kept.
+	 */
+
+	public void generateGraph(final String resultFileName,
+			final boolean keepDotFile) {
 
 		final String desktopPath = CommonPathResolver.getDesktopDirectory()
 				.getAbsolutePath() + "/";
 
-		final String dotFilePath = desktopPath + graphName + ".dot";
-		final String imageFilePath = desktopPath + graphName + ".png";
+		final String dotFilePath = desktopPath + resultFileName + ".dot";
+		final String imageFilePath = desktopPath + resultFileName + ".png";
 
 		System.out.println("Dot file path:" + dotFilePath);
 
@@ -107,4 +170,22 @@ public class ClassGraph {
 		return resultStr;
 	}
 
+	public Filter getFilter() {
+		return filter;
+	}
+
+	/**
+	 * Hide orphaned class that have no references
+	 */
+	public void hideOrphanedClasses() {
+
+		for (final ClassDescriptor classDescriptor : nameToClassMap.values())
+			classDescriptor.hideClassIfNoReferences();
+
+	}
+
+	public void setFilter(final Filter filter) {
+		this.filter = filter;
+	}
+
 }