6
6
package org.jetbrains.kotlinx.dataframe.plugin
7
7
8
8
import org.jetbrains.kotlin.fir.FirSession
9
+ import org.jetbrains.kotlinx.dataframe.plugin.extensions.KotlinTypeFacade
10
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.Interpreter
11
+ import org.jetbrains.kotlinx.dataframe.plugin.utils.Names.INTERPRETABLE_FQNAME
12
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Add
13
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddWithDsl
14
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.And0
15
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.And10
16
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Convert0
17
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Convert2
18
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Convert6
19
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameGroupBy
20
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropNulls0
21
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Exclude0
22
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Exclude1
23
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Explode0
24
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Expr0
25
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.From
26
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Group0
27
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AggregateDslInto
28
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByToDataFrame
29
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert0
30
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert1
31
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert2
32
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert3
33
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Into
34
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Into0
35
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Join0
36
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Match0
37
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Preserve0
38
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Preserve1
39
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Properties0
40
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Remove0
41
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Rename
42
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.RenameInto
43
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Select0
44
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.To0
45
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under0
46
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under1
47
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under2
48
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under3
49
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under4
50
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Ungroup0
51
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.With0
9
52
import org.jetbrains.kotlin.fir.declarations.findArgumentByName
10
53
import org.jetbrains.kotlin.fir.expressions.FirClassReferenceExpression
11
54
import org.jetbrains.kotlin.fir.expressions.FirFunctionCall
@@ -23,15 +66,10 @@ import org.jetbrains.kotlin.name.CallableId
23
66
import org.jetbrains.kotlin.name.ClassId
24
67
import org.jetbrains.kotlin.name.Name
25
68
import org.jetbrains.kotlin.name.StandardClassIds
26
- import org.jetbrains.kotlinx.dataframe.plugin.extensions.KotlinTypeFacade
27
- import org.jetbrains.kotlinx.dataframe.plugin.impl.Interpreter
28
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Add
29
69
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddDslNamedGroup
30
70
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddDslStringInvoke
31
71
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddId
32
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddWithDsl
33
72
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Aggregate
34
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AggregateDslInto
35
73
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AggregateRow
36
74
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.All0
37
75
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.All1
@@ -49,8 +87,6 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AllFrom2
49
87
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AllUpTo0
50
88
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AllUpTo1
51
89
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AllUpTo2
52
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.And0
53
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.And10
54
90
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ByName
55
91
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColGroups0
56
92
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColGroups1
@@ -64,11 +100,7 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColsOf1
64
100
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColsOf2
65
101
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColumnRange
66
102
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ConcatWithKeys
67
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Convert0
68
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Convert2
69
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Convert6
70
103
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameBuilderInvoke0
71
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameGroupBy
72
104
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameOf0
73
105
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameOf3
74
106
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameXs
@@ -79,11 +111,6 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropLast0
79
111
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropLast1
80
112
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropLast2
81
113
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropNa0
82
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropNulls0
83
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Exclude0
84
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Exclude1
85
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Explode0
86
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Expr0
87
114
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FillNulls0
88
115
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.First0
89
116
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.First1
@@ -93,8 +120,6 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FlattenDefault
93
120
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FrameCols0
94
121
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FrameCols1
95
122
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FrameCols2
96
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.From
97
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Group0
98
123
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByAdd
99
124
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByCount0
100
125
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByInto
@@ -113,38 +138,29 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByMinOf
113
138
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByReduceExpression
114
139
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByReduceInto
115
140
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByReducePredicate
141
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByXs
142
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.InsertAfter0
143
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Last0
144
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Last1
145
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Last2
116
146
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByStd0
117
147
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByStd1
118
148
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByStdOf
119
149
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupBySum0
120
150
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupBySum1
121
151
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupBySumOf
122
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByToDataFrame
123
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByXs
124
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert0
125
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert1
126
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert2
127
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Insert3
128
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.InsertAfter0
129
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Into
130
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Into0
131
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Join0
132
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Last0
133
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Last1
134
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Last2
135
152
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MapToFrame
136
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Match0
137
153
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Merge0
154
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeId
138
155
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeBy0
139
156
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeBy1
140
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeId
141
157
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeInto0
142
158
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Move0
143
159
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveAfter0
144
160
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveInto0
145
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToEnd0
146
161
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToStart0
147
162
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToStart1
163
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToEnd0
148
164
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveUnder0
149
165
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveUnder1
150
166
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.NameContains0
@@ -160,54 +176,39 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Named0
160
176
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.PairConstructor
161
177
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.PairToConstructor
162
178
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.PerRowCol
163
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Preserve0
164
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Preserve1
165
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Properties0
166
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Remove0
167
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Rename
168
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.RenameInto
169
179
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.RenameMapping
180
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrame
181
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameColumn
182
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameDefault
183
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameDsl
184
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameDslStringInvoke
185
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameFrom
186
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToTop
187
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.TrimMargin
188
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Update0
189
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.UpdateWith0
190
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCounts
170
191
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.RenameToCamelCase
171
192
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.RenameToCamelCaseClause
172
193
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Reorder
173
194
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ReorderColumnsByName
174
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Select0
175
195
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Single0
176
196
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Single1
177
197
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Single2
198
+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCols2
178
199
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Take0
179
200
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Take1
180
201
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Take2
181
202
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.TakeLast0
182
203
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.TakeLast1
183
204
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.TakeLast2
184
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.To0
185
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrame
186
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameColumn
187
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameDefault
188
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameDsl
189
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameDslStringInvoke
190
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameFrom
191
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToTop
192
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.TrimMargin
193
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under0
194
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under1
195
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under2
196
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under3
197
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Under4
198
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Ungroup0
199
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Update0
200
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.UpdateWith0
201
205
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCols0
202
206
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCols1
203
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCols2
204
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCounts
205
- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.With0
206
207
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.WithoutNulls0
207
208
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.WithoutNulls1
208
209
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.WithoutNulls2
209
210
import org.jetbrains.kotlinx.dataframe.plugin.utils.Names
210
- import org.jetbrains.kotlinx.dataframe.plugin.utils.Names.INTERPRETABLE_FQNAME
211
+
211
212
212
213
internal fun FirFunctionCall.loadInterpreter (session : FirSession ): Interpreter <* >? {
213
214
val interpreter = Stdlib .interpreter(this )
0 commit comments