Create Function From Usage: Translate ClassCandidateListCellRenderer to Kotlin

2014-09-22 20:32:14 +04:00
parent d1e43d6482
commit 34632f46f6
2 changed files with 25 additions and 35 deletions
@@ -1,35 +0,0 @@
-package org.jetbrains.jet.plugin.quickfix.createFromUsage.createFunction;
-
-import com.intellij.ide.util.PsiElementListCellRenderer;
-import org.jetbrains.annotations.NotNull;
-import org.jetbrains.annotations.Nullable;
-import org.jetbrains.jet.lang.psi.JetClass;
-import org.jetbrains.jet.plugin.presentation.JetClassPresenter;
-
-import javax.swing.*;
-import java.awt.*;
-
-class ClassCandidateListCellRenderer extends PsiElementListCellRenderer<JetClass> {
-    private final JetClassPresenter presenter = new JetClassPresenter();
-
-    @Override
-    public String getElementText(@NotNull JetClass element) {
-        return presenter.getPresentation(element).getPresentableText();
-    }
-
-    @Nullable
-    @Override
-    protected String getContainerText(JetClass element, String name) {
-        return presenter.getPresentation(element).getLocationString();
-    }
-
-    @Override
-    protected int getIconFlags() {
-        return 0;
-    }
-
-    @Override
-    public Component getListCellRendererComponent(JList list, Object value, int index, boolean isSelected, boolean cellHasFocus) {
-        return super.getListCellRendererComponent(list, ((ClassCandidate) value).getJetClass(), index, isSelected, cellHasFocus);
-    }
-}
@@ -0,0 +1,25 @@
+package org.jetbrains.jet.plugin.quickfix.createFromUsage.createFunction
+
+import com.intellij.ide.util.PsiElementListCellRenderer
+import org.jetbrains.jet.lang.psi.JetClass
+import org.jetbrains.jet.plugin.presentation.JetClassPresenter
+
+import javax.swing.*
+import java.awt.*
+
+class ClassCandidateListCellRenderer : PsiElementListCellRenderer<JetClass>() {
+    private val presenter = JetClassPresenter()
+
+    override fun getElementText(element: JetClass): String? =
+            presenter.getPresentation(element)?.getPresentableText()
+
+    protected override fun getContainerText(element: JetClass?, name: String?): String? =
+            element?.let { presenter.getPresentation(it) }?.getLocationString()
+
+    override fun getIconFlags(): Int = 0
+
+    public override fun getListCellRendererComponent(
+            list: JList<out Any?>, value: Any?, index: Int, isSelected: Boolean, cellHasFocus: Boolean
+    ): Component? =
+            super.getListCellRendererComponent(list, (value as ClassCandidate).jetClass, index, isSelected, cellHasFocus)
+}