upickle.implicits.ObjectContexts.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of upickle-implicits_native0.5_2.12 Show documentation

upickle-implicits

There is a newer version: 4.0.2

Show newest version

package upickle.implicits

import upickle.core.ObjVisitor


trait BaseCaseObjectContext {
  def storeAggregatedValue(currentIndex: Int, v: Any): Unit

  def visitKey(index: Int) = _root_.upickle.core.StringVisitor

  var currentIndex = -1

  def storeValueIfNotFound(i: Int, v: Any): Unit

  protected def errorMissingKeys(rawArgsLength: Int, mappedArgs: Array[String]): Unit

  protected def checkErrorMissingKeys(rawArgsBitset: Long): Boolean
}

abstract class CaseObjectContext[V](fieldCount: Int) extends ObjVisitor[Any, V] with BaseCaseObjectContext {
  var found = 0L

  def visitValue(v: Any, index: Int): Unit = {
    if (currentIndex != -1 && ((found & (1L << currentIndex)) == 0)) {
      storeAggregatedValue(currentIndex, v)
      found |= (1L << currentIndex)
    }
  }

  def storeValueIfNotFound(i: Int, v: Any) = {
    if ((found & (1L << i)) == 0) {
      found |= (1L << i)
      storeAggregatedValue(i, v)
    }
  }

  protected def errorMissingKeys(rawArgsLength: Int, mappedArgs: Array[String]) = {
    val keys = for {
      i <- 0 until rawArgsLength
      if (found & (1L << i)) == 0
    } yield mappedArgs(i)
    throw new _root_.upickle.core.Abort(
      "missing keys in dictionary: " + keys.mkString(", ")
    )
  }

  protected def checkErrorMissingKeys(rawArgsBitset: Long) = {
    found != rawArgsBitset
  }
}

abstract class HugeCaseObjectContext[V](fieldCount: Int) extends ObjVisitor[Any, V] with BaseCaseObjectContext {
  var found = new Array[Long](fieldCount / 64 + 1)

  def visitValue(v: Any, index: Int): Unit = {
    if (currentIndex != -1 && ((found(currentIndex / 64) & (1L << currentIndex)) == 0)) {
      storeAggregatedValue(currentIndex, v)
      found(currentIndex / 64) |= (1L << currentIndex)
    }
  }

  def storeValueIfNotFound(i: Int, v: Any) = {
    if ((found(i / 64) & (1L << i)) == 0) {
      found(i / 64) |= (1L << i)
      storeAggregatedValue(i, v)
    }
  }

  protected def errorMissingKeys(rawArgsLength: Int, mappedArgs: Array[String]) = {
    val keys = for {
      i <- 0 until rawArgsLength
      if (found(i / 64) & (1L << i)) == 0
    } yield mappedArgs(i)
    throw new _root_.upickle.core.Abort(
      "missing keys in dictionary: " + keys.mkString(", ")
    )
  }

  protected def checkErrorMissingKeys(rawArgsLength: Long) = {
    var bits = 0
    for (v <- found) bits += java.lang.Long.bitCount(v)
    bits != rawArgsLength
  }
}