connected resource projection

Author: Oron Port

core/src/main/scala/dfhdl/platforms/resources/Resource.scala

@@ -1,13 +1,28 @@
 package dfhdl.platforms.resources
 import scala.annotation.implicitNotFound
-import scala.collection.mutable.ListBuffer
+import scala.collection.mutable
+import dfhdl.compiler.ir.constraints.SigConstraint
 
 trait Resource extends ResourceContext:
-  private val owner: ResourceOwner = dfc.mutableDB.ResourceOwnershipContext.owner
-  private val connections = ListBuffer[Resource]()
+  private val connections = mutable.ListBuffer[Resource]()
   protected[resources] def connect(that: Resource): Unit =
     connections += that
+  private lazy val allConnections: List[Resource] =
+    val visited = mutable.Set[Resource]()
+    val result = mutable.ListBuffer[Resource]()
+    def dfs(res: Resource): Unit =
+      for (conn <- res.connections)
+        if (!visited.contains(conn))
+          visited += conn
+          result += conn
+          dfs(conn)
+    dfs(this)
+    result.distinct.toList
+  end allConnections
+  lazy val allSigConstraints: List[SigConstraint] =
+    allConnections.flatMap(_.directAndOwnerSigConstraints).distinct
   owner.addResource(this)
+end Resource
 
 private trait ResourceLP:
   import Resource.CanConnect

core/src/main/scala/dfhdl/platforms/resources/ResourceContext.scala

@@ -2,20 +2,32 @@ package dfhdl.platforms.resources
 import dfhdl.internals.*
 import dfhdl.core.*
 import scala.annotation.Annotation
-import dfhdl.compiler.ir.constraints.Constraint
+import dfhdl.compiler.ir.constraints.{Constraint, SigConstraint}
 import scala.collection.mutable.ListBuffer
 
 trait ResourceContext extends OnCreateEvents, HasDFC, HasClsMetaArgs:
   final lazy val dfc: DFC = __dfc
   final lazy val id: String = dfc.nameOpt.get
-  private var resourceConstraints = ListBuffer[Constraint]()
-  def getResourceConstraints: List[Constraint] =
-    resourceConstraints.toList ++ dfc.annotations.collect {
-      case constraint: Constraint => constraint
-    }
+  // the top-level resource owner is set to be itself
+  private[resources] val owner: ResourceOwner =
+    dfc.mutableDB.ResourceOwnershipContext.ownerOpt.getOrElse(this.asInstanceOf[ResourceOwner])
+  private[resources] val isTopResource: Boolean =
+    dfc.mutableDB.ResourceOwnershipContext.ownerOpt.isEmpty
+  private var resourceConstraints = ListBuffer.from(dfc.annotations.collect {
+    case constraint: Constraint => constraint
+  })
+  def getResourceConstraints: List[Constraint] = resourceConstraints.toList
+  // the constraints that are directly applied to this resource and its owners
+  private[resources] lazy val directAndOwnerSigConstraints: List[SigConstraint] =
+    val ownerConstraints = if (isTopResource) Nil else owner.directAndOwnerSigConstraints
+    (ownerConstraints ++ getResourceConstraints.collect {
+      case constraint: SigConstraint => constraint
+    }).distinct
   protected def __dfc: DFC =
     println("Severe error: missing DFHDL context!\nMake sure you enable the DFHDL compiler plugin.")
     sys.exit(1)
+  protected def injectConstraint(constraint: Constraint): Unit =
+    resourceConstraints += constraint
   protected def setClsNamePos(
       name: String,
       position: Position,