com.intellij.openapi.ui.popup.util.BaseListPopupStep.java Source code

Introduction

Here is the source code for com.intellij.openapi.ui.popup.util.BaseListPopupStep.java

Source

// Copyright 2000-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.intellij.openapi.ui.popup.util;

import com.intellij.openapi.ui.popup.ListPopupStep;
import com.intellij.openapi.ui.popup.ListSeparator;
import com.intellij.openapi.ui.popup.PopupStep;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import javax.swing.*;
import java.awt.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class BaseListPopupStep<T> extends BaseStep<T> implements ListPopupStep<T> {
    private String myTitle;
    private List<T> myValues;
    private List<? extends Icon> myIcons;
    private int myDefaultOptionIndex = -1;

    @SafeVarargs
    public BaseListPopupStep(@Nullable String title, T @NotNull... values) {
        this(title, values, new Icon[] {});
    }

    public BaseListPopupStep(@Nullable String title, List<? extends T> values) {
        this(title, values, new ArrayList<>());
    }

    public BaseListPopupStep(@Nullable String title, T[] values, Icon[] icons) {
        this(title, Arrays.asList(values), Arrays.asList(icons));
    }

    public BaseListPopupStep(@Nullable String aTitle, @NotNull List<? extends T> aValues, Icon aSameIcon) {
        List<Icon> icons = new ArrayList<>();
        for (int i = 0; i < aValues.size(); i++) {
            icons.add(aSameIcon);
        }
        init(aTitle, aValues, icons);
    }

    public BaseListPopupStep(@Nullable String title, @NotNull List<? extends T> values,
            List<? extends Icon> icons) {
        init(title, values, icons);
    }

    protected BaseListPopupStep() {
    }

    protected final void init(@Nullable String title, @NotNull List<? extends T> values,
            @Nullable List<? extends Icon> icons) {
        myTitle = title;
        myValues = new ArrayList<>(values);
        myIcons = icons;
    }

    @Override
    @Nullable
    public final String getTitle() {
        return myTitle;
    }

    @Override
    @NotNull
    public final List<T> getValues() {
        return myValues;
    }

    @Override
    @Nullable
    public PopupStep onChosen(T selectedValue, final boolean finalChoice) {
        return FINAL_CHOICE;
    }

    @Override
    public Icon getIconFor(T value) {
        int index = myValues.indexOf(value);
        if (index != -1 && myIcons != null && index < myIcons.size()) {
            return myIcons.get(index);
        } else {
            return null;
        }
    }

    @Nullable
    public Color getBackgroundFor(T value) {
        return null;
    }

    @Nullable
    public Color getForegroundFor(T value) {
        return null;
    }

    @Override
    @NotNull
    public String getTextFor(T value) {
        return value.toString();
    }

    @Override
    @Nullable
    public ListSeparator getSeparatorAbove(T value) {
        return null;
    }

    @Override
    public boolean isSelectable(T value) {
        return true;
    }

    @Override
    public boolean hasSubstep(T selectedValue) {
        return false;
    }

    @Override
    public void canceled() {
    }

    public void setDefaultOptionIndex(int aDefaultOptionIndex) {
        myDefaultOptionIndex = aDefaultOptionIndex;
    }

    @Override
    public int getDefaultOptionIndex() {
        return myDefaultOptionIndex;
    }
}