org.umlg.framework.ModelVisitor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of uml Show documentation
The newest version!
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);
		}
	}

}