gate.termraider.gui.HyponymyDebugger.java Source code

Introduction

Here is the source code for gate.termraider.gui.HyponymyDebugger.java

Source

/*
 *  Copyright (c) 2008--2014, The University of Sheffield. See the file
 *  COPYRIGHT.txt in the software or at http://gate.ac.uk/gate/COPYRIGHT.txt
 *
 *  This file is part of GATE (see http://gate.ac.uk/), and is free
 *  software, licenced under the GNU Library General Public License,
 *  Version 2, June 1991 (in the distribution as file licence.html,
 *  and also available at http://gate.ac.uk/gate/licence.html).
 *
 *  $Id: HyponymyDebugger.java 17718 2014-03-20 20:40:06Z adamfunk $
 */
package gate.termraider.gui;

import gate.creole.ANNIEConstants;
import gate.termraider.bank.HyponymyTermbank;
import gate.termraider.util.Term;
import gate.termraider.util.TermComparator;
import java.awt.BorderLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import javax.swing.Box;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTable;
import javax.swing.JTextField;
import javax.swing.table.AbstractTableModel;
import javax.swing.table.TableModel;
import org.apache.commons.lang.StringUtils;

public class HyponymyDebugger extends JPanel implements ANNIEConstants {

    private static final long serialVersionUID = 4949616706410088547L;

    private JScrollPane scrollPane;
    private HyponymyTermbank termbank;
    private JTable table;
    private JButton goButton;
    private JPanel controlPanel, placeholder;

    public HyponymyDebugger(HyponymyTermbank termbank) {
        this.termbank = termbank;
        setLayout(new BorderLayout());
        makeControlPanel();
        this.add(controlPanel, BorderLayout.NORTH);
        placeholder = new JPanel();
        this.add(placeholder, BorderLayout.CENTER);
    }

    private void makeControlPanel() {
        goButton = new JButton("generate debugging table");
        goButton.setToolTipText("This may take some time!");
        goButton.addActionListener(new HDGoButtonActionListener(this));

        controlPanel = new JPanel();
        controlPanel.setLayout(new BoxLayout(controlPanel, BoxLayout.X_AXIS));
        controlPanel.add(Box.createHorizontalGlue());
        controlPanel.add(goButton);
        controlPanel.add(Box.createHorizontalGlue());
    }

    protected void generateTable() {
        JTextField tempField = new JTextField("generating...");
        placeholder.add(tempField);

        TableModel tableModel = new HDTableModel(termbank);
        table = new JTable(tableModel);
        table.setDefaultRenderer(String.class, new MultilineCellRenderer());
        table.setAutoCreateRowSorter(true);
        scrollPane = new JScrollPane(table, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED,
                JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED);
        this.remove(placeholder);
        this.add(scrollPane, BorderLayout.CENTER);
    }

}

class HDTableModel extends AbstractTableModel {
    private static final long serialVersionUID = -1124137938074923640L;
    private String[] columnNames = { "term", "score", "docs", "docs", "hyponyms", "hyponyms", "heads" };
    private Map<Term, Set<String>> termDocuments, termHyponyms, termHeads;
    private List<Term> terms;
    private HyponymyTermbank termbank;

    public HDTableModel(HyponymyTermbank termbank) {
        this.termbank = termbank;
        this.termDocuments = termbank.getTermDocuments();
        this.termHeads = termbank.getTermHeads();
        this.termHyponyms = termbank.getTermHyponyms();
        terms = new ArrayList<Term>(termDocuments.keySet());
        Collections.sort(terms, new TermComparator());
    }

    public Class<?> getColumnClass(int i) {
        return String.class;
    }

    @Override
    public int getColumnCount() {
        return columnNames.length;
    }

    @Override
    public int getRowCount() {
        return termDocuments.size();
    }

    public String getColumnName(int c) {
        return columnNames[c];
    }

    public boolean isCellEditable(int r, int c) {
        return false;
    }

    @Override
    public Object getValueAt(int row, int column) {
        Term term = terms.get(row);
        String result = "";
        switch (column) {
        case 0:
            result = term.toString();
            break;
        case 1:
            result = termbank.getDefaultScores().get(term).toString();
            break;
        case 2:
            result = Integer.toString(termDocuments.get(term).size());
            break;
        case 3:
            result = StringUtils.join(termDocuments.get(term), '\n');
            break;
        case 4:
            result = Integer.toString(termHyponyms.get(term).size());
            break;
        case 5:
            result = StringUtils.join(termHyponyms.get(term), '\n');
            break;
        case 6:
            result = StringUtils.join(termHeads.get(term), '\n');
            break;
        }
        return result;
    }

}

class HDGoButtonActionListener implements ActionListener {
    private HyponymyDebugger viewer;

    public HDGoButtonActionListener(HyponymyDebugger viewer) {
        this.viewer = viewer;
    }

    @Override
    public void actionPerformed(ActionEvent arg0) {
        viewer.generateTable();
    }
}