org.umlg.javageneration.visitor.clazz.ClassCheckConstraintsBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-generation Show documentation
The newest version!
package org.umlg.javageneration.visitor.clazz;

import org.eclipse.uml2.uml.AssociationClass;
import org.eclipse.uml2.uml.Class;
import org.eclipse.uml2.uml.Constraint;
import org.umlg.framework.ModelLoader;
import org.umlg.framework.VisitSubclasses;
import org.umlg.framework.Visitor;
import org.umlg.generation.Workspace;
import org.umlg.java.metamodel.OJField;
import org.umlg.java.metamodel.OJPathName;
import org.umlg.java.metamodel.annotation.OJAnnotatedClass;
import org.umlg.java.metamodel.annotation.OJAnnotatedOperation;
import org.umlg.javageneration.util.UmlgGenerationUtil;
import org.umlg.javageneration.util.UmlgClassOperations;
import org.umlg.javageneration.visitor.BaseVisitor;

import java.util.List;

/**
 * Date: 2013/03/10
 * Time: 1:55 PM
 */
public class ClassCheckConstraintsBuilder extends BaseVisitor implements Visitor {

    public ClassCheckConstraintsBuilder(Workspace workspace) {
        super(workspace);
    }

    @Override
    @VisitSubclasses({Class.class, AssociationClass.class})
    public void visitBefore(Class clazz) {
        OJAnnotatedClass annotatedClass = findOJClass(clazz);
        addCheckConstraints(annotatedClass, clazz);

        //To change body of implemented methods use File | Settings | File Templates.
    }

    @Override
    public void visitAfter(Class element) {
        //To change body of implemented methods use File | Settings | File Templates.
    }

    private void addCheckConstraints(OJAnnotatedClass annotatedClass, Class clazz) {
        OJAnnotatedOperation checkConstraints = new OJAnnotatedOperation("checkClassConstraints", new OJPathName("java.util.List").addToGenerics(UmlgGenerationUtil.UmlgConstraintViolation));
        UmlgGenerationUtil.addOverrideAnnotation(checkConstraints);
        OJField result = new OJField("result", new OJPathName("java.util.List").addToGenerics(UmlgGenerationUtil.UmlgConstraintViolation));
        result.setInitExp("new ArrayList<" + UmlgGenerationUtil.UmlgConstraintViolation.getLast() + ">()");
        checkConstraints.getBody().addToLocals(result);
        List constraints = ModelLoader.INSTANCE.getConstraints(clazz);
        if (!clazz.getGeneralizations().isEmpty()) {
            checkConstraints.getBody().addToStatements("result.addAll(super.checkClassConstraints())");
        }
        for (Constraint constraint : constraints) {
            checkConstraints.getBody().addToStatements("result.addAll(" + UmlgClassOperations.checkClassConstraintName(constraint) + "())");
        }
        checkConstraints.getBody().addToStatements("return result");
        annotatedClass.addToOperations(checkConstraints);
    }

}