add constraints to all elements

OpenPonk · Nov 21, 2023 · 73305ce · 73305ce
1 parent 6bd1fda
commit 73305ce
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 2 deletions.
diff --git a/repository/OpenPonk-BormModel/BormActivity.class.st b/repository/OpenPonk-BormModel/BormActivity.class.st
@@ -12,11 +12,19 @@ Class {
 	#category : #'OpenPonk-BormModel-Models'
 }
 
-{ #category : #'as yet unclassified' }
+{ #category : #associating }
 BormActivity >> ==> anActivity [
 	^ BormFactory communicationFrom: self to: anActivity
 ]
 
+{ #category : #accessing }
+BormActivity >> allElements [
+	"should return all elements recursively (incl. sub-elements)"
+
+	^ super allElements , (self communications flatCollect: [ :each |
+		   (Set with: each) , each allElements ])
+]
+
 { #category : #accessing }
 BormActivity >> allEntities [
 	^ super allEntities , (self sent flatCollect: #allEntities)

diff --git a/repository/OpenPonk-BormModel/BormCommunication.class.st b/repository/OpenPonk-BormModel/BormCommunication.class.st
@@ -54,7 +54,15 @@ BormCommunication >> constraints [
 
 { #category : #accessing }
 BormCommunication >> dataFlows [
-	^ self elements
+
+	^ elements
+]
+
+{ #category : #accessing }
+BormCommunication >> elements [
+
+	self hasConstraint ifFalse: [ ^ super elements ].
+	^ super elements , { self constraint }
 ]
 
 { #category : #testing }

diff --git a/repository/OpenPonk-BormModel/BormProcessNode.class.st b/repository/OpenPonk-BormModel/BormProcessNode.class.st
@@ -28,6 +28,14 @@ BormProcessNode >> addTransition: aTransition [
 		ifFalse: [ self changed ]
 ]
 
+{ #category : #accessing }
+BormProcessNode >> allElements [
+	"should return all elements recursively (incl. sub-elements)"
+
+	^ super allElements , (self transitions flatCollect: [ :each |
+		   (Set with: each) , each allElements ])
+]
+
 { #category : #accessing }
 BormProcessNode >> allEntities [
 	^ (Set with: self entity) , (self outgoing flatCollect: #allEntities)

diff --git a/repository/OpenPonk-BormModel/BormTransition.class.st b/repository/OpenPonk-BormModel/BormTransition.class.st
@@ -42,6 +42,13 @@ BormTransition >> constraints [
 	^ constraint ifNil: [ #() ] ifNotNil: [ ^ {constraint} ]
 ]
 
+{ #category : #accessing }
+BormTransition >> elements [
+
+	self hasConstraint ifFalse: [ ^ #(  ) ].
+	^ { self constraint }
+]
+
 { #category : #accessing }
 BormTransition >> has: anEntity [
 	^ constraint = anEntity