X-Git-Url: http://www2.svjatoslav.eu/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Feu%2Fsvjatoslav%2Finspector%2Fjava%2Fstructure%2FClassDescriptor.java;h=e1e45af91d0f2fc65d70180a417fd49f981d8f52;hb=ff4200b065c2eb7c8448fa060e7610ab57f6aec1;hp=c0be547d8bbb4d65b084d9429ff7d1f267773095;hpb=16ba5341215d24fdcfa07072d52cf87e5a4375ed;p=javainspect.git

diff --git a/src/main/java/eu/svjatoslav/inspector/java/structure/ClassDescriptor.java b/src/main/java/eu/svjatoslav/inspector/java/structure/ClassDescriptor.java
index c0be547..e1e45af 100644
--- a/src/main/java/eu/svjatoslav/inspector/java/structure/ClassDescriptor.java
+++ b/src/main/java/eu/svjatoslav/inspector/java/structure/ClassDescriptor.java
@@ -57,7 +57,13 @@ public class ClassDescriptor implements GraphElement {
 	/**
 	 * Amount of field and method references pointing to this class.
 	 */
-	private int incomingReferencesCount = 0;
+	private int referencesCount = 0;
+
+	// for interface, counts amount of found implementations
+	private int implementationsCount = 0;
+
+	// counts amount of times this class is extended
+	private int extensionsCount = 0;
 
 	public ClassDescriptor(final Class<? extends Object> clazz,
 			final ClassGraph dump) {
@@ -87,14 +93,21 @@ public class ClassDescriptor implements GraphElement {
 
 		indexMethods(clazz);
 
-		for (final Class interfaceClass : clazz.getInterfaces())
-			interfaces.add(dump.addClass(interfaceClass));
+		for (final Class interfaceClass : clazz.getInterfaces()) {
+			final ClassDescriptor classDescriptor = dump
+					.addClass(interfaceClass);
+			classDescriptor.registerImplementation();
+			interfaces.add(classDescriptor);
+		}
 
 		superClass = dump.addClass(clazz.getSuperclass());
+		if (superClass != null)
+			superClass.registerExtension();
+
 	}
 
 	public boolean areReferencesShown() {
-		return incomingReferencesCount <= MAX_REFERECNES_COUNT;
+		return referencesCount <= MAX_REFERECNES_COUNT;
 	}
 
 	public void enlistFieldReferences(final StringBuffer result) {
@@ -295,11 +308,24 @@ public class ClassDescriptor implements GraphElement {
 		return null;
 	}
 
+	/**
+	 * Returns field with given name (case is ignored). Or <code>null</code> if
+	 * field is not found.
+	 */
+	public FieldDescriptor getFieldIgnoreCase(final String fieldToSearch) {
+
+		for (final String fieldName : nameToFieldMap.keySet())
+			if (fieldToSearch.equalsIgnoreCase(fieldName))
+				return nameToFieldMap.get(fieldName);
+
+		return null;
+	}
+
 	@Override
 	public String getGraphId() {
 		final String result = "class_"
 				+ fullyQualifiedName.replace('.', '_').replace(";", "")
-				.replace("[L", "").replace('$', '_');
+						.replace("[L", "").replace('$', '_');
 		return result;
 	}
 
@@ -310,25 +336,28 @@ public class ClassDescriptor implements GraphElement {
 		return interfaceColor;
 	}
 
-	public String getPackageName() {
+	private int getOutgoingReferencesCount() {
+		int result = 0;
 
-		final int i = fullyQualifiedName.lastIndexOf('.');
+		// count method references
+		for (final MethodDescriptor methodDescriptor : methods)
+			result += methodDescriptor.getOutsideVisibleReferencesCount();
 
-		if (i == -1)
-			return "";
+		// count field references
+		for (final FieldDescriptor fieldDescriptor : nameToFieldMap.values())
+			result += fieldDescriptor.getOutsideVisibleReferencesCount();
 
-		return fullyQualifiedName.substring(0, i).replace("[L", "");
-	}
+		// count implemented interfaces
+		for (final ClassDescriptor classDescriptor : interfaces)
+			if (classDescriptor.isVisible())
+				result++;
 
-	public String getParentClassesName() {
-		int i = fullyQualifiedName.lastIndexOf('.');
-		final String fullClassName = fullyQualifiedName.substring(i + 1);
+		// count superclass
+		if (superClass != null)
+			if (superClass.isVisible())
+				result++;
 
-		i = fullClassName.lastIndexOf('$');
-		if (i == -1)
-			return "";
-		final String parentClassesName = fullClassName.substring(0, i);
-		return parentClassesName.replace('$', '.');
+		return result;
 	}
 
 	// public String getReadableName() {
@@ -345,6 +374,27 @@ public class ClassDescriptor implements GraphElement {
 	// return fullyQualifiedName;
 	// }
 
+	public String getPackageName() {
+
+		final int i = fullyQualifiedName.lastIndexOf('.');
+
+		if (i == -1)
+			return "";
+
+		return fullyQualifiedName.substring(0, i).replace("[L", "");
+	}
+
+	public String getParentClassesName() {
+		int i = fullyQualifiedName.lastIndexOf('.');
+		final String fullClassName = fullyQualifiedName.substring(i + 1);
+
+		i = fullClassName.lastIndexOf('$');
+		if (i == -1)
+			return "";
+		final String parentClassesName = fullClassName.substring(0, i);
+		return parentClassesName.replace('$', '.');
+	}
+
 	public String getSuperClassColor() {
 		if (superClassColor == null)
 			superClassColor = Utils.getNextLightColor();
@@ -352,33 +402,36 @@ public class ClassDescriptor implements GraphElement {
 		return superClassColor;
 	}
 
+	/**
+	 * Checks if class has field with given name (case is ignored). Returns
+	 * <code>true</code> if such field is found.
+	 */
+	public boolean hasFieldIgnoreCase(final String fieldToSearch) {
+
+		for (final String fieldName : nameToFieldMap.keySet())
+			if (fieldToSearch.equalsIgnoreCase(fieldName))
+				return true;
+
+		return false;
+	}
+
 	public void hide() {
 		isShown = false;
 	}
 
-	public boolean hideClassIfNoReferences() {
+	public void hideClassIfNoReferences() {
 		if (!isVisible())
-			return false;
-
-		int outgoingVisibleReferencesCount = 0;
-
-		for (final MethodDescriptor methodDescriptor : methods)
-			outgoingVisibleReferencesCount += methodDescriptor
-			.getOutsideVisibleReferencesCount();
-
-		for (final FieldDescriptor fieldDescriptor : nameToFieldMap.values())
-			outgoingVisibleReferencesCount += fieldDescriptor
-			.getOutsideVisibleReferencesCount();
+			return;
 
-		final int totalReferencesCount = outgoingVisibleReferencesCount
-				+ incomingReferencesCount;
+		final int totalReferencesCount = getOutgoingReferencesCount()
+				+ referencesCount + extensionsCount + implementationsCount;
 
 		if (totalReferencesCount == 0) {
 			hide();
-			return true;
+			return;
 		}
 
-		return false;
+		return;
 	}
 
 	public void indexFields(final Field[] fields) {
@@ -415,8 +468,19 @@ public class ClassDescriptor implements GraphElement {
 		return isShown;
 	}
 
+	/**
+	 * Register event when another class is extending this one.
+	 */
+	public void registerExtension() {
+		extensionsCount++;
+	}
+
+	public void registerImplementation() {
+		implementationsCount++;
+	}
+
 	public void registerReference() {
-		incomingReferencesCount++;
+		referencesCount++;
 	}
 
 	public void setDistinctiveColor(final String distinctiveColor) {