Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

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

import org.gvsig.fmap.dal.store.simplereader.SimpleReaderPointAttributeEmulator;

import java.io.IOException;

import java.io.Reader;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DataTypes;

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

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

import org.gvsig.fmap.dal.store.csv.simplereaders.CSVReaderSuperCSV;

import org.gvsig.fmap.dal.store.csv.simplereaders.FixedLenReader;

import org.gvsig.fmap.dal.store.simplereader.*;

import org.gvsig.fmap.dal.store.simplereader.simplereaders.SimpleReader;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.type.GeometryType;

import org.gvsig.tools.locator.LocatorException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author fdiaz

 */

public class CSVFeatureTypeLoader extends SimpleReaderFeatureTypeLoader{

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

    public CSVFeatureTypeLoader(CSVStoreParameters parameters) {

        super(parameters);

    }

    protected String getProviderName() {

        return CSVStoreProvider.NAME;

    }

    protected SimpleReader getSimpleReader(Reader in) throws IOException {

        SimpleReader reader;

        if (CSVStoreParameters.getRawFieldsDefinition(parameters) != null) {

            reader = new FixedLenReader(in, (CSVStoreParameters) parameters);

        } else {

            reader = new CSVReaderSuperCSV(in, (CSVStoreParameters) parameters);

        }

        return reader;

    }

    @Override

    protected boolean isFirstLineHeader() {

        return CSVStoreParameters.isFirstLineHeader(parameters);

    }

    protected String[] getHeaders(SimpleReader reader) throws RuntimeException, IOException {

        String headers[];

        headers = CSVStoreParameters.getHeaders(parameters);

        if (headers == null) {

            if (CSVStoreParameters.isFirstLineHeader(parameters)) {

                headers = reader.getHeader();

                if (headers == null) {

                    if (CSVStoreParameters.getIgnoreErrors(parameters)) {

                        headers = getFixedHeaders(reader.getColumnsCount());

                    } else {

                        String msg = "Can't retrieve header from csv file '"

                                + parameters.getFile()

                                        .getAbsolutePath()

                                + "' and not specified in the parameters.";

                        LOGGER.warn(msg);

                        throw new RuntimeException(msg);

                    }

                }

            } else {

                headers = getFixedHeaders(reader.getColumnsCount());

            }

        } else {

            if (CSVStoreParameters.isFirstLineHeader(parameters)) {

                reader.getHeader(); // Skip and ignore the header of file

            }

        }

        return headers;

    }

    protected void fillFeatureType(EditableFeatureType fType, String headers[], AutomaticDetectionOfTypes.DetectedValue automaticTypes[]) {

        String fullFileName = parameters.getFile()==null? "":parameters.getFile().getAbsolutePath();

        String providerName = this.getProviderName();

        fType.setHasOID(true);

        FieldTypeParser[] fieldTypes = getFieldTypes(headers, automaticTypes);

        fillFeatureType(fType, fieldTypes);

        makeGeometry(fType, providerName, fullFileName);        

        declareGeometryType(fType, fullFileName);

    }

    private void declareGeometryType(EditableFeatureType fType, String fullFileName) throws LocatorException {

        String geometry_column = CSVStoreParameters.getGeometryColumn(parameters);

        if (!StringUtils.isEmpty(geometry_column)) {

            EditableFeatureAttributeDescriptor attr = (EditableFeatureAttributeDescriptor) fType.get(geometry_column);

            if (attr != null ) {

                if( attr.getType() != DataTypes.GEOMETRY ) {

                    attr.setDataType(DataTypes.GEOMETRY);

                }

                GeometryManager geommgr = GeometryLocator.getGeometryManager();

                GeometryType gt;

                try {

                    gt = geommgr.getGeometryType(

                            CSVStoreParameters.getGeometryType(parameters),

                            CSVStoreParameters.getGeometrySubType(parameters)

                    );

                    attr.setGeometryType(gt);

                } catch (Exception e) {

                    LOGGER.warn("Can't set geometry type for the calculated field in CSV file '" + fullFileName + "'.", e);

                }

                fType.setDefaultGeometryAttributeName(geometry_column);

            }

        }

    }

    private void makeGeometry(EditableFeatureType fType, String providerName, String fullFileName) throws LocatorException {

        String[] pointDimensionNames = CSVStoreParameters.getPointDimensionNames(parameters);

        if ( pointDimensionNames != null ) {

            SimpleReaderPointAttributeEmulator emulator = new SimpleReaderPointAttributeEmulator(pointDimensionNames);

            String columnName = CSVStoreParameters.getPointColumnName(parameters);

            if( StringUtils.isBlank(columnName) ) {

                columnName = "geom";

            }

            EditableFeatureAttributeDescriptor attr = fType.add(columnName, DataTypes.GEOMETRY, emulator);

            GeometryManager geommgr = GeometryLocator.getGeometryManager();

            GeometryType gt;

            try {

                if ( emulator.getFieldNames() != null && emulator.getFieldNames().length <= 2 ) {

                    gt = geommgr.getGeometryType(Geometry.TYPES.GEOMETRY, Geometry.SUBTYPES.GEOM2D);

                } else {

                    gt = geommgr.getGeometryType(Geometry.TYPES.GEOMETRY, Geometry.SUBTYPES.GEOM3D);

                }

                attr.setGeometryType(gt);

            } catch (Exception e) {

                LOGGER.warn("Can't set geometry type for the calculated field in '"+providerName+"' file '" + fullFileName + "'.", e);

            }

        }

    }

}