Application: gvSIG desktop » gvSIG raster

/* 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.multifile.app.panel;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JOptionPane;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.ui.mdiManager.WindowInfo;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

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

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

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

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

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

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

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.fmap.dal.coverage.store.parameter.RasterDataParameters;

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

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

import org.gvsig.fmap.dal.coverage.util.ProviderServices;

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.swing.FilesystemExplorerWizardPanel;

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

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

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

import org.gvsig.gui.beans.buttonspanel.ButtonsPanelEvent;

import org.gvsig.gui.beans.buttonspanel.ButtonsPanelListener;

import org.gvsig.gui.beans.swing.JFileChooser;

import org.gvsig.gui.beans.table.exceptions.NotInitializeException;

import org.gvsig.i18n.Messages;

import org.gvsig.raster.fmap.layers.DefaultFLyrRaster;

import org.gvsig.raster.fmap.layers.FLyrRaster;

import org.gvsig.raster.mainplugin.RasterMainPluginUtils;

import org.gvsig.raster.mainplugin.config.Configuration;

import org.gvsig.raster.multifile.io.MultiFileDataParameters;

import org.gvsig.raster.multifile.io.MultiFileFormat;

import org.gvsig.raster.multifile.io.MultiFileProvider;

import org.gvsig.raster.swing.RasterSwingLibrary;

import org.gvsig.tools.locator.LocatorException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Clase que maneja los eventos del panel BandSetupPanel. Gestiona el a?adir o

 * eliminar ficheros de la lista y contiene las acciones a realizar cuando en

 * panel registrable se pulsa aceptar, aplicar o cancelar.

 *

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

 */

public class BandSelectorListener implements ActionListener, ButtonsPanelListener {

	private Logger                log            = LoggerFactory.getLogger(BandSelectorListener.class);

	private BandSelectorPanel     bandSetupPanel = null;

	private JFileChooser          fileChooser    = null;

	private FLyrRaster            fLayer         = null;

	private boolean               enabled        = true;

	private List<File>            fileList       = null;

	/**

	 * Constructor

	 * @param bs Panel del selector de bandas

	 * @param lyr Capa raster

	 */

	public BandSelectorListener(BandSelectorPanel bs) {

		this.bandSetupPanel = bs;

		bs.getFileList().getJButtonAdd().addActionListener(this);

		bs.getFileList().getJButtonRemove().addActionListener(this);

		bs.getNumBandSelectorCombo().addActionListener(this);

	}

	/**

	 * Comprobar si la asignacion de color es correcta para las 4 bandas. No puede

	 * existir una banda con distintas interpretaciones de color. Se comprueban dos

	 * casos, asignaciones en escala de grises o en RGB.

	 * @param r

	 * @param g

	 * @param b

	 * @param a

	 * @return

	 */

	private boolean isCorrectAssignedBand(int r, int g, int b, int a) {

		// Si es gris es correcta la asignacion

		if ((r == g) && (r == b) && (r >= 0)) {

			// Si el alpha esta asignado a la misma banda es incorrecto

			if (r == a)

				return false;

			// En caso contrario es correcto

			return true;

		}

		// Si dos bandas coinciden, se dice que no es correcta la asignacion

		int list[] = { r, g, b, a };

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

			for (int j = 0; j <= 3; j++)

				if ((i != j) && (list[i] == list[j]) && (list[i] > -1))

					return false;

		return true;

	}

	/**

	 * Constructor

	 * @param bs Panel del selector de bandas

	 * @param lyr Capa raster

	 */

	public void init(FLyrRaster lyr) {

		fLayer = lyr;

	}

	/**

	 * Listener para la gesti?n de los botones de a?adir, eliminar fichero y

	 * el combo de selecci?n de bandas.

	 */

	public void actionPerformed(ActionEvent e) {

		if (e.getSource().equals(bandSetupPanel.getFileList().getJButtonAdd()))

			addFileBand();

		if (e.getSource().equals(bandSetupPanel.getFileList().getJButtonRemove()))

			delFileBand();

		if (e.getSource().equals(bandSetupPanel.getNumBandSelectorCombo())) {

			String vBands = (String) bandSetupPanel.getNumBandSelectorCombo().getSelectedItem();

			if (vBands != null) {

				if (vBands.compareTo("3") == 0)

					bandSetupPanel.resetMode(3);

				if (vBands.compareTo("2") == 0)

					bandSetupPanel.resetMode(2);

				if (vBands.compareTo("1") == 0)

					bandSetupPanel.resetMode(1);

			}

		}

		if (e.getSource().equals(bandSetupPanel.getSaveButton())) {

			int rBand = bandSetupPanel.getAssignedBand(RasterDataStore.RED_BAND);

			int gBand = bandSetupPanel.getAssignedBand(RasterDataStore.GREEN_BAND);

			int bBand = bandSetupPanel.getAssignedBand(RasterDataStore.BLUE_BAND);

			int aBand = bandSetupPanel.getAssignedBand(RasterDataStore.ALPHA_BAND);

			if (!isCorrectAssignedBand(rBand, gBand, bBand, aBand)) {

				RasterSwingLibrary.messageBoxError(Messages.getText("combinacion_no_asignable"), bandSetupPanel);

				return;

			}

			RasterSwingLibrary.messageBoxYesOrNot(Messages.getText("color_interpretation_continue"), this);

			RasterDataStore dataSource = fLayer.getDataStore();

			if(dataSource == null) {

				RasterSwingLibrary.messageBoxError(Messages.getText("error_carga_capa"), bandSetupPanel);

				return;

			}

			ColorInterpretation ci = dataSource.getColorInterpretation();

			try {

				// Combinaci?n GRAY

				if ((rBand == gBand) && (rBand == bBand) && (rBand >= 0)) {

					for (int iBand = 0; iBand < bandSetupPanel.getARGBTable().getRowCount(); iBand++) {

						ci.setColorInterpValue(iBand, ColorInterpretation.UNDEF_BAND);

					}

					ci.setColorInterpValue(rBand, ColorInterpretation.GRAY_BAND);

					ci.setColorInterpValue(aBand, ColorInterpretation.ALPHA_BAND);

				} else {

					// Combinaci?n RGB

					for (int iBand = 0; iBand < bandSetupPanel.getARGBTable().getRowCount(); iBand++)

						ci.setColorInterpValue(iBand, bandSetupPanel.getColorInterpretationByBand(iBand));

				}

				String fileName = fLayer.getDataStore().getName();

				RasterLocator.getManager().getProviderServices().saveObjectToRmfFile(fileName, ci);

			} catch (RmfSerializerException exc) {

				RasterSwingLibrary.messageBoxError(Messages.getText("error_salvando_rmf"), bandSetupPanel, exc);

			} catch (NotInitializeException exc) {

				RasterSwingLibrary.messageBoxError(Messages.getText("table_not_initialize"), bandSetupPanel, exc);

			}

		}

		boolean autoRefreshView = Configuration.getValue("general_auto_preview", Boolean.TRUE).booleanValue();

		if (!autoRefreshView)

			return;

		bandSetupPanel.onlyApply();

	}

	/**

	 * Checks if the new file is compatible with the old one

	 * @param file

	 * @return

	 * @throws LocatorException

	 * @throws NotSupportedExtensionException

	 * @throws RasterDriverException

	 * @throws CloseException

	 */

	private boolean checkNewFile(String file) throws LocatorException, NotSupportedExtensionException, RasterDriverException, CloseException {

		Rectangle2D extentOrigin = fLayer.getFullRasterExtent().toRectangle2D();

		ProviderServices provServ = RasterLocator.getManager().getProviderServices();

		RasterDataParameters storeParameters = provServ.createParameters(file);

		storeParameters.setSRS(fLayer.getDataStore().getProjection());

		RasterDataStore dataStore = RasterLocator.getManager().getProviderServices().open(storeParameters);

		Extent extentNewFile = dataStore.getExtent();

		// Comprobamos que el extent y tama?o del fichero a?adido sea igual al

		// fichero original. Si no es as? no abrimos la capa y mostramos un aviso

		double widthNewFile = (extentNewFile.getMax().getX() - extentNewFile.getMin().getX());

		double heightNewFile = (extentNewFile.getMax().getY() - extentNewFile.getMin().getY());

		if ((widthNewFile - extentOrigin.getWidth()) > 1.0 || (widthNewFile - extentOrigin.getWidth()) < -1.0 || (heightNewFile - extentOrigin.getHeight()) > 1.0

				|| (heightNewFile - extentOrigin.getHeight()) < -1.0) {

			return false;

		}

		if ((extentNewFile.getMax().getX() - extentNewFile.getMin().getX()) != extentOrigin.getWidth()

				|| (extentNewFile.getMax().getY() - extentNewFile.getMin().getY()) != extentOrigin.getHeight()) {

			return false;

		}

		dataStore.close();

		return true;

	}

	/**

	 * A?ade una banda al raster

	 * @param e

	 */

	private void addFileBand() {

		// String[] driverNames = null;

		// Creaci?n del dialogo para selecci?n de ficheros

		fileChooser = new JFileChooser(

				FilesystemExplorerWizardPanel.OPEN_LAYER_FILE_CHOOSER_ID,

				JFileChooser.getLastPath(FilesystemExplorerWizardPanel.OPEN_LAYER_FILE_CHOOSER_ID, null));

		fileChooser.setMultiSelectionEnabled(true);

		fileChooser.setAcceptAllFileFilterUsed(false);

		fileChooser.addChoosableFileFilter(new DriverFileFilter());

		int result = fileChooser.showOpenDialog(bandSetupPanel);

		if (result == JFileChooser.APPROVE_OPTION) {

			RasterDataStore dataStore = fLayer.getDataStore();

			File[] files = fileChooser.getSelectedFiles();

			JFileChooser.setLastPath(FilesystemExplorerWizardPanel.OPEN_LAYER_FILE_CHOOSER_ID, files[0]);

			fileList = new ArrayList<File>();

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

				//Checks that the file does not exist

				String[] uris = dataStore.getURIByProvider();

				boolean exists = false;

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

					if (uris[j].endsWith(files[i].getName())) {

						RasterSwingLibrary.messageBoxError( Messages.getText("fichero_existe") + ": " + files[i].getAbsolutePath(), bandSetupPanel);

						exists = true;

						break;

					}

				}

				if(!exists)

					fileList.add(files[i]);

			}

			for (int i = fileList.size() - 1; i >= 0; i--) {

				//Checks extents

				try {

					if(!checkNewFile(fileList.get(i).getAbsolutePath())) {

						JOptionPane.showMessageDialog(null, Messages.getText("extents_no_coincidentes") +  " " + files[i].getAbsolutePath(), "", JOptionPane.ERROR_MESSAGE);

						fileList.remove(i);

					}

				} catch (Exception e) {

					log.debug("Problems check the bounding boxes", e);

				}

			}

			if(dataStore.isTiled()) {

				if(!RasterSwingLibrary.messageBoxYesOrNot(Messages.getText("store_tiled"), this)) {

					return;

				}

			}

			if(dataStore.isMultiFile()) {

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

					try {

						dataStore.addFile(fileList.get(i).getAbsolutePath());

					} catch (InvalidSourceException e) {

						RasterSwingLibrary.messageBoxError(Messages.getText("addband_error"), bandSetupPanel, e);

					}

				}

				dataStore.setProvider(dataStore.getProvider());

				//It shows the files and bands in the panel

				try {

					bandSetupPanel.addFiles(dataStore);

				} catch (NotInitializeException e) {

					RasterSwingLibrary.messageBoxError("table_not_initialize", this, e);

				}

			} else {

				//New layer name

				WindowInfo wi = PluginServices.getMDIManager().getActiveWindow().getWindowInfo();

				LayerNameDialog dialog = new LayerNameDialog(new Point2D.Double(wi.getX(), wi.getY()), 300, 80, this);

				RasterMainPluginUtils.addWindow(dialog);

			}

		}

	}

	/**

	 * Catchs the events from LayerNameDialog to get the name of the new layer

	 */

	public void actionButtonPressed(ButtonsPanelEvent e) {

		String layerName = (String)e.getSource();

		RasterDataStore dataStore = fLayer.getDataStore();

		RasterDataParameters paramFirstFile = (RasterDataParameters)dataStore.getParameters();//(RasterDataParameters)((RasterProvider)dataStore.getProvider()).getDataParameters();

		ProviderServices provServ = RasterLocator.getManager().getProviderServices();

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

		try {

			//It creates the new Multifile provider

			DataServerExplorerParameters explParams = dataManager.createServerExplorerParameters(MultiFileProvider.NAME);

			DataServerExplorer explorer = dataManager.openServerExplorer(MultiFileProvider.NAME, explParams);

			MultiFileDataParameters newParamsMultifile = (MultiFileDataParameters)dataManager.createStoreParameters(explorer.getProviderName());

			String path = paramFirstFile.getURI().substring(0, paramFirstFile.getURI().lastIndexOf(File.separator) + 1);

			String fileURI = path + layerName + ".mff";

			String rmfURI = path + layerName + ".rmf";

			int counter = 0;

			while(new File(fileURI).exists() || new File(rmfURI).exists()) {

				fileURI = path + layerName + "_" + counter + ".mff";

				rmfURI = path + layerName + "_" + counter + ".rmf";

				counter ++;

			}

			if(counter > 0)

				layerName += "_" + counter;

			newParamsMultifile.setURI(fileURI);

			newParamsMultifile.addProvider(dataStore);

			DataStoreProvider provMultifile = dataManager.createProvider((DataStoreProviderServices)dataStore, newParamsMultifile);

			//And now it creates and adds the new ones

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

				ArrayList<RasterDataParameters> storeParametersList = provServ.createParametersList(fileList.get(i).getAbsolutePath());

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

					DataStoreProvider newFileProv = dataManager.createProvider((DataStoreProviderServices)dataStore, storeParametersList.get(j));

					newParamsMultifile.addProviderNotTiled(newFileProv);

				}

			}

			((DefaultFLyrRaster)fLayer).setName(layerName);

			//Assigns the MultifileProvider to the store

			dataStore.setProvider((CoverageStoreProvider)provMultifile);

			//It shows the files and bands in the panel

			try {

				bandSetupPanel.addFiles(dataStore);

			} catch (NotInitializeException ex) {

				RasterSwingLibrary.messageBoxError(Messages.getText("table_not_initialize"), this, ex);

			}

			ArrayList<File> uriList = new ArrayList<File>();

			uriList.add(new File(paramFirstFile.getURI()));

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

				uriList.add(fileList.get(i));

			}

			saveMultiFileLayer(layerName, paramFirstFile.getURI(), uriList);

		} catch (Exception exc) {

			RasterSwingLibrary.messageBoxError(Messages.getText("addband_error"), bandSetupPanel, exc);

		}

	}

	/**

	 * Saves the new layer in disk

	 * @param layerName

	 * @param file

	 * @param uriList

	 * @throws IOException

	 */

	private String saveMultiFileLayer(String layerName, String file, ArrayList<File> uriList) throws IOException {

		String path = file.substring(0, file.lastIndexOf(File.separator) + 1);

		path = path + layerName + ".mff";

		/*File filePath = new File(path);

		if(filePath.exists()) {

			RasterToolsUtil.messageBoxInfo("file_exists_rename", bandSetupPanel);

			filePath.renameTo(new File(path + "~"));

		}*/

		MultiFileFormat format = new MultiFileFormat();

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

			format.addFile(uriList.get(i));

		}

		format.setName(layerName);

		format.write(path);

		return path;

	}

	/**

	 * Elimina una banda del raster. Si queda solo un fichero o no se ha

	 * seleccionado ninguna banda no hace nada.

	 *

	 * @param e

	 */

	private void delFileBand() {

		Object[] objects = bandSetupPanel.getFileList().getJList().getSelectedValues();

		RasterDataStore dataStore = fLayer.getDataStore();

		for (int i = objects.length - 1; i >= 0; i--) {

			if (bandSetupPanel.getFileList().getNFiles() > 1) {

				String pathName = objects[i].toString();

				dataStore.removeFile(pathName);

				String file = pathName.substring(pathName.lastIndexOf(File.separator) + 1);

				file = file.substring(file.lastIndexOf("\\") + 1);

				bandSetupPanel.removeFile(file);

			}

		}

		setNewBandsPositionInRendering();

	}

	/**

	 * Acciones a ejecutar cuando se aplica

	 */

	public void apply() {

		if (enabled)

			setNewBandsPositionInRendering();

	}

	/**

	 * Asigna la posici?n de las bandas en el rederizado basandose en la selecci?n

	 * hecho en la tabla de bandas.

	 */

	public void setNewBandsPositionInRendering() {

		if (fLayer != null && fLayer.getRender() != null) {

			fLayer.getRender().setRenderBands(new int[]{bandSetupPanel.getAssignedBand(RasterDataStore.RED_BAND),

					bandSetupPanel.getAssignedBand(RasterDataStore.GREEN_BAND),

					bandSetupPanel.getAssignedBand(RasterDataStore.BLUE_BAND)});

			int alphaBand = bandSetupPanel.getAssignedBand(RasterDataStore.ALPHA_BAND);

			// Ultima transparencia aplicada en el renderizador

			Transparency gt = fLayer.getRender().getLastTransparency();

			if(gt != null)

				gt.setTransparencyBand(alphaBand);

			// Transparencia del dataset

			/*if(fLayer.getDataStore() != null) {

				Transparency t = fLayer.getDataStore().getTransparency();

				if(t != null)

					t.setTransparencyBand(alphaBand);

			}*/

			fLayer.getMapContext().invalidate();

		}

	}

	/**

	 * Activa o desactiva la acci?n del panel

	 * @param enabled true para activa y false para desactivar.

	 */

	public void setEnabledPanelAction(boolean enabled) {

		this.enabled = enabled;

	}

}