| gvSIG desktop 1

/*

 * Created on 16-feb-2004

 *

 * To change the template for this generated file go to

 * Window>Preferences>Java>Code Generation>Code and Comments

 */

package com.iver.cit.gvsig.fmap.drivers.shp;

/**

 */

import java.io.File;

import java.io.FileInputStream;

import java.io.InputStream;

import java.util.Calendar;

import java.util.zip.GZIPInputStream;

/**

 * Class to read and write data to a dbase III format file. Creation date:

 * (5/15/2001 5:15:13 PM)

 */

public class DbaseFile {

    // verbose output flag

    private static boolean ourVerbose = false;

    // Header information for the DBase File

    private DbaseFileHeader myHeader;

    // Record Information.

    // private Vector myRecordVect = new Vector();

    private LERandomAccessFile m_fichDbf;

    // The file name to read or write.

    private String myFileName;

    // Convenient place to store the field names.

    private String[] myFieldNames;

    // notify about warnings.

    private boolean myWarning = true;

    /**

     * DbaseFile constructor comment.

     */

    public DbaseFile() {

        super();

    }

    /**

     * DbaseFile constructor comment.

     *

     * @param inFileName DOCUMENT ME!

     */

    public DbaseFile(String inFileName) {

        super();

        setFileName(inFileName);

    }

    /**

     * Returns the array of field names.

     *

     * @return DOCUMENT ME!

     */

    public String[] getFieldNames() {

        return myFieldNames;

    }

    /**

     * Retrieves the file name to read or write.

     *

     * @return DOCUMENT ME!

     */

    public String getFileName() {

        return myFileName;

    }

    // Retrieve number of records in the DbaseFile

    public int getNumRecords() {

        return myHeader.getNumRecords();

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public int getNumFields() {

        return myHeader.getNumFields();

    }

    // Retrieve the record at the given index

    public Object[] getRecord(long inIndex) {

        long nRecordOffset = (myHeader.getRecordLength() * inIndex) +

            myHeader.getHeaderLength();

        // retrieve the record length

        int tempNumFields = myHeader.getNumFields();

        // storage for the actual values

        Object[] tempRow = new Object[tempNumFields];

        try {

            m_fichDbf.seek(nRecordOffset);

            // read the deleted flag

            char tempDeleted = (char) m_fichDbf.readByte();

            // read the record length

            int tempRecordLength = 1; // for the deleted character just read.

            // read the Fields

            for (int j = 0; j < tempNumFields; j++) {

                // find the length of the field.

                int tempFieldLength = myHeader.getFieldLength(j);

                tempRecordLength = tempRecordLength + tempFieldLength;

                // find the field type

                char tempFieldType = myHeader.getFieldType(j);

                //System.out.print("Reading Name="+myHeader.getFieldName(j)+" Type="+tempFieldType +" Length="+tempFieldLength);

                // read the data.

                Object tempObject = null;

                switch (tempFieldType) {

                case 'L': // logical data type, one character (T,t,F,f,Y,y,N,n)

                    char tempChar = (char) m_fichDbf.readByte();

                    if ((tempChar == 'T') || (tempChar == 't') ||

                            (tempChar == 'Y') || (tempChar == 'y')) {

                        tempObject = new Boolean(true);

                    } else {

                        tempObject = new Boolean(false);

                    }

                    break;

                case 'C': // character record.

                    byte[] sbuffer = new byte[tempFieldLength];

                    m_fichDbf.readFully(sbuffer);

                    tempObject = new String(sbuffer, "ISO-8859-1").trim();

                    break;

                case 'D': // date data type.

                    byte[] dbuffer = new byte[8];

                    m_fichDbf.readFully(dbuffer);

                    String tempString = new String(dbuffer, 0, 4);

                    try {

                        int tempYear = Integer.parseInt(tempString);

                        tempString = new String(dbuffer, 4, 2);

                        int tempMonth = Integer.parseInt(tempString) - 1;

                        tempString = new String(dbuffer, 6, 2);

                        int tempDay = Integer.parseInt(tempString);

                        Calendar c = Calendar.getInstance();

                        c.set(Calendar.YEAR, tempYear);

                        c.set(Calendar.MONTH, tempMonth);

                        c.set(Calendar.DAY_OF_MONTH, tempDay);

                        tempObject = c.getTime();

                    } catch (NumberFormatException e) {

                    }

                    break;

                case 'M': // memo field.

                    byte[] mbuffer = new byte[10];

                    m_fichDbf.readFully(mbuffer);

                    break;

                case 'N': // number

                case 'F': // floating point number

                    byte[] fbuffer = new byte[tempFieldLength];

                    m_fichDbf.readFully(fbuffer);

                    try {

                        tempString = new String(fbuffer);

                        tempObject = Double.valueOf(tempString.trim());

                    } catch (NumberFormatException e) {

                    }

                    break;

                default:

                    byte[] defbuffer = new byte[tempFieldLength];

                    m_fichDbf.readFully(defbuffer);

                    System.out.println("Do not know how to parse Field type " +

                        tempFieldType);

                }

                tempRow[j] = tempObject;

                //                                System.out.println(" Data="+tempObject);

            }

            // ensure that the full record has been read.

            if (tempRecordLength < myHeader.getRecordLength()) {

                byte[] tempbuff = new byte[myHeader.getRecordLength() -

                    tempRecordLength];

                m_fichDbf.readFully(tempbuff);

                /* if (tempTelling){

                        System.out.println("DBF File has "+(myHeader.getRecordLength()-tempRecordLength)+" extra bytes per record");

                        tempTelling = false;

                } */

            }

        } catch (Exception e) {

            e.printStackTrace();

        }

        return tempRow;

    }

    /**

     * Retrieve the name of the given column.

     *

     * @param inIndex DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public String getFieldName(int inIndex) {

        return myHeader.getFieldName(inIndex);

    }

    /**

     * Retrieve the type of the given column.

     *

     * @param inIndex DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public char getFieldType(int inIndex) {

        return myHeader.getFieldType(inIndex);

    }

    /**

     * Retrieve the length of the given column.

     *

     * @param inIndex DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public int getFieldLength(int inIndex) {

        return myHeader.getFieldLength(inIndex);

    }

        /**

         * Retrieve the value of the given column as string.

         *

         * @param idField DOCUMENT ME!

         * @param idRecord DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         */

        public Object getFieldValueAsObject(int idField, long idRecord) {

                Object[] tmpReg = getRecord(idRecord);

                return tmpReg[idField];

        }

    /**

     * DOCUMENT ME!

     *

     * @param idField DOCUMENT ME!

     * @param idRecord DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public double getFieldValueAsDouble(int idField, int idRecord) {

        Object[] tmpReg = getRecord(idRecord);

        return (double) Double.parseDouble(tmpReg[idField].toString());

    }

    /**

     * Retrieve the location of the decimal point.

     *

     * @param inIndex DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public int getFieldDecimalLength(int inIndex) {

        return myHeader.getFieldDecimalCount(inIndex);

    }

    /**

     * read the DBF file into memory.

     *

     * @throws Exception DOCUMENT ME!

     */

    public void Open() throws Exception {

        // create an input stream from the file

        if (myFileName == null) {

            throw new Exception("No file name specified");

        }

        InputStream fin = null;

        // check to see if it is a compressed file

        if (myFileName.toUpperCase().endsWith(".GZ")) {

            FileInputStream fileIn = new FileInputStream(myFileName);

            fin = (InputStream) new GZIPInputStream(fileIn);

        } else {

            fin = (InputStream) new FileInputStream(myFileName);

        }

        m_fichDbf = new LERandomAccessFile(myFileName, "r");

        // create the header to contain the header information.

        myHeader = new DbaseFileHeader();

        myHeader.readHeader(m_fichDbf);

        myFieldNames = new String[myHeader.getNumFields()];

        for (int i = 0; i < myHeader.getNumFields(); i++) {

            myFieldNames[i] = myHeader.getFieldName(i).trim();

        }

    }

    /**

     * Removes all data from the dataset

     */

    protected void Close() {

        try {

            m_fichDbf.close();

        } catch (Exception e) {

        }

    }

    /**

     * Set the file name to read or write.

     *

     * @param inFileName DOCUMENT ME!

     */

    public void setFileName(String inFileName) {

        if (inFileName == null) {

            return;

        }

        if (inFileName.length() == 0) {

            return;

        }

        // check if the file exists

        File tempFile = new File(inFileName);

        if (!tempFile.exists()) {

            if (!inFileName.toUpperCase().endsWith(".DBF")) {

                myFileName = inFileName + ".dbf";

            }

        } else {

            myFileName = inFileName;

        }

    }

    /**

     * Print warnings to system.out.

     *

     * @param inWarning DOCUMENT ME!

     */

    public void setWarnings(boolean inWarning) {

        myWarning = inWarning;

        if (myHeader == null) {

            myHeader = new DbaseFileHeader();

        }

        myHeader.setWarnings(inWarning);

    }

    // Set the verbose mode for output from this class

    public static void setVerbose(boolean inVerbose) {

        ourVerbose = inVerbose;

    }

}