| gvSIG desktop 1

*

* @author Jaume Dominguez Faus

* 		  Nacho Brodin

*/

public class FLyrWMS extends FLyrDefault implements InfoByPoint, RasterOperations {

	private boolean 					isPrinting = false;

	private final int 					maxTileDrawHeight = 1023;

	private final int 					maxTilePrintWidth = 1023;

	private final int 					maxTilePrintHeight = 1023;

    public URL 							host;

    public String 						m_Format;

	private String 						m_SRS;

	private String 						layerQuery;

	private String 						infoLayerQuery;

	private Dimension 					fixedSize;

	private boolean 					queryable = true;

	private VisualStatusWMS				visualStatus = new VisualStatusWMS();

	private class MyCancellable implements ICancellable

	{

		public boolean isCanceled() {

			return original.isCanceled();

		}

	}

	public FLyrWMS(){

		super();

	}

	public FLyrWMS(Map args) throws DriverIOException{

		FMapWMSDriver drv = null;

		String host = (String)args.get("host");

		String sSRS = (String)args.get("SRS");

		String sFormat = (String)args.get("Format");

		String[] sLayers = sLayer.split(",");

		try {

			this.setHost(new URL(host));

		} catch (MalformedURLException e) {

			//e.printStackTrace();

		}

		try {

			drv = this.getDriver();

		} catch (Exception e) {

			// e.printStackTrace();

		}

		if( sFormat == null || sSRS == null || fullExtent == null ) {

			if (!drv.connect(null))

			WMSLayerNode wmsNode = drv.getLayer(sLayer);

			if (wmsNode == null){

				throw new DriverIOException("The server '" + host + "' doesn't has the layer '" + sLayer + "'.");

			if( sFormat == null ) {

				sFormat = this.getGreatFormat(drv.getFormats());

			}

				fullExtent = drv.getLayersExtent(sLayers,(String)wmsNode.getAllSrs().get(0));

			}

		}

		this.setFullExtent(fullExtent);

		this.setFormat(sFormat);

		this.setLayerQuery(sLayer);

		this.setSRS(sSRS);

		this.setName(sLayer);

	}

	/**

	 * transparency.

	 * @param formats

	 * Arraywith all the formats supported by the server

	 * @return

	private String getGreatFormat(Vector formats){

	    for (int i=0 ; i<formats.size() ; i++){

	        String format = (String) formats.get(i);

	            return format;

	    	}

	    }

	    return (String)formats.get(0);

	}

	/**

	 * Clase que contiene los datos de visualizaci?n de WMS.

	 * @author Nacho Brodin (brodin_ign@gva.es)

	 */

	private class VisualStatusWMS{

		/**

		 * el ancho y alto del viewPort

		 */

		private	int							width = 0, height = 0;

		private int 						bandCount = 0;

		private int							dataType = DataBuffer.TYPE_UNDEFINED;

	}

	/**

	 * Devuelve el XMLEntity con la informaci?n necesaria para reproducir la

	 * capa.

		// Full extent

		xml.putProperty("fullExtent", StringUtilities.rect2String(fullExtent));

		// Host

		xml.putProperty("host", host.toExternalForm());

		// Part of the query that is not the host, or the

		// layer names, or other not listed bellow

		xml.putProperty("infoLayerQuery", infoLayerQuery);

		// Part of the query containing the layer names

		xml.putProperty("layerQuery", layerQuery);

		// Format

		xml.putProperty("format", m_Format);

		// SRS

		xml.putProperty("srs", m_SRS);

		if (status!=null)

			status = new StatusLayerRaster();

			status.getXMLEntity(xml, true, this);

		}

        // Transparency

        xml.putProperty("wms_transparency", wmsTransparency);

        // Styles

        if (styles!=null){

            String stylePr = "";

            	stylePr += " ";

            xml.putProperty("styles", stylePr);

        }

        if (dimensions!=null){

            String dim = "";

            for (int i = 0; i < dimensions.size(); i++) {

            	dim += " ";

            xml.putProperty("dimensions", dim);

        }

        // OnlineResources

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

        String strOnlines = "";

        		strOnlines += "~##SEP1##~";

        }

        xml.putProperty("onlineResources", strOnlines);

        // Queryable

        xml.putProperty("queryable", queryable);

        // fixedSize

        if (isSizeFixed()) {

        	xml.putProperty("fixedSize", true);

		super.setXMLEntity(xml);

		fullExtent = StringUtilities.string2Rect(xml.getStringProperty(

					"fullExtent"));

		// Host

		try {

			host = new URL(xml.getStringProperty("host"));

		// Part of the query containing the layer names

		layerQuery = xml.getStringProperty("layerQuery");

		// Format

		m_Format = xml.getStringProperty("format");

		// SRS

		m_SRS = xml.getStringProperty("srs");

		String claseStr = StatusLayerRaster.defaultClass;

		if (xml.contains("raster.class")) {

			claseStr = xml.getStringProperty("raster.class");

		}

		// Transparency

        if (xml.contains("wms_transparency"))

            wmsTransparency = xml.getBooleanProperty("wms_transparency");

        // Styles

        if (xml.contains("styles")){

            styles = new Vector();

            String[] stl = xml.getStringProperty("styles").split(",");

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

            	if (stl[i].equals(" "))

            		stl[i]="";

                styles.add(stl[i]);

            }

        }

        // Dimensions

        if (xml.contains("dimensions")){

            dimensions = new Vector();

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

            	if (dims[i].equals(" "))

            		dims[i]="";

                dimensions.add(dims[i]);

            }

        }

        // OnlineResources

        if (xml.contains("onlineResources")) {

        	String[] operations = xml.getStringProperty("onlineResources").split("~##SEP1##~");

					onlineResources.put(resources[0], resources[1]);

			}

        }

        // Queryable

        queryable = true; // let's assume that the layer is queryable by default

        if (xml.contains("queryable"))

        	queryable = xml.getBooleanProperty("queryable");

        // fixedSize

        if (xml.contains("fixedSize")) {

        			                  xml.getIntProperty("fixedHeight"));

        }

		if(status!=null)

			status.setXMLEntity(xml, this);

		else{

					exc.printStackTrace();

				}catch(InvocationTargetException exc){

					exc.printStackTrace();

			}

		}

		firstLoad = true;

	 */

	public XMLItem[] getInfo(Point p, double tolerance) throws DriverException {

		XMLItem[] item =  new XMLItem[1];

			if (queryable)

			{

				//TODO

				int col = (int) p.getX() / maxTilePrintWidth;

				int row = (int) p.getY() / maxTilePrintHeight;

				int tileIndex = (row*nCols) + col;

				ViewPort vp = tiles.getTileViewPort(viewPort, tileIndex);

				wmsStatus.setExtent(vp.getExtent());

				wmsStatus.setHeight(vp.getImageHeight());

				wmsStatus.setWidth(vp.getImageWidth());

				wmsStatus.setOnlineResource((String) onlineResources.get("GetFeatureInfo"));

				item[0] = new StringXMLItem(new String(getDriver()

				return item;

			}

			else

	}

	/**

	 * @see com.iver.cit.gvsig.fmap.layers.FLayer#draw(java.awt.image.BufferedImage,

	 * 		java.awt.Graphics2D, com.iver.cit.gvsig.fmap.ViewPort,

	 * 		com.iver.cit.gvsig.fmap.operations.Cancellable)

	 */

	public void draw(BufferedImage image, Graphics2D g, ViewPort viewPort,

			Cancellable cancel,double scale) throws DriverException {

		if (isWithinScale(scale)){

			Point2D p = viewPort.getOffset();

			// p will be (0, 0) when drawing a view or other when painting onto

//		long mem = r.totalMemory() - r.freeMemory();

//		System.err.println("Memoria total: " + (mem / 1024) +"KB");

	}

	private void drawFixedSize(Graphics2D g, ViewPort vp, Cancellable cancel) throws DriverException {

		// This is the extent that will be requested

		Rectangle2D bBox = getFullExtent();

		MyCancellable c = new MyCancellable(cancel);

			wmsStatus.setExtent( bBox );

			wmsStatus.setFormat( m_Format );

			wmsStatus.setHeight( fixedSize.height );

				return;

			String nameWorldFile = f.getPath() + getExtensionWorldFile();

			com.iver.andami.Utilities.createTemp(nameWorldFile, this.getDataWorldFile(bBox, fixedSize));

			if(status!=null && firstLoad){

				status.applyStatus(this);

				firstLoad = false;

			}

			// And finally, obtain the extent intersecting the view and the BBox

			// to draw to.

			Rectangle2D extent = new Rectangle2D.Double();

			Rectangle2D.intersect(vp.getAdjustedExtent(), bBox, extent);

			ViewPortData vpData = new ViewPortData(

				vp.getProjection(), new Extent(extent), fixedSize );

			vpData.setMat(vp.getAffineTransform());

			rasterProcess(g, vpData, f);

		} catch (ValidationException e) {

			if (!c.isCanceled())

				throw new DriverException(PluginServices.getText(this, "unknown_response_format"), e);

				this.setVisible(false);

			}

		}

	}

	/**

	 * This is the method used to draw a tile in a WMS mosaic layer.

	 */

	private void drawTile(Graphics2D g, ViewPort vp, Cancellable cancel) throws DriverException {

		// 1. Check if it is within borders

		Rectangle2D extent = getFullExtent();

        if ((vp.getExtent().getMinX() > extent.getMaxX()) ||

                (vp.getExtent().getMaxY() < extent.getMinY())) {

            return;

        }

        // 2. Compute extent to be requested.

        Rectangle2D bBox = new Rectangle2D.Double();

        Rectangle2D.intersect(vp.getExtent(), extent, bBox);

        // 3. Compute size in pixels

        int wImg = (int) Math.ceil(Math.abs(bBox.getWidth() * scalex) + 1);

        int hImg = (int) Math.ceil(Math.abs(bBox.getHeight() * scaley) + 1);

        Dimension sz = new Dimension(wImg, hImg);

        if ((wImg <= 0) || (hImg <= 0)) {

            return;

        }

        MyCancellable c = new MyCancellable(cancel);

		try {

			wmsStatus.setExtent( bBox );

			wmsStatus.setFormat(m_Format);

			wmsStatus.setHeight( hImg );

			wmsStatus.setDimensions(dimensions);

			wmsStatus.setTransparency(wmsTransparency);

			wmsStatus.setOnlineResource((String) onlineResources.get("GetMap"));

			File f = getDriver().getMap(wmsStatus, c);

			if (f == null)

				return;

			String nameWordFile = f.getPath() + getExtensionWorldFile();

			com.iver.andami.Utilities.createTemp(nameWordFile, this.getDataWorldFile(bBox, sz));

			if(status!=null && firstLoad){

				status.applyStatus(this);

				firstLoad = false;

			vpData.setMat(vp.getAffineTransform());

			rasterProcess(g, vpData, f);

		} catch (ValidationException e) {

			if (!c.isCanceled())

				throw new DriverException(PluginServices.getText(this, "unknown_response_format"), e);

				this.setVisible(false);

			}

		}

	}

	/**

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

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

	 */

	private String getExtensionWorldFile(){

		String extWorldFile = ".wld";

    		extWorldFile = ".jpgw";

    	return extWorldFile;

	}

	/**

	 * Calcula el contenido del fichero de georreferenciaci?n de una imagen.

	 * @param bBox Tama?o y posici?n de la imagen (en coordenadas de usuario)

    	data.append(""+bBox.getMinY()+"\n");

    	return data.toString();

	}

	/**

	 * Dibuja una imagen usando PxRaster

	 * @param g	Graphics2D en el que hay que dibujar.

	 * @param file La imagen en cuesti?n.

	 */

	private void rasterProcess(Graphics2D g, ViewPortData vpData, File file) {

		rasterFile = new GdalFile(vpData.getProjection(), file.getAbsolutePath());

		raster = new PxRaster(rasterFile, null, rasterFile.getExtent());

		//Recuperamos la pila de filtros si ya hubiese sido cargado antes

		if(this.filterStack!=null)

			raster.filterStack = this.filterStack;

		raster.setTransparency(false);

		//Asignamos transparencia y orden de bandas

		if(this.transparency==-1 && !firstLoad);

		else

			raster.setTransparency(this.transparency);

		raster.setBand(GeoRasterFile.RED_BAND,rband);

		raster.setBand(GeoRasterFile.GREEN_BAND, gband);

		raster.setBand(GeoRasterFile.BLUE_BAND, bband);

		//Despues del primer pxRaster asignamos el stackManager guardado para los siguientes.

		//propiedades cuando creamos un nuevo pxRaster

		if(this.stackManager != null)

		if(visualStatus != null){

			visualStatus.bandCount = raster.getBandCount();

			visualStatus.dataType = raster.getDataType();

		}

		raster.draw(g, vpData);

		//En el primer pxRaster de una imagen obtenemos el Stack Manager para poder modificarlo

		//si queremos desde las propiedades

		if(this.stackManager == null)

		if(this.filterStack == null)

			this.filterStack = raster.filterStack;

		//rasterFile.close();

	}

	/**

	 * @see com.iver.cit.gvsig.fmap.layers.FLayer#print(java.awt.Graphics2D,

	 * 		com.iver.cit.gvsig.fmap.ViewPort,

	 */

	public void print(Graphics2D g, ViewPort viewPort, Cancellable cancel,double scale)

		throws DriverException {

		isPrinting = true;

		if (!mustTilePrint) {

			draw(null, g, viewPort, cancel,scale);

	        // a hacer lo mismo que hace EcwFile: chunkear.

	        // Llamamos a drawView con cuadraditos m�s peque�os

	        // del BufferedImage ni caso, cuando se imprime viene con null

			Tiling tiles = new Tiling(maxTilePrintWidth, maxTilePrintHeight, g.getClipRect());

			tiles.setAffineTransform((AffineTransform) viewPort.getAffineTransform().clone());

			for (int tileNr=0; tileNr < tiles.getNumTiles(); tileNr++) {

	    isPrinting = false;

		}

	}

	public void _print(Graphics2D g, ViewPort viewPort, Cancellable cancel,double scale)

		throws DriverException {

		draw(null, g, viewPort, cancel,scale);

	 * @throws IOException

	 */

	private FMapWMSDriver getDriver()

			UnsupportedVersionException, IOException {

		return FMapWMSDriverFactory.getFMapDriverForURL(host);

	}

	/**

	 * Devuelve el FMapWMSDriver.

	 *

	public void setFullExtent(Rectangle2D fullExtent) {

		this.fullExtent = fullExtent;

	}

	public HashMap getProperties() {

		HashMap info = new HashMap();

        String[] layerNames = getLayerQuery().split(",");

        }

        return null;

	}

	/**

	 * Asignar el estado del raster

	 * @param status

	public void setStatus(StatusRasterInterface status){

		this.status = status;

	}

	/**

	 * Obtiene el estado del raster

	 * @return

	public StatusRasterInterface getStatus(){

		return this.status;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#getAttributes()

	 */

		}

		return  null;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#getFilterStack()

	 */

	public double getHeight() {

		return visualStatus.height;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#getMaxX()

	 */

	public double getMaxX() {

		return visualStatus.maxX;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#getMaxY()

	 */

	public double getMaxY() {

		return visualStatus.maxY;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#getMinX()

	 */

	public double getMinX() {

		return visualStatus.minX;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#getMinY()

	 */

		this.posX = x;

		this.posY = y;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#setPosWC(double, double)

	 */

		this.posXWC = x;

		this.posYWC = y;

	}

	/* (non-Javadoc)

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#setRGB(int, int, int)

	 */

	public GeoRasterFile getGeoRasterFile() {

		return rasterFile;

	}

	public void setTransparency(int trans) {

		this.transparency = trans;

	}

	/**

	 * Sets the R-band.

	 * Asigna la banda R.

	 * @param r

	 */

	public void setBandR(int r){

		this.rband = r;

	}

	/**

	 * Sets the G-band.

	 * Asigna la banda G

	 * @param g

	 */

	public void setBandG(int g){

		this.gband = g;

	}

	/**

	 * Sets the B-band.

	 * Asigna la banda B

	 * @param b

	 */

    		}

    	}

    }

    /**

     * Sets the dimension vector that is a list of key-value pairs containing

     * the name of the dimension and the value for it

    /**

     * Sets the set of URLs that should be accessed for each operation performed

     * to the server.

     * @param onlineResources

     */

	public void setOnlineResources(Hashtable onlineResources) {

		this.onlineResources = onlineResources;

	}

	/**

	 * When a server is not fully featured and it only can serve constant map

	 * sizes this value must be set. It expresses the size in pixels (width, height)

	public void setFixedSize(Dimension sz) {

		fixedSize = sz;

	}

	/**

	 * Tells whether if this layer must deal with the server with the constant-size

	 * limitations or not.

	 * layer(s).

	 * @param version, The desired version of the resulting document. (1.1.0)

	 * @return String containing the xml.

	 */

	public String toMapContext(String mapContextVersion) {

		XmlBuilder xml = new XmlBuilder();

			drv = getDriver();

		} catch (Exception e) {

			return xml.toString();

		String[] layerNames = getLayerQuery().split(",");

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

			WMSLayerNode layer = drv.getLayer(layerNames[i]);

			HashMap xmlAttrs = new HashMap();

			// <Layer>

			xmlAttrs.put(WebMapContextTags.HIDDEN, !isVisible()+"");

			xmlAttrs.put(WebMapContextTags.QUERYABLE, queryable+"");

				xmlAttrs.put(WebMapContextTags.SERVER_TITLE, drv.getServiceTitle());

				xml.openTag(WebMapContextTags.SERVER, xmlAttrs);

				xmlAttrs.clear();

					// <OnlineResource>

					xmlAttrs.put(WebMapContextTags.XLINK_TYPE, "simple");

					xmlAttrs.put(WebMapContextTags.XLINK_HREF, getHost().toString());

					xml.writeTag(WebMapContextTags.ONLINE_RESOURCE, xmlAttrs);

					xmlAttrs.clear();

					// </OnlineResource>

				xml.closeTag();

				// </Server>

				// <Name>

				xml.writeTag(WebMapContextTags.NAME, layer.getName().trim());

				// </Name>

				// <Title>

				xml.writeTag(WebMapContextTags.TITLE, getName().trim());

				// </Title>

				// <Abstract>

				if (layer.getAbstract() != null)

					xml.writeTag(WebMapContextTags.ABSTRACT, layer.getAbstract());

				// </Abstract>

				// <SRS> (a list of available SRS for the enclosing layer)

				String[] strings = (String[]) layer.getAllSrs().toArray(new String[0]);

				String mySRS = strings[0];

				}

				xml.writeTag(WebMapContextTags.SRS, mySRS);

				// </SRS>

				// <FormatList>

				xml.openTag(WebMapContextTags.FORMAT_LIST);

					strings = (String[]) drv.getFormats().toArray(new String[0]);

							xml.writeTag(WebMapContextTags.FORMAT, str, WebMapContextTags.CURRENT, "1");

						else

							xml.writeTag(WebMapContextTags.FORMAT, str);

					}

				xml.closeTag();

				// </FormatList>

				// <StyleList>

				xml.openTag(WebMapContextTags.STYLE_LIST);

					if (layer.getStyles().size()>0) {

						for (int j = 0; j < layer.getStyles().size(); j++) {

							// <Style>

								xmlAttrs.put(WebMapContextTags.CURRENT, "1");

							xml.openTag(WebMapContextTags.STYLE, xmlAttrs);

							xmlAttrs.clear();

								// <Name>

								xml.writeTag(WebMapContextTags.NAME, st.name);

								// </Name>

								// <Title>

								xml.writeTag(WebMapContextTags.TITLE, st.title);

								// </Title>

								// <LegendURL width="180" format="image/gif" height="50">

									// <OnlineResource xlink:type="simple" xlink:href="http://globe.digitalearth.gov/globe/en/icons/colorbars/NATIONAL.gif"/>

									// </OnlineResource>

								// </LegendURL>

							xml.closeTag();

							// </Style>

						}

					} else {

						// Create fake style (for compatibility issues)

						xmlAttrs.put(WebMapContextTags.CURRENT, "1");

							// <Name>

							xml.writeTag(WebMapContextTags.NAME, "default");

							// </Name>

							// <Title>

							xml.writeTag(WebMapContextTags.TITLE, "default");

							// </Title>

//							// <LegendURL width="180" format="image/gif" height="50">

//							xmlAttrs.put(WebMapContextTags.WIDTH, "0");

//							xmlAttrs.put(WebMapContextTags.HEIGHT, "0");

		}

		return xml.getXML();

	}

		return new ImageIcon(PluginServices.getPluginServices("com.iver.cit.gvsig.wms").getClassLoader().getResource("images/icoLayer.png"));

	}

	 * @see com.iver.cit.gvsig.fmap.layers.RasterOperations#getTileSize()

	 */

	public int[] getTileSize() {

		return size;

	}

	public boolean isTiled() {

		return mustTileDraw;

	}

}