de.sciss.fscape.stream.Buf.scala Maven / Gradle / Ivy

Go to download
/*
 *  Buf.scala
 *  (FScape)
 *
 *  Copyright (c) 2001-2016 Hanns Holger Rutz. All rights reserved.
 *
 *  This software is published under the GNU General Public License v2+
 *
 *
 *  For further information, please contact Hanns Holger Rutz at
 *  [email protected]
 */

package de.sciss.fscape.stream

import java.util.concurrent.atomic.AtomicInteger

trait BufLike {
  def release()(implicit ctrl: Control): Unit
  def acquire(): Unit

  def assertAllocated(): Unit

  var size: Int

  def allocCount(): Int

  def at(idx: Int): Any
}

object BufD {
  def apply(elems: Double*): BufD = {
    val arr = elems.toArray
    new BufD(arr, size = arr.length, borrowed = false)
  }

  def alloc(size: Int): BufD = {
    new BufD(new Array[Double](size), size = size, borrowed = true)
  }
}
final class BufD private(val buf: Array[Double], var size: Int, borrowed: Boolean)
  extends BufLike {

  private[this] val _allocCount = if (borrowed) new AtomicInteger(1) else null

  def assertAllocated(): Unit = require(!borrowed || _allocCount.get() > 0)

  def allocCount(): Int = _allocCount.get()

  def at(idx: Int): Any = buf(idx)

  def acquire(): Unit = if (borrowed) {
    /* val oldCount = */ _allocCount.getAndIncrement()
    // require(oldCount >= 0)
  }

  def release()(implicit ctrl: Control): Unit = if (borrowed) {
    val newCount = _allocCount.decrementAndGet()
    require(newCount >= 0)
    if (newCount == 0) ctrl.returnBufD(this)
  }

  override def toString: String =
    if (size == 1) buf(0).toString else s"BufD(size = $size)@${hashCode.toHexString}"
}

object BufI {
  def apply(elems: Int*): BufI = {
    val arr = elems.toArray
    new BufI(arr, size = arr.length, borrowed = false)
  }

  def alloc(size: Int): BufI = {
    new BufI(new Array[Int](size), size = size, borrowed = true)
  }
}
final class BufI private(val buf: Array[Int], var size: Int, borrowed: Boolean) extends BufLike {
  private[this] val _allocCount = if (borrowed) new AtomicInteger(1) else null

  def assertAllocated(): Unit = require(!borrowed || _allocCount.get() > 0)

  def allocCount(): Int = _allocCount.get()

  def at(idx: Int): Any = buf(idx)

  def acquire(): Unit = if (borrowed)
    _allocCount.getAndIncrement()

  def release()(implicit ctrl: Control): Unit = if (borrowed) {
    val newCount = _allocCount.decrementAndGet()
    require(newCount >= 0)
    if (newCount == 0) ctrl.returnBufI(this)
  }

  override def toString =
    if (size == 1) buf(0).toString else s"BufI(size = $size)@${hashCode.toHexString}"
}

object BufL {
  def apply(elems: Long*): BufL = {
    val arr = elems.toArray
    new BufL(arr, size = arr.length, borrowed = false)
  }

  def alloc(size: Int): BufL = {
    new BufL(new Array[Long](size), size = size, borrowed = true)
  }
}
final class BufL private(val buf: Array[Long], var size: Int, borrowed: Boolean) extends BufLike {
  private[this] val _allocCount = if (borrowed) new AtomicInteger(1) else null

  def assertAllocated(): Unit = require(!borrowed || _allocCount.get() > 0)

  def allocCount(): Int = _allocCount.get()

  def at(idx: Int): Any = buf(idx)

  def acquire(): Unit = if (borrowed)
    _allocCount.getAndIncrement()

  def release()(implicit ctrl: Control): Unit = if (borrowed) {
    val newCount = _allocCount.decrementAndGet()
    require(newCount >= 0)
    if (newCount == 0) ctrl.returnBufL(this)
  }

  override def toString =
    if (size == 1) buf(0).toString else s"BufL(size = $size)@${hashCode.toHexString}"
}