net.sf.nakeduml.emf.extraction.GeneralizationExtractor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metamodel Show documentation
A uml code generator and execution engine
The newest version!
package net.sf.nakeduml.emf.extraction;

import net.sf.nakeduml.feature.StepDependency;
import net.sf.nakeduml.feature.visit.VisitBefore;
import net.sf.nakeduml.metamodel.core.INakedClassifier;
import net.sf.nakeduml.metamodel.core.INakedGeneralization;
import net.sf.nakeduml.metamodel.core.INakedInterface;
import net.sf.nakeduml.metamodel.core.INakedPowerType;
import net.sf.nakeduml.metamodel.core.INakedPowerTypeInstance;
import net.sf.nakeduml.metamodel.core.internal.NakedGeneralizationImpl;
import net.sf.nakeduml.metamodel.core.internal.NakedInterfaceRealizationImpl;
import net.sf.nakeduml.metamodel.core.internal.NakedPowerTypeInstanceImpl;
import net.sf.nakeduml.metamodel.name.SingularNameWrapper;

import org.eclipse.uml2.uml.Classifier;
import org.eclipse.uml2.uml.Enumeration;
import org.eclipse.uml2.uml.Generalization;
import org.eclipse.uml2.uml.GeneralizationSet;
import org.eclipse.uml2.uml.InterfaceRealization;

@StepDependency(phase = EmfExtractionPhase.class, requires = NameSpaceExtractor.class, after = NameSpaceExtractor.class)
public class GeneralizationExtractor extends AbstractExtractorFromEmf {
	@VisitBefore
	public void visitInterfaceRealization(InterfaceRealization r, NakedInterfaceRealizationImpl impl) {
		Classifier parent = r.getContract();
		Classifier child = r.getImplementingClassifier();
		INakedInterface nakedParent = (INakedInterface) getNakedPeer(parent);
		INakedClassifier nakedChild = (INakedClassifier) getNakedPeer(child);
		if (nakedParent == null) {
			System.out.println("Contract is not in model:" + parent.getName());
			return;
		}
		if (nakedChild == null) {
			System.out.println("Implementation is not in model:" + child.getName());
			return;
		}
		impl.setContract(nakedParent);
		impl.setImplementingClassifier(nakedChild);
		nakedChild.addInterface(impl);
	}

	@VisitBefore
	public void visitGeneralization(Generalization g) {
		Classifier parent = g.getGeneral();
		Classifier child = g.getSpecific();
		INakedClassifier nakedParent = (INakedClassifier) getNakedPeer(parent);
		INakedClassifier nakedChild = (INakedClassifier) getNakedPeer(child);
		if (nakedParent == null) {
			System.out.println("General is not in model:" + parent);
			return;
		}
		if (nakedChild == null) {
			System.out.println("Specific is not in model:" + child);
			return;
		}
		INakedGeneralization nakedGeneralization = new NakedGeneralizationImpl();
		initialize(nakedGeneralization, g, g.getOwner());
		nakedGeneralization.setParentAndChild(nakedParent, nakedChild);
		if (!g.getGeneralizationSets().isEmpty()) {
			GeneralizationSet generalizationSet = g.getGeneralizationSets().get(0);
			if (generalizationSet.getPowertype() instanceof Enumeration) {
				// TODO do this during analysis
				Enumeration powerType = (Enumeration) generalizationSet.getPowertype();
				INakedPowerTypeInstance ptl = new NakedPowerTypeInstanceImpl();
				SingularNameWrapper nameWrapper = new SingularNameWrapper(child.getName(), null);
				ptl.initialize(getId(g) + "EnumerationLiteral", nameWrapper.getDecapped().toString());
				this.workspace.putModelElement(ptl);
				nakedGeneralization.setPowerTypeLiteral(ptl);
				INakedPowerType ptw = (INakedPowerType) getNakedPeer(powerType);
				ptl.setOwnerElement(ptw);
				INakedClassifier representedSupertype = (INakedClassifier) getNakedPeer(parent);
				ptw.setRepresentedSupertype(representedSupertype);
			}
		}
	}
}