diff --git a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/add.kt b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/add.kt index 3ed50ad08..50cc00d03 100644 --- a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/add.kt +++ b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/add.kt @@ -207,6 +207,7 @@ public class AddDsl( public infix fun ColumnReference.into(column: KProperty): Boolean = into(column.name) + @Interpretable("AddDslStringInvoke") public operator fun String.invoke(body: AddDsl.() -> Unit): Unit = group(this, body) public infix fun AnyColumnGroupAccessor.from(body: AddDsl.() -> Unit): Unit = group(this, body) diff --git a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/addId.kt b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/addId.kt index 14a5e7840..d29f697ab 100644 --- a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/addId.kt +++ b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/addId.kt @@ -3,6 +3,8 @@ package org.jetbrains.kotlinx.dataframe.api import org.jetbrains.kotlinx.dataframe.AnyCol import org.jetbrains.kotlinx.dataframe.AnyFrame import org.jetbrains.kotlinx.dataframe.DataFrame +import org.jetbrains.kotlinx.dataframe.annotations.Interpretable +import org.jetbrains.kotlinx.dataframe.annotations.Refine import org.jetbrains.kotlinx.dataframe.columns.ColumnAccessor // region DataColumn @@ -15,6 +17,8 @@ public fun AnyCol.addId(columnName: String = "id"): AnyFrame = toDataFrame().add public fun DataFrame.addId(column: ColumnAccessor): DataFrame = insert(column) { index() }.at(0) +@Refine +@Interpretable("AddId") public fun DataFrame.addId(columnName: String = "id"): DataFrame = insert(columnName) { index() }.at(0) // endregion diff --git a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/groupBy.kt b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/groupBy.kt index cfad7fd1d..3f59ef8f2 100644 --- a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/groupBy.kt +++ b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/groupBy.kt @@ -29,6 +29,7 @@ import kotlin.reflect.KProperty * * `df.add("columnName") { "someColumn"() + 15 }.groupBy("columnName")` */ +@Refine @Interpretable("DataFrameGroupBy") public fun DataFrame.groupBy(moveToTop: Boolean = true, cols: ColumnsSelector): GroupBy = groupByImpl(moveToTop, cols) diff --git a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/map.kt b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/map.kt index c79717fd1..6907c9e0b 100644 --- a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/map.kt +++ b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/map.kt @@ -7,6 +7,8 @@ import org.jetbrains.kotlinx.dataframe.DataFrame import org.jetbrains.kotlinx.dataframe.DataRow import org.jetbrains.kotlinx.dataframe.RowExpression import org.jetbrains.kotlinx.dataframe.Selector +import org.jetbrains.kotlinx.dataframe.annotations.Interpretable +import org.jetbrains.kotlinx.dataframe.annotations.Refine import org.jetbrains.kotlinx.dataframe.columns.ColumnReference import org.jetbrains.kotlinx.dataframe.columns.FrameColumn import org.jetbrains.kotlinx.dataframe.impl.columnName @@ -102,6 +104,8 @@ public fun ColumnsContainer.mapToColumn( body: AddExpression, ): DataColumn = mapToColumn(column.columnName, type, infer, body) +@Refine +@Interpretable("MapToFrame") public fun DataFrame.mapToFrame(body: AddDsl.() -> Unit): AnyFrame { val dsl = AddDsl(this) body(dsl) diff --git a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/move.kt b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/move.kt index 2f6efc217..6fd3f3b59 100644 --- a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/move.kt +++ b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/move.kt @@ -5,6 +5,8 @@ import org.jetbrains.kotlinx.dataframe.AnyColumnReference import org.jetbrains.kotlinx.dataframe.ColumnSelector import org.jetbrains.kotlinx.dataframe.ColumnsSelector import org.jetbrains.kotlinx.dataframe.DataFrame +import org.jetbrains.kotlinx.dataframe.annotations.Interpretable +import org.jetbrains.kotlinx.dataframe.annotations.Refine import org.jetbrains.kotlinx.dataframe.columns.ColumnReference import org.jetbrains.kotlinx.dataframe.columns.ColumnWithPath import org.jetbrains.kotlinx.dataframe.columns.toColumnSet @@ -18,6 +20,7 @@ import kotlin.reflect.KProperty // region move +@Interpretable("Move0") public fun DataFrame.move(columns: ColumnsSelector): MoveClause = MoveClause(this, columns) public fun DataFrame.move(vararg cols: String): MoveClause = move { cols.toColumnSet() } @@ -120,6 +123,8 @@ public fun MoveClause.under( public fun MoveClause.to(columnIndex: Int): DataFrame = moveTo(columnIndex) +@Refine +@Interpretable("ToTop") public fun MoveClause.toTop( newColumnName: ColumnsSelectionDsl.(ColumnWithPath) -> String = { it.name() }, ): DataFrame = into { newColumnName(it).toColumnAccessor() } diff --git a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/update.kt b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/update.kt index e5c6ba12a..7c483713b 100644 --- a/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/update.kt +++ b/core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/update.kt @@ -233,6 +233,7 @@ public class Update( * * @param [columns] The [Columns Selector][org.jetbrains.kotlinx.dataframe.ColumnsSelector] used to select the columns of this [DataFrame][org.jetbrains.kotlinx.dataframe.DataFrame] to update. */ +@Interpretable("Update0") public fun DataFrame.update(columns: ColumnsSelector): Update = Update(this, null, columns) /**