Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.sextante.app.extension.core;

import org.gvsig.app.project.ProjectManager;

import org.gvsig.app.project.documents.table.TableDocument;

import org.gvsig.app.project.documents.table.TableManager;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureSet;

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

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.tools.dispose.DisposeUtils;

import es.unex.sextante.dataObjects.AbstractTable;

import es.unex.sextante.dataObjects.IRecordsetIterator;

public class gvTable extends AbstractTable {

        private Object m_BaseDataObject;

        private String m_sName;

        private String m_sFilename;

        private FeatureSet featureSet;

        private FeatureType featureType;

        private FeatureStore featureStore;

        public String getName() {

                if (m_BaseDataObject instanceof TableDocument){

                        TableDocument table = (TableDocument) m_BaseDataObject;

                        return table.getName();

                }

                else{

                        return m_sName;

                }

        }

        public void create(Object obj) {

                if (obj instanceof TableDocument){

                        try {

                                featureStore =((TableDocument)obj).getStore();

                                featureSet = featureStore.getFeatureSet(((TableDocument)obj).getQuery());

                                featureType = featureStore.getDefaultFeatureType();

                        } catch (DataException e) {

                                e.printStackTrace();

                        }

                }

        }

        public void create(String sName, String sFilename, Class[] types, String[] sFields) {

//                m_sFilename = sFilename;

//                TableMemoryDriver table = new TableMemoryDriver(sFields, DataTools.getgvSIGTypes(types));

//                m_sName = sName;

//                m_BaseDataObject = table;

        }

        public void create(String sName, Object inputParams, Class[] types, String[] sFields) {

        }

        public void addRecord(Object[] record) {

//                if (m_BaseDataObject instanceof TableMemoryDriver) {

//                        TableMemoryDriver table = (TableMemoryDriver) m_BaseDataObject;

//                        table.addRow(DataTools.getGVSIGValues(record));

//                }

        }

        public IRecordsetIterator iterator() {

                try {

                        return new gvRecordsetIterator(featureSet);

                } catch (DataException e) {

                        throw new RuntimeException(e);

                }

        }

        public String getFieldName(int i) {

                return ((FeatureAttributeDescriptor)featureType.get(i)).getName();

        }

        public Class getFieldType(int i) {

                return DataTools.getTypeClass(((FeatureAttributeDescriptor) featureType

                                .get(i)).getDataType().getType());

        }

        public int getFieldCount() {

                return featureType.size();

        }

        public long getRecordCount() {

                try {

                        return featureSet.getSize();

                } catch (DataException e) {

                        e.printStackTrace();

                }

                return 0;

        }

        public void postProcess() {

                try {

                        TableDocument table = (TableDocument) ProjectManager.getInstance().createDocument(

                                        TableManager.TYPENAME, 

                                        m_sName

                        );

                        table.setStore( featureStore);

                } catch (Exception e) {

                        return;

                }

                TableDocument table = FileTools.openTable(m_sFilename, m_sName);

                create(table);

        }

        public void open() {

        }

        public void close() {

                DisposeUtils.dispose(featureSet);

        }

        public String getFilename() {

                return m_sFilename;

        }

        public void setName(String name) {

                if (m_BaseDataObject instanceof TableDocument){

                        TableDocument table = (TableDocument) m_BaseDataObject;

                        table.setName(name);

                }

                else{

                        m_sName = name;

                }

        }

        public Object getBaseDataObject() {

                return m_BaseDataObject;

        }

}