| gvSIG desktop 1

package org.gvsig.data.datastores.vectorial.driver.dbf;

import java.io.File;

import java.text.DateFormat;

import java.text.ParseException;

import java.util.Date;

import java.util.Locale;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.datastores.vectorial.driver.AbstractDriver;

import org.gvsig.data.datastores.vectorial.driver.IDriverParameters;

import org.gvsig.data.datastores.vectorial.driver.IDriverStoreParameters;

import org.gvsig.data.datastores.vectorial.driver.IFeaturesReader;

import org.gvsig.data.datastores.vectorial.driver.IFeaturesWriter;

import org.gvsig.data.datastores.vectorial.driver.dbf.filedbf.DbaseFile;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.InitializeWriterException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.vectorial.DefaultAttributeDescriptor;

import org.gvsig.data.vectorial.DefaultFeatureType;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureAttributeDescriptor;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureID;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.expansionadapter.FeatureManager;

import org.gvsig.metadata.IMetadata;

import org.gvsig.metadata.IMetadataManager;

import org.gvsig.metadata.MetadataManager;

public class DBFDriver  extends AbstractDriver {

        public static final String DRIVER_NAME = "dbf";

        private static String tempDirectoryPath = System.getProperty("java.io.tmpdir");

        protected static Locale ukLocale = new Locale("en", "UK"); // English, UK version

    protected DbaseFile dbf = new DbaseFile();

    private IFeatureType featureType;

//    private DefaultAttributeDescriptor dad=new DefaultAttributeDescriptor();

        private File file;

        protected IMetadata metadata;

    public DBFDriver() {

        }

        public void init(IDriverParameters parameters) throws InitializeException {

                super.init(parameters);

                DBFDriverParameters DBFParameters=(DBFDriverParameters)parameters;

                file=DBFParameters.getDBFFile();

                dbf.open(file);

                featureType=new DefaultFeatureType();

                    for (int i = 0; i < dbf.getFieldCount(); i++) {

                featureType.add(createFeatureAttribute(i));

            }

            //Para el writer

//                    int aux = (int)(Math.random() * 1000);

//                    fTemp = new File(tempDirectoryPath + "/tmpDbf" + aux + ".dbf");

//                    dbfWriter.setFile(fTemp);

        }

        private IFeatureAttributeDescriptor createFeatureAttribute(int i) {

        char fieldType = dbf.getFieldType(i);

        DefaultAttributeDescriptor dad=new DefaultAttributeDescriptor();

        dad.setOrdinal(i);

        dad.setName(dbf.getFieldName(i));

        dad.setSize(dbf.getFieldLength(i));

        if (fieldType == 'L') {

                dad.setType(IFeatureAttributeDescriptor.TYPE_BOOLEAN);

        } else if ((fieldType == 'F') || (fieldType == 'N')) {

                int precision = dbf.getFieldDecimalLength(i);

                if (precision > 0){

                        dad.setType(IFeatureAttributeDescriptor.TYPE_DOUBLE);

                        dad.setPrecision(precision);

                } else{

                        dad.setType(IFeatureAttributeDescriptor.TYPE_INT);

                }

        } else if (fieldType == 'C') {

                dad.setType(IFeatureAttributeDescriptor.TYPE_STRING);

        } else if (fieldType == 'D') {

                dad.setType(IFeatureAttributeDescriptor.TYPE_DATE);

        } else {

//            throw new BadFieldDriverException(getName(),null,String.valueOf(fieldType));

        }

              return dad;

    }

        public void open() {

        }

        public void close() throws CloseException {

                dbf.close();

        }

        public void dispose() {

                // TODO Apéndice de método generado automáticamente

        }

        public IFeaturesWriter getFeaturesWriter() {

                IFeaturesWriter writer = new DBFFeaturesWriter();

//                writer.init(this);

                return writer;

        }

        public IFeaturesReader getFeatutesReader() {

                IFeaturesReader reader = new DBFFeaturesReader();

                reader.init(this);

                return reader;

        }

        public IFeature getFeatureByID(IFeatureID id) {

                return getFeatureByPosition(featureType,((DBFFeatureID)id).getIndex());

        }

        public IFeature getFeatureByPosition(IFeatureType ft,long position) {

//                 Field Type (C  or M)

//        char cfieldType = fieldTypes[fieldId];

//        int fieldType = getFieldType(fieldId);

                IFeature feature=new DBFFeature(ft,this,position);

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

                        IFeatureAttributeDescriptor descriptor= (IFeatureAttributeDescriptor)ft.get(i);

                        if (dbf.getFieldCount()<=i)

                                return feature;

                        String value=dbf.getStringFieldValue((int) position, i);

                        value=value.trim();

                        String fieldType=descriptor.getDataType();

                        if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_STRING)){

                                feature.set(i,value);

                        }else if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_DOUBLE)){

                                try{

                                        Double.parseDouble(value);

                                }catch (NumberFormatException e) {

                                        value="0";

                                }

                                feature.set(i,Double.parseDouble(value));

                        }else if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_INT)){

                                feature.set(i,Integer.parseInt(value));

                        }else if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_FLOAT)){

                                feature.set(i,Float.parseFloat(value));

                        }else if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_LONG)){

                                feature.set(i,Long.parseLong(value));

                        }else if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_BOOLEAN)){

                                feature.set(i,Boolean.parseBoolean(value));

                        }else if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_BYTE)){

                                feature.set(i,Byte.parseByte(value));

                        }else if (fieldType.equals(IFeatureAttributeDescriptor.TYPE_DATE)){

                                 String year = value.substring(0, 4);

                            String month = value.substring(4, 6);

                            String day = value.substring(6, 8);

                            DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT, ukLocale);

                            /* Calendar c = Calendar.getInstance();

                            c.clear();

                            c.set(Integer.parseInt(year), Integer.parseInt(month),

                                Integer.parseInt(day));

                            c.set(Calendar.MILLISECOND, 0); */

                            String strAux = month + "/" + day + "/" + year;

                            Date dat=null;

                            try {

                                dat = df.parse(strAux);

                            } catch (ParseException e) {

//                                throw new ReadDriverException(getName(),e);

                            }

                                feature.set(i,dat);

                        }else{

                                feature.set(i,null);

                        }

                }

                return feature;

        }

        public IFeatureType getFeatureType() {

                return featureType;

        }

        public IFeatureCollection getDataCollection(FeatureManager featureManager,IFeatureType type, String filter, String order) throws ReadException {

                if (type==null){

                        type=getFeatureType();

                }

                return getFeatutesReader().getFeatures(featureManager,type,filter,order);

        }

        public String getName() {

                return DRIVER_NAME;

        }

        public boolean isEditable() {

                if (file.canWrite()) return true;

                return false;

        }

        public boolean canAlterFeatureType() {

                return true;

        }

        public long getFeatureCount(){

                return dbf.getRecordCount();

        }

        public IMetadata getMetadata() {

                if (metadata==null){

                        IMetadataManager manager=MetadataManager.getManager();

                        metadata=manager.create(DRIVER_NAME);

                }

                return metadata;

        }

}