/**
* This method is only useful if the graph has been built, and {@link #generateArrows(ArrayList)}
* has been called on all classes. This method removes any arrows that point to classes that are
* super classes of classes that this class has arrows to.
*/
public void removeExtraArrows() {
ArrayList<UMLClass> pointedTo = new ArrayList<UMLClass>();
for (UMLArrow arrow : this.arrows) {
pointedTo.add(arrow.getEndClass());
}
// If the used class is in a field, ignore it.
for (UMLClass firstClass : new ArrayList<UMLClass>(pointedTo)) {
for (UMLClass secondClass : new ArrayList<UMLClass>(pointedTo)) {
if (firstClass.checkExtendsOrImplements(secondClass)) {
pointedTo.remove(secondClass);
} else if (secondClass.checkExtendsOrImplements(firstClass)) {
pointedTo.remove(firstClass);
}
}
}
// Add back in extends/implements arrows
for (UMLArrow arrow : this.arrows) {
if (arrow.extendsOrImplements()) {
pointedTo.add(arrow.getEndClass());
}
}
for (UMLArrow arrow : new ArrayList<UMLArrow>(this.arrows)) {
if ((!pointedTo.contains(arrow.getEndClass())) && arrow.isUsesArrow()) {
this.arrows.remove(arrow);
}
}
}
public ArrayList<UMLClass> getAllUsedClasses() {
ArrayList<UMLClass> usedClasses = new ArrayList<UMLClass>();
for (UMLArrow arrow : this.arrows) {
if (arrow.isUsesArrow()) {
usedClasses.add(arrow.getEndClass());
}
}
return usedClasses;
}
// TODO Move these methods into UMLGraph where it makes more sense?
// TODO Have a seperate array stored for all of the arrows?
public void removeRedundantUsesArrows() {
ArrayList<UMLClass> extendsOrImplements = this.getAllExtendsOrImplements();
ArrayList<UMLClass> thisUsed = this.getAllUsedClasses();
// Go through everything this class extends or implements (directly or
// indirectly)
for (UMLClass extendedClass : extendsOrImplements) {
ArrayList<UMLClass> otherUsed = extendedClass.getAllUsedClasses();
// Go through all of the classes that this class uses
for (UMLClass usedClass : thisUsed) {
boolean removeArrow = true;
// If the other class uses the same thing as this class, check
// to see
// if all instances where that is used is the same in both
// classes
// or not used in this class. If so, remove the arrow from this
// class.
if (otherUsed.contains(usedClass)) {
// Check if the used class is within a field
boolean foundField = false;
for (UMLField field : this.getFields()) {
if (field.getType().getFullBaseDataType().equals(usedClass.getName())) {
foundField = true;
removeArrow = false;
}
}
// If the used class is in a field, don't remove the arrow,
// so skip the arrow checks.
if (!foundField) {
// Check everything for type, and if they are the same
// in the two
// If everything is the same, remove this class' used
// arrow for that thing
for (UMLMethod thisMethod : this.methods) {
boolean wasUsed = false;
// Check to see if this method uses the used class
// at all
for (TypeData d : thisMethod.getClassesUsed()) {
if (d.getFullBaseDataType().equals(usedClass.getName())) {
wasUsed = true;
break;
}
}
// The class was used in the method. Check to see if
// there is an identical method in the other.
if (wasUsed) {
boolean isSame = false;
for (UMLMethod extendedMethod : extendedClass.getMethods()) {
// If the methods have the same signature
if (thisMethod.sameSignature(extendedMethod)) {
isSame = true;
break;
}
}
if (!isSame) {
removeArrow = false;
break;
}
}
}
}
if (removeArrow) {
for (UMLArrow arrow : new ArrayList<UMLArrow>(this.arrows)) {
if (arrow.isUsesArrow()
&& arrow.getEndClass().getName().equals(usedClass.getName())) {
this.arrows.remove(arrow);
break;
}
}
}
}
}
}
}
