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.csv;

import java.util.Iterator;

import java.util.List;

import java.util.NoSuchElementException;

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

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

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

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

import org.gvsig.fmap.dal.feature.spi.AbstractFeatureProviderIterator;

import org.gvsig.fmap.dal.feature.spi.AbstractFeatureSetProvider;

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.tools.exception.BaseException;

/**

 * @author jjdelcerro

 *

 */

public class CSVSetProvider extends AbstractFeatureSetProvider {

    private static class CSVFeatureProvider extends AbstractFeatureProviderLoadedOnDemand {

        protected CSVStoreProvider store;

        public CSVFeatureProvider(CSVStoreProvider store, FeatureType type) {

            super(type);

            this.store = store;

        }

        @Override

        protected void doLoad() {

            try {

                CSVStoreProvider.RowToFeatureTranslator translator = this.store.getRowToFeatureTranslator();

                long index = this.getOID();

                List<String> row = this.store.getRowByIndex(index);

                translator.translate(index, row, this);

            } catch (Exception ex) {

                throw new ReadRuntimeException(getStoreFullName(), this.getOID(), ex);

            }

        }

        @Override

        public Long getOID() {

            return (Long) super.getOID();

        }

        public void setOID(long oid) {

            super.setOID(oid);

        }

        private String getStoreFullName() {

            try {

                return this.store.getFullName();

            } catch (Exception ex) {

                return "unknown";

            }

        }

    }

    private class CSVIterator extends AbstractFeatureProviderIterator {

        protected long index;

        protected FeatureType type;

        protected long count;

        protected final FeatureQuery query;

        protected Iterator spatialIndexIt;

        public CSVIterator(CSVStoreProvider store, FeatureQuery query, FeatureType type,

                long startOn) throws DataException {

            super(store);

            this.index = startOn;

            this.type = type;

            this.query = query;

            this.count = store.getFeatureCount();

            if( this.query!=null ) {

                this.spatialIndexIt = createSpatialIterator(

                        store.getFeatureStore().getDefaultFeatureTypeQuietly(),

                        query,

                        store.getSpatialIndex()

                );

            }

        }

        @Override

        protected CSVStoreProvider getFeatureStoreProvider() {

            return (CSVStoreProvider) super.getFeatureStoreProvider();

        }

        @Override

        protected boolean internalHasNext() {

            if( this.spatialIndexIt != null ) {

                return this.spatialIndexIt.hasNext();

            }

            return this.count > index;

        }

        @Override

        protected FeatureProvider internalNext() {

            CSVFeatureProvider data = new CSVFeatureProvider(getStore(), type);

            if( this.spatialIndexIt != null ) {

                Object oid = this.spatialIndexIt.next();

                data.setOID(oid);

            } else {

                if (index >= this.count) {

                    throw new NoSuchElementException();

                }

                data.setOID(index++);

            }

            return data;

        }

        @Override

        public void remove() {

            throw new UnsupportedOperationException();

        }

        protected void internalDispose() {

            type = null;

        }

        @Override

        protected void doDispose() throws BaseException {

            // Nothing to do

        }

    }

    private class FastCSVIterator extends CSVIterator {

        protected FeatureProvider data;

        public FastCSVIterator(CSVStoreProvider store, FeatureQuery query, FeatureType type,

                long startOn) throws DataException {

            super(store, query, type, startOn);

            this.data = new CSVFeatureProvider(store, type);

        }

        @Override

        protected FeatureProvider internalNext() {

            if( this.spatialIndexIt != null ) {

                Object oid = this.spatialIndexIt.next();

                data.setOID(oid);

            } else {

                if (index >= this.count) {

                    throw new NoSuchElementException();

                }

                data.setOID(index++);

            }

            return data;

        }

        @Override

        protected void internalDispose() {

            super.internalDispose();

            data = null;

        }

    }

    public CSVSetProvider(CSVStoreProvider store, FeatureQuery query,

            FeatureType featureType)

            throws DataException {

        super(store, query, featureType);

    }

    @Override

    protected CSVStoreProvider getStore() {

        return (CSVStoreProvider) super.getStore();

    }

    protected String getStoreFullName() {

        try {

            return this.getStore().getFullName();

        } catch (Throwable th) {

            return "unknown";

        }

    }

    @Override

    public boolean canFilter() {

        return false;

    }

    @Override

    public boolean canIterateFromIndex() {

        return true;

    }

    @Override

    public boolean canOrder() {

        return false;

    }

    @Override

    public long getSize() throws DataException {

        return getStore().getFeatureCount();

    }

    @Override

    public boolean isEmpty() throws DataException {

        return getSize() == 0;

    }

    @Override

    protected void doDispose() throws BaseException {

    }

    @Override

    protected AbstractFeatureProviderIterator createIterator(long index)

            throws DataException {

        return new CSVIterator(

                getStore(),

                getQuery(),

                getProviderFeatureType(),

                index

        );

    }

    @Override

    protected AbstractFeatureProviderIterator createFastIterator(long index)

            throws DataException {

        return new FastCSVIterator(

                getStore(),

                getQuery(),

                getProviderFeatureType(),

                index

        );

    }

}