net.sf.nakeduml.javageneration.basicjava.simpleactions.StructuralFeatureReader Maven / Gradle / Ivy

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

import net.sf.nakeduml.javageneration.NakedStructuralFeatureMap;
import net.sf.nakeduml.javageneration.bpm.actions.ActionMap;
import net.sf.nakeduml.javageneration.util.OJUtil;
import net.sf.nakeduml.javametamodel.OJBlock;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedOperation;
import net.sf.nakeduml.metamodel.actions.INakedReadStructuralFeatureAction;
import net.sf.nakeduml.metamodel.activities.INakedOutputPin;
import nl.klasse.octopus.codegen.umlToJava.maps.StructuralFeatureMap;
import nl.klasse.octopus.oclengine.IOclEngine;

public class StructuralFeatureReader extends SimpleActionBuilder{
	public StructuralFeatureReader(IOclEngine oclEngine,INakedReadStructuralFeatureAction action){
		super(oclEngine, action);
	}
	@Override
	public void implementActionOn(OJAnnotatedOperation operation,OJBlock block){
		INakedOutputPin result = node.getResult();
		StructuralFeatureMap resultMap = buildResultVariable(operation, result);
		ActionMap actionMap = new ActionMap(node);
		OJBlock fs = buildLoopThroughTarget(operation, block, actionMap);
		String call = actionMap.targetName() + "." + new NakedStructuralFeatureMap(node.getFeature()).getter() + "()";
		NakedStructuralFeatureMap featureMap = OJUtil.buildStructuralFeatureMap(node.getFeature());
		if(resultMap.isCollection()){
			if(featureMap.isMany()){
				call = resultMap.umlName() + ".addAll(" + call + ")";
			}else{
				call = resultMap.umlName() + ".add(" + call + ")";
			}
		}else{
			// TODO what if the feature is Many? Force the result map to be many!
			call = resultMap.umlName() + "=" + call;
		}
		fs.addToStatements(call);
	}
}