2 * JavaInspect - Utility to visualize java software
3 * Copyright (C) 2013, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of version 2 of the GNU General Public License
7 * as published by the Free Software Foundation.
10 package eu.svjatoslav.inspector.java.structure;
13 import java.io.IOException;
14 import java.io.PrintWriter;
15 import java.util.HashMap;
18 import eu.svjatoslav.commons.file.CommonPathResolver;
19 import eu.svjatoslav.inspector.java.methods.Clazz;
20 import eu.svjatoslav.inspector.java.methods.ProjectScanner;
22 public class ClassGraph {
25 * Maps class fully qualified names to class descriptors.
27 Map<String, ClassDescriptor> nameToClassMap = new HashMap<String, ClassDescriptor>();
29 public Filter filter = new Filter();
34 public ClassGraph(final Class<? extends Object> clazz) {
38 public ClassGraph(final Object root) {
39 addClass(root.getClass());
42 public ClassDescriptor addClass(final Class<? extends Object> clazz) {
47 final String className = clazz.getName();
49 if (nameToClassMap.containsKey(className))
50 return nameToClassMap.get(className);
52 return new ClassDescriptor(clazz, this);
55 public ClassDescriptor addObject(final Object object) {
56 return addClass(object.getClass());
59 public void addProject(final String path) {
60 final ProjectScanner projectScanner = new ProjectScanner(new File(path));
61 for (final Clazz clazz : projectScanner.getAllClasses())
63 System.out.println("Class full name: " + clazz.getFullName());
64 final Class c = this.getClass().forName(clazz.getFullName());
66 } catch (final Exception exception) {
67 System.out.println("cannot add class: "
68 + exception.getMessage());
72 public void generateGraph(final String graphName) {
73 generateGraph(graphName, false);
76 public void generateGraph(final String graphName, final boolean keepDotFile) {
78 final String desktopPath = CommonPathResolver.getDesktopDirectory()
79 .getAbsolutePath() + "/";
81 final String dotFilePath = desktopPath + graphName + ".dot";
82 final String imageFilePath = desktopPath + graphName + ".png";
84 System.out.println("Dot file path:" + dotFilePath);
87 // write DOT file to disk
88 final PrintWriter out = new PrintWriter(dotFilePath);
92 // execute GraphViz to visualize graph
95 .exec(new String[] { "dot", "-Tpng", dotFilePath, "-o",
96 imageFilePath }).waitFor();
97 } catch (final InterruptedException e) {
103 final File dotFile = new File(dotFilePath);
106 } catch (final IOException e) {
107 System.err.println(e);
111 private String getDot() {
112 final StringBuffer result = new StringBuffer();
114 result.append("digraph Java {\n");
115 result.append("graph [rankdir=LR, overlap = false, concentrate=true];\n");
117 for (final Map.Entry<String, ClassDescriptor> entry : nameToClassMap
119 result.append(entry.getValue().getDot());
121 result.append("}\n");
123 final String resultStr = result.toString();