net.sf.nakeduml.javageneration.basicjava.RedefinitionImplementor Maven / Gradle / Ivy

Go to download
package net.sf.nakeduml.javageneration.basicjava;

import net.sf.nakeduml.feature.visit.VisitBefore;
import net.sf.nakeduml.javageneration.AbstractJavaProducingVisitor;
import net.sf.nakeduml.javageneration.NakedStructuralFeatureMap;
import net.sf.nakeduml.javageneration.util.OJUtil;
import net.sf.nakeduml.javametamodel.OJBlock;
import net.sf.nakeduml.javametamodel.OJClass;
import net.sf.nakeduml.javametamodel.OJField;
import net.sf.nakeduml.javametamodel.OJIfStatement;
import net.sf.nakeduml.javametamodel.OJOperation;
import net.sf.nakeduml.javametamodel.OJParameter;
import net.sf.nakeduml.javametamodel.OJPathName;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedField;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedOperation;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotationValue;
import net.sf.nakeduml.metamodel.core.INakedClassifier;
import net.sf.nakeduml.metamodel.core.INakedEntity;
import net.sf.nakeduml.metamodel.core.INakedInterface;
import net.sf.nakeduml.metamodel.core.INakedProperty;
import net.sf.nakeduml.metamodel.core.INakedStructuredDataType;

//TODO if the property is not derived, remove the field 
//TODO replace references to the field with the redefining field's getter
//TODO downcast parameters to the correct type 
public class RedefinitionImplementor extends AbstractJavaProducingVisitor {
	@VisitBefore(matchSubclasses = true)
	public void property(INakedProperty p) {
		visitProperty(p.getOwner(), p);
	}

	@VisitBefore(matchSubclasses = true, match = { INakedEntity.class, INakedStructuredDataType.class })
	public void classifier(INakedClassifier c) {
		for (INakedProperty p : c.getEffectiveAttributes()) {
			if (p.getOwner() instanceof INakedInterface) {
				visitProperty(c, p);
			}
		}
	}

	private void visitProperty(INakedClassifier owner, INakedProperty p) {
		if (p.isNavigable()) {
			NakedStructuralFeatureMap map = new NakedStructuralFeatureMap(p);
			OJClass c = findJavaClass(owner);
			for (INakedProperty redefinedProperty : p.getRedefinedProperties()) {
				implementRedefinition(map, c, redefinedProperty);
			}
		}
	}

	public void implementRedefinition(NakedStructuralFeatureMap redefiningMap, OJClass c, INakedProperty redefinedProperty) {
		NakedStructuralFeatureMap redefinedMap = new NakedStructuralFeatureMap(redefinedProperty);
		OJField f = c.findField(redefinedMap.umlName());
		if (f != null) {
			c.removeFromFields(f);
			redefineOperation(c, redefinedMap.setter(), redefiningMap.setter(), redefinedMap.javaTypePath());
			redefineOperation(c, redefinedMap.adder(), redefiningMap.adder(), redefinedMap.javaBaseTypePath());
			redefineOperation(c, redefinedMap.remover(), redefiningMap.remover(), redefinedMap.javaBaseTypePath());
		}
		OJAnnotatedOperation o = (OJAnnotatedOperation) OJUtil.findOperation(c, redefinedMap.getter());
		// might exist if the modeler defined an
		// attribute with the same name or if it was inherited from an interface
		if (o == null) {
			o = new OJAnnotatedOperation();
			o.setName(redefinedMap.getter());
			o.setReturnType(redefinedMap.javaTypePath());
			c.addToOperations(o);
			o.setBody(null);
		}
		if (o.getBody() == null || redefinedProperty.getOwner() instanceof INakedInterface) {
			// force implementation of interface redefinition
			o.setBody(new OJBlock());
			if (redefinedMap.isMany()) {
				if (redefiningMap.isOne()) {
					wrapInCollection(c, o, redefiningMap, redefinedMap);
				} else {
					convertToCorrectCollectionType(c, o, redefiningMap, redefinedMap);
				}
			} else {
				o.getBody().addToStatements("return " + (redefiningMap.getter() + "()"));
			}
		}
	}

	private void redefineOperation(OJClass c, String redefinedOperationName, String redefiningOperationName, OJPathName paramType) {
		OJAnnotatedOperation redefinedOperation = (OJAnnotatedOperation) OJUtil.findOperation(c, redefinedOperationName);
		OJAnnotatedOperation redefiningOperation = (OJAnnotatedOperation) OJUtil.findOperation(c, redefiningOperationName);
		if (redefiningOperation != null) {
			if (redefinedOperation == null) {
				redefinedOperation = new OJAnnotatedOperation();
				redefinedOperation.addParam("in", paramType);
				redefinedOperation.setName(redefinedOperationName);
				c.addToOperations(redefinedOperation);
			}
			redefinedOperation.setBody(new OJBlock());
			OJParameter parm = redefinedOperation.getParameters().get(0);
			OJAnnotatedField tmpField = new OJAnnotatedField(parm.getName(), redefiningOperation.getParameters().get(0).getType());
			parm.setName(parm.getName() + "New");
			OJPathName collectionType = getRawType(tmpField.getType());
			if (!collectionType.equals(tmpField.getType())) {
				suppressTypeChecks(redefinedOperation);
			}
			tmpField.setInitExp("(" + collectionType.getLast() + ")" + parm.getName());
			redefinedOperation.getBody().addToLocals(tmpField);
			redefinedOperation.getBody().addToStatements(redefiningOperationName + "(" + tmpField.getName() + ")");
		}
	}

	public OJPathName getRawType(OJPathName pathName) {
		OJPathName collectionType = pathName.getDeepCopy();
		collectionType.removeAllFromElementTypes();
		return collectionType;
	}

	private void convertToCorrectCollectionType(OJClass c, OJAnnotatedOperation o, NakedStructuralFeatureMap redefiningMap,
			NakedStructuralFeatureMap redefinedMap) {
		if (redefinedMap.getFeature().getType() != redefiningMap.getFeature().getType()) {
			c.addToImports(redefinedMap.javaTypePath());
			OJPathName collectionType = getRawType(redefiningMap.javaTypePath());
			suppressTypeChecks(o);
			o.getBody().addToStatements("return " + "(" + collectionType + ")" + (redefiningMap.getter() + "()"));
		} else {
			o.getBody().addToStatements("return " + (redefiningMap.getter() + "()"));
		}
	}

	public void suppressTypeChecks(OJAnnotatedOperation o) {
		OJAnnotationValue suppress = new OJAnnotationValue(new OJPathName("SuppressWarnings"), "unchecked");
		suppress.addStringValue("rawtypes");
		o.putAnnotation(suppress);
	}

	private void wrapInCollection(OJClass c, OJOperation o, NakedStructuralFeatureMap redefiningMap, NakedStructuralFeatureMap redefinedMap) {
		OJAnnotatedField result = new OJAnnotatedField("result", redefinedMap.javaTypePath());
		c.addToImports(redefinedMap.javaTypePath());
		c.addToImports(redefinedMap.javaDefaultTypePath());
		result.setInitExp(redefinedMap.javaDefaultValue());
		o.getBody().addToLocals(result);
		OJIfStatement ifNotNull = new OJIfStatement((redefiningMap.getter() + "()") + "!=null", "result.add("
				+ (redefiningMap.getter() + "()") + ")");
		o.getBody().addToStatements(ifNotNull);
		o.getBody().addToStatements("return result");
	}
}