Skip to main content
summaryrefslogtreecommitdiffstats
blob: 769e3dc8ff00ec5a821b0a9d78d253ef6eec2f9a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/*******************************************************************************
 * Copyright (c) 2012, 2015 Willink Transformations and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     E.D.Willink - initial API and implementation
 *******************************************************************************/
package org.eclipse.qvtd.doc.bigmde2016.tests.qvtc;

import java.util.Collections;
import java.util.List;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.jdt.annotation.Nullable;
import org.eclipse.qvtd.doc.bigmde2016.tests.SimpleUMLGenerator;

import org.eclipse.qvtd.xtext.qvtcore.tests.uml2rdbms.simpleuml.Association;
import org.eclipse.qvtd.xtext.qvtcore.tests.uml2rdbms.simpleuml.Attribute;
import org.eclipse.qvtd.xtext.qvtcore.tests.uml2rdbms.simpleuml.Class;
import org.eclipse.qvtd.xtext.qvtcore.tests.uml2rdbms.simpleuml.Package;
import org.eclipse.qvtd.xtext.qvtcore.tests.uml2rdbms.simpleuml.PrimitiveDataType;
import org.eclipse.qvtd.xtext.qvtcore.tests.uml2rdbms.simpleuml.SimpleumlFactory;

public class QVTcSimpleUMLGenerator extends SimpleUMLGenerator<Package, Class, PrimitiveDataType, Attribute, Association>
{ 
	@Override
	protected @NonNull Association createAssociation(@NonNull Package parentPackage, @NonNull String name, @NonNull Class fromClass, @NonNull Class toClass) {
		Association anAssociation = SimpleumlFactory.eINSTANCE.createAssociation();
		anAssociation.setName(name);
		anAssociation.setSource(fromClass);
		anAssociation.setDestination(toClass);
		parentPackage.getElements().add(anAssociation);
		return anAssociation;
	}

	@Override
	protected @NonNull Class createClass(@NonNull Package parentPackage, @NonNull String name) {
		Class aClass = SimpleumlFactory.eINSTANCE.createClass();
		aClass.setName(name);
//		aClass.getStereotype().add("persistent");
		if (parentPackage != null) {
			parentPackage.getElements().add(aClass);
		}
		return aClass;
	}

	@Override
	protected @NonNull List<@NonNull ? extends EObject> createModel(@NonNull List<Package> rootPackages, @NonNull String name) {
		Package aModel = SimpleumlFactory.eINSTANCE.createPackage();
		aModel.setName(name);
		aModel.getElements().addAll(rootPackages);
		return Collections.singletonList(aModel);
	}

	public @NonNull Package createPackage(@Nullable Package parentPackage, @NonNull String name) {
		Package aPackage = SimpleumlFactory.eINSTANCE.createPackage();
		aPackage.setName(name);
		if (parentPackage != null) {
			parentPackage.getElements().add(aPackage);
		}
		return aPackage;
	}

	@Override
	protected @NonNull Attribute createProperty(@NonNull Class aClass, @NonNull String name, @NonNull PrimitiveDataType type) {
		Attribute aProperty = SimpleumlFactory.eINSTANCE.createAttribute();
		aProperty.setName(name);
		aProperty.setType(type);
		if (aClass != null) {
			aClass.getAttributes().add(aProperty);
		}
		return aProperty;
	}

	@Override
	protected @NonNull PrimitiveDataType createType(@NonNull Package parentPackage, @NonNull String name) {
		PrimitiveDataType aType = SimpleumlFactory.eINSTANCE.createPrimitiveDataType();
		aType.setName(name);
		if (parentPackage != null) {
			parentPackage.getElements().add(aType);
		}
		return aType;
	}
}

Back to the top