// See LICENSE for license details.

package chisel3.util

import chisel3._
import chisel3.core.ChiselAnnotation
import firrtl.transforms.{BlackBoxInline, BlackBoxResource, BlackBoxSourceHelper}

trait HasBlackBoxResource extends BlackBox {
  self: Module =>

  def setResource(blackBoxResource: String): Unit = {
    annotate(ChiselAnnotation(self, classOf[BlackBoxSourceHelper], BlackBoxResource(blackBoxResource).serialize))
  }
}

trait HasBlackBoxInline extends BlackBox {
  self: Module =>

  def setInline(blackBoxName: String, blackBoxInline: String): Unit = {
    annotate(ChiselAnnotation(
      self, classOf[BlackBoxSourceHelper], BlackBoxInline(blackBoxName, blackBoxInline).serialize))
  }
}