/***
* Copyright (c) 2008, Endless Loop Software, Inc.
*
* This file is part of EgoNet.
*
* EgoNet is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* EgoNet 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
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.egonet.graph;
import org.jdesktop.layout.GroupLayout;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.swing.*;
import javax.swing.table.TableColumnModel;
import java.awt.Color;
import java.awt.Dimension;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.ItemEvent;
import java.util.*;
import org.egonet.gui.interview.EgoClient;
import org.egonet.gui.table.*;
import org.egonet.model.QuestionList;
import org.egonet.model.Study;
import org.egonet.model.answer.*;
import org.egonet.model.question.AlterQuestion;
import org.egonet.model.question.Question;
import org.egonet.model.question.Selection;
public class NodeShapePanel extends JPanel {
final private static Logger logger = LoggerFactory.getLogger(NodeShapePanel.class);
private JLabel questionLabel;
private JComboBox<Question> questionCombo;
private JComboBox<NodeProperty.NodeShape> shapeCombo;
private JTable table;
private GroupLayout layout;
private GraphRenderer graphRenderer;
private JButton applyButton;
List<Selection> selectionList = new ArrayList<Selection>();
private EgoClient egoClient;
public NodeShapePanel(EgoClient egoClient, GraphRenderer renderer) {
this.egoClient = egoClient;
this.graphRenderer = renderer;
layout = new GroupLayout(this);
this.setLayout(layout);
layout.setAutocreateGaps(true);
layout.setAutocreateContainerGaps(true);
createComponents();
}
private void createComponents() {
questionLabel = new JLabel("Choose question to shape");
questionLabel.setVisible(true);
// create questionCombo
List<Question> qList = new ArrayList<Question>();
Study study = egoClient.getInterview().getStudy();
QuestionList questionList = study.getQuestions();
for (Long key : questionList.keySet()) {
Question currentQuestion = questionList.get(key);
if (currentQuestion instanceof AlterQuestion) {
// populate the list box with only questions that have choices
// as answers
if (currentQuestion.answerType.equals(CategoricalAnswer.class)
|| currentQuestion.answerType.equals(TextAnswer.class))
qList.add(currentQuestion);
}
}
questionCombo = new JComboBox<Question>(qList.toArray(new Question[0]));
questionCombo.setVisible(true);
questionCombo.setEnabled(true);
questionCombo.setPreferredSize(new Dimension(20, 20));
questionCombo.setMaximumSize(new Dimension(20, 30));
questionCombo.setAutoscrolls(true);
questionCombo.addItemListener(new java.awt.event.ItemListener() {
public void itemStateChanged(ItemEvent e) {
if (e.getStateChange() == ItemEvent.SELECTED) {
createTable();
drawPanel();
}
}
});
// create shape combo for table cell
shapeCombo = new JComboBox<NodeProperty.NodeShape>(NodeProperty.NodeShape.values());
shapeCombo.setPreferredSize(new Dimension(20, 20));
shapeCombo.setMaximumSize(new Dimension(20, 30));
// shapeCombo.setSelectedIndex(0);
// create apply button
applyButton = new JButton("Apply Shape");
applyButton.setVisible(true);
applyButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
updateNodeShape();
}
});
createTable();
drawPanel();
}
private Object[][] getRowData() {
if(questionCombo.getSelectedIndex() == -1)
return new Object[0][0];
Question question = (Question) questionCombo.getSelectedItem();
// logger.info("Question examining:" + question.UniqueId);
if (question.answerType.equals(CategoricalAnswer.class)) {
int noOfRows = question.getSelections().size();
Object[][] rowData = new Object[noOfRows][2];
/* change the list of selections based on the selected question */
if (!selectionList.isEmpty()) {
selectionList.removeAll(selectionList);
}
for (Selection selection : question.getSelections()) {
selectionList.add(selection);
}
for (int i = 0; i < noOfRows; i++) {
rowData[i][0] = selectionList.get(i);
}
// populate the shapes
int maxNoOfShapes = NodeProperty.NodeShape.values().length;
for (int i = 0; i < noOfRows; i++) {
if (i < maxNoOfShapes) {
NodeProperty.NodeShape shape = NodeProperty.NodeShape
.values()[i];
rowData[i][1] = shape;
} else {
NodeProperty.NodeShape shape = NodeProperty.NodeShape
.values()[i % maxNoOfShapes];
rowData[i][1] = shape;
}
}
return rowData;
} else {
// logger.info("Populating text answers!!!");
if (!selectionList.isEmpty()) {
selectionList.removeAll(selectionList);
}
Answer[] answers = egoClient.getInterview().get_answers();
for (int i = 0; i < answers.length; i++) {
// logger.info("Question examining:"
// + answers[i].questionId + "," + question.UniqueId);
if (answers[i].getQuestionId().equals(question.UniqueId)) {
if (answers[i].string == null
|| isPresentInSelectionList(answers[i].string)) {
continue;
}
Selection selection = new Selection();
selection.setString(answers[i].string);
// logger.info("Selection:" + selection);
selectionList.add(selection);
}
}
Object[][] rowData = new Object[selectionList.size()][2];
for (int i = 0; i < selectionList.size(); i++) {
rowData[i][0] = selectionList.get(i);
}
int noOfRows = selectionList.size();
// populate the shapes
int maxNoOfShapes = NodeProperty.NodeShape.values().length;
for (int i = 0; i < noOfRows; i++) {
if (i < maxNoOfShapes) {
NodeProperty.NodeShape shape = NodeProperty.NodeShape
.values()[i];
rowData[i][1] = shape;
} else {
NodeProperty.NodeShape shape = NodeProperty.NodeShape
.values()[i % maxNoOfShapes];
rowData[i][1] = shape;
}
}
return rowData;
}
}
private boolean isPresentInSelectionList(String str) {
for (int i = 0; i < selectionList.size(); i++) {
if (selectionList.get(i).getString().equals(str))
return true;
}
return false;
}
private void createTable() {
Object[][] rowData = getRowData();
table = new JTable(new PropertyTableModel(rowData));
table.setPreferredScrollableViewportSize(table.getPreferredSize());
table.setRowHeight(25);
table.setVisible(true);
table.setBorder(BorderFactory.createCompoundBorder(BorderFactory
.createLineBorder(Color.black), getBorder()));
TableColumnModel columnModel = table.getColumnModel();
LabelRenderer selectionRenderer = new LabelRenderer();
columnModel.getColumn(0).setCellRenderer(selectionRenderer);
DefaultCellEditor shapeEditor = new DefaultCellEditor(shapeCombo);
columnModel.getColumn(1).setCellEditor(shapeEditor);
columnModel.getColumn(1).setCellRenderer(new TableComboBoxRenderer<NodeProperty.NodeShape>(NodeProperty.NodeShape.values()));
columnModel.getColumn(0).setMaxWidth(200);
columnModel.getColumn(1).setMaxWidth(150);
}
private void updateNodeShape() {
Question question = (Question) questionCombo.getSelectedItem();
if (question.answerType.equals(CategoricalAnswer.class)) {
for (int i = 0; i < question.getSelections().size(); i++) {
Selection selection = question.getSelections().get(i);
GraphQuestionSelectionPair graphQuestion = new GraphQuestionSelectionPair(question,selection);
NodeProperty nodeProperty = new NodeProperty();
nodeProperty.setShape((NodeProperty.NodeShape) table
.getValueAt(i, 1));
nodeProperty.setProperty(NodeProperty.NodePropertyType.Shape);
graphRenderer.addQAsettings(graphQuestion, nodeProperty);
graphRenderer.updateGraphSettings();
}
}
else if (question.answerType.equals(TextAnswer.class)) {
logger.info("Applying labels for text questions");
for (int i =0;i < selectionList.size() ; i++) {
Selection selection = selectionList.get(i);
NodeProperty nodeProperty = new NodeProperty();
nodeProperty.setShape((NodeProperty.NodeShape) table
.getValueAt(i, 1));
nodeProperty.setProperty(NodeProperty.NodePropertyType.Shape);
GraphQuestionSelectionPair graphQuestion = new GraphQuestionSelectionPair(question, selection);
graphRenderer.addQAsettings(graphQuestion, nodeProperty);
graphRenderer.updateGraphSettings();
}
}
}
private void drawPanel() {
this.removeAll();
GroupLayout.SequentialGroup hGroup = layout.createSequentialGroup();
hGroup.add(layout.createParallelGroup().add(questionLabel).add(
questionCombo).add(table.getTableHeader()).add(table).add(
applyButton));
layout.setHorizontalGroup(hGroup);
GroupLayout.SequentialGroup vGroup = layout.createSequentialGroup();
vGroup.add(layout.createParallelGroup(GroupLayout.BASELINE).add(
questionLabel));
vGroup.add(layout.createParallelGroup(GroupLayout.BASELINE).add(
questionCombo));
vGroup.add(layout.createParallelGroup(GroupLayout.BASELINE).add(
table.getTableHeader()));
vGroup.add(layout.createParallelGroup(GroupLayout.BASELINE).add(table));
vGroup.add(layout.createParallelGroup(GroupLayout.BASELINE).add(
applyButton));
layout.setVerticalGroup(vGroup);
}
}