[Compiler plugin] df.convert { }.to<Type>()

Author: koperagen

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

@@ -94,6 +94,7 @@ public inline fun <T, C, reified R> Convert<T, C?>.notNull(
 public class Convert<T, out C>(internal val df: DataFrame<T>, internal val columns: ColumnsSelector<T, C>) {
     public fun <R> cast(): Convert<T, R> = Convert(df, columns as ColumnsSelector<T, R>)
 
+    @Refine
     @Interpretable("To0")
     public inline fun <reified D> to(): DataFrame<T> = to(typeOf<D>())
 

plugins/kotlin-dataframe/testData/box/convert_to.kt

@@ -0,0 +1,10 @@
+import org.jetbrains.kotlinx.dataframe.*
+import org.jetbrains.kotlinx.dataframe.annotations.*
+import org.jetbrains.kotlinx.dataframe.api.*
+import org.jetbrains.kotlinx.dataframe.io.*
+
+fun box(): String {
+    val df = dataFrameOf("col")("1", "2").convert { col }.to<Int>()
+    val i: Int = df.col[0]
+    return "OK"
+}

plugins/kotlin-dataframe/tests-gen/org/jetbrains/kotlin/fir/dataframe/DataFrameBlackBoxCodegenTestGenerated.java

@@ -58,6 +58,12 @@ public void testConvertToDataFrame() {
     runTest("testData/box/convertToDataFrame.kt");
   }
 
+  @Test
+  @TestMetadata("convert_to.kt")
+  public void testConvert_to() {
+    runTest("testData/box/convert_to.kt");
+  }
+
   @Test
   @TestMetadata("dataFrameOf.kt")
   public void testDataFrameOf() {