Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2016 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 2

 * 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.raster.tilecache.provider;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import java.util.Set;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

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

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

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

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

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

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

import org.gvsig.fmap.dal.raster.RasterQuery;

import org.gvsig.fmap.dal.raster.spi.AbstractRasterStoreProvider;

import org.gvsig.fmap.dal.raster.spi.RasterCacheStoreProvider;

import org.gvsig.fmap.dal.raster.spi.RasterStoreProvider;

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.raster.lib.buffer.api.BandPageManager;

import org.gvsig.raster.lib.buffer.api.Buffer;

import org.gvsig.raster.lib.buffer.api.BufferDimensions;

import org.gvsig.raster.lib.buffer.api.BufferLocator;

import org.gvsig.raster.lib.buffer.api.BufferManager;

import org.gvsig.raster.lib.buffer.api.NoData;

import org.gvsig.raster.lib.buffer.api.PageManager;

import org.gvsig.raster.lib.buffer.api.TileStruct;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferException;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dispose.impl.AbstractDisposable;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.exception.DynFieldNotFoundException;

import org.gvsig.tools.exception.BaseException;

/**

 * @author fdiaz

 *

 */

public abstract class AbstractTileCacheStoreProvider extends AbstractRasterStoreProvider implements

    RasterCacheStoreProvider {

    private TileCacheStructImage tileStructImage = null;

    protected Envelope envelope = null;

    protected Map<Integer, List<PageManager>> pageManagersPerZoomLevel = null;

    protected RasterStoreProvider innerProvider;

    protected int bandsNumber;

    /**

     * @param params

     * @param storeServices

     * @param metadata

     */

    public AbstractTileCacheStoreProvider(DataStoreParameters params, DataStoreProviderServices storeServices,

        DynObject metadata) {

        super(params, storeServices, metadata);

    }

    @Override

    public Buffer createBuffer(RasterQuery rasterQuery) throws BufferException {

        BufferManager bufferManager = BufferLocator.getBufferManager();

        int[] bandDataTypes;

        List<PageManager> pageManagers = new ArrayList<PageManager>();

        NoData[] bandNoData;

        // FIXME: Ver qu? deber?amos hacer aqu? con la rasterQuery

        // if (rasterQuery != null) {

        // List<BandQuery> bands;

        // bands = rasterQuery.getBands();

        // }

        bandDataTypes = new int[bandsNumber];

        bandNoData = new NoData[bandsNumber];

        double pixelSize = rasterQuery.getPixelSize();

        int zoomLevel = 0;

        if (pixelSize != 0.0) {

            zoomLevel = tileStructImage.getZoomLvlForPixelSize(pixelSize);

        }

        List<PageManager> pageManagersInZoom;

        pageManagersInZoom = getPageManagersInZoom(zoomLevel);

        for (int i = 0; i < tileStructImage.getBandNumber(); i++) {

            bandDataTypes[i] = this.getBandDescriptor(i).getDataType();

            bandNoData[i] = this.getNoData(rasterQuery, i);

            pageManagers.add(pageManagersInZoom.get(i));

        }

        envelope = tileStructImage.getEnvelope();

        Buffer buffer =

            bufferManager.createBuffer(tileStructImage.getRows(zoomLevel), tileStructImage.getColumns(zoomLevel),

                bandDataTypes, bandNoData, (IProjection) innerProvider.getDynValue(DataStore.METADATA_CRS), envelope,

                pageManagers);

        return buffer;

    }

    // @Override

    // public BandInfo getBandInfo(int band) {

    // return innerProvider.getBandInfo(band);

    // }

    @Override

    public int getBands() {

        return bandsNumber;

    }

    @Override

    public DataServerExplorer getExplorer() throws ReadException, ValidateDataParametersException {

        return null;

    }

    @Override

    public void open() throws OpenException {

        if (this.tileStructImage != null) {

            return;

        }

    }

    protected List<PageManager> getPageManagersInZoom(int zoomLevel) {

        List<PageManager> pageManagersInZoom = this.pageManagersPerZoomLevel.get(zoomLevel);

        if (pageManagersInZoom == null) {

            pageManagersInZoom = new ArrayList<PageManager>();

            for (int i = 0; i < tileStructImage.getBandNumber(); i++) {

                pageManagersInZoom.add(new TileCacheBandPageManager(tileStructImage, zoomLevel, i));

            }

            this.pageManagersPerZoomLevel.put(zoomLevel, pageManagersInZoom);

        }

        return pageManagersInZoom;

    }

    @Override

    public Object getDynValue(String name) throws DynFieldNotFoundException {

        if (DataStore.METADATA_ENVELOPE.equalsIgnoreCase(name)) {

            return this.envelope;

        }

        if (DataStore.METADATA_CRS.equalsIgnoreCase(name)) {

            return this.innerProvider.getDynValue(name);

        }

        return super.getDynValue(name);

    }

    @Override

    public TileStruct getTileStruct() {

        return innerProvider.getTileStruct();

    }

    protected void setTileStructImage(TileCacheStructImage tileStructImage) {

        this.tileStructImage = tileStructImage;

        for (int i = 0; i < this.tileStructImage.getBandNumber(); i++) {

            this.getBandDescriptor(i).setDataType(this.tileStructImage.getDataTypes()[i]);

        }

    }

    /**

     * @return TileStructImage

     */

    public TileStructImage getTileStructImage() {

        return tileStructImage;

    }

    @Override

    public void deleteAll() {

        tileStructImage.forgetCache();

    }

    @Override

    public void close() throws CloseException {

        this.tileStructImage = null;

    }

    @Override

    public String getName() {

        return this.innerProvider.getName();

    }

    @Override

    public String getFullName() {

        return this.innerProvider.getFullName();

    }

    @Override

    protected void doDispose() throws BaseException {

        super.doDispose();

        if (this.tileStructImage != null) {

            DisposeUtils.dispose((AbstractDisposable) tileStructImage);

            this.tileStructImage = null;

        }

        Set<Entry<Integer, List<PageManager>>> entrySet = pageManagersPerZoomLevel.entrySet();

        for (Iterator<Entry<Integer, List<PageManager>>> iterator = entrySet.iterator(); iterator.hasNext();) {

            Entry<Integer, List<PageManager>> entry = (Entry<Integer, List<PageManager>>) iterator.next();

            List<PageManager> pageManagers = entry.getValue();

            for (Iterator<PageManager> iterator2 = pageManagers.iterator(); iterator2.hasNext();) {

                PageManager pageManager = (PageManager) iterator2.next();

                DisposeUtils.dispose(pageManager);

                iterator2.remove();

            }

            iterator.remove();

        }

    }

    @Override

    public void apply(RasterStoreProvider provider, IProjection crs, RasterQuery query) throws DataException {

        this.innerProvider = provider;

        pageManagersPerZoomLevel = new HashMap<Integer, List<PageManager>>();

    }

    @Override

    public RasterStoreProvider getInnerProvider() {

        return innerProvider;

    }

    @Override

    public BufferDimensions getDimensions() throws InitializeException {

        return innerProvider.getDimensions();

    }

    @Override

    public boolean isTiled() {

        return true;

    }

}