Refactor set builder benchmark generators

Author: Abduqodiri Qurbonzoda

benchmarks-libraries/src/jmh/java/benchmarks/immutableSet/capsule/builder/utils.kt

@@ -19,14 +19,15 @@
 package benchmarks.immutableSet.capsule.builder
 
 import benchmarks.immutableSize
+import benchmarks.IntWrapper
 
-fun <E> persistentSetBuilderAdd(
-        elements: List<E>,
+fun persistentSetBuilderAdd(
+        elements: List<IntWrapper>,
         immutablePercentage: Double
-): io.usethesource.capsule.Set.Transient<E> {
+): io.usethesource.capsule.Set.Transient<IntWrapper> {
     val immutableSize = immutableSize(elements.size, immutablePercentage)
 
-    var set = io.usethesource.capsule.core.PersistentTrieSet.of<E>()
+    var set = io.usethesource.capsule.core.PersistentTrieSet.of<IntWrapper>()
     for (index in 0 until immutableSize) {
         set = set.__insert(elements[index])
     }

benchmarks-libraries/src/jmh/java/benchmarks/immutableSet/clojure/builder/utils.kt

@@ -19,9 +19,10 @@
 package benchmarks.immutableSet.clojure.builder
 
 import benchmarks.immutableSize
+import benchmarks.IntWrapper
 
-fun <E> persistentSetBuilderAdd(
-        elements: List<E>,
+fun persistentSetBuilderAdd(
+        elements: List<IntWrapper>,
         immutablePercentage: Double
 ): clojure.lang.ATransientSet {
     val immutableSize = immutableSize(elements.size, immutablePercentage)

benchmarks-libraries/src/jmh/java/benchmarks/immutableSet/kotlin/builder/utils.kt

@@ -19,14 +19,15 @@
 package benchmarks.immutableSet.kotlin.builder
 
 import benchmarks.immutableSize
+import benchmarks.IntWrapper
 
-fun <E> persistentSetBuilderAdd(
-        elements: List<E>,
+fun persistentSetBuilderAdd(
+        elements: List<IntWrapper>,
         immutablePercentage: Double
-): kotlinx.collections.immutable.PersistentSet.Builder<E> {
+): kotlinx.collections.immutable.PersistentSet.Builder<IntWrapper> {
     val immutableSize = immutableSize(elements.size, immutablePercentage)
 
-    var set = kotlinx.collections.immutable.persistentHashSetOf<E>()
+    var set = kotlinx.collections.immutable.persistentHashSetOf<IntWrapper>()
     for (index in 0 until immutableSize) {
         set = set.add(elements[index])
     }

benchmarks-libraries/src/jmh/java/benchmarks/immutableSet/kotlinOrdered/builder/utils.kt

@@ -19,14 +19,15 @@
 package benchmarks.immutableSet.kotlinOrdered.builder
 
 import benchmarks.immutableSize
+import benchmarks.IntWrapper
 
-fun <E> persistentSetBuilderAdd(
-        elements: List<E>,
+fun persistentSetBuilderAdd(
+        elements: List<IntWrapper>,
         immutablePercentage: Double
-): kotlinx.collections.immutable.PersistentSet.Builder<E> {
+): kotlinx.collections.immutable.PersistentSet.Builder<IntWrapper> {
     val immutableSize = immutableSize(elements.size, immutablePercentage)
 
-    var set = kotlinx.collections.immutable.persistentSetOf<E>()
+    var set = kotlinx.collections.immutable.persistentSetOf<IntWrapper>()
     for (index in 0 until immutableSize) {
         set = set.add(elements[index])
     }

benchmarks-libraries/src/main/kotlin/generators/benchmarkSourceGenerator.kt

@@ -156,10 +156,10 @@ private val setImpls = listOf(
         VavrOrderedSetImplementation
 )
 private val setBuilderImpls = listOf(
-        SetBuilderKotlinBenchmark(),
-        SetBuilderKotlinOrderedBenchmark(),
-        SetBuilderCapsuleBenchmark(),
-        SetBuilderClojureBenchmark()
+        KotlinSetBuilderImplementation,
+        KotlinOrderedSetBuilderImplementation,
+        CapsuleSetBuilderImplementation,
+        ClojureSetBuilderImplementation
 )
 
 fun generateBenchmarks() {
@@ -236,12 +236,20 @@ fun generateBenchmarks() {
             setImpls.filterIsInstance<SetIterateBenchmark>().map { SetIterateBenchmarkGenerator(it) },
             setImpls.filterIsInstance<SetRemoveBenchmark>().map { SetRemoveBenchmarkGenerator(it) }
     )*/
-    val setBuilderBenchmarks = listOf(
+    val setBuilderBenchmarks = setBuilderImpls.map { listOf(
+            SetBuilderAddBenchmarkGenerator(it),
+            SetBuilderContainsBenchmarkGenerator(it),
+            SetBuilderRemoveBenchmarkGenerator(it),
+            SetBuilderUtilsGenerator(it)
+    ) } + setBuilderImpls.filter(SetBuilderImplementation::isIterable).map { listOf(
+            SetBuilderIterateBenchmarkGenerator(it)
+    ) }
+    /*listOf(
             setBuilderImpls.filterIsInstance<SetBuilderAddBenchmark>().map { SetBuilderAddBenchmarkGenerator(it) },
             setBuilderImpls.filterIsInstance<SetBuilderContainsBenchmark>().map { SetBuilderContainsBenchmarkGenerator(it) },
             setBuilderImpls.filterIsInstance<SetBuilderIterateBenchmark>().map { SetBuilderIterateBenchmarkGenerator(it) },
             setBuilderImpls.filterIsInstance<SetBuilderRemoveBenchmark>().map { SetBuilderRemoveBenchmarkGenerator(it) }
-    )
+    )*/
 
 
     val allBenchmarks = (listBenchmarks + listBuilderBenchmarks + mapBenchmarks + mapBuilderBenchmarks + setBenchmarks + setBuilderBenchmarks).flatten()
@@ -257,16 +265,12 @@ fun generateBenchmarks() {
 }
 
 fun generateUtils() {
-    val setBuilderUtils = setBuilderImpls.filterIsInstance<SetBuilderBenchmarkUtils>().map { SetBuilderUtilsGenerator(it) }
-
     val commonUtils = listOf(
             IntWrapperGenerator(),
             CommonUtilsGenerator()
     )
 
-    val utils = setBuilderUtils + commonUtils
-
-    utils.forEach { util ->
+    commonUtils.forEach { util ->
         val path = util.getPackage().replace('.', '/') + "/" + util.outputFileName + ".kt"
         val file = File(BENCHMARKS_ROOT + path)
         file.parentFile?.mkdirs()

benchmarks-libraries/src/main/kotlin/generators/immutableSetBuilder/addGenerator.kt

@@ -19,12 +19,7 @@ package generators.immutableSetBuilder
 import generators.BenchmarkSourceGenerator
 import java.io.PrintWriter
 
-interface SetBuilderAddBenchmark {
-    val packageName: String
-    fun setBuilderType(E: String): String
-}
-
-class SetBuilderAddBenchmarkGenerator(private val impl: SetBuilderAddBenchmark) : BenchmarkSourceGenerator() {
+class SetBuilderAddBenchmarkGenerator(private val impl: SetBuilderImplementation) : BenchmarkSourceGenerator() {
     override val outputFileName: String = "Add"
 
     override fun getPackage(): String {
@@ -45,15 +40,15 @@ open class Add {
     @Param("0.0", "50.0")
     var immutablePercentage: Double = 0.0
 
-    private var elements = listOf<IntWrapper>()
+    private var elements = listOf<$setBuilderElementType>()
 
     @Setup(Level.Trial)
     fun prepare() {
         elements = generateElements(hashCodeType, size)
     }
 
     @Benchmark
-    fun add(): ${impl.setBuilderType("IntWrapper")} {
+    fun add(): ${impl.type()} {
         return persistentSetBuilderAdd(elements, immutablePercentage)
     }
 
@@ -66,7 +61,7 @@ open class Add {
     }
         """.trimIndent()
         )
-        if (impl is SetBuilderIterateBenchmark) {
+        if (impl.isIterable) {
             out.println("""
     @Benchmark
     fun addAndIterate(bh: Blackhole) {

benchmarks-libraries/src/main/kotlin/generators/immutableSetBuilder/containsGenerator.kt

@@ -19,12 +19,7 @@ package generators.immutableSetBuilder
 import generators.BenchmarkSourceGenerator
 import java.io.PrintWriter
 
-interface SetBuilderContainsBenchmark {
-    val packageName: String
-    fun emptyOf(E: String): String
-}
-
-class SetBuilderContainsBenchmarkGenerator(private val impl: SetBuilderContainsBenchmark) : BenchmarkSourceGenerator() {
+class SetBuilderContainsBenchmarkGenerator(private val impl: SetBuilderImplementation) : BenchmarkSourceGenerator() {
     override val outputFileName: String = "Contains"
 
     override fun getPackage(): String {
@@ -45,8 +40,8 @@ open class Contains {
     @Param("0.0", "50.0")
     var immutablePercentage: Double = 0.0
 
-    private var elements = listOf<IntWrapper>()
-    private var builder = ${impl.emptyOf("IntWrapper")}
+    private var elements = listOf<$setBuilderElementType>()
+    private var builder = ${impl.empty()}
 
     @Setup(Level.Trial)
     fun prepare() {

benchmarks-libraries/src/main/kotlin/generators/immutableSetBuilder/impl/capsule.kt

@@ -18,24 +18,28 @@ package generators.immutableSetBuilder.impl
 
 import generators.immutableSetBuilder.*
 
-class SetBuilderCapsuleBenchmark:
-        SetBuilderContainsBenchmark,
-        SetBuilderIterateBenchmark,
-        SetBuilderAddBenchmark,
-        SetBuilderRemoveBenchmark,
-        SetBuilderBenchmarkUtils
-{
-    override val packageName: String = "capsule.builder"
-
-    override fun setBuilderType(E: String): String = "io.usethesource.capsule.Set.Transient<$E>"
-
-    override fun emptyOf(E: String): String = "io.usethesource.capsule.core.PersistentTrieSet.of<$E>().asTransient()"
-    override fun immutableOf(E: String): String = "io.usethesource.capsule.core.PersistentTrieSet.of<$E>()"
-
-    override val addOperation: String = "add"
-    override fun immutableAddOperation(set: String, element: String): String = "$set.__insert($element)"
-
-    override val removeOperation: String = "__remove"
-
-    override fun builderOperation(set: String): String = "$set.asTransient()"
+object CapsuleSetBuilderImplementation: SetBuilderImplementation{
+    override val packageName: String
+            = "capsule.builder"
+
+    override fun type(): String
+            = "io.usethesource.capsule.Set.Transient<$setBuilderElementType>"
+    override fun empty(): String
+            = "io.usethesource.capsule.core.PersistentTrieSet.of<$setBuilderElementType>().asTransient()"
+
+    override fun addOperation(builder: String, element: String): String
+            = "$builder.add($element)"
+    override fun removeOperation(builder: String, element: String): String
+            = "$builder.__remove($element)"
+
+    override val isIterable: Boolean
+            = true
+
+    override fun builderOperation(immutable: String): String
+            = "$immutable.asTransient()"
+
+    override fun immutableEmpty(): String
+            = "io.usethesource.capsule.core.PersistentTrieSet.of<$setBuilderElementType>()"
+    override fun immutableAddOperation(immutable: String, element: String): String
+            = "$immutable.__insert($element)"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableSetBuilder/impl/clojure.kt

@@ -18,23 +18,28 @@ package generators.immutableSetBuilder.impl
 
 import generators.immutableSetBuilder.*
 
-class SetBuilderClojureBenchmark:
-        SetBuilderContainsBenchmark,
-        SetBuilderAddBenchmark,
-        SetBuilderRemoveBenchmark,
-        SetBuilderBenchmarkUtils
-{
-    override val packageName: String = "clojure.builder"
-
-    override fun setBuilderType(E: String): String = "clojure.lang.ATransientSet"
-
-    override fun emptyOf(E: String): String = "clojure.lang.PersistentHashSet.EMPTY.asTransient() as clojure.lang.ATransientSet"
-    override fun immutableOf(E: String): String = "clojure.lang.PersistentHashSet.EMPTY"
-
-    override val addOperation: String = "conj"
-    override fun immutableAddOperation(set: String, element: String): String = "$set.cons($element) as clojure.lang.PersistentHashSet"
-
-    override val removeOperation: String = "disjoin"
-
-    override fun builderOperation(set: String): String = "$set.asTransient() as clojure.lang.ATransientSet"
+object ClojureSetBuilderImplementation: SetBuilderImplementation{
+    override val packageName: String
+            = "clojure.builder"
+
+    override fun type(): String
+            = "clojure.lang.ATransientSet"
+    override fun empty(): String
+            = "clojure.lang.PersistentHashSet.EMPTY.asTransient() as clojure.lang.ATransientSet"
+
+    override fun addOperation(builder: String, element: String): String
+            = "$builder.conj($element)"
+    override fun removeOperation(builder: String, element: String): String
+            = "$builder.disjoin($element)"
+
+    override val isIterable: Boolean
+            = false
+
+    override fun builderOperation(immutable: String): String
+            = "$immutable.asTransient() as clojure.lang.ATransientSet"
+
+    override fun immutableEmpty(): String
+            = "clojure.lang.PersistentHashSet.EMPTY"
+    override fun immutableAddOperation(immutable: String, element: String): String
+            = "$immutable.cons($element) as clojure.lang.PersistentHashSet"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableSetBuilder/impl/kotlin.kt

@@ -18,24 +18,28 @@ package generators.immutableSetBuilder.impl
 
 import generators.immutableSetBuilder.*
 
-class SetBuilderKotlinBenchmark:
-        SetBuilderContainsBenchmark,
-        SetBuilderIterateBenchmark,
-        SetBuilderAddBenchmark,
-        SetBuilderRemoveBenchmark,
-        SetBuilderBenchmarkUtils
-{
-    override val packageName: String = "kotlin.builder"
-
-    override fun setBuilderType(E: String): String = "kotlinx.collections.immutable.PersistentSet.Builder<$E>"
-
-    override fun emptyOf(E: String): String = "kotlinx.collections.immutable.persistentHashSetOf<$E>().builder()"
-    override fun immutableOf(E: String): String = "kotlinx.collections.immutable.persistentHashSetOf<$E>()"
-
-    override val addOperation: String = "add"
-    override fun immutableAddOperation(set: String, element: String): String = "$set.add($element)"
-
-    override val removeOperation: String = "remove"
-
-    override fun builderOperation(set: String): String = "$set.builder()"
+object KotlinSetBuilderImplementation: SetBuilderImplementation{
+    override val packageName: String
+            = "kotlin.builder"
+
+    override fun type(): String
+            = "kotlinx.collections.immutable.PersistentSet.Builder<$setBuilderElementType>"
+    override fun empty(): String
+            = "kotlinx.collections.immutable.persistentHashSetOf<$setBuilderElementType>().builder()"
+
+    override fun addOperation(builder: String, element: String): String
+            = "$builder.add($element)"
+    override fun removeOperation(builder: String, element: String): String
+            = "$builder.remove($element)"
+
+    override val isIterable: Boolean
+            = true
+
+    override fun builderOperation(immutable: String): String
+            = "$immutable.builder()"
+
+    override fun immutableEmpty(): String
+            = "kotlinx.collections.immutable.persistentHashSetOf<$setBuilderElementType>()"
+    override fun immutableAddOperation(immutable: String, element: String): String
+            = "$immutable.add($element)"
 }