Application: gvSIG desktop » gvSIG raster

error_connecting=Error conectando con el servidor

malformed_url=URL mal formada

error_connecting=Error connecting with the server

malformed_url=Malformed URL

org.gvsig.raster.wcs.io.DefaultWCSIOLibrary

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* 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.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2009 IVER T.I   {{Task}}

*/

package org.gvsig.raster.wcs.io;

import java.awt.geom.Point2D;

import java.io.IOException;

import java.io.InputStream;

import java.net.InetAddress;

import java.net.MalformedURLException;

import java.net.URL;

import java.net.URLConnection;

import java.net.UnknownHostException;

import java.util.Hashtable;

import java.util.List;

import org.gvsig.compat.net.ICancellable;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.NewDataStoreParameters;

import org.gvsig.fmap.dal.coverage.exception.RemoteServiceException;

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

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

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

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

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

import org.gvsig.i18n.Messages;

/**

 * Explorer for a WCS server

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class WCSServerExplorer implements DataServerExplorerProvider {

	public static final String          NAME                     = "WCSRemoteServerExplorer";

	private WCSConnector                connector                = null;

	private WCSServerExplorerParameters parameters               = null;

	public WCSServerExplorer(

			WCSServerExplorerParameters parameters,

			DataServerExplorerProviderServices services)

			throws InitializeException {

		this.parameters = parameters;

	}

	/**

	 * Gets the provider's name

	 * @return

	 */

	public String getDataStoreProviderName() {

		return WCSProvider.NAME;

	}

	/*

	 * (non-Javadoc)

	 * @see org.gvsig.fmap.dal.coverage.store.remote.RemoteServerExplorerProvider#getDescription()

	 */

	public String getDescription() {

		return WCSProvider.DESCRIPTION;

	}

	public DataStoreParameters getStoreParameters() {

		DataManager manager = DALLocator.getDataManager();

		WCSDataParametersImpl params = null;

		try {

			params = (WCSDataParametersImpl) manager.createStoreParameters(this.getDataStoreProviderName());

			/*if(WCSProvider.TILED) {

				TileDataParameters tileParams = (TileDataParameters) manager.createStoreParameters("Tile Store");

				tileParams.setDataParameters(params);

				return tileParams;

			} */

		} catch (InitializeException e) {

			e.printStackTrace();

		} catch (ProviderNotRegisteredException e) {

			e.printStackTrace();

		}

		params.setURI(parameters.getHost());

		return params;

	}

	public boolean add(String provider, NewDataStoreParameters parameters,

			boolean overwrite) throws DataException {

		return false;

	}

	public boolean canAdd() {

		return false;

	}

	public boolean canAdd(String storeName) throws DataException {

		return false;

	}

	public NewDataStoreParameters getAddParameters(String storeName)

			throws DataException {

		return null;

	}

	public List getDataStoreProviderNames() {

		return null;

	}

	/*

	 * (non-Javadoc)

	 * @see org.gvsig.fmap.dal.DataServerExplorer#getParameters()

	 */

	public DataServerExplorerParameters getParameters() {

		return parameters;

	}

	public List list() throws DataException {

		return null;

	}

	public List list(int mode) throws DataException {

		return null;

	}

	public void remove(DataStoreParameters parameters) throws DataException {

	}

	public void dispose() {

	}

	public String getProviderName() {

		return null;

	}

	/**

	 * Gets the online resources

	 * @return

	 */

	public Hashtable getOnlineResources() {

		/*if(connector != null) {

			return connector.getOnlineResources();

		}*/

		return null;

	}

	//**********************************************

	//Connector

	//**********************************************

	/**

	 * Connects to the server and throws a getCapabilities. This loads 

	 * the basic information to make requests.

	 * @throws RemoteServiceException 

	 */

	public void connect(ICancellable cancellable) throws RemoteServiceException {

		URL url = null;

		boolean override = false;

		try {

			url = new URL(parameters.getHost());

		} catch (Exception e) {

			throw new RemoteServiceException(Messages.getText("malformed_url"), e);

		}

        try {

        	connector = WCSProvider.getConnectorFromURL(url);

        	if (!connector.connect(override, cancellable))

        		throw new RemoteServiceException(Messages.getText("error_connecting"));

        } catch (IOException e) {

			throw new RemoteServiceException(Messages.getText("error_connecting"), e);

		}

	}

	/**

	 * Checks if the network and host are reachable

	 * @param timeout for the host

	 * @return true if both are reachable and false if they are not

	 */

	public boolean isHostReachable(int timeout) {

		URL url = null;

		try {

			url = new URL(parameters.getHost());

			URLConnection con = url.openConnection();

			if(con == null)

				return false;

			con.connect();

			InputStream stream = con.getInputStream();

			if(stream == null)

				return false;

		} catch (MalformedURLException e) {

			return false;

		} catch (IOException e) {

			return false;

		}

		return true;

	}

	/**

	 * Checks if the network and host are reachable

	 * @return true if both are reachable and false if they are not

	 */

	public boolean isHostReachable() {

		int timeout = 10000;

		return isHostReachable(timeout);

	}

	/**

	 * Returns true if this provider is connected to the server

	 * @return

	 */

	public boolean isConnected() {

		if(connector != null)

			return true;

		return false;

	}

	/**

	 * Gets the description of this service

	 * @return

	 */

	public String getAbstract() {

		if(connector != null)

			return connector.getDescription();

		return null;

	}

	/**

	 * Gets the server title

	 * @return

	 */

	public String getServerType() {

		if (getVersion() == null) 

			return "WCS";

        return "WCS "+ getVersion();

	}

	/**

	 * Gets the protocol supported by the server

	 * @return

	 */

	public String getVersion() {

		if(connector != null) {

			return (connector.getVersion() == null) ? "" : connector.getVersion();

		}

		return null;

	}

	/**

	 * Gets the host URI

	 * @return

	 */

	public String getHost() {

		return parameters.getHost();

	}

	/**

	 * Gets the title

	 * @return

	 */

	public String getTitle() {

		return null;

	}

	public DataServerExplorerProviderServices getServerExplorerProviderServices() {

		// TODO Auto-generated method stub

		return null;

	}

	/*

	 * (non-Javadoc)

	 * @see org.gvsig.fmap.dal.coverage.explorer.WCSServerExplorer#getMaxResolution(java.lang.String)

	 */

	public Point2D getMaxResolution(String layerName) {

		return connector.getMaxResolution(layerName);

	}

	/**

	 * Gets the coverage list

	 * @return

	 */

	public WCSLayerNode[] getCoverageList() {

		return connector.getLayerList();

	}

	/**

	 * Gets a layer searching by its name 

	 * @return

	 */

	public WCSLayerNode getCoverageByName(String name) {

		WCSLayerNode[] list = getCoverageList();

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

			if(list[i].getName().compareTo(name) == 0)

				return list[i];

		}

		return null;

	}

}

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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.

 *

 */

package org.gvsig.raster.wcs.io;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.io.BufferedOutputStream;

import java.io.DataInputStream;

import java.io.DataOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.OutputStreamWriter;

import java.net.ConnectException;

import java.net.HttpURLConnection;

import java.net.URL;

import java.security.KeyManagementException;

import java.security.NoSuchAlgorithmException;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.Iterator;

import java.util.Set;

import java.util.prefs.Preferences;

import javax.net.ssl.HttpsURLConnection;

import javax.net.ssl.SSLContext;

import javax.net.ssl.TrustManager;

import javax.net.ssl.X509TrustManager;

import org.gvsig.compat.net.ICancellable;

import org.gvsig.fmap.dal.coverage.exception.RemoteServiceException;

import org.gvsig.remoteclient.exceptions.ServerErrorException;

import org.gvsig.remoteclient.exceptions.WCSException;

import org.gvsig.remoteclient.utils.BoundaryBox;

import org.gvsig.remoteclient.wcs.WCSClient;

import org.gvsig.remoteclient.wcs.WCSCoverage;

import org.gvsig.remoteclient.wcs.WCSStatus;

import org.gvsig.remoteclient.wcs.WCSCoverage.AxisDescription;

import org.gvsig.remoteclient.wcs.WCSCoverage.RectifiedGrid;

/**

 * Connector between a WCS data provider and a WCSClient. 

 *

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

@SuppressWarnings("unchecked")

public class WCSConnector  {

	private WCSClient                     client;

	private Hashtable                     coverages;

    private WCSLayerNode[]                layerList;

    public WCSConnector(URL url) throws ConnectException, IOException {

    	client = new WCSClient(url.toString());

    }

	/**

	 * Returns the string "WCSDriver", which is the driver's name.

	 *

	 * Devuelve "WCSDriver", el nombre del driver.

	 * @return String

	 */

	public String getName() { 

		return "WCSDriver"; 

	}

	/**

	 * Sets the server that we want to connect to.

	 *

	 * Establece el servidor al que se quiere conectar.

	 *

	 * @param host

	 * @throws IOException

	 */

	public void setHost(String host) throws IOException{

		client = new WCSClient(host);

	}

	/**

	 * Returns a human-readable string containing the server's name.

	 *

	 * Devuelve el nombre legible del servidor devuelto por ?ste.

	 *

	 * @return String

	 */

	public String getLabel() {

		return client.getServiceTitle();

	}

	/**

	 * Returns a string containing the server's WCS version number.

	 *

	 * Devuelve el n?mero de versi?n WCS del servidor

	 *

	 * @return String

	 */

	public String getVersion(){

		return client.getVersion();

	}

	/**

	 * <p>

	 * Returns name and description of the server. It is supposed to be used

	 * as the source of the abstract field in your application's interface.

	 * </p>

	 * <p>

	 * Devuelve nombre y descripci?n (abstract) del servidor.

	 * </p>

	 * @return String

	 */

	public String getDescription(){

		return client.getDescription();

	}

	/**

	 * Returns the layer descriptor for a given coverage name.

	 * @param layerName

	 * @return WCSLayer

	 */

	public WCSLayerNode getLayer(String layerName) {

		getLayerList();

		return (WCSLayerNode) coverages.get(layerName);

	}

	/**

	 * Returns an array of WCSLayer's with the descriptors of all coverages

	 * @return WCSLayer[]

	 */

	public WCSLayerNode[] getLayerList() {

		if (coverages == null || coverages.isEmpty()) {

			// the WCSLayer collection will be built

			coverages = new Hashtable();

			Hashtable wcsCoverages  = client.getCoverageList();

			int sz = wcsCoverages.size();

			// Create an array with the WCSCoverages

			WCSCoverage[] coverageList = new WCSCoverage[sz];

			Iterator it = wcsCoverages.keySet().iterator();

			int i = 0;

			while (it.hasNext()) {

				coverageList[i] = (WCSCoverage) wcsCoverages.get(it.next());

				i++;

			}

			// Create a WCSLayer array from the previous WCSCoverage array

			layerList = new WCSLayerNode[sz];

			for (int j = 0; j < layerList.length; j++) {

				WCSLayerNode lyr = new WCSLayerNode();

				WCSCoverage cov = coverageList[j];

				// name

				lyr.setName(cov.getName());

				// title

				lyr.setTitle(cov.getTitle());

				// description

				lyr.setDescription(cov.getAbstract());

				// srs

				lyr.addAllSrs(cov.getAllSrs());

				// native srs

				lyr.setNativeSRS(cov.getNativeSRS());

				// extents

				Set k = cov.getBBoxes().keySet();

				if (!k.isEmpty()) {

					it = k.iterator();

					while (it.hasNext()) {

						String srs = (String) it.next();

						BoundaryBox bBox = cov.getBbox(srs);

						Rectangle2D r = new Rectangle2D.Double(

								bBox.getXmin(),

								bBox.getYmin(),

								bBox.getXmax() - bBox.getXmin(),

								bBox.getYmax() - bBox.getYmin()

						);

						lyr.addExtent(srs, r);

					}

				}

				// formats

				lyr.setFormats(cov.getFormats());

				// time positions

				lyr.setTimePositions(cov.getTimePositions());

				RectifiedGrid rf = cov.getRectifiedGrid();

				int w = rf.getHighGridEnvelopLimits()[0][0] - rf.getLowGridEnvelopLimits()[0][0];

				int h = rf.getHighGridEnvelopLimits()[0][1] - rf.getLowGridEnvelopLimits()[0][1]; 

				lyr.setWidth(w);

				lyr.setHeight(h);

				// max res

				lyr.setMaxRes(new Point2D.Double(cov.getResX(), cov.getResY()));

				// interpolations

				lyr.setInterpolationMethods(cov.getInterpolationMethods());

				// parameters

				k = cov.axisPool.keySet();

				if (!k.isEmpty()) {

					it = k.iterator();

					while (it.hasNext()) {

						AxisDescription ad = (AxisDescription) cov.axisPool.get(it.next());

						FMapWCSParameter p = new FMapWCSParameter();

						p.setName(ad.getName());

						p.setLabel(ad.getLabel());

						p.setType(ad.getInterval() == null ? FMapWCSParameter.VALUE_LIST : FMapWCSParameter.INTERVAL);

						if (p.getType() == FMapWCSParameter.VALUE_LIST) {

							p.setValueList(ad.getSingleValues());

						} /*else {

							p.setInterval(ad.getInterval());

						}*/

						lyr.addParameter(p);

					}

				}

				layerList[j] = lyr;

				coverages.put(lyr.getName(), lyr);

			}

		}

		return layerList;

	}

	/**

	 * Establishes the connection to the WCS server. Connecting to a WCS is

	 * an abstraction.<br>

	 * <p>

	 * Actually, it sends a GetCapabilities and a general DescribeCoverage

	 * request (not a coverage-specific DescribeCoverage request) to read the

	 * necessary data for building further GetCoverage requests.

	 * </p>

	 * @param override

	 * @throws IOException.

	 */

	public boolean connect(boolean override, ICancellable cancel)

			throws IOException {

		coverages = null;

		setHost(client.getHost());

		return client.connect(override, cancel);

	}

	/**

	 * No close operation is needed since WCS service it is a non-session based

	 * protocol. So, this does nothing and you can omit it.<br>

	 */

	public void close() {

//		connected = false;

	}

	/**

	 * Returns the label of an specific coverage given by the coverage name

	 * @param coverage name (string)

	 * @return string

	 */

	public String getLabel(String coverageName) {

		return client.getLabel(coverageName);

	}

	/**

	 * Returns the coverage's MAX extent from the server.

	 * @return Rectangle2D

	 * @throws IOException

	 */

	public Rectangle2D getFullExtent(String coverageName, String srs)

			throws IOException {

		return client.getExtent(coverageName, srs);

	}

	/**

	 * Returns the max resolution of a specific coverage given by the coverage's name.

	 * @param coverage name (string)

	 * @return double

	 */

	public Point2D getMaxResolution(String coverageName) {

		if (coverages.containsKey(coverageName)) {

			return ((WCSLayerNode) coverages.get(coverageName)).getMaxRes();

		}

		return null;

	}

	/**

	 * Gets the maximum width in pixels of this coverage

	 * @param coverageName

	 * @return

	 */

	public int getWidth(String coverageName) {

		if (coverages.containsKey(coverageName)) {

			return ((WCSLayerNode) coverages.get(coverageName)).getWidth();

		}

		return 0;

	}

	/**

	 * Gets the maximum height in pixels of this coverage

	 * @param coverageName

	 * @return

	 */

	public int getHeight(String coverageName) {

		if (coverages.containsKey(coverageName)) {

			return ((WCSLayerNode) coverages.get(coverageName)).getHeight();

		}

		return 0;

	}

	/**

	 * Returns an ArrayList containing a set of Strings with the coverage's SRSs.

	 * @param coverage name (string)

	 * @return ArrayList

	 */

	public ArrayList getSRSs(String coverageName) {

		if (coverages.containsKey(coverageName)) {

			return ((WCSLayerNode) coverages.get(coverageName)).getSRSs();

		}

		return null;

	}

	/**

	 * Returns a String containing a description of an specific coverage.

	 * @param coverage name (string)

	 * @return string

	 */

	public String getCoverageDescription(String coverageName) {

		if (coverages.containsKey(coverageName)) {

			return ((WCSLayerNode) coverages.get(coverageName)).getDescription();

		}

		return null;

	}

	/**

	 * Returns an ArrayList containing strings for the time positions of an

	 * specific coverage given by the coverage's name.

	 * @param coverage name (string)

	 * @return ArrayList

	 */

	public ArrayList getTimes(String coverageName) {

		if (coverages.containsKey(coverageName)) {

			return ((WCSLayerNode) coverages.get(coverageName)).getTimePositions();

		}

		return null;

	}

	/**

	 * Sends a GetCoverage request to the client.

	 * @param status

	 * @return

	 * @throws WCSException

	 */

	public File getCoverage(WCSStatus status, ICancellable cancel) throws RemoteServiceException {

		try {

			return client.getCoverage(status, cancel);

		} catch (ServerErrorException e) {

			throw new RemoteServiceException(getName(), e);

		} catch (org.gvsig.remoteclient.exceptions.WCSException e) {

			throw new RemoteServiceException(e.getMessage(), e);

		}

	}

	/**

	 * Sends a GetCoverage request to the client.

	 * @param status

	 * @return

	 * @throws WCSException

	 */

	public void getCoverageURL(WCSStatus status, ICancellable cancel, File file) throws RemoteServiceException {

		try {

			URL url = client.getCoverageURL(status, cancel);

			downloadFile(url, file, cancel);

			String exceptionMessage = client.getExceptionMessage(file);

			if(exceptionMessage != null)

				throw new RemoteServiceException(exceptionMessage);

		} catch(IOException e) {

			throw new RemoteServiceException("WMS: error downloading the file. File:" + file.getAbsolutePath() + "...." + e.getMessage());

		} catch (ServerErrorException e) {

			throw new RemoteServiceException("WMS Unexpected server error."+e.getMessage());

		}  catch (org.gvsig.remoteclient.exceptions.WMSException e) {

			throw new RemoteServiceException(e.getMessage());

		}

	}

	public void downloadFile(URL url, File dstFile, ICancellable cancel) throws IOException {

		Preferences prefs = Preferences.userRoot().node( "gvsig.downloader" );

		// by default 1 minute (60000 milliseconds.

		int timeout = prefs.getInt("timeout", 60000);

		DataOutputStream dos;

		DataInputStream is;

		OutputStreamWriter os = null;

		HttpURLConnection connection = null;

		//If the used protocol is HTTPS

		if (url.getProtocol().equals("https")) {

			try {

				disableHttsValidation();

			} catch (KeyManagementException e) {

				e.printStackTrace();

			} catch (NoSuchAlgorithmException e) {

				e.printStackTrace();

			}

		}

		connection = (HttpURLConnection)url.openConnection();

		connection.setConnectTimeout(timeout);

		is = new DataInputStream(url.openStream());

		dos = new DataOutputStream( new BufferedOutputStream(new FileOutputStream(dstFile)));

		byte[] buffer = new byte[1024 * 4];

		long readed = 0;

		for (int i = is.read(buffer); i > 0; i = is.read(buffer)){

			dos.write(buffer, 0, i);

			readed += i;

			if(cancel != null && cancel.isCanceled())

				return;

		}

		if(os != null) {

			os.close();

		}

		dos.close();

		is.close();

		is = null;

		dos = null;

	}

	/**

	 * This method disables the Https certificate validation.

	 * @throws KeyManagementException

	 * @throws NoSuchAlgorithmException

	 */

	private void disableHttsValidation() throws KeyManagementException, NoSuchAlgorithmException{

		// Create a trust manager that does not validate certificate chains

		TrustManager[] trustAllCerts = new TrustManager[] {

				new X509TrustManager() {

					public java.security.cert.X509Certificate[] getAcceptedIssuers() {

						return null;

					}

					public void checkClientTrusted(

							java.security.cert.X509Certificate[] certs, String authType) {

					}

					public void checkServerTrusted(

							java.security.cert.X509Certificate[] certs, String authType) {

					}

				}

		};

		// Install the all-trusting trust manager

		SSLContext sc = SSLContext.getInstance("SSL");

		sc.init(null, trustAllCerts, new java.security.SecureRandom());

		HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory());

	}

}

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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.

 *

 */

package org.gvsig.raster.wcs.io;

import java.awt.geom.AffineTransform;

import java.awt.geom.NoninvertibleTransformException;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.IOException;

import java.net.URL;

import java.util.Hashtable;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

import org.gvsig.fmap.dal.coverage.datastruct.BandList;

import org.gvsig.fmap.dal.coverage.datastruct.Extent;

import org.gvsig.fmap.dal.coverage.exception.BandAccessException;

import org.gvsig.fmap.dal.coverage.exception.BandNotFoundInListException;

import org.gvsig.fmap.dal.coverage.exception.FileNotOpenException;

import org.gvsig.fmap.dal.coverage.exception.InvalidSetViewException;

import org.gvsig.fmap.dal.coverage.exception.NotSupportedExtensionException;

import org.gvsig.fmap.dal.coverage.exception.ProcessInterruptedException;

import org.gvsig.fmap.dal.coverage.exception.RasterDriverException;

import org.gvsig.fmap.dal.coverage.exception.RemoteServiceException;

import org.gvsig.fmap.dal.coverage.store.props.ColorInterpretation;

import org.gvsig.fmap.dal.coverage.store.props.HistogramComputer;

import org.gvsig.fmap.dal.coverage.store.props.Transparency;

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

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

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

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

import org.gvsig.metadata.MetadataLocator;

import org.gvsig.raster.cache.tile.TileCacheLocator;

import org.gvsig.raster.cache.tile.TileCacheManager;

import org.gvsig.raster.cache.tile.exception.TileGettingException;

import org.gvsig.raster.cache.tile.provider.TileListener;

import org.gvsig.raster.cache.tile.provider.TileServer;

import org.gvsig.raster.impl.DefaultRasterManager;

import org.gvsig.raster.impl.buffer.DefaultRasterQuery;

import org.gvsig.raster.impl.datastruct.BandListImpl;

import org.gvsig.raster.impl.datastruct.DatasetBandImpl;

import org.gvsig.raster.impl.datastruct.ExtentImpl;

import org.gvsig.raster.impl.provider.DefaultRasterProvider;

import org.gvsig.raster.impl.provider.RasterProvider;

import org.gvsig.raster.impl.provider.RemoteRasterProvider;

import org.gvsig.raster.impl.store.DefaultStoreFactory;

import org.gvsig.raster.impl.store.properties.DataStoreColorInterpretation;

import org.gvsig.raster.impl.store.properties.DataStoreTransparency;

import org.gvsig.raster.impl.store.properties.RemoteDataStoreStatistics;

import org.gvsig.raster.impl.store.properties.RemoteStoreHistogram;

import org.gvsig.raster.util.DefaultProviderServices;

import org.gvsig.raster.wcs.io.downloader.WCSTileServer;

import org.gvsig.remoteclient.wcs.WCSStatus;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.task.TaskStatus;

/**

 * Clase que representa al driver de acceso a datos de wcs.

 *

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class WCSProvider extends DefaultRasterProvider implements RemoteRasterProvider {

	public static String                NAME                     = "Wcs Store";

	public static String                DESCRIPTION              = "Wcs Raster file";

	public static final String          METADATA_DEFINITION_NAME = "WcsStore";

	private Extent                      viewRequest              = null;

	private static Hashtable<URL, WCSConnector>    

	                                    drivers                  = new Hashtable<URL, WCSConnector> ();

	private boolean                     open                     = false;

	private DataStoreTransparency       fileTransparency         = null;

	private File                        lastRequest              = null;

	private DefaultRasterProvider       lastRequestProvider      = null; 

	public static void register() {

		DataManagerProviderServices dataman = (DataManagerProviderServices) DALLocator.getDataManager();

		if (dataman != null && !dataman.getStoreProviders().contains(NAME)) {

			dataman.registerStoreProvider(NAME,

					WCSProvider.class, WCSDataParametersImpl.class);

		}

		if (!dataman.getExplorerProviders().contains(WCSServerExplorer.NAME)) {

			dataman.registerExplorerProvider(WCSServerExplorer.NAME, WCSServerExplorer.class, WCSServerExplorerParameters.class);

		}

		dataman.registerStoreFactory(NAME, DefaultStoreFactory.class);

	}

	public WCSProvider() throws NotSupportedExtensionException {

		super();

	}

	/**

	 * Constructor. Abre el dataset.

	 * @param proj Proyecci?n

	 * @param fName Nombre del fichero

	 * @throws NotSupportedExtensionException

	 */

	public WCSProvider(String params) throws InitializeException {

		super(params);

		if(params instanceof String) {

			WCSDataParametersImpl p = new WCSDataParametersImpl();

			p.setURI((String)params);

			super.init(p, null, ToolsLocator.getDynObjectManager()

					.createDynObject(

							MetadataLocator.getMetadataManager().getDefinition(

									DataStore.METADATA_DEFINITION_NAME)));

			init(p, null);

		}

	}

	public WCSProvider(WCSDataParametersImpl params,

			DataStoreProviderServices storeServices) throws InitializeException {

		super(params, storeServices, ToolsLocator.getDynObjectManager()

				.createDynObject(

						MetadataLocator.getMetadataManager().getDefinition(

								DataStore.METADATA_DEFINITION_NAME)));

		init(params, storeServices);

	}

	/**

	 * Gets the connector from the URL

	 * @return

	 * @throws RemoteServiceException

	 */

	public WCSConnector getConnector() throws RemoteServiceException {

		WCSDataParametersImpl p = (WCSDataParametersImpl)parameters;

		URL url = null;

		try {

			url = new URL(p.getURI());

		} catch (Exception e) {

			throw new RemoteServiceException("Malformed URL",e);

		}

		try {

			return WCSProvider.getConnectorFromURL(url);

		} catch (IOException e) {

			throw new RemoteServiceException("Error getting the connector",e);

		}

	}

	/**

	 * Crea las referencias al fichero y carga

	 * las estructuras con la informaci?n y los metadatos.

	 * @param proj Proyecci?n

	 * @param param Parametros de carga

	 * @throws NotSupportedExtensionException

	 * @throws RasterDriverException 

	 */

	public void init (DataStoreParameters params,

			DataStoreProviderServices storeServices) throws InitializeException {

		setParam(storeServices, params);

		open = true;

		try {

			loadInitialInfo();

		} catch (RasterDriverException e) {

			throw new InitializeException(e.getMessage(), e);

		}

		stats = new RemoteDataStoreStatistics(this);

	}

	public static final WCSConnector getConnectorFromURL(URL url) throws IOException {

		WCSConnector drv = (WCSConnector) drivers.get(url);

		if (drv == null) {

			drv = new WCSConnector(url);

			drivers.put(url, drv);

		}

		return drv;

	}

	/**

	 * Obtiene el objeto que contiene que contiene la interpretaci?n de

	 * color por banda

	 * @return

	 */

	public ColorInterpretation getColorInterpretation() {

		if(super.getColorInterpretation() == null) {

			ColorInterpretation colorInterpretation = new DataStoreColorInterpretation();

			colorInterpretation.initColorInterpretation(getBandCount());

			if(getBandCount() == 3) {

				colorInterpretation.setColorInterpValue(0, DataStoreColorInterpretation.RED_BAND);

				colorInterpretation.setColorInterpValue(1, DataStoreColorInterpretation.GREEN_BAND);

				colorInterpretation.setColorInterpValue(2, DataStoreColorInterpretation.BLUE_BAND);

			} else {

				for (int i = 0; i < getBandCount(); i++) {

					colorInterpretation.setColorInterpValue(i, DataStoreColorInterpretation.GRAY_BAND);

				}

			}

			setColorInterpretation(colorInterpretation);

		}

		return super.getColorInterpretation();

	}

	/**

	 * Gets WCS parameters

	 * @return

	 */

	public WCSDataParameters getParameters() {

		return (WCSDataParameters)parameters;

	}

	public AffineTransform getAffineTransform() {

		Extent e = getExtent();

		double resolutionX = e.width() / getWidth();

		double resolutionY = e.height() / getHeight();

		ownTransformation = new AffineTransform(

				resolutionX, 

				0, 

				0, 

				-resolutionY, 

				e.getULX() - (resolutionX / 2),

				e.getULY() - (resolutionY / 2));

		externalTransformation = (AffineTransform) ownTransformation.clone();

		return ownTransformation;

	}

	/**

	 * Calcula el extent en coordenadas del mundo real

	 * @return Extent

	 */

	public Extent getExtent() {

		WCSDataParametersImpl p = (WCSDataParametersImpl)parameters;

		try {

			Rectangle2D r = getConnector().getFullExtent(p.getCoverageName(), p.getSRSCode());

			if(r == null)

				return null;

			return new ExtentImpl(r.getX(),  r.getY() + r.getHeight(), r.getX() + r.getWidth(), r.getY());

		} catch (RemoteServiceException e1) {

			e1.printStackTrace();

		} catch (IOException e) {

			e.printStackTrace();

		}

		return null;

	}

	public Rectangle2D getLayerExtent(String layerName, String srs) throws RemoteServiceException {

		return null;

	}

	public RasterProvider load() {

		return this;

	}

	public boolean isOpen() {

		return open;

	}

	public void close() {

		open = false;

	}

	public Transparency getTransparency() {

		if(fileTransparency == null)

			fileTransparency = new DataStoreTransparency();

		return fileTransparency;

	}

	public String translateFileName(String fileName) {

		return fileName;

	}

	public void setView(Extent e) {

		viewRequest = e;

	}

	public Extent getView() {

		return viewRequest;

	}

	public double getWidth() {

		WCSDataParametersImpl p = (WCSDataParametersImpl)parameters;

		try {

			return getConnector().getWidth(p.getCoverageName());

		} catch (RemoteServiceException e) {

			e.printStackTrace();

		}

		return 0;

	}

	public double getHeight() {

		WCSDataParametersImpl p = (WCSDataParametersImpl)parameters;

		try {

			return getConnector().getHeight(p.getCoverageName());

		} catch (RemoteServiceException e) {

			e.printStackTrace();

		}

		return 0;

	}

	public Object readCompleteLine(int line, int band)

		throws InvalidSetViewException, FileNotOpenException, RasterDriverException {

		return null;

	}

	/**

	 * When the remote layer has fixed size this method downloads the file and return its reference. 

	 * File layer has in the long side FIXED_SIZE pixels and the bounding box is complete. This file could be

	 * useful to build an histogram or calculate statistics. This represents a sample of data.

	 * @return

	 * @throws RasterDriverException

	 */

	public File getFileLayer() throws RasterDriverException {

		Extent e = getExtent();

		Rectangle2D bBox = new Rectangle2D.Double(e.getULX(), e.getLRY(), e.width(), e.height());

		WCSStatus wcsStatus = loadWCSStatus(bBox);

		return downloadFile(wcsStatus, e.getULX(), e.getULY(), e.getLRX(), e.getLRY(), 

				(int)getWidth(), (int)getHeight());

	}

	/**

	 * Reads a complete block of data and returns an tridimensional array of the right type. This function is useful

	 * to read a file very fast without setting a view. In a WCS service when the size is fixed then it will read the

	 * entire image but when the source hasn't pixel size it will read a sample of data. This set of data will have

	 * the size defined in FIXED_SIZE. 

	 * 

	 * @param pos Posici?n donde se empieza  a leer

	 * @param blockHeight Altura m?xima del bloque leido

	 * @return Object que es un array tridimendional del tipo de datos del raster. (Bandas X Filas X Columnas)

	 * @throws InvalidSetViewException

	 * @throws FileNotOpenException

	 * @throws RasterDriverException

	 */

	public Object readBlock(int pos, int blockHeight, double scale) 

	throws InvalidSetViewException, FileNotOpenException, RasterDriverException, ProcessInterruptedException {

		File lastFile = getFileLayer();

		BandList bandList = new BandListImpl();

		for (int i = 0; i < 3; i++) {

			try {

				bandList.addBand(new DatasetBandImpl(lastFile.getPath(), pos, Buffer.TYPE_BYTE, 3));

			} catch (BandNotFoundInListException e1) {

			}

		}

		bandList.setDrawableBands(new int[]{0, 1, 2});

		try {

			lastRequestProvider = openLastRequest();

			return lastRequestProvider.readBlock(pos, blockHeight, scale);

		} catch (ProviderNotRegisteredException exc) {

			throw new RasterDriverException("Error building GdalDriver", exc);

		} catch (InitializeException exc) {

			throw new RasterDriverException("Error building GdalDriver", exc);

		}

	}

	public double getLastRequestHeight() throws RasterDriverException {

		if(lastRequestProvider == null) {

			try {

				lastRequestProvider = openLastRequest();

			} catch (ProviderNotRegisteredException e) {

				throw new RasterDriverException("Error building GdalDriver", e);

			} catch (InitializeException e) {

				throw new RasterDriverException("Error building GdalDriver", e);

			}

		}

		return lastRequestProvider.getHeight();

	}

	public double getLastRequestWidth() throws RasterDriverException {

		if(lastRequestProvider == null) {

			try {

				lastRequestProvider = openLastRequest();

			} catch (ProviderNotRegisteredException e) {

				throw new RasterDriverException("Error building GdalDriver", e);

			} catch (InitializeException e) {

				throw new RasterDriverException("Error building GdalDriver", e);

			}

		}

		return lastRequestProvider.getWidth();

	}

	public File getLastRequest() {

		return lastRequest;

	}

	public Buffer getBufferLastRequest() throws ProcessInterruptedException, RasterDriverException {

		try {

			lastRequestProvider = openLastRequest();

			BandList bandList = new BandListImpl();

			for (int i = 0; i < lastRequestProvider.getBandCount(); i++) {

				try {

					bandList.addBand(new DatasetBandImpl(lastRequest.getPath(), i, Buffer.TYPE_BYTE, 3));

				} catch (BandNotFoundInListException e1) {

				}

			}

			bandList.setDrawableBands(new int[]{0, 1, 2});

			Buffer newBuffer = DefaultRasterManager.getInstance().createBuffer(lastRequestProvider.getDataType()[0], (int)lastRequestProvider.getWidth(), (int)lastRequestProvider.getHeight(), lastRequestProvider.getBandCount(), true);

			DefaultRasterQuery q = (DefaultRasterQuery)RasterLocator.getManager().createQuery();

			q.setAreaOfInterest(lastRequestProvider.getExtent(), 

					(int)lastRequestProvider.getWidth(), 

					(int)lastRequestProvider.getHeight());

			q.setBandList(bandList);

			q.setBuffer(newBuffer);

			q.setAdjustToExtent(true);

			return lastRequestProvider.getDataSet(q);

		} catch (ProviderNotRegisteredException e) {

			throw new RasterDriverException("Error building GdalDriver", e);

		} catch (InitializeException e) {

			throw new RasterDriverException("Error building GdalDriver", e);

		}

	}

	/**

	 * Opens the last request downloaded

	 * @return

	 * @throws ProviderNotRegisteredException

	 * @throws InitializeException

	 * @throws RasterDriverException

	 */

	private DefaultRasterProvider openLastRequest() throws ProviderNotRegisteredException, InitializeException, RasterDriverException {

		if(lastRequestProvider != null)

			lastRequestProvider.close();

		File lastFile = getFileLayer();

		lastRequestProvider = DefaultProviderServices.loadProvider(new File(lastFile.getPath()));

		return lastRequestProvider;

	}

	public Object getData(int x, int y, int band)

		throws InvalidSetViewException, FileNotOpenException, RasterDriverException {

		int w = 25;

		int h = 25;

		Point2D p1 = rasterToWorld(new Point2D.Double(x, y));

		Point2D p2 = rasterToWorld(new Point2D.Double(x + w, y + h));

		Rectangle2D bBox = new Rectangle2D.Double(p1.getX(), p1.getY(), Math.abs(p1.getX() - p2.getX()), Math.abs(p1.getY() - p2.getY()));

		WCSDataParametersImpl p = (WCSDataParametersImpl)parameters;

		WCSStatus wcsStatus = new WCSStatus();

		wcsStatus.setCoveraName(p.getCoverageName());

		wcsStatus.setSrs(p.getSRSCode());

		wcsStatus.setFormat(p.getFormat());

		wcsStatus.setOnlineResource(p.getOnlineResource() != null ? (String) p.getOnlineResource().get("GetCoverage") : null);

		wcsStatus.setExtent(bBox);

		wcsStatus.setHeight(h);

		wcsStatus.setWidth(w);

		wcsStatus.setDepth(p.getDepth());

		wcsStatus.setParameters(p.getParameter());

		wcsStatus.setTime(p.getTime());

		lastRequest = downloadFile(wcsStatus, p1.getX(), p1.getY(), p2.getX(), p2.getY(), w, h);

		DefaultRasterProvider driver = null;

		try {

			driver = DefaultProviderServices.loadProvider(new File(lastRequest.getPath()));

			Buffer buf = DefaultRasterManager.getInstance().createBuffer(getDataType()[0], w, h, getBandCount(), true);

			BandList bandList = new BandListImpl();

			int[] drawable = new int[getBandCount()];

			for (int i = 0; i < getBandCount(); i++) {

				try {

					bandList.addBand(new DatasetBandImpl(lastRequest.getPath(), i, getDataType()[i], getBandCount()));

					drawable[i] = i;

				} catch (BandNotFoundInListException e1) {

				}

			}

			bandList.setDrawableBands(drawable);

			DefaultRasterQuery q = (DefaultRasterQuery)RasterLocator.getManager().createQuery();

			q.setAreaOfInterest(0, 0, 25, 25);

			q.setBandList(bandList);

			q.setBuffer(buf);

			buf = driver.getDataSet(q);

			driver.close();

			if(buf.getDataType() == Buffer.TYPE_BYTE) {

				return new Integer(buf.getElemByte(0, 0, band));

			}

			if(buf.getDataType() == Buffer.TYPE_SHORT) {

				return new Integer(buf.getElemShort(0, 0, band));

			}

			if(buf.getDataType() == Buffer.TYPE_INT) {

				return new Integer(buf.getElemInt(0, 0, band));

			}

			if(buf.getDataType() == Buffer.TYPE_FLOAT) {

				return new Float(buf.getElemFloat(0, 0, band));

			}

			if(buf.getDataType() == Buffer.TYPE_DOUBLE) {

				return new Double(buf.getElemDouble(0, 0, band));

			}

		} catch(InitializeException e) {

			throw new RasterDriverException("", e);

		} catch (ProviderNotRegisteredException e) {

			throw new RasterDriverException("", e);

		} catch (ProcessInterruptedException e) {

		}

		return null;

	}

	/**

	 * Gets the georeferencing file name form a raster file

	 * @param file

	 * a raster file

	 * @return

	 * a georeferencing file

	 */

	private String getWorldFile(String file){		

		String worldFile = file;

		int index = file.lastIndexOf(".");

		if (index > 0) {			

			worldFile = file.substring(0, index) + getExtensionWorldFile();

		} else {

			worldFile = file + getExtensionWorldFile();

		}

		return worldFile;

	}

	/**

	 * Obtiene la extensi?n del fichero de georreferenciaci?n

	 * @return String con la extensi?n del fichero de georreferenciaci?n dependiendo

	 * del valor del formato obtenido del servidor. Por defecto asignaremos un .wld

	 */

	private String getExtensionWorldFile() {

		WCSDataParametersImpl p = (WCSDataParametersImpl)parameters;

		String extWorldFile = ".wld";

		if(p.getFormat().equals("image/tif") || p.getFormat().equals("image/tiff")) {

			extWorldFile = ".tfw";

		}

		return extWorldFile;

	}

	private WCSStatus loadWCSStatus(Rectangle2D bBox) {

		WCSDataParametersImpl p = (WCSDataParametersImpl)parameters;

		WCSStatus wcsStatus = new WCSStatus();

		wcsStatus.setCoveraName(p.getCoverageName());

		wcsStatus.setSrs(p.getSRSCode());

		wcsStatus.setFormat(p.getFormat());

		wcsStatus.setOnlineResource(p.getOnlineResource() != null ? (String) p.getOnlineResource().get("GetCoverage") : null);

		wcsStatus.setExtent(bBox);

		wcsStatus.setHeight(p.getHeight());

		wcsStatus.setWidth(p.getWidth());

		wcsStatus.setDepth(p.getDepth());

		wcsStatus.setParameters(p.getParameter());

		wcsStatus.setTime(p.getTime());

		return wcsStatus;

	}

	public boolean needEnhanced() {

		return (getDataType()[0] != Buffer.TYPE_BYTE || 

				(getBandCount() == 1 && getDataType()[0] == Buffer.TYPE_BYTE));

	}

	/**

	 * This function downloads the file and creates the georeferencing file

	 * @param wcsStatus

	 * @param ulx

	 * @param uly

	 * @param lrx

	 * @param lry

	 * @param w

	 * @param h

	 * @return

	 * @throws RasterDriverException

	 */

	private File downloadFile(WCSStatus wcsStatus, double ulx, double uly, double lrx, double lry, int w, int h) throws RasterDriverException {

		try {

			lastRequest = getConnector().getCoverage(wcsStatus, ((WCSDataParametersImpl)parameters).getCancellable());

		} catch (RemoteServiceException e) {

			throw new RasterDriverException(e.getMessage(), e);

		}

		String nameWorldFile = getWorldFile(lastRequest.getPath());

		try {

			fileUtil.createWorldFile(nameWorldFile, new ExtentImpl(ulx, uly, lrx, lry), w, h);

		} catch (IOException e) {

			throw new RasterDriverException("Error downloading file", e);

		}

		return lastRequest;

	}

	/**

	 * Assigns the list of bands RGB and read a window of data

	 * @param rasterBuf

	 * @param bandList

	 * @param lastFile

	 * @param ulx

	 * @param uly

	 * @param lrx

	 * @param lry

	 * @return

	 * @throws RasterDriverException

	 * @throws ProcessInterruptedException

	 */

	public Buffer getBuffer(Buffer rasterBuf, BandList bandList, File lastFile, 

			double ulx, double uly, double lrx, double lry) throws RasterDriverException, ProcessInterruptedException {

		try {

			//El nombre de fichero que ha puesto en el bandList es el del servidor y no el del fichero en disco

			String serverName = bandList.getBand(0).getFileName();

			for (int i = 0; i < bandList.getBandCount(); i++) {

				bandList.getBand(i).setFileName(lastFile.getPath());

			}

			DefaultRasterProvider driver = DefaultProviderServices.loadProvider(new File(lastFile.getPath()));

			DefaultRasterQuery q = (DefaultRasterQuery)RasterLocator.getManager().createQuery();

			q.setAreaOfInterest(new ExtentImpl(ulx, uly, lrx, lry));

			q.setBandList(bandList);

			q.setBuffer(rasterBuf);

			Buffer buf = driver.getDataSet(q);

			for (int i = 0; i < bandList.getBandCount(); i++) {

				bandList.getBand(i).setFileName(serverName);

			}

			return buf;

		} catch (ProviderNotRegisteredException e) {

			throw new RasterDriverException("Error building GdalDriver", e);

		} catch (InitializeException e) {

			throw new RasterDriverException("Error building GdalDriver", e);

		}

	}

	public void getWindow(Extent ex, int bufWidth, int bufHeight, 

			BandList bandList, TileListener listener, TaskStatus status) throws ProcessInterruptedException, RasterDriverException {

		Buffer raster = DefaultRasterManager.getInstance().createBuffer(getDataType()[0], bufWidth, bufHeight, 3, true);

		getWindow(ex, bufWidth, bufHeight, bandList, raster, true, null);

		raster.setDataExtent(ex.toRectangle2D());

		TileCacheManager m = TileCacheLocator.getManager();

		org.gvsig.raster.cache.tile.Tile t = m.createTile(-1, 0, 0);

		t.setData(new Object[]{raster});

		t.setUl(new Point2D.Double(ex.getULX(), ex.getULY()));

		t.setLr(new Point2D.Double(ex.getLRX(), ex.getLRY()));

		t.setDownloaderParams("AffineTransform", getAffineTransform());

		t.setDownloaderParams("Tiling", new Boolean(false));