| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2005 IVER T.I. and Generalitat Valenciana.

 *

 * This program 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 2

 * of the License, or (at your option) any later version.

 *

 * This program 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, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

package org.gvsig.symbology.gui.layerproperties;

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.beans.PropertyChangeEvent;

import javax.swing.AbstractCellEditor;

import javax.swing.JCheckBox;

import javax.swing.JComponent;

import javax.swing.JLabel;

import javax.swing.JOptionPane;

import javax.swing.JScrollPane;

import javax.swing.JTable;

import javax.swing.table.DefaultTableColumnModel;

import javax.swing.table.DefaultTableModel;

import javax.swing.table.TableCellEditor;

import javax.swing.table.TableCellRenderer;

import javax.swing.table.TableColumn;

import javax.swing.table.TableColumnModel;

import org.gvsig.gui.beans.swing.GridBagLayoutPanel;

import org.gvsig.gui.beans.swing.JBlank;

import org.gvsig.gui.beans.swing.JButton;

import org.gvsig.gui.beans.swing.celleditors.BooleanTableCellEditor;

import org.gvsig.gui.beans.swing.cellrenderers.BooleanTableCellRenderer;

import org.gvsig.symbology.fmap.labeling.FeatureDependentLabeled;

import org.gvsig.symbology.fmap.labeling.parse.LabelExpressionParser;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.iver.andami.PluginServices;

import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;

import com.iver.cit.gvsig.fmap.rendering.styling.labeling.ILabelingMethod;

import com.iver.cit.gvsig.fmap.rendering.styling.labeling.LabelClass;

public class FeatureDependent extends AbstractLabelingMethodPanel implements ActionListener{

	private static final long serialVersionUID = 5493451803343695650L;

	private static int NAME_FIELD_INDEX = 0;

	private static int PREVIEW_FIELD_INDEX = 1;

	private static int FILTER_FIELD_INDEX = 2;

	private static int LABEL_EXPRESSION_FIELD_INDEX = 3;

	private static int VISIBLE_FIELD_INDEX = 4;

	private JTable tblClasses = null;

	private JButton btnMoveUpClass;

	private JButton btnAddClass;

	private JCheckBox chkLabel;

	private JCheckBox chkDefinePriorities;

	private JScrollPane scrlPan;

	private boolean openEditor = false;

	private JButton btnMoveDownClass;

	private JButton btnDelClass;

	private String[] fieldNames;

	private int[] fieldTypes;

	private JCheckBox getChkDefinePriorities() {

		if (chkDefinePriorities == null) {

			chkDefinePriorities = new JCheckBox(PluginServices.getText(this, "label_priority"));

			chkDefinePriorities.addActionListener(this);

			chkDefinePriorities.setName("CHK_DEFINE_PRIORITIES");

		}

		return chkDefinePriorities;

	}

	private JButton getBtnDelClass() {

		if (btnDelClass == null) {

			btnDelClass = new JButton(PluginServices.getText(this, "delete"));

			btnDelClass.setName("BTNDELCLASS");

			btnDelClass.addActionListener(this);

		}

		return btnDelClass;

	}

	private JButton getBtnAddClass() {

		if (btnAddClass == null) {

			btnAddClass = new JButton(PluginServices.getText(this, "add"));

			btnAddClass.setName("BTNADDCLASS");

			btnAddClass.addActionListener(this);

		}

		return btnAddClass;

	}

	private JButton getBtnMoveUpClass() {

		if (btnMoveUpClass == null) {

			btnMoveUpClass = new JButton(PluginServices.getText(this, "move_up"));

			btnMoveUpClass.setName("BTNMOVEUPCLASS");

			btnMoveUpClass.addActionListener(this);

		}

		return btnMoveUpClass;

	}

	private JButton getBtnMoveDownClass() {

		if (btnMoveDownClass == null) {

			btnMoveDownClass = new JButton(PluginServices.getText(this, "move_down"));

			btnMoveDownClass.setName("BTNMOVEDOWNCLASS");

			btnMoveDownClass.addActionListener(this);

		}

		return btnMoveDownClass;

	}

	@Override

	protected void initializePanel() {

		setLayout(new BorderLayout(10, 10));

		GridBagLayoutPanel aux2 = new GridBagLayoutPanel();

		aux2.addComponent(getBtnAddClass());

		aux2.addComponent(getBtnDelClass());

		aux2.addComponent(new JBlank(10, 10));

		aux2.addComponent(getBtnMoveUpClass());

		aux2.addComponent(getBtnMoveDownClass());

		add(aux2, BorderLayout.EAST);

		add(getChkDefinePriorities(), BorderLayout.NORTH);

		add(getCenterScrl(), BorderLayout.CENTER);

	}

	private class LabelClassCellEditor extends AbstractCellEditor implements TableCellEditor {

		private static final long serialVersionUID = 6399823783851437400L;

		public Component getTableCellEditorComponent(JTable table, Object value, boolean isSelected, int row, int column) {

			if (openEditor) {

				LabelClass oldLc = (LabelClass) getTblClasses().getValueAt(row, PREVIEW_FIELD_INDEX);

				LabelClassProperties lcProp = new LabelClassProperties(fieldNames, fieldTypes);

				oldLc.setTexts(new String[] {oldLc.getName()});

				lcProp.setLabelClass(oldLc);

				PluginServices.getMDIManager()

				.addWindow(lcProp);

				LabelClass newLc = lcProp.getLabelClass();

				LabelClassTableModel m = (LabelClassTableModel) getTblClasses().getModel();

				if (!(oldLc.getName().equals(newLc.getName())))

					if(!checSameLablClassName(m,newLc.getName())){

						m.setValueAt(newLc.getLabelExpression(), row, LABEL_EXPRESSION_FIELD_INDEX);

						m.setValueAt(newLc.getName(), row, NAME_FIELD_INDEX);

						m.setValueAt(newLc, row, PREVIEW_FIELD_INDEX);

						m.setValueAt(newLc.getSQLQuery(), row, FILTER_FIELD_INDEX);

						m.setValueAt(newLc.isVisible(), row, VISIBLE_FIELD_INDEX);

						fireEditingStopped(); //Make the renderer reappear.

					}

					else JOptionPane.showMessageDialog(tblClasses, PluginServices.getText(this, "cannot_exist_two_label_classes_with_the_same_name")+"\n",PluginServices.getText(this,"error"),JOptionPane.ERROR_MESSAGE);

			}

			openEditor = false;

			return null;

		}

		public Object getCellEditorValue() {

			return null;

		}

	}

	private class LabelClassTableModel extends DefaultTableModel {

		private static final long serialVersionUID = -9152998982339430209L;

		private String[] classesTableFieldNames = new String[] {

				PluginServices.getText(this, "name"),

				PluginServices.getText(this, "preview"),

				PluginServices.getText(this, "filter"),

				PluginServices.getText(this, "label_expression"),

				PluginServices.getText(this, "visible"),

		};

		public LabelClassTableModel() {

			super();

			LabelClass[] labelClasses = method.getLabelClasses();

			Object[][] values = new Object[labelClasses.length][classesTableFieldNames.length];

			for (int i = 0; i < values.length; i++) {

				values[i][PREVIEW_FIELD_INDEX] = labelClasses[i];

				values[i][NAME_FIELD_INDEX] = labelClasses[i].getName();

				values[i][FILTER_FIELD_INDEX] = labelClasses[i].getSQLQuery();

				values[i][LABEL_EXPRESSION_FIELD_INDEX] = labelClasses[i].getLabelExpression();

				values[i][VISIBLE_FIELD_INDEX] = labelClasses[i].isVisible();

			}

			setDataVector(values, classesTableFieldNames);

		}

		public String getColumnName(int col) {

			return classesTableFieldNames[col];

		}

		@Override

		public boolean isCellEditable(int row, int column) {

			return true;

		}

		public LabelClass getLabelAtRow(int row) {

			return (LabelClass) getValueAt(row, PREVIEW_FIELD_INDEX);

		}

		public LabelClass[] toLabelClassArray() {

			LabelClass[] classes = new LabelClass[getRowCount()];

			for (int i = 0; i < classes.length; i++) {

				classes[i] = getLabelAtRow(i);

				if (getChkDefinePriorities().isSelected()) {

					classes[i].setPriority(i);

				}

			}

			return classes;

		}

	}

	@Override

	public String getName() {

		return PluginServices.getText(

				this,

		"define_classes_of_features_and_label_each_differently")+".";

	}

	@Override

	public Class<? extends ILabelingMethod> getLabelingMethodClass() {

		return FeatureDependentLabeled.class;

	}

	@Override

	public void fillPanel(ILabelingMethod method, SelectableDataSource dataSource)

	throws ReadDriverException {

		fieldNames = dataSource.getFieldNames();

		fieldTypes = new int[fieldNames.length];

		for (int i = 0; i < fieldTypes.length; i++) {

			fieldTypes[i] = dataSource.getFieldType(i);

		}

		if (method == null) {

			this.method = new FeatureDependentLabeled();

		}

//		getTblClasses().addColumn(new TableColumn());

//		scrlPan.setViewportView(getTblClasses());

		repaint();

	}

	private JScrollPane getCenterScrl() {

		if (scrlPan == null) {

			scrlPan = new JScrollPane(getTblClasses());

//			scrlPan.setViewportView(getTblClasses());

			scrlPan.setPreferredSize(new Dimension(180, 300));

		}

		return scrlPan;

	}

	private JTable getTblClasses() {

		if (tblClasses == null) {

			tblClasses = new JTable();

			tblClasses.setRowHeight(50);

			tblClasses.addMouseListener(new MouseAdapter() {

				int prevRow =-1;

				@Override

				public void mouseClicked(MouseEvent e) {

					int col = getTblClasses().getColumnModel().getColumnIndexAtX(e.getX());

					int row = (int) ((e.getY()-getTblClasses().getLocation().getY()) / getTblClasses().getRowHeight());

					if(!(row > getTblClasses().getRowCount()-1 || col > getTblClasses().getColumnCount()-1))

					{

						openEditor = (row == prevRow && e.getButton() == MouseEvent.BUTTON1 && e.getClickCount() == 2

						) ;

						prevRow = row;

						if (openEditor)

							getTblClasses().editCellAt(row, col);

					}

				}

			});

			tblClasses.setModel(new LabelClassTableModel());

			TableColumnModel cm = tblClasses.getColumnModel();

			tblClasses.getColumnModel().getColumn(PREVIEW_FIELD_INDEX).setCellRenderer(new TableCellRenderer() {

				public Component getTableCellRendererComponent(JTable table,

						Object value, boolean isSelected, boolean hasFocus,

						int row, int column) {

					LabelClassPreview lcPr = new LabelClassPreview();

					lcPr.setLabelClass((LabelClass) value);

					return lcPr;

				}

			});

			tblClasses.getColumnModel().getColumn(VISIBLE_FIELD_INDEX).setCellRenderer(new BooleanTableCellRenderer(false));

			tblClasses.getColumnModel().getColumn(LABEL_EXPRESSION_FIELD_INDEX).setCellRenderer(new TableCellRenderer() {

				public Component getTableCellRendererComponent(JTable table,

						Object value, boolean isSelected, boolean hasFocus,

						int row, int column) {

					String expr = null;

					if (value != null)

						expr = (String) value;

					if (expr == null)

						expr = LabelExpressionParser.tokenFor(LabelExpressionParser.EOEXPR);

					expr = expr.replaceAll(LabelExpressionParser.tokenFor(LabelExpressionParser.EOFIELD), " | ");

					expr = expr.substring(0, expr.length()-1);

					return new JLabel("<html><p>"+expr+"</p></html>", JLabel.CENTER);

				}

			});

			// the editors

			for (int i = 0; i < tblClasses.getColumnModel().getColumnCount(); i++) {

				if (i!= VISIBLE_FIELD_INDEX) {

					tblClasses.getColumnModel().getColumn(i).setCellEditor(new LabelClassCellEditor());

				} else {

					tblClasses.getColumnModel().getColumn(VISIBLE_FIELD_INDEX).

					setCellEditor(new BooleanTableCellEditor(tblClasses));

				}

			}

			((DefaultTableModel)tblClasses.getModel()).fireTableDataChanged();

		}

		return tblClasses;

	}

	private void swapClass(int classIndex, int targetPos,int numOfElements) {

		LabelClass[] classes = ((LabelClassTableModel) getTblClasses().getModel()).toLabelClassArray();

		LabelClass aux = classes[targetPos];

		classes[targetPos] = classes[classIndex];

		classes[classIndex] = aux;

	}

	public void actionPerformed(ActionEvent e) {

		if (method == null) return;

		JComponent c = (JComponent)e.getSource();

		if (c.equals(btnAddClass)) {

			LabelClass newClass = new LabelClass();

			String name = PluginServices.getText(this, "labeling")+

			String.valueOf(1);

			LabelClassTableModel mod = ((LabelClassTableModel) getTblClasses().getModel());

			int count = 0;

			while(checSameLablClassName(mod,name)){

				count++;

				name = PluginServices.getText(this, "labeling")+

				String.valueOf(count);

			}

			newClass.setName(name);

			mod.addRow(new Object[] {newClass.getName(), newClass, newClass.getSQLQuery(), newClass.getLabelExpression(), newClass.isVisible()});

//			tblClasses.setModel(mod);

		} else if (c.equals(btnDelClass)) {

			LabelClassTableModel mod = ((LabelClassTableModel) getTblClasses().getModel());

			if (mod.getRowCount()>=1) {

				int[] sRows = getTblClasses().getSelectedRows();

				for (int i = sRows.length-1; i >= 0 ; i--) {

					mod.removeRow(sRows[i]);

				}

			}

		} else if (c.equals(chkDefinePriorities)) {

		} else  if (c.equals(chkLabel)) {

			LabelClassTableModel mod = ((LabelClassTableModel) getTblClasses().getModel());

			int[] sRows = getTblClasses().getSelectedRows();

			for (int i = sRows.length-1; i >= 0 ; i--) {

				LabelClass lc = mod.getLabelAtRow(i);

				lc.setVisible(chkLabel.isSelected());

			}

		} else if (c.equals(btnMoveUpClass)) {

			int[] indices = getTblClasses().getSelectedRows();

			if (indices.length>0) {

				int classIndex = indices[0];

				int targetPos = Math.max(0, classIndex-1);

				swapClass(classIndex, targetPos,indices.length);

			}

		} else if (c.equals(btnMoveDownClass)) {

			int[] indices = getTblClasses().getSelectedRows();

			if (indices.length>0) {

				int classIndex = indices[indices.length-1];

				int targetPos = Math.min(getTblClasses().getRowCount()-1, classIndex+1);

				swapClass(classIndex, targetPos,indices.length);

			}

		}else if (c.equals(btnDelClass)) {

			int[] indices = getTblClasses().getSelectedRows();

			if (indices.length>0) {

				int classIndex = indices[0];

				int targetPos = Math.min(getTblClasses().getRowCount(), classIndex);

				swapClass(classIndex, targetPos,indices.length);

			}

		}

		method.setDefinesPriorities(chkDefinePriorities.isSelected());

		method.clearAllClasses();

		LabelClass[] classes = ((LabelClassTableModel)tblClasses.getModel()).toLabelClassArray();

		for (int i = 0; i < classes.length; i++) {

			method.addLabelClass(classes[i]);

		}

	}

	private boolean checSameLablClassName(LabelClassTableModel mod, String name) {

		for (int i = 0; i < mod.getRowCount(); i++) {

			if(name.equals(mod.getLabelAtRow(i).getName()))

				return true;

		}

		return false;

	}

	public void propertyChange(PropertyChangeEvent evt) {

		// TODO Auto-generated method stub

//		throw new Error("Not yet implemented!");

	}

}