ajet/df-research
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init research

Browse Source
This commit is contained in:
Adam Jeniski
2026-02-08 11:20:43 -10:00
commit bdf064f54d
3041 changed files with 1592200 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dataframe/plugins/expressions-converter/testData/box/any.kt
Vendored
+16
View File
@@ -0,0 +1,16 @@
package org.jetbrains.kotlinx.dataframe.explainer
import org.jetbrains.kotlinx.dataframe.*
import org.jetbrains.kotlinx.dataframe.api.*
object PluginCallback {
fun doAction(any: Any) {}
}
fun box(): String {
val a: GroupBy<*, *> = dataFrameOf("a")(1).groupBy("a")
a.also {
PluginCallback.doAction(it)
}
return "OK"
}
dataframe/plugins/expressions-converter/testData/box/any.txt
Vendored
+25
View File
@@ -0,0 +1,25 @@
package
package org {
package org.jetbrains {
package org.jetbrains.kotlinx {
package org.jetbrains.kotlinx.dataframe {
package org.jetbrains.kotlinx.dataframe.explainer {
public fun box(): kotlin.String
public object PluginCallback {
private constructor PluginCallback()
public final fun doAction(/*0*/ any: kotlin.Any): kotlin.Unit
public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
}
}
}
}
}
}
dataframe/plugins/expressions-converter/testData/box/df.kt
Vendored
+156
View File
@@ -0,0 +1,156 @@
package org.jetbrains.kotlinx.dataframe.explainer
import org.jetbrains.kotlinx.dataframe.*
import org.jetbrains.kotlinx.dataframe.api.*
@TransformDataFrameExpressions
fun callChain(df: DataFrame<*>) {
val df1 = df
.filter { "age"<Int>() > 20 }
.groupBy("value")
.sum()
}
interface Person
class Wrapper {
val df = dataFrameOf("name", "age", "city", "weight")(
"Alice", 15, "London", 54,
"Bob", 45, "Dubai", 87,
"Charlie", 20, "Moscow", null,
"Charlie", 40, "Milan", null,
"Bob", 30, "Tokyo", 68,
"Alice", 20, null, 55,
"Charlie", 30, "Moscow", 90
)
val typed: DataFrame<Person> = df.cast()
@TransformDataFrameExpressions
fun ff() {
val name by column<String>()
val aggregated = typed.groupBy { name() }.aggregate {
(if (name().first().startsWith("A")) first() else null) into "agg"
}["agg"]
}
}
@TransformDataFrameExpressions
fun aggregateDf() {
val df = dataFrameOf("firstName", "lastName", "age", "city", "weight", "isHappy")(
"Alice", "Cooper", 15, "London", 54, true,
"Bob", "Dylan", 45, "Dubai", 87, true,
"Charlie", "Daniels", 20, "Moscow", null, false,
"Charlie", "Chaplin", 40, "Milan", null, true,
"Bob", "Marley", 30, "Tokyo", 68, true,
"Alice", "Wolf", 20, null, 55, false,
"Charlie", "Byrd", 30, "Moscow", 90, true
).group("firstName", "lastName").into("name")
df.groupBy("city").aggregate {
count() into "total"
count { "age"<Int>() > 18 } into "adults"
median("age") into "median age"
min("age") into "min age"
maxBy("age")["name"] into "oldest"
}
}
@TransformDataFrameExpressions
fun move() {
val df = dataFrameOf("firstName", "lastName", "age", "city", "weight", "isHappy")(
"Alice", "Cooper", 15, "London", 54, true,
"Bob", "Dylan", 45, "Dubai", 87, true,
"Charlie", "Daniels", 20, "Moscow", null, false,
"Charlie", "Chaplin", 40, "Milan", null, true,
"Bob", "Marley", 30, "Tokyo", 68, true,
"Alice", "Wolf", 20, null, 55, false,
"Charlie", "Byrd", 30, "Moscow", 90, true
).group("firstName", "lastName").into("name")
df.move("age", "weight").into { pathOf("info", it.name()) }
}
fun interface PluginCallback {
fun doAction(
source: String,
name: String,
df: Any,
id: String,
receiverId: String?,
containingClassFqName: String?,
containingFunName: String?,
statementIndex: Int
)
}
object PluginCallbackProxy : PluginCallback {
var action: PluginCallback = PluginCallback { _, _, _, _, _, _, _, _ -> Unit }
override fun doAction(source: String, name: String, df: Any, id: String, receiverId: String?, containingClassFqName: String?, containingFunName: String?, statemenIndex: Int) {
action.doAction(source, name, df, id, receiverId, containingClassFqName, containingFunName, statemenIndex)
}
}
//fun callChainTransformed(df: DataFrame<*>) {
// val df1 = df
// .filter { "age"<Int>() > 20 }
// .also { PluginCallbackProxy.doAction(""".filter { "age"<Int>() > 20 }""", "filter", it) }
//}
//fun callChainTransformed(df: DataFrame<*>) {
// val df1 = df
// .filter { "age"<Int>() > 20 }
// .also { PluginCallbackProxy.action(""".filter { "age"<Int>() > 20 }""", it) }
// .groupBy("something")
// .sum()
// .also { PluginCallbackProxy.action(""".groupBy("something").sum()""", it) }
//}
annotation class TransformDataFrameExpressions
fun box(): String {
val age by columnOf(10, 21, 30, 1)
val value by columnOf("a", "b", "c", "c")
val df = dataFrameOf(age, value)
val expressions = mutableListOf<String>()
PluginCallbackProxy.action = PluginCallback { source, _, df, id, receiverId, containingClassFqName, containingFunName, statementIndex ->
println("== Call ==")
expressions += source
if (df is AnyFrame) {
println(source)
df.print()
println(id)
println(receiverId)
println(containingClassFqName)
println(containingFunName)
println("statementIndex = ${statementIndex}")
} else {
println(df::class)
}
println("== End ==")
}
println("CallChain")
callChain(df)
println("expressions = ${expressions}")
expressions.clear()
println("ff")
Wrapper().ff()
println("expressions = ${expressions}")
expressions.clear()
// callChainTransformed(df)
println("aggregateDf")
aggregateDf()
println("expressions = ${expressions}")
expressions.clear()
println("move")
move()
println("expressions = ${expressions}")
expressions.clear()
return "OK"
}
dataframe/plugins/expressions-converter/testData/box/df.txt
Vendored
+59
View File
@@ -0,0 +1,59 @@
package
package org {
package org.jetbrains {
package org.jetbrains.kotlinx {
package org.jetbrains.kotlinx.dataframe {
package org.jetbrains.kotlinx.dataframe.explainer {
@org.jetbrains.kotlinx.dataframe.explainer.TransformDataFrameExpressions public fun aggregateDf(): kotlin.Unit
public fun box(): kotlin.String
@org.jetbrains.kotlinx.dataframe.explainer.TransformDataFrameExpressions public fun callChain(/*0*/ df: org.jetbrains.kotlinx.dataframe.DataFrame<*>): kotlin.Unit
@org.jetbrains.kotlinx.dataframe.explainer.TransformDataFrameExpressions public fun move(): kotlin.Unit
public interface Person {
public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
}
public fun interface PluginCallback {
public abstract fun doAction(/*0*/ source: kotlin.String, /*1*/ name: kotlin.String, /*2*/ df: kotlin.Any, /*3*/ id: kotlin.String, /*4*/ receiverId: kotlin.String?, /*5*/ containingClassFqName: kotlin.String?, /*6*/ containingFunName: kotlin.String?, /*7*/ statementIndex: kotlin.Int): kotlin.Unit
public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
}
public object PluginCallbackProxy : org.jetbrains.kotlinx.dataframe.explainer.PluginCallback {
private constructor PluginCallbackProxy()
public final var action: org.jetbrains.kotlinx.dataframe.explainer.PluginCallback
public open override /*1*/ fun doAction(/*0*/ source: kotlin.String, /*1*/ name: kotlin.String, /*2*/ df: kotlin.Any, /*3*/ id: kotlin.String, /*4*/ receiverId: kotlin.String?, /*5*/ containingClassFqName: kotlin.String?, /*6*/ containingFunName: kotlin.String?, /*7*/ statemenIndex: kotlin.Int): kotlin.Unit
public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
}
public final annotation class TransformDataFrameExpressions : kotlin.Annotation {
public constructor TransformDataFrameExpressions()
public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
}
public final class Wrapper {
public constructor Wrapper()
public final val df: org.jetbrains.kotlinx.dataframe.DataFrame<*>
public final val typed: org.jetbrains.kotlinx.dataframe.DataFrame<org.jetbrains.kotlinx.dataframe.explainer.Person>
public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
@org.jetbrains.kotlinx.dataframe.explainer.TransformDataFrameExpressions public final fun ff(): kotlin.Unit
public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
}
}
}
}
}
}
dataframe/plugins/expressions-converter/testData/box/test.kt
Vendored
+21
View File
@@ -0,0 +1,21 @@
package org.jetbrains.kotlinx.dataframe.explainer
fun print(any: Any) {}
fun printM(pair: Pair<String, Any>) {
val (text, obj) = pair
println("# ${text}: $obj")
}
fun ir() {
val df = Any()
printM("Any()" to df)
}
fun box(): String {
val df = Any()
val df1 = 1 + 2
print(df)
print(df1)
return "OK"
}
dataframe/plugins/expressions-converter/testData/box/test.txt
Vendored
+20
View File
@@ -0,0 +1,20 @@
package
package org {
package org.jetbrains {
package org.jetbrains.kotlinx {
package org.jetbrains.kotlinx.dataframe {
package org.jetbrains.kotlinx.dataframe.explainer {
public fun box(): kotlin.String
public fun ir(): kotlin.Unit
public fun print(/*0*/ any: kotlin.Any): kotlin.Unit
public fun printM(/*0*/ pair: kotlin.Pair<kotlin.String, kotlin.Any>): kotlin.Unit
}
}
}
}
}