/*
* Copyright 2000-2015 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.ui.stripe;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.project.DumbAwareAction;
import javax.swing.*;
import javax.swing.event.TreeModelListener;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreePath;
import java.beans.EventHandler;
import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
/**
* @author Sergey.Malenkov
*/
public class TreeUpdater<Painter extends ErrorStripePainter> extends Updater<Painter> {
private final JTree myTree;
private final TreeModelListener myTreeModelListener = EventHandler.create(TreeModelListener.class, this, "update");
private final PropertyChangeListener myPropertyChangeListener = new PropertyChangeListener() {
@Override
public void propertyChange(PropertyChangeEvent event) {
Object oldValue = event.getOldValue();
if (oldValue instanceof TreeModel) ((TreeModel)oldValue).removeTreeModelListener(myTreeModelListener);
Object newValue = event.getNewValue();
if (newValue instanceof TreeModel) ((TreeModel)newValue).addTreeModelListener(myTreeModelListener);
}
};
public TreeUpdater(Painter painter, JScrollPane pane, JTree tree) {
super(painter, pane);
myTree = tree;
myTree.addPropertyChangeListener(JTree.TREE_MODEL_PROPERTY, myPropertyChangeListener);
TreeModel model = myTree.getModel();
if (model != null) model.addTreeModelListener(myTreeModelListener);
new DumbAwareAction() {
@Override
public void actionPerformed(AnActionEvent event) {
selectNext(myTree.getMaxSelectionRow());
}
}.registerCustomShortcutSet(getNextErrorShortcut(), myTree, this);
new DumbAwareAction() {
@Override
public void actionPerformed(AnActionEvent event) {
selectPrevious(myTree.getMinSelectionRow());
}
}.registerCustomShortcutSet(getPreviousErrorShortcut(), myTree, this);
}
@Override
public void dispose() {
myTree.removePropertyChangeListener(JTree.TREE_MODEL_PROPERTY, myPropertyChangeListener);
TreeModel model = myTree.getModel();
if (model != null) model.removeTreeModelListener(myTreeModelListener);
super.dispose();
}
@Override
protected void onSelect(Painter painter, int index) {
if (0 <= index) {
myTree.setSelectionRow(index);
myTree.scrollRowToVisible(index);
}
}
protected void update(Painter painter) {
update(painter, myTree);
}
protected void update(Painter painter, JTree tree) {
int count = tree.getRowCount();
painter.setErrorStripeCount(count);
for (int index = 0; index < count; index++) {
TreePath path = tree.getPathForRow(index);
update(painter, index, path == null ? null : path.getLastPathComponent());
}
}
}