net.sf.nakeduml.javageneration.NakedOperationMap Maven / Gradle / Ivy

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

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import net.sf.nakeduml.javametamodel.OJPathName;
import net.sf.nakeduml.metamodel.commonbehaviors.INakedBehavior;
import net.sf.nakeduml.metamodel.core.IParameterOwner;
import nl.klasse.octopus.codegen.umlToJava.maps.OperationMap;
import nl.klasse.octopus.model.IClassifier;
import nl.klasse.octopus.model.IParameter;
import nl.klasse.octopus.model.internal.types.OperationImpl;

public class NakedOperationMap extends OperationMap {
	private IParameterOwner parameterOwner;

	public NakedOperationMap(final IParameterOwner operation) {
		super(new OperationImpl(operation.getName()) {
			@Override
			public IClassifier getReturnType() {
				return operation.getReturnParameter() != null ? operation.getReturnParameter().getType() : null;
			}

			@Override
			public List getParameters() {
				return operation.getParameters();
			}
		});
		this.parameterOwner = operation;
		if (super.operation.getReturnType() == null) {
			// this.operationTypeMap=new ClassifierMap();
		} else {
			this.operationTypeMap = new NakedClassifierMap(super.operation.getReturnType());
		}
	}

	@Override
	public String javaOperName() {
		if (parameterOwner instanceof INakedBehavior) {
			INakedBehavior behaviour = (INakedBehavior) parameterOwner;
			if(behaviour.getSpecification()!=null){
				return behaviour.getSpecification().getName();
			}else{
				return super.javaOperName();
			}
		} else {
			return super.javaOperName();
		}
	}

	Map params = null;

	private Map getParamMap() {
		if (params == null) {
			params = new HashMap();
			for (IParameter p : getOperation().getParameters()) {
				params.put(p, new NakedClassifierMap(p.getType()));
			}
		}
		return params;
	}

	public IParameterOwner getOperation() {
		return this.parameterOwner;
	}

	@Override
	public OJPathName javaParamTypePath(IParameter elem) {
		return getParamMap().get(elem).javaTypePath();
	}
}