com.intellij.execution.junit2.configuration.MethodListDlg.java Source code

Introduction

Here is the source code for com.intellij.execution.junit2.configuration.MethodListDlg.java

Source

/*
 * Copyright 2000-2009 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.intellij.execution.junit2.configuration;

import java.awt.BorderLayout;
import java.awt.event.MouseEvent;
import java.util.Comparator;

import javax.swing.JComponent;
import javax.swing.JList;
import javax.swing.JPanel;
import javax.swing.ListSelectionModel;

import com.intellij.execution.ExecutionBundle;
import com.intellij.ide.structureView.impl.StructureNodeRenderer;
import com.intellij.openapi.ui.DialogWrapper;
import com.intellij.openapi.util.Condition;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.PsiSubstitutor;
import com.intellij.psi.util.PsiFormatUtil;
import com.intellij.ui.ColoredListCellRenderer;
import com.intellij.ui.DoubleClickListener;
import com.intellij.ui.ListScrollingUtil;
import com.intellij.ui.ScrollPaneFactory;
import com.intellij.ui.SimpleTextAttributes;
import com.intellij.ui.SortedListModel;
import com.intellij.ui.components.JBList;

// Author: dyoma

public class MethodListDlg extends DialogWrapper {
    private final PsiClass myClass;
    private static final Comparator<PsiMethod> METHOD_NAME_COMPARATOR = new Comparator<PsiMethod>() {
        @Override
        public int compare(final PsiMethod psiMethod, final PsiMethod psiMethod1) {
            return psiMethod.getName().compareToIgnoreCase(psiMethod1.getName());
        }
    };
    private final SortedListModel<PsiMethod> myListModel = new SortedListModel<PsiMethod>(METHOD_NAME_COMPARATOR);
    private final JList myList = new JBList(myListModel);
    private final JPanel myWholePanel = new JPanel(new BorderLayout());

    public MethodListDlg(final PsiClass psiClass, final Condition<PsiMethod> filter, final JComponent parent) {
        super(parent, false);
        myClass = psiClass;
        createList(psiClass.getAllMethods(), filter);
        myWholePanel.add(ScrollPaneFactory.createScrollPane(myList));
        myList.setCellRenderer(new ColoredListCellRenderer() {
            @Override
            protected void customizeCellRenderer(final JList list, final Object value, final int index,
                    final boolean selected, final boolean hasFocus) {
                final PsiMethod psiMethod = (PsiMethod) value;
                append(PsiFormatUtil.formatMethod(psiMethod, PsiSubstitutor.EMPTY, PsiFormatUtil.SHOW_NAME, 0),
                        StructureNodeRenderer.applyDeprecation(psiMethod, SimpleTextAttributes.REGULAR_ATTRIBUTES));
                final PsiClass containingClass = psiMethod.getContainingClass();
                if (!myClass.equals(containingClass)) {
                    append(" (" + containingClass.getQualifiedName() + ")", StructureNodeRenderer
                            .applyDeprecation(containingClass, SimpleTextAttributes.GRAY_ATTRIBUTES));
                }
            }
        });
        myList.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
        new DoubleClickListener() {
            @Override
            protected boolean onDoubleClick(MouseEvent e) {
                MethodListDlg.this.close(OK_EXIT_CODE);
                return true;
            }
        }.installOn(myList);

        ListScrollingUtil.ensureSelectionExists(myList);
        setTitle(ExecutionBundle.message("choose.test.method.dialog.title"));
        init();
    }

    private void createList(final PsiMethod[] allMethods, final Condition<PsiMethod> filter) {
        for (int i = 0; i < allMethods.length; i++) {
            final PsiMethod method = allMethods[i];
            if (filter.value(method)) {
                myListModel.add(method);
            }
        }
    }

    @Override
    protected JComponent createCenterPanel() {
        return myWholePanel;
    }

    public PsiMethod getSelected() {
        return (PsiMethod) myList.getSelectedValue();
    }
}