66package org.jetbrains.kotlinx.dataframe.plugin
77
88import 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
952import org.jetbrains.kotlin.fir.declarations.findArgumentByName
1053import org.jetbrains.kotlin.fir.expressions.FirClassReferenceExpression
1154import org.jetbrains.kotlin.fir.expressions.FirFunctionCall
@@ -23,15 +66,10 @@ import org.jetbrains.kotlin.name.CallableId
2366import org.jetbrains.kotlin.name.ClassId
2467import org.jetbrains.kotlin.name.Name
2568import 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
2969import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddDslNamedGroup
3070import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddDslStringInvoke
3171import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddId
32- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AddWithDsl
3372import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Aggregate
34- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AggregateDslInto
3573import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AggregateRow
3674import org.jetbrains.kotlinx.dataframe.plugin.impl.api.All0
3775import org.jetbrains.kotlinx.dataframe.plugin.impl.api.All1
@@ -49,8 +87,6 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AllFrom2
4987import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AllUpTo0
5088import org.jetbrains.kotlinx.dataframe.plugin.impl.api.AllUpTo1
5189import 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
5490import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ByName
5591import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColGroups0
5692import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColGroups1
@@ -64,11 +100,7 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColsOf1
64100import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColsOf2
65101import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ColumnRange
66102import 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
70103import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameBuilderInvoke0
71- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameGroupBy
72104import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameOf0
73105import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameOf3
74106import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DataFrameXs
@@ -79,11 +111,6 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropLast0
79111import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropLast1
80112import org.jetbrains.kotlinx.dataframe.plugin.impl.api.DropLast2
81113import 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
87114import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FillNulls0
88115import org.jetbrains.kotlinx.dataframe.plugin.impl.api.First0
89116import org.jetbrains.kotlinx.dataframe.plugin.impl.api.First1
@@ -93,8 +120,6 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FlattenDefault
93120import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FrameCols0
94121import org.jetbrains.kotlinx.dataframe.plugin.impl.api.FrameCols1
95122import 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
98123import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByAdd
99124import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByCount0
100125import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByInto
@@ -113,38 +138,29 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByMinOf
113138import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByReduceExpression
114139import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByReduceInto
115140import 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
116146import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByStd0
117147import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByStd1
118148import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupByStdOf
119149import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupBySum0
120150import org.jetbrains.kotlinx.dataframe.plugin.impl.api.GroupBySum1
121151import 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
135152import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MapToFrame
136- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Match0
137153import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Merge0
154+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeId
138155import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeBy0
139156import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeBy1
140- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeId
141157import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MergeInto0
142158import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Move0
143159import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveAfter0
144160import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveInto0
145- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToEnd0
146161import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToStart0
147162import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToStart1
163+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveToEnd0
148164import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveUnder0
149165import org.jetbrains.kotlinx.dataframe.plugin.impl.api.MoveUnder1
150166import org.jetbrains.kotlinx.dataframe.plugin.impl.api.NameContains0
@@ -160,54 +176,39 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Named0
160176import org.jetbrains.kotlinx.dataframe.plugin.impl.api.PairConstructor
161177import org.jetbrains.kotlinx.dataframe.plugin.impl.api.PairToConstructor
162178import 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
169179import 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
170191import org.jetbrains.kotlinx.dataframe.plugin.impl.api.RenameToCamelCase
171192import org.jetbrains.kotlinx.dataframe.plugin.impl.api.RenameToCamelCaseClause
172193import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Reorder
173194import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ReorderColumnsByName
174- import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Select0
175195import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Single0
176196import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Single1
177197import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Single2
198+ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCols2
178199import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Take0
179200import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Take1
180201import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Take2
181202import org.jetbrains.kotlinx.dataframe.plugin.impl.api.TakeLast0
182203import org.jetbrains.kotlinx.dataframe.plugin.impl.api.TakeLast1
183204import 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
201205import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ValueCols0
202206import 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
206207import org.jetbrains.kotlinx.dataframe.plugin.impl.api.WithoutNulls0
207208import org.jetbrains.kotlinx.dataframe.plugin.impl.api.WithoutNulls1
208209import org.jetbrains.kotlinx.dataframe.plugin.impl.api.WithoutNulls2
209210import org.jetbrains.kotlinx.dataframe.plugin.utils.Names
210- import org.jetbrains.kotlinx.dataframe.plugin.utils.Names.INTERPRETABLE_FQNAME
211+
211212
212213internal fun FirFunctionCall.loadInterpreter (session : FirSession ): Interpreter <* >? {
213214 val interpreter = Stdlib .interpreter(this )
0 commit comments