Application: gvSIG desktop

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

 *

 * Copyright (C) 2004 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 com.iver.cit.gvsig.gui;

import java.awt.Component;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.beans.PropertyChangeEvent;

import java.beans.PropertyChangeListener;

import java.io.File;

import java.io.IOException;

import java.text.DateFormat;

import java.text.NumberFormat;

import java.text.ParseException;

import java.util.BitSet;

import javax.swing.DefaultListSelectionModel;

import javax.swing.JFileChooser;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.table.AbstractTableModel;

import org.apache.log4j.Logger;

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.gdbms.engine.data.driver.DriverException;

import com.hardcode.gdbms.engine.values.NullValue;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.ui.mdiFrame.MainFrame;

import com.iver.andami.ui.mdiManager.SingletonView;

import com.iver.andami.ui.mdiManager.ViewInfo;

import com.iver.cit.gvsig.fmap.FMap;

import com.iver.cit.gvsig.fmap.core.DefaultFeature;

import com.iver.cit.gvsig.fmap.core.IRow;

import com.iver.cit.gvsig.fmap.drivers.DriverIOException;

import com.iver.cit.gvsig.fmap.edition.EditionException;

import com.iver.cit.gvsig.fmap.edition.writers.shp.ShpWriter;

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

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

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

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

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

import com.iver.cit.gvsig.gui.tables.EditionTable;

import com.iver.cit.gvsig.gui.tables.PopupMenu;

import com.iver.cit.gvsig.project.ProjectTable;

import com.iver.utiles.SimpleFileFilter;

import com.iver.utiles.swing.jtable.FieldSelectionEvent;

import com.iver.utiles.swing.jtable.FieldSelectionListener;

import com.iver.utiles.swing.jtable.JTable;

import com.iver.utiles.swing.jtable.SelectionHeaderSupport;

import com.iver.utiles.swing.jtable.TextFieldCellEditor;

/**

 * DOCUMENT ME!

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class Table extends JPanel implements SingletonView,EditionTable {

    private static Logger logger = Logger.getLogger(Table.class.getName());

    private javax.swing.JScrollPane jScrollPane = null;

    private JTable table = null;

    private ProjectTable model = null;

    private DateFormat df = DateFormat.getDateInstance();

    private NumberFormat nf = NumberFormat.getInstance();

    private JLabel jLabelStatus = null;

    private ViewInfo m_viewInfo = null;

    private FMap fmap;

    private boolean updating = false;

    private TableSelectionListener selectionListener = new TableSelectionListener();

    private long numReg=0;

    private SelectionHeaderSupport headerSelectionSupport = new SelectionHeaderSupport();

    private long[] orderIndexes = null;

    private long[] orderIndexesInverted = null;

    //private DataWare dw;

    private IRow[] rowsCopied=null;

    /**

     * This is the default constructor

     */

    public Table() {

        super();

        initialize();

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public ProjectTable getModel() {

        return model;

    }

    public BitSet getSelectedFieldIndices(){

        return headerSelectionSupport.getSelectedColumns();

    }

    public int[] getSelectedRowIndices(){

            return getTable().getSelectedRows();

    }

    private void refreshControls(){

            try{

            MainFrame mF = PluginServices.getMainFrame();

                        if (mF != null)

                PluginServices.getMDIManager().getViewInfo(Table.this).setTitle(PluginServices.getText(this, "Tabla")+": " + model.getName());

                if (model.getAssociatedTable() != null){

                        this.fmap = ((FLayer)model.getAssociatedTable()).getFMap();

                }else{

                        this.fmap = null;

                }

                SelectableDataSource dataSource = (SelectableDataSource)model.getModelo().getRecordset();

                logger.debug("dataSource.start()");

                dataSource.start();

                AbstractTableModel dataModel = new DataSourceDataModel(model);

                        getTable().setModel(dataModel);

                        headerSelectionSupport.setTableHeader(getTable().getTableHeader());

                        headerSelectionSupport.addFieldSelectionListener(new FieldSelectionListener() {

                public void fieldSelected(FieldSelectionEvent e) {

                    if (PluginServices.getMainFrame() != null)

                        PluginServices.getMainFrame().enableControls();

                }

            });

                        model.getModelo().getRecordset().addSelectionListener(selectionListener);

                updateSelection();

            }catch (DriverException e){

                    NotificationManager.addError("No se pudo leer la informaci?n", e);

            } catch (DriverLoadException e) {

                    NotificationManager.addError("No se pudo leer la informaci?n", e);

                }

    }

    /**

     * DOCUMENT ME!

     *

     * @param table DOCUMENT ME!

     */

    public void setModel(ProjectTable table) {

        //SelectableDataSource dataSource;

        model = table;

        //Gesti?n del nombre de la ventana

        model.addPropertyChangeListener(new PropertyChangeListener() {

                        public void propertyChange(PropertyChangeEvent evt) {

                                if (evt.getPropertyName().equals("name")){

                                        PluginServices.getMDIManager().getViewInfo(Table.this).setTitle(PluginServices.getText(this, "Tabla")+": " + (String)evt.getNewValue());

                                }else if (evt.getPropertyName().equals("model")){

                                        refreshControls();

                                }

                        }

                });

        refreshControls();

    }

    /**

     *

     */

    private void updateSelection() {

            updating = true;

            try {

        DefaultListSelectionModel sm = (DefaultListSelectionModel) getTable()

                .getSelectionModel();

        sm.clearSelection();

        BitSet bs = (model.getModelo().getRecordset()).getSelection();

        sm.setValueIsAdjusting(true);

        if (orderIndexes != null){

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

                if (bs.get(i)){

                    sm.addSelectionInterval((int) orderIndexesInverted[i], (int) orderIndexesInverted[i]);

                }

            }

        }else{

            for (int i = bs.nextSetBit(0); i >= 0; i = bs.nextSetBit(i + 1)) {

                sm.addSelectionInterval(i, i);

            }

        }

        sm.setValueIsAdjusting(false);

                try {

                        numReg=model.getModelo().getRowCount();

                } catch (IOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        jLabelStatus.setText(" " + ((SelectableDataSource)model.getModelo().getRecordset()).getSelection().cardinality() + " / " + numReg +" "+

            PluginServices.getText(this, "registros_seleccionados_total") + ".");

                } catch (DriverIOException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } catch (DriverLoadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        updating = false;

    }

    /**

     * This method initializes this

     */

    private void initialize() {

        jLabelStatus = new JLabel();

        setLayout(new java.awt.BorderLayout());

        jLabelStatus.setText("");

        jLabelStatus.setName("");

        jLabelStatus.setPreferredSize(new java.awt.Dimension(100, 18));

        add(getJScrollPane(), java.awt.BorderLayout.CENTER);

        this.add(jLabelStatus, java.awt.BorderLayout.SOUTH);

    }

    /**

     * This method initializes table

     *

     * @return javax.swing.JTable

     */

    public javax.swing.JTable getTable() {

        if (table == null) {

            table = new JTable();

            table.setAutoResizeMode(javax.swing.JTable.AUTO_RESIZE_OFF);

            table.setSelectionModel(new DefaultListSelectionModel());

            table.getTableHeader().addMouseListener(new MouseHandler());

            table.addMouseListener(new MouseRow());

            table.getSelectionModel().addListSelectionListener(new ListSelectionListener() {

                    public void valueChanged(ListSelectionEvent e) {

                            if (updating) return;

                            SelectableDataSource dataModel=null;

                                                try {

                                                        dataModel = Table.this.model.getModelo().getRecordset();

                                                } catch (DriverLoadException e1) {

                                                        // TODO Auto-generated catch block

                                                        e1.printStackTrace();

                                                }

                            DefaultListSelectionModel model = (DefaultListSelectionModel) table.getSelectionModel();

                        BitSet selection = dataModel.getSelection();

                        for (int i = e.getFirstIndex();

                                i <= e.getLastIndex(); i++) {

                            if (orderIndexes != null){

                                selection.set((int) orderIndexes[i], model.isSelectedIndex(i));

                            }else{

                                selection.set(i, model.isSelectedIndex(i));

                            }

                        }

                        if (e.getValueIsAdjusting() == false) {

                                if (fmap != null){

                                        fmap.endAtomicEvent();

                                }

                            dataModel.fireSelectionEvents();

                        }else{

                                if (fmap != null)

                                fmap.beginAtomicEvent();

                        }

                                                jLabelStatus.setText(" " + selection.cardinality() + " / " + numReg + " " +

                                                    PluginServices.getText(this, "registros_seleccionados_total") + ".");

                    }

                });

            TextFieldCellEditor ce = new TextFieldCellEditor();

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

                table.getColumnModel().getColumn(i).setCellEditor(ce);

            }

        }

        return table;

    }

    /**

     * This method initializes jScrollPane

     *

     * @return javax.swing.JScrollPane

     */

    private javax.swing.JScrollPane getJScrollPane() {

        if (jScrollPane == null) {

            jScrollPane = new javax.swing.JScrollPane();

            jScrollPane.setViewportView(getTable());

        }

        return jScrollPane;

    }

    /**

     * @see com.iver.mdiApp.ui.MDIManager.SingletonView#getViewModel()

     */

    public Object getViewModel() {

        return model;

    }

    /**

     * @see com.iver.mdiApp.ui.MDIManager.View#getViewInfo()

     */

    public ViewInfo getViewInfo() {

            ViewInfo m_viewInfo = new ViewInfo(ViewInfo.ICONIFIABLE | ViewInfo.MAXIMIZABLE |

                ViewInfo.RESIZABLE);

            m_viewInfo.setTitle(PluginServices.getText(this, "Tabla"));

            m_viewInfo.setWidth(300);

            m_viewInfo.setHeight(200);

        return m_viewInfo;

    }

    /**

     * DOCUMENT ME!

     *

     * @author Fernando Gonz?lez Cort?s

     */

    public class DataSourceDataModel extends AbstractTableModel {

        //private SelectableDataSource dataSource;

        private ProjectTable pt;

        private String[] aliases;

        private int[] mapping;

        private int dif=0;

        /**

         * Crea un nuevo DataSourceDataModel.

         *

         * @param ds DOCUMENT ME!

         */

        public DataSourceDataModel(ProjectTable pt) {

            this.pt = pt;

            //try {

                                //dataSource = pt.getModelo().getRecordset();

                        //} catch (DriverLoadException e) {

                                // TODO Auto-generated catch block

                        //        e.printStackTrace();

                        //}

            aliases = pt.getAliases();

            mapping = pt.getMapping();

        }

        /**

         * Returns the name of the field.

         *

         * @param col index of field

         *

         * @return Name of field

         */

        public String getColumnName(int col) {

                return aliases[mapping[col]];

        }

        /**

         * Returns the number of fields.

         *

         * @return number of fields

         */

        public int getColumnCount() {

            return mapping.length;

        }

        /**

         * Returns number of rows.

         *

         * @return number of rows.

         */

        public int getRowCount() {

            try {

                return (int) getModel().getModelo().getRowCount();

            } catch (DriverIOException e) {

                    return 0;

                        } catch (IOException e) {

                                return 0;

                        }

        }

        /**

         * DOCUMENT ME!

         *

         * @param row DOCUMENT ME!

         * @param col DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         */

        public Object getValueAt(int row, int col) {

            try {

                //if (dw != null){

                    //return dw.getFieldValue(row, mapping[col]);

                //}else{

                    if (orderIndexes != null){

                        return  getModel().getModelo().getRow(row).getAttribute(col);//getRecordset().getFieldValue(orderIndexes[row], mapping[col]);

                    }else{

                            Object obj=  getModel().getModelo().getRow(row).getAttribute( mapping[col]);//getRecordset().getFieldValue(row, mapping[col]);

                            //if (obj instanceof NullValue)

                            //        return null;

                            return obj;

                    }

                //}

            }catch (DriverIOException e) {

                     return ValueFactory.createValue("").toString();

            } catch (IOException e) {

                     return ValueFactory.createValue("").toString();

                        }

        }

        public boolean isCellEditable(int rowIndex, int columnIndex) {

            return false;//(dw != null);

        }

        public void setValueAt(Object aValue, int rowIndex, int columnIndex) {

            Value v;

            try {

                v = ValueFactory.createValueByType(aValue.toString(), getModel().getModelo().getRecordset().getFieldType(columnIndex));

                //dw.setFieldValue(rowIndex, mapping[columnIndex], v);

            } catch (DriverException e1) {

                throw new RuntimeException(e1);

            } catch (ParseException e) {

                throw new RuntimeException(e);

            } catch (DriverLoadException e) {

                     throw new RuntimeException(e);

                        }

        }

    }

    public class TableSelectionListener implements SelectionListener{

                /**

                 * @see com.iver.cit.gvsig.fmap.layers.LegendListener#selectionChanged(com.iver.cit.gvsig.fmap.layers.LayerEvent)

                 */

                public void selectionChanged(SelectionEvent e) {

                        updateSelection();

                        Table.this.repaint();

                }

    }

    /**

     * @param ds

     * @throws IOException

     */

    public void setOrder(long[] indexes) throws IOException {

        orderIndexes = indexes;

        orderIndexesInverted = new long[orderIndexes.length];

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

            orderIndexesInverted[(int) indexes[i]] = i;

        }

        updating=true;

        ((DataSourceDataModel)getTable().getModel()).fireTableDataChanged();

        updating=false;

        updateSelection();

    }

  /*  public void startEditing() throws DriverException{

       // dw = model.getModelo().getDataWare(DataSourceFactory.DATA_WARE_COHERENT_ROW_ORDER);

       // dw.beginTrans();

    }

*/

    /**

     * @throws DriverException

     *

     */

  /*  public void cancelEditing() throws DriverException {

        dw.rollBackTrans();

        dw = null;

    }

*/

    /**

     * @throws DriverException

     *

     */

  /*  public void stopEditing() throws DriverException {

       // dw.commitTrans();

       // dw = null;

       // model.getModelo().stop();

       // model.getModelo().start();

    }

*/

    /**

     * @return

     */

  /*  public boolean isEditing() {

        return dw != null;

    }

*/

    /**

     * Quita los campos seleccionados

     */

    public void clearSelectedFields()

    {

        headerSelectionSupport.clearSelectedColumns();

    }

    private class MouseHandler extends MouseAdapter {

        public void mouseClicked(MouseEvent e) {

           /* JTableHeader h = (JTableHeader) e.getSource();

            TableColumnModel columnModel = h.getColumnModel();

            int viewColumn = columnModel.getColumnIndexAtX(e.getX());

            int column = columnModel.getColumn(viewColumn).getModelIndex();

            if (column != -1) {

            }*/

        }

    }

    private class MouseRow extends MouseAdapter {

        public void mouseClicked(MouseEvent e) {

                PluginServices.getMainFrame().enableControls();

                if (e.getButton()==MouseEvent.BUTTON3){

                        new PopupMenu(e.getPoint());

                }

        }

    }

    public Value[] getValueRow(int index){

            DataSourceDataModel dsdm=(DataSourceDataModel)getTable().getModel();

            Value [] values=new Value[dsdm.getColumnCount()];

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

                    values[i]=(Value)dsdm.getValueAt(index,i);

            }

            return values;

    }

   /* private void refresh() throws DriverException{

            //dw.commitTrans();

            //model.getModelo().stop();

            //dw.beginTrans();

                //DataSourceDataModel dsdm=(DataSourceDataModel)getTable().getModel();

                //dsdm.fireTableDataChanged();

    }*/

  /*  public void addEmptyRow() throws DriverException{

            ValueCollection valuePK=new ValueCollection();

                valuePK.setValues(new Value[]{ValueFactory.createValue(dw.getRowCount())});

                dw.insertEmptyRow(valuePK);

                refresh();

        }

*/

/*        public void copySelectedRows() throws DriverException{

                int[] sel=getSelectedRowIndices();

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

                        rowsCopied.add(getValueRow(sel[i]));

                }

        }

*/

/*        public void addSelectionToEnd() throws DriverException {

                for (int i=0;i<rowsCopied.size();i++){

                        dw.insertFilledRow((Value[])rowsCopied.get(i));

                }

                refresh();

        }

        */

        /*public void delSelectionRow() throws DriverException{

                int[] sel=getSelectedRowIndices();

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

                        dw.deleteRow(sel[i]);

                }

                refresh();

        }

        */

        /*public boolean isCopy(){

                return !rowsCopied.isEmpty();

        }

        */

        /*

    public void addSelectionToRow() throws DriverException {

                int[] sel=getSelectedRowIndices();

                        dw.insertFilledRow((Value[])rowsCopied.get(i),sel[0]);

                refresh();

        }

        */

        public void startEditing() throws EditionException {

                getModel().getModelo().startEdition();

        }

         public void stopEditing() {

                try {

                    JFileChooser jfc = new JFileChooser();

                    SimpleFileFilter filterShp = new SimpleFileFilter("shp", "Ficheros .shp");

                    jfc.setFileFilter(filterShp);

                    if (jfc.showSaveDialog((Component) PluginServices.getMainFrame()) == JFileChooser.APPROVE_OPTION) {

                                    File newFile = jfc.getSelectedFile();

                                        //VectorialEditableAdapter vea = (VectorialEditableAdapter) layer.getSource();

                                        // File newFile = vea.getDriver().

                            ;

                                    ShpWriter writer = new ShpWriter(newFile,(FLyrVect) getModel().getAssociatedTable());

                            getModel().getModelo().stopEdition(writer);

                            //layer.setSource(getModel().getModelo().getOriginalAdapter());

                            //layer.setEditing(false);

                            //vista.hideConsole();

                        }

                    //}

                } catch (Exception e) {

                         NotificationManager.addError("No se pudo guardar la edici?n", e);

                }

            }

        public void hideColumns(int[] index) {

                // TODO Auto-generated method stub

        }

        public void setUneditableColumns(int[] index) {

                // TODO Auto-generated method stub

        }

        public void setDefaultValues(int[] numColumns, Value[] values) {

                // TODO Auto-generated method stub

        }

        public Value getDefaultValue() {

                // TODO Auto-generated method stub

                return null;

        }

        public int[] getSelectedColumns() {

                return null;

        }

        public void cancelEditing() throws IOException {

                getModel().getModelo().cancelEdition();

        }

        public boolean isEditing() {

                return getModel().getModelo().isEditing();

        }

        public void refresh() {

                updating=true;

        ((DataSourceDataModel)getTable().getModel()).fireTableDataChanged();

        updating=false;

        updateSelection();

        }

        public void addRow(IRow[] rows) throws DriverIOException, IOException {

                if (rows==null){

                        IRow row=new DefaultFeature(null,null);

                        getModel().getModelo().addRow(row);

                }else{

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

                                getModel().getModelo().addRow(rows[i]);

                        }

                }

                refresh();

        }

        public void copyRow() throws DriverIOException, IOException {

                int[] index=getSelectedRowIndices();

                rowsCopied=new IRow[index.length];

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

                        rowsCopied[i]=getModel().getModelo().getRow(index[i]);

                }

        }

        public void cutRow() throws DriverIOException, IOException {

                int[] index=getSelectedRowIndices();

                rowsCopied=new IRow[index.length];

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

                        rowsCopied[i]=getModel().getModelo().getRow(index[i]);

                }

                removeRow();

        }

        public void removeRow() throws DriverIOException, IOException {

                int[] index=getSelectedRowIndices();

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

                        getModel().getModelo().removeRow(index[i]);

                        //getTable().removeRowSelectionInterval(index[i],index[i]+1);

                }

                refresh();

        }

        public void addColumn() {

                // TODO Auto-generated method stub

                refresh();

        }

        public void removeColumn() {

                // TODO Auto-generated method stub

                refresh();

        }

        public boolean isCopied() {

                return (rowsCopied!=null);

        }

        public void pasteRow() throws DriverIOException, IOException {

                addRow(rowsCopied);

        }

}