SearchDialog.java

/*
 * SearchDialog.java
 *
 * This file is part of TDA - Thread Dump Analysis Tool.
 *
 * TDA is free software; you can redistribute it and/or modify
 * it under the terms of the Lesser GNU General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * TDA is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * Lesser GNU General Public License for more details.
 *
 * You should have received a copy of the Lesser GNU General Public License
 * along with TDA; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * $Id: SearchDialog.java,v 1.10 2008-01-09 09:31:35 irockel Exp $
 */

package com.pironet.tda;

import java.awt.FlowLayout;
import java.awt.GridLayout;
import java.awt.Rectangle;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;

import javax.swing.JButton;
import javax.swing.JComponent;
import javax.swing.JDialog;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JTable;
import javax.swing.JTextField;
import javax.swing.JTree;
import javax.swing.JViewport;
import javax.swing.table.TableModel;
import javax.swing.text.Position;
import javax.swing.tree.TreePath;

import com.pironet.tda.utils.ColoredTable;
import com.pironet.tda.utils.TableSorter;
import com.pironet.tda.utils.ThreadsTableModel;

import fr.loicmathieu.bobbin.gui.LinesTableModel;

/**
 *
 * @author irockel
 */
public class SearchDialog extends JDialog
implements ActionListener {

	private static String SEARCH = "search";
	private static String CANCEL = "cancel";

	private JTextField searchField;

	private JComponent searchComp;
	private JTable searchTable;

	public SearchDialog(JFrame owner, JComponent comp) {
		super(owner, "Search this category... ");
		setLayout(new FlowLayout(FlowLayout.LEFT));

		//Create everything.
		searchField = new JTextField(10);
		searchField.setActionCommand(SEARCH);
		searchField.addActionListener(this);
		setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE);

		JLabel label = new JLabel("Enter search string: ");
		label.setLabelFor(searchField);

		searchComp = comp;

		JComponent buttonPane = createButtonPanel();

		//Lay out everything.
		JPanel textPane = new JPanel(new FlowLayout(FlowLayout.TRAILING));
		textPane.add(label);
		textPane.add(searchField);

		add(textPane);
		add(buttonPane);
	}

	protected JComponent createButtonPanel() {
		JPanel p = new JPanel(new GridLayout(0,1));
		JButton searchButton = new JButton("Search");

		searchButton.setActionCommand(SEARCH);
		searchButton.addActionListener(this);

		p.add(searchButton);

		return p;
	}

	public void actionPerformed(ActionEvent e) {
		String cmd = e.getActionCommand();

		if (SEARCH.equals(cmd)) {
			if(searchComp instanceof JTree) {
				//FIXME : search in monitors didn't work
				TreePath searchPath = ((JTree) searchComp).getNextMatch(searchField.getText(), 0, Position.Bias.Forward);

				if (searchPath != null) {
					((JTree) searchComp).setSelectionPath(searchPath);
					Rectangle view = ((JTree) searchComp).getPathBounds(searchPath);
					((JViewport) searchComp.getParent()).scrollRectToVisible(view);
					dispose();
					searchComp.requestFocusInWindow();
				} else {
					JOptionPane.showMessageDialog(getOwner(),
							searchField.getText() + " not found!",
							"Search Error",
							JOptionPane.ERROR_MESSAGE);
					resetFocus();
				}
			} else if (searchComp instanceof ColoredTable) {
				ColoredTable theTable = (ColoredTable) searchComp;
				int row = -1;
				TableModel tableModel = ((TableSorter) theTable.getModel()).getTableModel();
				if(tableModel instanceof ThreadsTableModel){
					ThreadsTableModel ttm = (ThreadsTableModel) tableModel;
					row = ttm.searchRowWithName(theTable.getSelectedRow(), searchField.getText());
				}
				else {
					LinesTableModel ltm = (LinesTableModel) tableModel;
					row = ltm.searchRowWithName(theTable.getSelectedRow(), searchField.getText());
				}
				theTable.getSelectionModel().setSelectionInterval(row, row);
				theTable.scrollRectToVisible(theTable.getCellRect(row, 0, true));
			}
		}
	}

	//Must be called from the event-dispatching thread.
	protected void resetFocus() {
		searchField.requestFocusInWindow();
	}

	public void reset() {
	}

}