Automated commit of generated code

Author: github-actions[bot]

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/constructors.kt

@@ -393,7 +393,7 @@ public class DataFrameBuilder(private val header: List<String>) {
 
     @JvmName("invoke1")
     internal fun withValues(values: Iterable<Any?>): DataFrame<*> =
-        withValuesImpl(header, values.asList()).map { (name, values) ->
+        (header to values.asList()).withValuesImpl().map { (name, values) ->
             DataColumn.createByInference(name, values)
         }.toDataFrame()
 

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/gather.kt

@@ -4,6 +4,8 @@ import org.jetbrains.kotlinx.dataframe.ColumnsSelector
 import org.jetbrains.kotlinx.dataframe.DataFrame
 import org.jetbrains.kotlinx.dataframe.RowValueFilter
 import org.jetbrains.kotlinx.dataframe.annotations.AccessApiOverload
+import org.jetbrains.kotlinx.dataframe.annotations.Interpretable
+import org.jetbrains.kotlinx.dataframe.annotations.Refine
 import org.jetbrains.kotlinx.dataframe.columns.ColumnAccessor
 import org.jetbrains.kotlinx.dataframe.columns.ColumnReference
 import org.jetbrains.kotlinx.dataframe.columns.toColumnSet
@@ -15,6 +17,7 @@ import kotlin.reflect.typeOf
 
 // region gather
 
+@Interpretable("Gather0")
 public fun <T, C> DataFrame<T>.gather(selector: ColumnsSelector<T, C>): Gather<T, C, String, C> =
     Gather(
         df = this,
@@ -44,30 +47,76 @@ public fun <T, C> DataFrame<T>.gather(vararg columns: KProperty<C>): Gather<T, C
 
 // endregion
 
+@Interpretable("GatherWhere")
 public fun <T, C, K, R> Gather<T, C, K, R>.where(filter: RowValueFilter<T, C>): Gather<T, C, K, R> =
-    copy(filter = this.filter and filter)
+    Gather(
+        df = df,
+        columns = columns,
+        filter = this.filter and filter,
+        keyType = keyType,
+        keyTransform = keyTransform,
+        valueTransform = valueTransform,
+        explode = explode,
+    )
 
+@Interpretable("GatherChangeType")
 public fun <T, C, K, R> Gather<T, C?, K, R>.notNull(): Gather<T, C, K, R> = where { it != null } as Gather<T, C, K, R>
 
-public fun <T, C, K, R> Gather<T, C, K, R>.explodeLists(): Gather<T, C, K, R> = copy(explode = true)
+@Interpretable("GatherExplodeLists")
+public fun <T, C, K, R> Gather<T, C, K, R>.explodeLists(): Gather<T, C, K, R> =
+    Gather(
+        df = df,
+        columns = columns,
+        filter = filter,
+        keyType = keyType,
+        keyTransform = keyTransform,
+        valueTransform = valueTransform,
+        explode = true,
+    )
 
+@Interpretable("GatherMap")
 public inline fun <T, C, reified K, R> Gather<T, C, *, R>.mapKeys(
     noinline transform: (String) -> K,
 ): Gather<T, C, K, R> =
-    copy(keyTransform = transform as ((String) -> Nothing), keyType = typeOf<K>()) as Gather<T, C, K, R>
+    Gather(
+        df = df,
+        columns = columns,
+        filter = filter,
+        keyType = typeOf<K>(),
+        keyTransform = transform,
+        valueTransform = valueTransform,
+        explode = explode,
+    )
 
+@Interpretable("GatherMap")
 public fun <T, C, K, R> Gather<T, C, K, *>.mapValues(transform: (C) -> R): Gather<T, C, K, R> =
-    copy(valueTransform = transform as ((C) -> Nothing)) as Gather<T, C, K, R>
+    Gather(
+        df = df,
+        columns = columns,
+        filter = filter,
+        keyType = keyType,
+        keyTransform = keyTransform,
+        valueTransform = transform,
+        explode = explode,
+    )
 
-public data class Gather<T, C, K, R>(
+public class Gather<T, C, K, R>(
+    @PublishedApi
     internal val df: DataFrame<T>,
+    @PublishedApi
     internal val columns: ColumnsSelector<T, C>,
+    @PublishedApi
     internal val filter: RowValueFilter<T, C>? = null,
+    @PublishedApi
     internal val keyType: KType? = null,
+    @PublishedApi
     internal val keyTransform: ((String) -> K),
+    @PublishedApi
     internal val valueTransform: ((C) -> R)? = null,
+    @PublishedApi
     internal val explode: Boolean = false,
 ) {
+    @Interpretable("GatherChangeType")
     public fun <P> cast(): Gather<T, P, K, P> {
         // TODO: introduce GatherWithTransform to avoid this error
         require(valueTransform == null) { "Cast is not allowed to be called after `mapValues`" }
@@ -77,6 +126,8 @@ public data class Gather<T, C, K, R>(
 
 // region into
 
+@Refine
+@Interpretable("GatherInto")
 public fun <T, C, K, R> Gather<T, C, K, R>.into(keyColumn: String, valueColumn: String): DataFrame<T> =
     gatherImpl(keyColumn, valueColumn)
 
@@ -100,6 +151,8 @@ public fun <T, C, K, R> Gather<T, C, K, R>.into(keyColumn: KProperty<K>, valueCo
 
 // region keysInto
 
+@Refine
+@Interpretable("GatherKeysInto")
 public fun <T, C, K, R> Gather<T, C, K, R>.keysInto(keyColumn: String): DataFrame<T> = gatherImpl(keyColumn, null)
 
 @Deprecated(
@@ -120,6 +173,8 @@ public fun <T, C, K, R> Gather<T, C, K, R>.keysInto(keyColumn: KProperty<K>): Da
 
 // region valuesInto
 
+@Refine
+@Interpretable("GatherValuesInto")
 public fun <T, C, K, R> Gather<T, C, K, R>.valuesInto(valueColumn: String): DataFrame<T> = gatherImpl(null, valueColumn)
 
 @Deprecated(

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/exceptions/ColumnNotFoundException.kt

@@ -2,4 +2,4 @@ package org.jetbrains.kotlinx.dataframe.exceptions
 
 public class ColumnNotFoundException(public val columnName: String, public override val message: String) :
     RuntimeException(),
-    DataFrameException
+    DataFrameError

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/exceptions/DataFrameException.kt renamed to core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/exceptions/DataFrameError.kt

@@ -1,8 +1,9 @@
 package org.jetbrains.kotlinx.dataframe.exceptions
 
 /**
- * If DataFrame function used by compiler plugin as implementation detail throws this exception, [message] will be reported as warning
+ * If DataFrame function used by compiler plugin as implementation detail throws exception
+ * that implements this interface, [message] will be reported as warning
  */
-public interface DataFrameException {
+public interface DataFrameError {
     public val message: String
 }

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/exceptions/DuplicateColumnNamesException.kt

@@ -2,7 +2,7 @@ package org.jetbrains.kotlinx.dataframe.exceptions
 
 public class DuplicateColumnNamesException(public val allColumnNames: List<String>) :
     IllegalArgumentException(),
-    DataFrameException {
+    DataFrameError {
 
     public val duplicatedNames: List<String> = allColumnNames
         .groupBy { it }

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/api/constructors.kt

@@ -1,10 +1,16 @@
 package org.jetbrains.kotlinx.dataframe.impl.api
 
-internal fun <T> withValuesImpl(header: List<String>, values: List<T>): List<Pair<String, List<T>>> {
+import org.jetbrains.kotlinx.dataframe.exceptions.DataFrameError
+
+/**
+ * Public API to be re-used in compiler plugin implementation
+ */
+public fun <T> Pair<List<String>, List<T>>.withValuesImpl(): List<Pair<String, List<T>>> {
+    val (header, values) = this
     val ncol = header.size
 
-    require(header.isNotEmpty() && values.size.rem(ncol) == 0) {
-        "Number of values ${values.size} is not divisible by number of columns $ncol"
+    if (!(header.isNotEmpty() && values.size.rem(ncol) == 0)) {
+        throw WrongNumberOfValuesException(values.size, ncol)
     }
 
     val nrow = values.size / ncol
@@ -16,3 +22,9 @@ internal fun <T> withValuesImpl(header: List<String>, values: List<T>): List<Pai
         header[col] to colValues
     }
 }
+
+internal class WrongNumberOfValuesException(size: Int, ncol: Int) :
+    IllegalArgumentException(),
+    DataFrameError {
+    override val message = "Number of values $size is not divisible by number of columns $ncol"
+}