Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.fmap.dal.store.dbf;

import java.io.File;

import java.io.IOException;

import java.io.RandomAccessFile;

import java.nio.channels.FileChannel;

import java.nio.channels.WritableByteChannel;

import java.util.Objects;

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

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

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

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

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

import org.gvsig.fmap.dal.store.dbf.utils.DbaseCodepage;

import org.gvsig.fmap.dal.store.dbf.utils.DbaseFileHeader;

import org.gvsig.fmap.dal.store.dbf.utils.DbaseFileWriter;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.dal.store.dbf.utils.LogUtils;

public class DBFFeatureWriter {

        private static final Logger LOGGER = LoggerFactory.getLogger(DBFFeatureWriter.class);

        protected FeatureType featuretype;

        private DbaseFileWriter dbfWriter = null;

        protected String name;

        private File dbfFile;

        private DbaseFileHeader myHeader;

        private FileChannel dbfChannel;

        protected DBFFeatureWriter(String name) {

                this.name= name;

        }

        public void begin(DBFStoreParameters storeParameters,

                        FeatureType featureType, long numRows) throws DataException {

                String charset = storeParameters.getEffectiveEncodingName();        

                if (charset==null) {

                        // set a safe encoding in case none has been defined

                        charset = "UTF-8";

                }

                try {

                        myHeader = DbaseFileHeader.fromFeatureType(featureType, charset);

                } catch (DataException e1) {

                        throw new WriteException(this.name, e1);

                }

                dbfFile = storeParameters.getDBFFile();

                // .cpg will be redundant if LDID was already set, but we still

                // want to write it with the hopes of increasing the range of programs

                // that will correctly interpret the charset

                DbaseCodepage cpWriter = new DbaseCodepage(dbfFile);

                cpWriter.write(charset);

                dbfChannel = null;

                try {

                        dbfChannel = (FileChannel) getWriteChannel(dbfFile

                                        .getAbsolutePath());

                } catch (IOException e) {

                        throw new WriteException(this.name, e);

                }

                try{

                        this.dbfWriter = new DbaseFileWriter(myHeader, dbfChannel, true);

                        } catch (InitializeException e) {

                                throw new WriteException(this.name, e);

                        }

                this.featuretype=featureType;

                dbfFile = storeParameters.getDBFFile();

        }

        public void append(Feature feature) throws DataException {

            if(feature.isBroken()){

                throw new IllegalStateException("The feature '"+Objects.toString(feature.getReference())+"' is broken.");

            }

            // TODO use FeatureProvider

            dbfWriter.append(feature);

        }

        public void end() throws DataException {

                dbfWriter.close();

        }

        public void dispose() {

                dbfWriter = null;

        }

        protected static WritableByteChannel getWriteChannel(String path)

        throws IOException {

                WritableByteChannel channel;

                File f = new File(path);

                LogUtils.log(DBFFeatureWriter.class, "begin getWriteChannel");

                LogUtils.log(DBFFeatureWriter.class, "exists("+f.getAbsolutePath()+")");

                if (!f.exists()) {

                        LogUtils.log(DBFFeatureWriter.class, "createNewFile("+f.getAbsolutePath()+")");

                        if (!f.createNewFile()) {

                            LOGGER.warn("Error al crear el fichero "+ f.getAbsolutePath());

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

                        }

                }

                LogUtils.log(DBFFeatureWriter.class, "RandomAccessFile('"+f.getAbsolutePath()+"', \"rw\")");

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

                LogUtils.log(DBFFeatureWriter.class, "getChanel()");

                channel = raf.getChannel();

                LogUtils.log(DBFFeatureWriter.class, "end getWriteChannel");

                return channel;

        }

        protected int getRowCount() {

                return this.myHeader.getNumRecords();

        }

}