org.umlg.framework.ModelVisitor Maven / Gradle / Ivy
package org.umlg.framework;
import org.eclipse.uml2.uml.Element;
import org.eclipse.uml2.uml.PackageImport;
import org.eclipse.uml2.uml.Profile;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.Arrays;
import java.util.List;
public class ModelVisitor {
@SuppressWarnings("unchecked")
public static void visitModel(Element element, @SuppressWarnings("rawtypes") Visitor visitor) {
Type[] types = visitor.getClass().getGenericInterfaces();
if (types.length != 1) {
throw new IllegalStateException(String.format("Visitor must have one and only one generic interface, visitor %s has %s", visitor.getClass()
.getSimpleName(), types.length));
}
ParameterizedType t = (ParameterizedType) types[0];
Type[] paramTypes = t.getActualTypeArguments();
if (paramTypes.length != 1) {
throw new IllegalStateException();
}
// This is the type the visitor is visiting
Type visitingType = paramTypes[0];
try {
Class visitingTypeClass = (Class) visitingType;
Method method = visitor.getClass().getMethod("visitBefore", visitingTypeClass);
VisitFilter visitFilter = method.getAnnotation(VisitFilter.class);
Class filter = (visitFilter != null ? visitFilter.value() : DataTypePropertyFilter.class);
try {
if (!filter.newInstance().filter(element)) {
if (method != null && method.isAnnotationPresent(VisitSubclasses.class)) {
VisitSubclasses visitSubclasses = method.getAnnotation(VisitSubclasses.class);
if (visitingTypeClass.isAssignableFrom(element.getClass())) {
List> umlClasses = Arrays.asList(visitSubclasses.value());
for (Class umlClass : umlClasses) {
Class[] elementInterfaces = element.getClass().getInterfaces();
for (Class elementInterface : elementInterfaces) {
if (umlClass.equals(elementInterface) || umlClass.equals(Element.class)) {
visitor.visitBefore(element);
break;
}
}
}
}
} else {
Class[] elementInterfaces = element.getClass().getInterfaces();
for (Class elementInterface : elementInterfaces) {
if (visitingTypeClass.equals(elementInterface)) {
visitor.visitBefore(element);
break;
}
}
}
}
} catch (Exception e) {
throw new RuntimeException(e);
}
if (element instanceof PackageImport) {
PackageImport pi = (PackageImport)element;
if (!(pi.getImportedPackage() instanceof Profile)) {
org.eclipse.uml2.uml.Package p = pi.getImportedPackage();
for (Element e : p.getOwnedElements()) {
visitModel(e, visitor);
}
}
} else {
for (Element e : element.getOwnedElements()) {
visitModel(e, visitor);
}
}
} catch (SecurityException e1) {
throw new RuntimeException(e1);
} catch (NoSuchMethodException e1) {
throw new RuntimeException(e1);
}
}
}