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.io.File;

import java.io.IOException;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException;

import org.apache.commons.io.FileUtils;

import org.cresques.cts.IProjection;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.dal.DataStoreNotification;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.FileHelper;

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

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

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

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

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

import org.gvsig.fmap.dal.resource.file.FileResource;

import org.gvsig.fmap.dal.resource.spi.ResourceConsumer;

import org.gvsig.fmap.dal.resource.spi.ResourceProvider;

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

import org.gvsig.metadata.MetadataLocator;

import org.gvsig.metadata.MetadataManager;

import org.gvsig.metadata.exceptions.MetadataException;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.DynObjectEncoder;

import org.gvsig.tools.dynobject.DynObjectManager;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.gvsig.tools.task.TaskStatusManager;

/**

 * Provider for Tile Cache

 *

 * @author dmartinezizquierdo

 *

 */

public class TileCacheStoreProvider extends AbstractTileCacheStoreProvider implements //RasterCacheStoreProvider,

    ResourceConsumer {

    private static final Logger logger = LoggerFactory.getLogger(TileCacheStoreProvider.class);

    public static String NAME = "TileCache";

    public static String DESCRIPTION = "Tile cache provider";

    public static final String METADATA_DEFINITION_NAME = NAME;

    private ResourceProvider resource;

    private final SimpleTaskStatus taskStatus;

    protected static void registerMetadataDefinition() throws MetadataException {

        MetadataManager manager = MetadataLocator.getMetadataManager();

        if (manager.getDefinition(METADATA_DEFINITION_NAME) == null) {

            manager.addDefinition(METADATA_DEFINITION_NAME,

                TileCacheStoreProviderParameters.class.getResourceAsStream("TileCacheMetadata.xml"),

                TileCacheStoreProviderParameters.class.getClassLoader());

        }

    }

    /**

     * @param params

     * @param storeServices

     * @throws InitializeException

     */

    public TileCacheStoreProvider(TileCacheStoreProviderParameters params, DataStoreProviderServices storeServices)

        throws InitializeException {

        super(params, storeServices, FileHelper.newMetadataContainer(METADATA_DEFINITION_NAME));

        TaskStatusManager manager = ToolsLocator.getTaskStatusManager();

        this.taskStatus = manager.createDefaultSimpleTaskStatus("TileCache");

        this.init(params, storeServices);

    }

    protected TileCacheStoreProvider(TileCacheStoreProviderParameters params, DataStoreProviderServices storeServices,

        DynObject metadata) throws InitializeException {

        super(params, storeServices, metadata);

        TaskStatusManager manager = ToolsLocator.getTaskStatusManager();

        this.taskStatus = manager.createDefaultSimpleTaskStatus("TileCache");

        this.init(params, storeServices);

    }

    protected void init(TileCacheStoreProviderParameters params,

        DataStoreProviderServices storeServices) throws InitializeException {

        bandsNumber = ((RasterStoreProviderServices)storeServices).getProvider().getBands();

        if (params == null) {

            throw new InitializeException(

                new NullPointerException("params is null"));

        }

    }

    public TileCacheStoreProviderParameters getParameters() {

        return (TileCacheStoreProviderParameters) super.getParameters();

    }

    @Override

    public ResourceProvider getResource() {

        return this.resource;

    }

    @Override

    public Object getSourceId() {

        File folder = this.getParameters().getFolder();

        if(folder == null){

            folder = this.getParameters().getRootFolder();

        }

        return folder;

    }

    @Override

    public String getProviderName() {

        return NAME;

    }

    @Override

    public boolean closeResourceRequested(ResourceProvider resource) {

        return true;

    }

    @Override

    public void resourceChanged(ResourceProvider resource) {

        this.getStoreServices().notifyChange(DataStoreNotification.RESOURCE_CHANGED, resource);

    }

    @Override

    protected void doDispose() throws BaseException {

        super.doDispose();

        resource.removeConsumer(this);

    }

    @Override

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

        super.apply(provider, crs, query);

        DataStoreParameters params = getParameters();

        File folder = (File) params.getDynValue(TileCacheStoreProviderParameters.TILECACHE_FOLDER_PARAMTER_NAME);

        if(folder == null){

            folder = (File) params.getDynValue(TileCacheStoreProviderParameters.TILECACHE_ROOTFOLDER_PARAMTER_NAME);

            DynObjectManager dynObjectManager = ToolsLocator.getDynObjectManager();

            DynObjectEncoder encoder = dynObjectManager.createSimpleDynObjectEncoder();

            MessageDigest md = null;

            String hex = "";

            try {

                md = java.security.MessageDigest.getInstance("MD5");

                md.update(encoder.encode(provider.getParameters()).getBytes());

                hex = convertToHex(md.digest());

            } catch (NoSuchAlgorithmException e) {

                logger.warn("Can't get a MD5 instance.", e);

            }

            folder = new File(folder.getAbsolutePath() + File.separator + hex+"-"+getName());

            try {

                FileUtils.write(new File(folder,"parameters.txt"), encoder.encode(provider.getParameters()));

            } catch (IOException e) {

                logger.warn("Can't create parameters file of store '"+getName()+"' for the cache.",e);

            }

        }

        resource = this.createResource(

            FileResource.NAME,

            new Object[] { folder.getAbsolutePath() }

        );

        resource.addConsumer(this);

        setTileStructImage(new FileTileCacheStructImage(folder, crs, provider, query));

    }

    /**

     * Converts an array of bytes to hexadecimal string

     *

     * @param data

     * @return

     */

    private String convertToHex(byte[] data) {

        StringBuffer buf = new StringBuffer();

        for (int i = 0; i < data.length; i++) {

            int halfbyte = (data[i] >>> 4) & 0x0F;

            int two_halfs = 0;

            do {

                if ((0 <= halfbyte) && (halfbyte <= 9))

                    buf.append((char) ('0' + halfbyte));

                else

                    buf.append((char) ('a' + (halfbyte - 10)));

                halfbyte = data[i] & 0x0F;

            } while (two_halfs++ < 1);

        }

        return buf.toString();

    }

}