New J2K: class tree mapper

2017-10-23 00:25:05 +03:00
parent e33df530cd
commit e3097d8c1f
1 changed files with 9 additions and 0 deletions
@@ -19,6 +19,9 @@ package org.jetbrains.kotlin.j2k
 import com.intellij.lang.java.JavaLanguage
 import com.intellij.psi.*
 import com.intellij.psi.impl.source.tree.java.PsiLiteralExpressionImpl
+import org.jetbrains.kotlin.j2k.ast.CommentsAndSpacesInheritance
+import org.jetbrains.kotlin.j2k.ast.Modifiers
+import org.jetbrains.kotlin.j2k.ast.assignPrototype
 import org.jetbrains.kotlin.j2k.tree.*
 import org.jetbrains.kotlin.j2k.tree.impl.*

@@ -41,6 +44,12 @@ class JavaToJKTreeBuilder {
        fun result(): JKExpression = resultExpression!!
    }

+    private object ClassTreeMapper {
+        fun PsiClass.toJK(): JKClass{
+            return JKClassImpl(with(ModifierMapper){modifierList.toJK()}, JKNameIdentifierImpl(this.name!!))
+        }
+    }
+
    private object ModifierMapper {
        fun PsiModifierList?.toJK(): JKModifierList {