scala.scalanative.optimizer.pass.ExternHoisting.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tools_2.11 Show documentation

tools

The newest version!

package scala.scalanative
package optimizer
package pass

import analysis.ClassHierarchy._
import analysis.ClassHierarchyExtractors._
import nir._

/** Hoists external members from external modules to top-level scope. */
class ExternHoisting(implicit top: Top) extends Pass {
  private def stripName(n: Global): Global = {
    val id = n.id
    assert(id.startsWith("extern."))
    Global.Member(Global.Top("__extern"), id.substring(7)) // strip extern. prefix
  }

  override def onDefns(defns: Seq[Defn]): Seq[Defn] = {
    val buf = scala.collection.mutable.UnrolledBuffer.empty[Defn]

    defns.foreach {
      case defn @ Defn.Declare(attrs, name, _) if attrs.isExtern =>
        buf += super.onDefn(defn.copy(name = stripName(name)))
      case defn @ Defn.Define(attrs, name, _, _) if attrs.isExtern =>
        buf += super.onDefn(defn.copy(name = stripName(name)))
      case defn @ Defn.Const(attrs, name, _, _) if attrs.isExtern =>
        buf += super.onDefn(defn.copy(name = stripName(name)))
      case defn @ Defn.Var(attrs, name, _, _) if attrs.isExtern =>
        buf += super.onDefn(defn.copy(name = stripName(name)))
      case defn =>
        buf += super.onDefn(defn)
    }

    buf.toSeq
  }

  override def onVal(value: Val) = value match {
    case Val.Global(n @ Global.Member(_, id), ty)
        if id.startsWith("extern.__") =>
      Val.Global(stripName(n), ty)
    case Val.Global(n @ Ref(node), ty) if node.attrs.isExtern =>
      Val.Global(stripName(n), ty)
    case _ =>
      super.onVal(value)
  }
}

object ExternHoisting extends PassCompanion {
  override def apply(config: tools.Config, top: Top) =
    new ExternHoisting()(top)
}