| gvSIG desktop 1

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

import java.io.File;

import java.io.IOException;

import java.io.RandomAccessFile;

import java.nio.channels.FileChannel;

import java.nio.channels.WritableByteChannel;

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

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

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

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

import org.gvsig.data.datastores.vectorial.driver.exception.UnsupportedEncodingDriverException;

import org.gvsig.data.exception.InitializeWriterException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.WriteException;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureType;

public class DBFFeaturesWriter implements IFeaturesWriter {

        private DbaseFileWriter dbfWrite;

        private DbaseFileHeader myHeader;

        private IFeatureType featureType;

        private File dbfFile;

        private int numRows;

        public void setFile(File file){

                dbfFile=file;

        }

        public void init(IDriver driver) {

                DBFDriverParameters parameters=(DBFDriverParameters)driver.getParameters();

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

        }

        public void postProcess() throws OpenException, WriteException {

                try {

                        myHeader.setNumRecords(numRows);

                        dbfWrite = new DbaseFileWriter(myHeader,

                                        (FileChannel) getWriteChannel(dbfFile.getAbsolutePath()));

                } catch (IOException e) {

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

                }

        }

        public void preProcess() throws OpenException, InitializeWriterException {

                myHeader = DbaseFileHeader.createDbaseHeader(featureType);

                try {

                        dbfWrite = new DbaseFileWriter(myHeader,

                                        (FileChannel) getWriteChannel(dbfFile.getAbsolutePath()));

                } catch (IOException e) {

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

                }

        }

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

                dbfWrite.write(feature);

                numRows++;

        }

        public void updateFeatureType(IFeatureType featureType) {

                this.featureType=featureType;

        }

        protected static WritableByteChannel getWriteChannel(String path) throws IOException {

                WritableByteChannel 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);

                        }

                }

                RandomAccessFile raf = new RandomAccessFile(f, "rw");

                channel = raf.getChannel();

                return channel;

        }

        public static void create(DBFDriverParameters parameters, IFeatureType featureType)throws OpenException, InitializeWriterException {

                DbaseFileHeader myHeader = DbaseFileHeader.createDbaseHeader(featureType);

                try {

                        DbaseFileWriter dbfWrite = new DbaseFileWriter(myHeader,

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

                } catch (IOException e) {

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

                }

        }

}