src/main/scala/firrtl/passes/memlib/DecorateMems.scala


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

// See LICENSE for license details.

package firrtl
package passes
package memlib

import firrtl.options.PreservesAll
import firrtl.stage.Forms

class CreateMemoryAnnotations(reader: Option[YamlFileReader]) extends Transform
    with DependencyAPIMigration
    with PreservesAll[Transform] {

  override def prerequisites = Forms.MidForm
  override def optionalPrerequisites = Seq.empty
  override def optionalPrerequisiteOf = Forms.MidEmitters

  def execute(state: CircuitState): CircuitState = reader match {
    case None => state
    case Some(r) =>
      import CustomYAMLProtocol._
      val configs = r.parse[Config]
      val oldAnnos = state.annotations
      val (as, pins) = configs.foldLeft((oldAnnos, Seq.empty[String])) { case ((annos, pins), config) =>
        (annos, pins :+ config.pin.name)
      }
      state.copy(annotations = PinAnnotation(pins.toSeq) +: as)
  }
}