Application: gvSIG desktop » gvSIG raster

/* 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.gdal.provider;

import java.io.File;

import java.io.IOException;

import java.util.List;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

import org.cresques.cts.ICRSFactory;

import org.cresques.cts.IProjection;

import org.gdal.gdal.Dataset;

import org.gdal.gdal.gdal;

import org.gdal.gdalconst.gdalconstConstants;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.crs.CRSFactory;

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

import org.gvsig.tools.dynobject.DynClass_v2;

import org.gvsig.tools.dynobject.Tags;

/**

 * Parameters to create a GDAL Provider

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalFileStoreParameters extends RasterGdalStoreParameters implements

OpenRasterStoreParameters, FilesystemStoreParameters{

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

    /**

     * Definition parameters name used to open a store

     */

    public static final String PARAMETERS_DEFINITION_NAME = "RasterGdalFileStoreParameters";

    /**

     * Parameter name to identify the file

     */

    public static final String FILE_PARAMTER_NAME = "file";

    /**

     * Constructor

     * @param parametersDefinitionName

     * @param name

     */

    public RasterGdalFileStoreParameters(String parametersDefinitionName, String name) {

        super(parametersDefinitionName,name);

    }

    @Override

    public boolean isValid() {

        if (this.getFile() != null){

            try {

                gdal.Open(this.getFile().getAbsolutePath(),gdalconstConstants.GA_ReadOnly);

            } catch (Exception e) {

                return false;

            }

            return true;

        }

        return false;

    }

    @Override

    public File getFile() {

        return (File) this.getDynValue(FILE_PARAMTER_NAME);

    }

    @Override

    public void setFile(File file) {

        this.setDynValue(FILE_PARAMTER_NAME, file);

        if( !file.exists() ) {

            return;

        }

        if (getCRS()==null){

            loadPRJ(file);

        }

        loadWLD(file);

        if (getCRS()==null){

            Dataset gdalDataset = null;

            try{

            gdalDataset = gdal.Open(this.getFile().getAbsolutePath(),gdalconstConstants.GA_ReadOnly);

            String projection=gdalDataset.GetProjection();

            if (StringUtils.isNotEmpty(projection)){

                ICRSFactory crsFactory = CRSFactory.getCRSFactory();

                IProjection crs=crsFactory.get(ICRSFactory.FORMAT_WKT, projection);

                setCRS(crs);

            }

            } finally {

                if(gdalDataset!=null){

                    gdalDataset.delete();

                }

            }

        }

    }

    private void loadPRJ(File file){

        File prjFile = new File(FilenameUtils.removeExtension(file.getAbsolutePath())+".prj");

        if (prjFile.exists()) {

            try {

                String contentFile = FileUtils.readFileToString(prjFile);

                if (StringUtils.isNotEmpty(contentFile)){

                    IProjection crs=CRSFactory.getCRSFactory().get(ICRSFactory.FORMAT_WKT_ESRI, contentFile);

                    setCRS(crs);

                }

            } catch (IOException e) {

                logger.warn("Couldn't read prj file");

            }

        }

    }

    private void loadWLD(File file){

        String extWorldFile = null;

        File wldFile=null;

        Tags paramsTags=((DynClass_v2)this.getDynClass()).getTags();

        if (paramsTags!=null && paramsTags.has(RasterGdalFileStoreParameters.WLD_EXTENSION_TAG_NAME)){

            extWorldFile=(String)paramsTags.get(RasterGdalFileStoreParameters.WLD_EXTENSION_TAG_NAME);

        }

        //If it is a tif or jpg/jpeg we try to use specific extensions, if they exist

        if (StringUtils.isNotEmpty(extWorldFile)){

            wldFile = new File(FilenameUtils.removeExtension(file.getAbsolutePath())+FilenameUtils.EXTENSION_SEPARATOR_STR+extWorldFile);

        }

        //by default we try to find a .wld file

        if (wldFile==null||!wldFile.exists()){

            extWorldFile = ".wld";

            wldFile = new File(FilenameUtils.removeExtension(file.getAbsolutePath())+extWorldFile);

        }

        if (wldFile.exists()) {

            try {

                List<String> lines = FileUtils.readLines(wldFile);

                if (lines!=null && lines.size()==6){

                    setWldParams(lines);

                }

            } catch (IOException e) {

                logger.warn("Couldn't read wld file");

            }

        }

    }

}