change the parents of an EolUnionType

epsilonlabs · Dec 3, 2024 · e598e37 · e598e37
1 parent a17459d
commit e598e37
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/...pse.epsilon.eol.engine/src/org/eclipse/epsilon/eol/staticanalyser/types/EolUnionType.java b/...pse.epsilon.eol.engine/src/org/eclipse/epsilon/eol/staticanalyser/types/EolUnionType.java
@@ -37,8 +37,17 @@ public boolean isAbstract() {
 	}
 
 	@Override
-	public List<EolType> getParentTypes(){
-		return new ArrayList<EolType>(containedTypes); 
+	public List<EolType> getParentTypes() {
+		Set<EolType> parentsOfContained = containedTypes.stream().flatMap(t -> t.getParentTypes().stream())
+				.collect(Collectors.toSet());
+		if (parentsOfContained.size() == 1) {
+			return new ArrayList<EolType>(parentsOfContained);
+		} else {
+			EolUnionType t = new EolUnionType(parentsOfContained);
+			List<EolType> r = new ArrayList<EolType>();
+			r.add(t);
+			return r;
+		}
 	}
 
 	@Override