moved to chunkImpl

Author: Jolanrensen

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/chunked.kt

@@ -6,29 +6,18 @@ import org.jetbrains.kotlinx.dataframe.DataRow
 import org.jetbrains.kotlinx.dataframe.columns.ColumnGroup
 import org.jetbrains.kotlinx.dataframe.columns.FrameColumn
 import org.jetbrains.kotlinx.dataframe.columns.ValueColumn
+import org.jetbrains.kotlinx.dataframe.impl.api.chunkedImpl
 import org.jetbrains.kotlinx.dataframe.impl.getListType
-import org.jetbrains.kotlinx.dataframe.impl.splitByIndices
 import org.jetbrains.kotlinx.dataframe.nrow
 import org.jetbrains.kotlinx.dataframe.type
 
-/**
- * Creates a [FrameColumn] from [this] by splitting the dataframe into
- * smaller ones, based on the given [startIndices].
- */
-public fun <T> DataFrame<T>.chunked(startIndices: Iterable<Int>, name: String = "groups"): FrameColumn<T> =
-    DataColumn.createFrameColumn(
-        name = name,
-        groups = this.splitByIndices(startIndices.asSequence()).toList(),
-        schema = lazy { this.schema() },
-    )
-
 /**
  * Creates a [FrameColumn] from [this] by splitting the dataframe into
  * smaller ones, with their number of rows at most [size].
  */
 public fun <T> DataFrame<T>.chunked(size: Int, name: String = "groups"): FrameColumn<T> {
     val startIndices = (0 until nrow step size)
-    return this.chunked(startIndices, name)
+    return this.chunkedImpl(startIndices, name)
 }
 
 public fun <T> DataColumn<T>.chunked(size: Int): ValueColumn<List<T>> {

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/api/chunked.kt

@@ -0,0 +1,18 @@
+package org.jetbrains.kotlinx.dataframe.impl.api
+
+import org.jetbrains.kotlinx.dataframe.DataColumn
+import org.jetbrains.kotlinx.dataframe.DataFrame
+import org.jetbrains.kotlinx.dataframe.api.schema
+import org.jetbrains.kotlinx.dataframe.columns.FrameColumn
+import org.jetbrains.kotlinx.dataframe.impl.splitByIndices
+
+/**
+ * Creates a [FrameColumn] from [this] by splitting the dataframe into
+ * smaller ones, based on the given [startIndices].
+ */
+internal fun <T> DataFrame<T>.chunkedImpl(startIndices: Iterable<Int>, name: String = "groups"): FrameColumn<T> =
+    DataColumn.createFrameColumn(
+        name = name,
+        groups = this.splitByIndices(startIndices.asSequence()).toList(),
+        schema = lazy { this.schema() },
+    )

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/api/groupBy.kt

@@ -6,7 +6,6 @@ import org.jetbrains.kotlinx.dataframe.DataRow
 import org.jetbrains.kotlinx.dataframe.api.GroupBy
 import org.jetbrains.kotlinx.dataframe.api.GroupedDataRow
 import org.jetbrains.kotlinx.dataframe.api.cast
-import org.jetbrains.kotlinx.dataframe.api.chunked
 import org.jetbrains.kotlinx.dataframe.api.getColumnsWithPaths
 import org.jetbrains.kotlinx.dataframe.api.getRows
 import org.jetbrains.kotlinx.dataframe.api.indices
@@ -62,7 +61,7 @@ internal fun <T> DataFrame<T>.groupByImpl(moveToTop: Boolean, columns: ColumnsSe
     }
 
     val groupedColumnName = keyColumnsDf.nameGenerator().addUnique(GroupBy.groupedColumnAccessor.name())
-    val groupedColumn = sorted.chunked(startIndices.asIterable(), groupedColumnName)
+    val groupedColumn = sorted.chunkedImpl(startIndices.asIterable(), groupedColumnName)
 
     val df = keyColumnsDf + groupedColumn
     return GroupByImpl(df, groupedColumn, columns)

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/readJson.kt

@@ -23,7 +23,6 @@ import org.jetbrains.kotlinx.dataframe.DataRow
 import org.jetbrains.kotlinx.dataframe.api.JsonPath
 import org.jetbrains.kotlinx.dataframe.api.KeyValueProperty
 import org.jetbrains.kotlinx.dataframe.api.cast
-import org.jetbrains.kotlinx.dataframe.api.chunked
 import org.jetbrains.kotlinx.dataframe.api.columnOf
 import org.jetbrains.kotlinx.dataframe.api.concat
 import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
@@ -38,6 +37,7 @@ import org.jetbrains.kotlinx.dataframe.columns.ColumnGroup
 import org.jetbrains.kotlinx.dataframe.columns.FrameColumn
 import org.jetbrains.kotlinx.dataframe.impl.ColumnNameGenerator
 import org.jetbrains.kotlinx.dataframe.impl.DataCollectorBase
+import org.jetbrains.kotlinx.dataframe.impl.api.chunkedImpl
 import org.jetbrains.kotlinx.dataframe.impl.asList
 import org.jetbrains.kotlinx.dataframe.impl.columns.createColumnGuessingType
 import org.jetbrains.kotlinx.dataframe.impl.commonType
@@ -301,7 +301,7 @@ internal fun fromJsonListAnyColumns(
 
                 else ->
                     parsed.unwrapUnnamedColumns()
-                        .chunked(
+                        .chunkedImpl(
                             startIndices = startIndices,
                             name = ARRAY_COLUMN_NAME, // will be erased
                         )
@@ -645,7 +645,7 @@ internal fun fromJsonListArrayAndValueColumns(
                                 )
                             }
 
-                            else -> parsed.unwrapUnnamedColumns().chunked(startIndices, colName)
+                            else -> parsed.unwrapUnnamedColumns().chunkedImpl(startIndices, colName)
                         }
                         UnnamedColumn(res)
                     }