com.avsystem.commons.jiop.JCanBuildFrom.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of commons-shared_2.12.0-RC1 Show documentation
AVSystem commons library for Scala
The newest version!
package com.avsystem.commons
package jiop

import com.avsystem.commons.jiop.BasicJavaInterop._

import scala.collection.generic.CanBuildFrom
import scala.collection.mutable

trait JCanBuildFrom[-Elem, +To] extends CanBuildFrom[Nothing, Elem, To]
object JCanBuildFrom extends JCanBuildFroms

final class JCollectionCBF[A, C <: JCollection[A]](creator: => C) extends JCanBuildFrom[A, C] {
  def apply(from: Nothing) = apply()
  def apply() = new mutable.Builder[A, C] {
    val coll = creator

    def +=(elem: A): this.type = {
      coll.add(elem)
      this
    }

    def clear(): Unit =
      coll.clear()

    def result(): C =
      coll
  }
}

final class JMapCBF[K, V, M <: JMap[K, V]](creator: => M) extends JCanBuildFrom[(K, V), M] {
  def apply(from: Nothing) = apply()
  def apply() = new mutable.Builder[(K, V), M] {
    val map = creator

    def +=(elem: (K, V)): this.type = {
      map.put(elem._1, elem._2)
      this
    }

    def clear(): Unit =
      map.clear()

    def result(): M =
      map
  }
}

trait JCanBuildFroms extends JCollectionCanBuildFroms with JMapCanBuildFroms

trait JCollectionCanBuildFroms extends JIterableCBF

trait JIterableCBF extends JSetCBF with JLinkedListCBF {
  // for JIterable, JCollection, JList and JArrayList
  implicit def jArrayListCBF[A]: JCollectionCBF[A, JArrayList[A]] =
    new JCollectionCBF(new JArrayList)
}

trait JSetCBF extends JSortedSetCBF with JLinkedHashSetCBF {
  // for JSet and JHashSet
  implicit def jHashSetCBF[A]: JCollectionCBF[A, JHashSet[A]] =
    new JCollectionCBF(new JHashSet)
}

trait JLinkedHashSetCBF {
  implicit def jLinkedHashSetCBF[A]: JCollectionCBF[A, JLinkedHashSet[A]] =
    new JCollectionCBF(new JLinkedHashSet)
}

trait JSortedSetCBF {
  // for JSortedSet, JNavigableSet and JTreeSet
  implicit def jTreeSetCBF[A: Ordering]: JCollectionCBF[A, JTreeSet[A]] =
    new JCollectionCBF(new JTreeSet(Ordering[A]))
}

trait JLinkedListCBF {
  implicit def jLinkedListCBF[A]: JCollectionCBF[A, JLinkedList[A]] =
    new JCollectionCBF(new JLinkedList)
}

trait JMapCanBuildFroms extends JSortedMapCBF with JLinkedHashMapCBF {
  // for JMap and JHashMap
  implicit def jHashMapCBF[K, V]: JMapCBF[K, V, JHashMap[K, V]] =
    new JMapCBF(new JHashMap)
}

trait JLinkedHashMapCBF {
  implicit def jLinkedHashMapCBF[K, V]: JMapCBF[K, V, JLinkedHashMap[K, V]] =
    new JMapCBF(new JLinkedHashMap)
}

trait JSortedMapCBF {
  // for JSortedMap, JNavigableMap and JTreeMap
  implicit def jTreeMapCBF[K: Ordering, V]: JMapCBF[K, V, JTreeMap[K, V]] =
    new JMapCBF(new JTreeMap(Ordering[K]))
}