org.scalameter.MeasureBuilder.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalameter-core_2.11 Show documentation

scalameter-core

The newest version!

package org.scalameter






class MeasureBuilder[T, U](
  val ctx: Context,
  val warmer: Warmer,
  val measurer: Measurer[U],
  val regen: () => T,
  val setup: T => Unit,
  val teardown: T => Unit,
  val resultFunction: Seq[Quantity[U]] => Quantity[U]
) {
  def config(kvs: KeyValue*) = new MeasureBuilder(ctx ++ Context(kvs: _*), warmer,
    measurer, regen, setup, teardown, resultFunction)

  def withWarmer(w: Warmer) =
    new MeasureBuilder(ctx, w, measurer, regen, setup, teardown, resultFunction)

  def withMeasurer(m: Measurer[U]) =
    new MeasureBuilder(ctx, warmer, m, regen, setup, teardown, resultFunction)

  def withMeasurer[V](m: Measurer[V], a: Seq[Quantity[V]] => Quantity[V]) =
    new MeasureBuilder(ctx, warmer, m, regen, setup, teardown, a)

  def setUp(b: T => Unit) =
    new MeasureBuilder(ctx, warmer, measurer, regen, b, teardown, resultFunction)

  def tearDown(b: T => Unit) =
    new MeasureBuilder(ctx, warmer, measurer, regen, setup, b, resultFunction)

  def measureWith[S](b: T => S): Quantity[U] = {
    val oldctx = dyn.currentContext.value
    try {
      dyn.currentContext.value = ctx

      val x = regen()
      warmer match {
        case Warmer.Zero => // do nothing
        case _ =>
          for (i <- warmer.warming(ctx, () => setup(x), () => teardown(x))) b(x)
      }
  
      if (ctx(Key.exec.requireGC)) compat.Platform.collectGarbage()
  
      val measurements = measurer.measure[T](
        ctx, ctx(Key.exec.benchRuns),
        setup, teardown, regen, b
      )
  
      resultFunction(measurements)
    } finally {
      dyn.currentContext.value = oldctx
    }
  }

  def measure[S](b: =>S): Quantity[U] = {
    measureWith(_ => b)
  }
}


object MeasureBuilder {
  val doNothing = (u: Unit) => {}
  val unitRegen = () => {}
  val average = Aggregator.average
  val timeMeasurer = new Measurer.Default
}