| gvSIG desktop 1

package com.iver.cit.gvsig.project.documents.table.gui.tablemodel;

import java.awt.Component;

import java.util.Date;

import javax.swing.JOptionPane;

import javax.swing.table.AbstractTableModel;

import org.apache.log4j.Logger;

import org.gvsig.fmap.data.DataException;

import org.gvsig.fmap.data.ReadException;

import org.gvsig.fmap.data.feature.Feature;

import org.gvsig.fmap.data.feature.FeatureCollection;

import org.gvsig.fmap.data.feature.FeatureStore;

import com.iver.andami.PluginServices;

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

/**

 * DOCUMENT ME!

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class DataSourceDataModel extends AbstractTableModel {

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

        //private SelectableDataSource dataSource;

    ProjectTable pt;

    boolean hasAssociatedLayer=false;

    private Feature lastRowEdited;

        private int lastNumRow=-1;

        private String order;

        private FeatureCollection featureCollection = null;

    /**

     * Crea un nuevo DataSourceDataModel.

     *

     * @param pt DOCUMENT ME!

     */

    public DataSourceDataModel(ProjectTable pt) {

        this.pt = pt;

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

                hasAssociatedLayer=true;

        }

        try {

                        featureCollection=(FeatureCollection)pt.getModel().getDataCollection(pt.getModel().getDefaultFeatureType(),null,order);

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        //try {

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

        //} catch (DriverLoadException e) {

        // TODO Auto-generated catch block

        //        e.printStackTrace();

        //}

    }

    /**

     * Returns the name of the field.

     *

     * @param col index of field

     *

     * @return Name of field

     */

    public String getColumnName(int col) {

//            if (col==0)

//                    return " ";

//            col--;

            if (col>=pt.getMapping().length) {

                    return null;

            }

            int i=pt.getMapping()[col];

            return pt.getAliases()[i];

    }

    /**

     * Returns the number of fields.

     *

     * @return number of fields

     */

    public int getColumnCount() {

        return pt.getMapping().length;

    }

    /**

     * Returns number of rows.

     *

     * @return number of rows.

     */

    public int getRowCount() {

        try {

                FeatureStore fs = pt.getModel();

            return ((FeatureCollection)fs.getDataCollection()).size();

        } catch (ReadException e) {

            return 0;

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @param row DOCUMENT ME!

     * @param col DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public Object getValueAt(int row, int col) {

//            FeatureStore fs=pt.getModelo();

//            if (hasAssociatedLayer) {

//                    Object stoppingEditing =((FLayer)pt.getAssociatedTable()).getProperty("stoppingEditing");

//

//                    if (stoppingEditing != null && ((Boolean)(stoppingEditing)).booleanValue()) {

//                            return new String("");

//                    }

//            }

//       if (col==0){

//               return new Integer(row);

//       }

//       col--;

       int numRow;

//       long[] orderIndexes=pt.getOrderIndexes();

//       if ( orderIndexes != null && row < orderIndexes.length) {

//               numRow=(int)orderIndexes[row];

//       }else {

               numRow=row;

//       }

       Object obj =null;

////TODO            try {

//                    if (hasAssociatedLayer)

//                            fs.getRecordset().start();

                    Feature rowEdited=null;

                        if (lastNumRow==numRow){

                                rowEdited=lastRowEdited;

                        }else{

                                try {

                                        rowEdited=featureCollection.getFeature(numRow);

                                } catch (ReadException e) {

                                        e.printStackTrace();

                                }

                                lastRowEdited=rowEdited;

                                lastNumRow=numRow;

                        }

                    obj= rowEdited.get(pt.getMapping()[col]);

                        //CHEMA: Problema con las fechas:

                        // Esto deberia solucionarse en el ValueFactory.createValueByType

                        if (obj instanceof Date) {

                                Date date = (Date) obj;

                                String str = "" + (date.getMonth() +1) + "/" + date.getDate()+ "/" +(date.getYear() +1900);

                                obj =  str;

                        }

//                        if (hasAssociatedLayer)

//                                pt.getModelo().getRecordset().stop();

//                        return obj;

////TODO                } catch (ReadException e1) {

////                        logger.error("Error reading data row= "+row+" column= "+col,e1);

////                        obj= new String("");

////                }

//                if (hasAssociatedLayer)

//                        try {

//                                ies.getRecordset().stop();

//                        } catch (ReadDriverException e1) {

//                                logger.error("Error reading data row= "+row+" column= "+col,e1);

//                                obj= ValueFactory.createValue("");

//                        }

                return obj;

    }

    /**

     * DOCUMENT ME!

     *

     * @param rowIndex DOCUMENT ME!

     * @param columnIndex DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean isCellEditable(int rowIndex, int columnIndex) {

//            if (columnIndex==0)

//            return false;

//            try {

//                        if (pt.getModelo().getFieldType(columnIndex)==Types.STRUCT)

//                                return true;

//                } catch (ReadDriverException e) {

//                        e.printStackTrace();

//                }

            return pt.getModel().isEditing();

    }

    /**

     * DOCUMENT ME!

     *

     * @param aValue DOCUMENT ME!

     * @param rowIndex DOCUMENT ME!

     * @param columnIndex DOCUMENT ME!

     *

     * @throws RuntimeException DOCUMENT ME!

     */

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

//             if (columnIndex==0)

//             throw new UnsupportedOperationException("Row Number is a read-only column");

//             columnIndex--;

//            Object v;

             int numFeature;

         if (pt.getOrderIndexes() != null) {

                 numFeature=(int)pt.getOrderIndexes()[rowIndex];

         }else {

                 numFeature=rowIndex;

         }

            if (getValueAt(rowIndex,columnIndex)==null || getValueAt(rowIndex,columnIndex).toString().equals(aValue))

                return;

        try {

                FeatureStore fs = pt.getModel();

//                try{

//                v = ValueFactory.createValueByType(aValue.toString(),

//                    des.getRecordset().getFieldType(columnIndex));

//            } catch (ParseException e) {

//                    if (aValue.equals("")){

//                            v = ValueFactory.createNullValue();

//                    } else {

//                            JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),"formato_del_campo_incorrecto");

//                            return;

//                    }

//                }

                Feature feature = featureCollection.getFeature(numFeature); //.getAttribute(columnIndex);

            feature.editing();

            feature.set(columnIndex,aValue);

            fs.update(feature);

//            Value[] values = row.getAttributes();

//            values[columnIndex] = v;

//            IRow newRow = null;

//

//            if (row.getLinkedRow() instanceof IFeature) {

//                IGeometry geometry = ((DefaultFeature) row.getLinkedRow()).getGeometry();

//                newRow = new DefaultFeature(geometry, values,row.getID());

//            } else {

//                newRow = new DefaultRow(values,row.getID());

//            }

//            des.modifyRow(numRow, newRow,"Editar valor", EditionEvent.ALPHANUMERIC);

        } catch (ReadException e1) {

            throw new RuntimeException(e1);

        } catch (NumberFormatException e) {

                /*        NotificationManager.addError(PluginServices.-getText(this,"numero_incorrecto")+

                                        "\n"+PluginServices.-getText(this,"fila")+" = "+rowIndex+

                                        "\n"+PluginServices.-getText(this,"columna")+ " = "+columnIndex,e);

                */

                } catch (DataException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

    }

    public void setOrder(String s){

            order=s;

    }

}