From e3097d8c1ff6cdb98587b39f2246cfb270e2ee33 Mon Sep 17 00:00:00 2001
From: knize <kirill.knize@gmail.com>
Date: Mon, 23 Oct 2017 00:25:05 +0300
Subject: [PATCH] New J2K: class tree mapper

---
 .../org/jetbrains/kotlin/j2k/JavaToJKTreeBuilder.kt      | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/j2k/newSrc/org/jetbrains/kotlin/j2k/JavaToJKTreeBuilder.kt b/j2k/newSrc/org/jetbrains/kotlin/j2k/JavaToJKTreeBuilder.kt
index d3302faeae5..85f5892d23d 100644
--- a/j2k/newSrc/org/jetbrains/kotlin/j2k/JavaToJKTreeBuilder.kt
+++ b/j2k/newSrc/org/jetbrains/kotlin/j2k/JavaToJKTreeBuilder.kt
@@ -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 {