| gvSIG desktop 1

package com.hardcode.gdbms.driver.csv;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileReader;

import java.io.IOException;

import java.util.ArrayList;

import com.hardcode.driverManager.Driver;

import com.hardcode.gdbms.engine.data.DriverException;

import com.hardcode.gdbms.engine.data.FileDriver;

import com.hardcode.gdbms.engine.values.IntValue;

import com.hardcode.gdbms.engine.values.StringValue;

import com.hardcode.gdbms.engine.values.Value;

/**

 * Driver para ficheros csv, en el que la primera fila se toma como la que

 * define los nombres de los campos

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class CSVDriver implements Driver, FileDriver {

    private BufferedReader reader;

    private ArrayList lineas;

    /**

     * @see com.hardcode.gdbms.driver.Driver#getName()

     */

    public String getName() {

        return "csv";

    }

    /**

     * @see com.hardcode.gdbms.data.DataSource#getFieldName(int)

     */

    public String getFieldName(int fieldId) throws DriverException {

        String[] campos = (String[]) lineas.get(0);

        return campos[fieldId];

    }

    /**

     * @see com.hardcode.gdbms.data.DataSource#getIntFieldValue(int, int)

     */

    public Value getFieldValue(long rowIndex, int fieldId)

        throws DriverException {

        String[] campos = (String[]) lineas.get((int) (rowIndex + 1));

        if (fieldId == 0) {

            IntValue value = new IntValue();

            value.setValue(Integer.parseInt(campos[fieldId]));

            return value;

        }

        StringValue value = new StringValue();

        value.setValue(campos[fieldId]);

        return value;

    }

    /**

     * @see com.hardcode.gdbms.data.DataSource#getFieldCount()

     */

    public int getFieldCount() throws DriverException {

        String[] campos = (String[]) lineas.get(0);

        return campos.length;

    }

    /**

     * @see com.hardcode.gdbms.data.DataSource#open(java.io.File)

     */

    public void open(File file) throws IOException {

        reader = new BufferedReader(new FileReader(file));

        lineas = new ArrayList();

        String aux;

        while ((aux = reader.readLine()) != null) {

            String[] campos = aux.split(";");

            lineas.add(campos);

        }

    }

    /**

     * @see com.hardcode.gdbms.data.DataSource#close()

     */

    public void close() throws IOException {

        reader.close();

    }

    /**

     * @see com.hardcode.gdbms.data.DataSource#getRowCount()

     */

    public long getRowCount() {

        return lineas.size() - 1;

    }

        /**

         * @see com.hardcode.gdbms.engine.data.FileDriver#fileAccepted(java.io.File)

         */

        public boolean fileAccepted(File f) {

                return f.getAbsolutePath().toUpperCase().endsWith("CSV");

        }

}