Application: gvSIG desktop

/*

 * To change this license theHeader, 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.simplereaders;

import java.io.File;

import java.io.IOException;

import java.io.Reader;

import java.nio.charset.Charset;

import java.util.List;

import org.apache.commons.io.FilenameUtils;

import org.gvsig.fmap.dal.store.csv.CSVStoreParameters;

import org.gvsig.fmap.dal.store.csv.xml.GfsFile;

import org.gvsig.fmap.dal.store.csv.xml.XMLFileAsList;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author gvSIG Team

 */

public class GMLReader extends AbstractSimpleReader {

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

    private XMLFileAsList gml;

    private int columns;

    private String[] header;

    private int currentElement = 0;

    private final CSVStoreParameters parameters;

    private GfsFile gfs;

    public GMLReader(Reader reader, CSVStoreParameters csvParameters) throws IOException {

        File gmlFile = csvParameters.getFile();

        File gmlIdx = new File(FilenameUtils.removeExtension(gmlFile.getName())+".gmlidx");

        File gfsFile = new File(FilenameUtils.removeExtension(gmlFile.getName())+".gfs");

        gfs = new GfsFile();

        if(gmlIdx.exists()){

            //TODO: Comparar fechas del gfs y gml

            gfs.load(gfsFile);

        } else {

            gfs.fetch(gmlFile);

            gfs.save(gfsFile);

        }

        this.gml = new XMLFileAsList(

                gmlFile, 

                gmlIdx, 

                Charset.forName(CSVStoreParameters.getCharset(csvParameters)), 

                gfs.getBaseElementPath(),

                gfs.getGeometryElementPaths(),

                gfs.getPropertiesPaths()

        );

        this.columns = -1;

        this.header = null;

        this.parameters = csvParameters;

    }

    @Override

    public String[] getHeader() throws IOException {

        if (this.header == null) {

            String[] theHeader = new String[gfs.size()];

            int i = 0;

            for (GfsFile.PropertyDefn item : gfs) {

                theHeader[i++] = item.getName();

            }

            this.header = theHeader;

        }

        return this.header;

    }

    @Override

    public int getColumnsCount() throws IOException {

        if (this.columns <= 0) {

            this.columns = this.getHeader().length;

        }

        return this.columns;

    }

    @Override

    public List<String> read() throws IOException {

        List<String> values = this.read(currentElement);

        if (values == null) {

            return null;

        }

        currentElement += 1;

        return values;

    }

    public List<String> read(int rowNumber) throws IOException {

        if (rowNumber < gml.size()) {

            List<String> values = gml.get(rowNumber);

            return values;

        }

        return null;

    }

    @Override

    public void close() throws IOException {

        this.gml.close();

        this.gml = null;

        this.gfs = null;

    }

    @Override

    public List<String> skip(int lines) throws IOException {

        this.currentElement += lines;

        if (this.currentElement > this.gml.size()) {

            return null;

        }

        return read(this.currentElement);

    }

    @Override

    public int getLine() {

        if (this.gml == null) {

            return 0;

        }

        return this.currentElement;

    }

    @Override

    public List<String> nextRowValues() {

        try {

            return this.read();

        } catch (IOException ex) {

            throw new RuntimeException(ex);

        }

    }

}