Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2007 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.rastertools.georeferencing;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.util.ArrayList;

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

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

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

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

import org.gvsig.raster.datastruct.GeoPoint;

import org.gvsig.raster.util.RasterToolsUtil;

import org.gvsig.rastertools.georeferencing.ui.launcher.GeorefLauncherDialog;

import org.gvsig.rastertools.georeferencing.ui.table.GCPTablePanel;

import org.gvsig.rastertools.georeferencing.ui.zoom.InvalidRequestException;

import org.gvsig.rastertools.georeferencing.ui.zoom.ViewMapRequestManager;

import org.gvsig.rastertools.georeferencing.ui.zoom.ViewRasterRequestManager;

import org.gvsig.rastertools.georeferencing.ui.zoom.layers.GCPsGraphicLayer;

import org.gvsig.rastertools.georeferencing.ui.zoom.layers.GPGraphic;

import org.gvsig.rastertools.georeferencing.ui.zoom.layers.ZoomCursorGraphicLayer;

import org.gvsig.rastertools.georeferencing.ui.zoom.tools.ToolEvent;

import org.gvsig.rastertools.georeferencing.ui.zoom.tools.ToolListener;

import org.gvsig.rastertools.georeferencing.view.BaseZoomView;

import org.gvsig.rastertools.georeferencing.view.ViewDialog;

import org.gvsig.rastertools.georeferencing.view.ZoomMapDialog;

import org.gvsig.rastertools.georeferencing.view.ZoomRasterDialog;

import com.iver.andami.PluginServices;

import com.iver.andami.ui.mdiFrame.MDIFrame;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.cit.gvsig.ProjectExtension;

import com.iver.cit.gvsig.fmap.MapControl;

import com.iver.cit.gvsig.project.Project;

import com.iver.cit.gvsig.project.documents.view.ProjectView;

import com.iver.cit.gvsig.project.documents.view.ProjectViewFactory;

import com.iver.cit.gvsig.project.documents.view.gui.IView;

/**

 * Clase principal para la georreferenciaci?n. Se encarga de la inicializaci?n

 * de la funcionalidad. Las acciones a realizar son:

 * <UL>

 * <LI>Lanzamiento del dialogo de inicializaci?n.</LI>

 * <LI>Carga de la capa raster a georreferenciar.</LI>

 * <LI>Creaci?n de las ventanas si no han sido creadas previamente.</LI>

 * <LI>Asignaci?n de tama?os y posiciones a las ventanas.</LI>

 * </UL>

 * 26/12/2007

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

 */

public class Georeferencing implements ButtonsPanelListener, ToolListener {

        public static final int               UNDEFINED                      = -1;

        //Algoritmo

        public static final int               POLYNOMIAL                     = 0;

        public static final int               AFFINE                         = 1;

        //Tipo de georreferenciacion

        public static final int               WITH_MAP                       = 0;

        public static final int               WITHOUT_MAP                    = 1;

        private static int                    DEFAULT_DEGREE                 = 3;

        private GeorefLauncherDialog          initDialog                     = null;

        private FLyrRasterSE                  lyrToGeoref                    = null;

        private String                        selectedView                   = null;

        private static GCPTablePanel          table                          = null;

        private static final int              smallWindowsHeight             = 100;

        private static final double           smallWindowsWidthPercent       = 0.5;

        private ViewDialog                    viewRaster                     = null;

        private ViewDialog                    viewMap                        = null;

        private ZoomMapDialog                 zoomMap                        = null;

        private ZoomRasterDialog              zoomPixel                      = null;

        private ViewRasterRequestManager      viewRasterRequestManager       = null;

        private ViewMapRequestManager         viewMapRequestManager          = null;

        private ViewRasterRequestManager      zoomRasterRequestManager       = null;

        private ViewMapRequestManager         zoomMapRequestManager          = null;

        private ApplicationControlsListener   buttonsListener                = null;

        public Georeferencing() {

        }

        public void initialize() {

                launcher();

        }

        /**

         * Colocaci?n inicial de las ventanas del espacio de trabajo. Carga la configuraci?n

         * de ventanas con cartograf?a de referencia.

         */

        private void initWindowsWithoutMap() {

                MDIFrame p = (MDIFrame)PluginServices.getMainFrame();

                int totalWidth = p.getWidth(); 

                int wViews = totalWidth;

                int hViews = p.getHeight() - 178 - smallWindowsHeight;

                int posYViews = p.getHeight() - 138 - smallWindowsHeight;

                int smallWindowsWidth = (int)(totalWidth * 0.25);

                int tableWidth = (int)(totalWidth * 0.75);

                //Si hay ventanas de georreferenciaci?n abiertas las cerramos

                closeAllWindows();

                //Inicializaci?n de la tabla

                table = new GCPTablePanel(0, posYViews, tableWidth, smallWindowsHeight);

                buttonsListener = new ApplicationControlsListener(table, this);

                table.initialize(buttonsListener);

                //Inicializaci?n de la ventana con la vista raster

                viewRaster = new ViewDialog(0, 0, wViews, hViews, this);

                viewRaster.setShowInfo(true);

                viewRasterRequestManager = new ViewRasterRequestManager(viewRaster, lyrToGeoref);

                viewRaster.setExtensionRequest(viewRasterRequestManager);

                //Inicializaci?n de la ventana del zoom de la vista raster

                zoomPixel = new ZoomRasterDialog(tableWidth, posYViews, smallWindowsWidth, smallWindowsHeight);

                zoomRasterRequestManager = new ViewRasterRequestManager(zoomPixel, lyrToGeoref);

                zoomPixel.setExtensionRequest(zoomRasterRequestManager);

                //A?adimos las ventanas a gvSIG

                PluginServices.getMDIManager().addWindow(viewRaster);

                PluginServices.getMDIManager().addWindow(table);

                PluginServices.getMDIManager().addWindow(zoomPixel);

                try {

                        //Inicializamos la vista de raster

                        viewRasterRequestManager.initRequest(lyrToGeoref.getFullExtent());

                        viewRaster.setZoomCursorSize(zoomPixel.getCanvasWidth() / 3, zoomPixel.getCanvasHeight() / 3);

                        //Inicializamos el zoom de raster

                        Rectangle2D ext = viewRaster.getCursorAdjustedWorldCoordinates(zoomPixel.getCanvasWidth(), zoomPixel.getCanvasHeight());

                        zoomRasterRequestManager.initRequest(lyrToGeoref.getFullExtent());

                        zoomRasterRequestManager.request(ext);

                        /*zoomRasterRequestManager.request(new Rectangle2D.Double(ext.getX(), 

                                                                                        viewRaster.getViewExtent().getMaxY() - (viewRaster.getViewExtent().getMinY() - ext.getY()), 

                                                                                        ext.getWidth(), 

                                                                                        ext.getHeight()));*/

                } catch (InvalidRequestException e1) {

                        RasterToolsUtil.messageBoxError("error_setview_preview", this);

                        PluginServices.getMDIManager().closeWindow(zoomPixel);

                        PluginServices.getMDIManager().closeWindow(viewRaster);

                        PluginServices.getMDIManager().closeWindow(table);

                }catch (Exception e1) {

                        RasterToolsUtil.messageBoxError("error_setview_preview", this);

                        PluginServices.getMDIManager().closeWindow(zoomPixel);

                        PluginServices.getMDIManager().closeWindow(viewRaster);

                        PluginServices.getMDIManager().closeWindow(table);

                }

        }

        /**

         * Colocaci?n inicial de las ventanas del espacio de trabajo. Carga la configuraci?n

         * de ventanas con cartograf?a de referencia.

         */

        private void initWindowsWithMap() {

                MDIFrame p = (MDIFrame)PluginServices.getMainFrame();

                int wViews = p.getWidth() >> 1;

                int hViews = p.getHeight() - 178 - smallWindowsHeight;

                int posYViews = p.getHeight() - 138 - smallWindowsHeight;

                int smallWindowsWidth = (int)(p.getWidth() * (smallWindowsWidthPercent * 0.5));

                int tableWidth = (int)(p.getWidth() * (1 - smallWindowsWidthPercent));

                //Si hay ventanas de georreferenciaci?n abiertas las cerramos

                closeAllWindows();

                MapControl mapControl = null;

                IWindow[] windowList = PluginServices.getMDIManager().getAllWindows();

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

                        if(windowList[i] instanceof IView && windowList[i].getWindowInfo().getTitle().endsWith(": " + selectedView)) 

                                mapControl = ((IView)windowList[i]).getMapControl();

                }

                if(mapControl == null) {

                        RasterToolsUtil.messageBoxError("error_lookingfor_view", this);

                        return;

                }

                //Inicializamos el control de tabla

                table = new GCPTablePanel(smallWindowsWidth, posYViews, tableWidth, smallWindowsHeight);

                buttonsListener = new ApplicationControlsListener(table, this);

                table.initialize(buttonsListener);

                //Inicializaci?n de la ventana con la vista mapa

                viewMap = new ViewDialog(0, 0, wViews, hViews, this);

                viewMap.setShowInfo(true);

                viewMapRequestManager = new ViewMapRequestManager(viewMap, mapControl);

                viewMap.setExtensionRequest(viewMapRequestManager);

                //Inicializaci?n de la ventana con la vista raster

                viewRaster = new ViewDialog(wViews, 0, wViews, hViews, this);

                //viewRaster.setMinxMaxyUL(false);

                viewRaster.setShowInfo(true);

                viewRasterRequestManager = new ViewRasterRequestManager(viewRaster, lyrToGeoref);

                viewRaster.setExtensionRequest(viewRasterRequestManager);

                //Inicializaci?n de la ventana del zoom de la vista raster

                zoomPixel = new ZoomRasterDialog(smallWindowsWidth + tableWidth, posYViews, smallWindowsWidth, smallWindowsHeight);

                zoomPixel.setMinxMaxyUL(false);

                zoomRasterRequestManager = new ViewRasterRequestManager(zoomPixel, lyrToGeoref);

                zoomPixel.setExtensionRequest(zoomRasterRequestManager);

                //Inicializaci?n de la ventana del zoom de la vista de referencia

                zoomMap = new ZoomMapDialog(0, posYViews, smallWindowsWidth, smallWindowsHeight);

                zoomMapRequestManager = new ViewMapRequestManager(zoomMap, mapControl);

                zoomMap.setExtensionRequest(zoomMapRequestManager);

                //A?adimos las ventanas a gvSIG

                PluginServices.getMDIManager().addWindow(viewRaster);

                PluginServices.getMDIManager().addWindow(viewMap);

                PluginServices.getMDIManager().addWindow(table);

                PluginServices.getMDIManager().addWindow(zoomMap);

                PluginServices.getMDIManager().addWindow(zoomPixel);

                try {

                        //Inicializamos la vista de mapa

                        viewMapRequestManager.initRequest(mapControl.getMapContext().getLayers().getFullExtent());

                        viewMap.setZoomCursorSize(zoomMap.getCanvasWidth() / 3, zoomMap.getCanvasHeight() / 3);

                        //Inicializamos la vista de raster

                        viewRasterRequestManager.initRequest(lyrToGeoref.getFullExtent());

                        viewRaster.setZoomCursorSize(zoomPixel.getCanvasWidth() / 3, zoomPixel.getCanvasHeight() / 3);

                        //Inicializamos el zoom de mapa

                        Rectangle2D ext = viewMap.getCursorAdjustedWorldCoordinates(zoomMap.getCanvasWidth(), zoomMap.getCanvasHeight());

                        zoomMapRequestManager.initRequest(ext);

                        //Inicializamos el zoom de raster

                        ext = viewRaster.getCursorAdjustedWorldCoordinates(zoomPixel.getCanvasWidth(), zoomPixel.getCanvasHeight());

                        zoomRasterRequestManager.initRequest(lyrToGeoref.getFullExtent());

                        zoomRasterRequestManager.request(ext);

                        createGCPLayer(viewRaster);

                } catch (InvalidRequestException e1) {

                        RasterToolsUtil.messageBoxError("error_setview_preview", this);

                }catch (Exception e1) {

                        RasterToolsUtil.messageBoxError("error_setview_preview", this);

                }

        }

        /**

         * Cierra todas las ventanas de georreferenciaci?n

         */

        public void closeAllWindows() {

                if(zoomMap != null)

                        PluginServices.getMDIManager().closeWindow(zoomMap);

                if(zoomPixel != null)

                        PluginServices.getMDIManager().closeWindow(zoomPixel);

                if(viewRaster != null)

                        PluginServices.getMDIManager().closeWindow(viewRaster);

                if(viewMap != null)

                        PluginServices.getMDIManager().closeWindow(viewMap);

                if(table != null)

                        PluginServices.getMDIManager().closeWindow(table);

        }

        /**

         * Crea una capa de puntos de control y la a?ade a la vista

         * que se le indica por par?metro

         * @param view

         */

        public void createGCPLayer(BaseZoomView view) {

                GCPsGraphicLayer viewRasterGCPs = new GCPsGraphicLayer(GPGraphic.PIXEL, view.getControl());

                view.addGraphicLayer(viewRasterGCPs);

                //Puntos de prueba

                GeoPoint gp1 = new GeoPoint(new Point2D.Double(100, 100), new Point2D.Double(645344, 4927206));

                GPGraphic gpg1 = new GPGraphic(GPGraphic.PIXEL, gp1);

                GeoPoint gp2 = new GeoPoint(new Point2D.Double(150, 150), new Point2D.Double(646213, 4925620));

                GPGraphic gpg2 = new GPGraphic(GPGraphic.PIXEL, gp2);

                viewRasterGCPs.addPixelGeoPoint(gpg1);

                viewRasterGCPs.addPixelGeoPoint(gpg2);

        }

        /**

         * Lanzador del cuadro de inicializaci?n de propiedades de georreferenciaci?n.

         */

        private void launcher() {

                Project p = ((ProjectExtension) PluginServices.getExtension(ProjectExtension.class)).getProject();

                ArrayList docs = p.getDocumentsByType(ProjectViewFactory.registerName);

                String[] viewList = new String[docs.size()];

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

                        viewList[i] = (((ProjectView)docs.get(i)).getName());

                initDialog = new GeorefLauncherDialog(viewList, DEFAULT_DEGREE, this);

                PluginServices.getMDIManager().addWindow(initDialog);

        }

        /**

         * Captura de eventos de los botones Aceptar y Cancelar del cuadro

         * inicial con las propiedades de georreferenciaci?n.

         */

        public void actionButtonPressed(ButtonsPanelEvent e) {

                // Bot?n de Aceptar

                if (e.getButton() == ButtonsPanel.BUTTON_ACCEPT) {

                        if(initDialog != null) {

                                selectedView = initDialog.getSelectedView();

                                lyrToGeoref = initDialog.getLayer();

                                //Inicializamos la matriz de transformaci?n en la imagen que vamos a georreferenciar para

                                //que nos de coordenadas pixel

                                //lyrToGeoref.setAffineTransform(new AffineTransform(1, 0, 0, -1, 0, 0));

                                if(lyrToGeoref == null) 

                                        RasterToolsUtil.messageBoxError("layer_not_loaded", this);

                                else {

                                        PluginServices.getMDIManager().closeWindow(initDialog);

                                        if(initDialog.getType() == Georeferencing.WITH_MAP)

                                                initWindowsWithMap();

                                        if(initDialog.getType() == Georeferencing.WITHOUT_MAP)

                                                initWindowsWithoutMap();

                                }

                        }

                }

                // Bot?n de Cerrar

                if (e.getButton() == ButtonsPanel.BUTTON_CANCEL) {

                        if(initDialog != null)

                                PluginServices.getMDIManager().closeWindow(initDialog);

                }

        }

        /**

         * Evento de finalizaci?n de la capa con el 

         * cursor gr?fico que controla el ?rea de la miniimagen asociada

         */

        public void endAction(ToolEvent ev) {

                if(ev.getSource() instanceof ZoomCursorGraphicLayer) {

                        try {        

                                Rectangle2D ext = null;

                                if(viewRaster != null) {

                                        ext = viewRaster.getCursorAdjustedWorldCoordinates(zoomPixel.getCanvasWidth(), zoomPixel.getCanvasHeight());

                                        zoomRasterRequestManager.request(ext);

                                }

                                if(viewMap != null) {

                                        ext = viewMap.getCursorAdjustedWorldCoordinates(zoomMap.getCanvasWidth(), zoomMap.getCanvasHeight());

                                        zoomMapRequestManager.request(ext);

                                }

                        } catch (InvalidRequestException e1) {

                                RasterToolsUtil.messageBoxError("error_setview_preview", this);

                        }

                }

        }

}