| gvSIG desktop 1

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

import java.io.File;

import java.io.IOException;

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

import org.gvsig.data.datastores.vectorial.file.dbf.utils.DbaseFileHeader;

import org.gvsig.data.datastores.vectorial.file.dbf.utils.DbaseFileWriter;

import org.gvsig.data.exception.InitializeWriterException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.exception.UnsupportedEncodingException;

import org.gvsig.data.exception.WriteException;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureStore;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.datasources.common.ICharset;

import org.gvsig.datasources.common.IRandomAccessFile;

import org.gvsig.datasources.common.IRandomFileChannel;

import org.gvsig.datasources.impljme.Impl;

public class DBFFeaturesWriter implements IFeaturesWriter {

        private DbaseFileWriter dbfWrite;

        private DbaseFileHeader myHeader;

        private IFeatureType featureType;

        private File dbfFile;

        protected int numRows;

        private ICharset charset = Impl.getCharsetForName("ISO-8859-1");

        public void setFile(File file){

                dbfFile=file;

        }

        public void init(IFeatureStore store) throws InitializeWriterException {

                DBFStoreParameters parameters=(DBFStoreParameters)store.getParameters();

                dbfFile=((DBFStoreParameters)parameters).getDBFFile();

        }

        public void postProcess() throws OpenException, WriteException {

                try {

                        myHeader.setNumRecords(numRows);

                        dbfWrite = new DbaseFileWriter(myHeader,

                                        getWriteChannel(dbfFile.getAbsolutePath()));

                } catch (IOException e) {

                        throw new WriteException(DBFStore.DATASTORE_NAME,e);

                }

        }

        public void preProcess() throws WriteException, ReadException {

                myHeader = DbaseFileHeader.createDbaseHeader(featureType);

                myHeader.setNumRecords(0);

                numRows = 0;

                try {

                        dbfWrite = new DbaseFileWriter(myHeader,

                                        getWriteChannel(dbfFile.getAbsolutePath()));

                        dbfWrite.setCharset(charset);

                } catch (IOException e) {

                        throw new InitializeWriterException("DBF Feature Writer",e);

                }

        }

        public void insertFeature(IFeature feature) throws WriteException, UnsupportedEncodingException {

                dbfWrite.write(feature);

                numRows++;

        }

        public void updateFeatureType(IFeatureType featureType) {

                this.featureType=featureType;

        }

        protected static IRandomFileChannel getWriteChannel(String path) throws IOException {

                IRandomFileChannel channel;

                File f = new File(path);

                if (!f.exists()) {

//                        System.out.println("Creando fichero " + f.getAbsolutePath());

                        if (!f.createNewFile()) {

                                System.err.print("Error al crear el fichero "

                                                + f.getAbsolutePath());

                                throw new IOException("Cannot create file " + f);

                        }

                }

                IRandomAccessFile raf = Impl.getRandomAccessFile(f, "rw");

                channel = raf.getChannel();

                return channel;

        }

        public static void create(DBFStoreParameters parameters, IFeatureType featureType)throws OpenException, WriteException {

                DbaseFileHeader myHeader = DbaseFileHeader.createDbaseHeader(featureType);

                try {

                        IRandomFileChannel filechannel=

                                getWriteChannel(parameters.getDBFFile().getAbsolutePath());

                        DbaseFileWriter dbfWrite = new DbaseFileWriter(myHeader,filechannel

                                        );

                        filechannel.close();

                } catch (IOException e) {

                        throw new InitializeWriterException("DBF Feature Writer",e);

                }

        }

}