Add util methods

ftsrg · Feb 11, 2022 · 2c3e069 · 2c3e069
1 parent 97769d0
commit 2c3e069
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/plugins/core/hu.bme.mit.gamma.util/src/hu/bme/mit/gamma/util/GammaEcoreUtil.xtend b/plugins/core/hu.bme.mit.gamma.util/src/hu/bme/mit/gamma/util/GammaEcoreUtil.xtend
@@ -252,6 +252,21 @@ class GammaEcoreUtil {
 		return contents
 	}
 
+	def <T extends EObject> T getFirstOfAllContentsOfType(EObject object, Class<T> type) {
+		val contents = newLinkedList
+		contents += object.eContents
+		while (!contents.empty) {
+			val content = contents.poll
+			if (type.isInstance(content)) {
+				return content as T
+			}
+			else {
+				contents += content.eContents
+			}
+		}
+		return null
+	}
+
 	def boolean containsTransitively(EObject potentialContainer, EObject object) {
 		if (potentialContainer === null || object === null) {
 			return false
@@ -318,6 +333,10 @@ class GammaEcoreUtil {
 	def EObject normalLoad(String parentFolder, String fileName, ResourceSet resourceSet) {
 		return URI.createFileURI(parentFolder + File.separator + fileName).normalLoad(resourceSet)
 	}
+
+	def void resolveAll(ResourceSet resourceSet) {
+		EcoreUtil.resolveAll(resourceSet)
+	}
 
 	def Resource normalSave(ResourceSet resourceSet, EObject rootElem, URI uri) {
 		val resource = resourceSet.createResource(uri)

diff --git a/plugins/core/hu.bme.mit.gamma.util/src/hu/bme/mit/gamma/util/JavaUtil.xtend b/plugins/core/hu.bme.mit.gamma.util/src/hu/bme/mit/gamma/util/JavaUtil.xtend
@@ -37,6 +37,14 @@ class JavaUtil {
 		return IterableExtensions.flatten(inputs).toList
 	}
 
+	def <T> T getFirstOfType(Iterable<? super T> collection, Class<T> clazz) {
+		for (element : collection) {
+			if (clazz.isInstance(element)) {
+				return element as T
+			}
+		}
+	}
+
 	def boolean isUnique(Iterable<?> collection) {
 		val set = newHashSet
 		for (element : collection) {