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.Point2D;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

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.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.ViewRasterRequestManager;

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.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.andami.ui.mdiManager.WindowInfo;

import com.iver.cit.gvsig.ProjectExtension;

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

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 static GCPTablePanel          table                          = null;

        private static ZoomMapDialog          zoomMap                        = null;

        private static ZoomRasterDialog       zoomPixel                      = null;

        private static ViewDialog             view                           = null;

        private static final int              smallWindowsHeight             = 100;

        private static final double           smallWindowsWidthPercent       = 0.5;

        private ViewRasterRequestManager      viewRequestManager             = null;

        private ViewRasterRequestManager      viewRequestManagerZoomPixel    = null;

        public Georeferencing() {

        }

        public void initialize() {

                launcher();

        }

        /**

         * Colocaci?n inicial de las ventanas del espacio de trabajo

         */

        private void initWindows() {

                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));

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

                BufferedImage imgView = null;

                ViewPort vp = null;

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

                        if(windowList[i] instanceof IView) {

                                WindowInfo info = windowList[i].getWindowInfo();

                                info.setX(0);

                                info.setY(0);

                                info.setWidth(wViews);

                                info.setHeight(posYViews);

                                imgView = ((IView)windowList[i]).getMapControl().getImage();

                                vp = ((IView)windowList[i]).getMapControl().getViewPort();

                        }

                }

                if(zoomMap != null)

                        PluginServices.getMDIManager().closeWindow(zoomMap);

                if(zoomPixel != null)

                        PluginServices.getMDIManager().closeWindow(zoomPixel);

                if(view != null)

                        PluginServices.getMDIManager().closeWindow(view);

                if(table != null)

                        PluginServices.getMDIManager().closeWindow(table);

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

                //Inicializaci?n de la vista raster

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

                view.setShowInfo(true);

                viewRequestManager = new ViewRasterRequestManager(view, lyrToGeoref);

                view.setExtensionRequest(viewRequestManager);

                //Inicializaci?n del zoom de la vista raster

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

                viewRequestManagerZoomPixel = new ViewRasterRequestManager(zoomPixel, lyrToGeoref);

                zoomPixel.setExtensionRequest(viewRequestManagerZoomPixel);

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

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

                /*GeoPoint gp = new GeoPoint(new Point2D.Double(100, 100), null);

                GPGraphic gpg = new GPGraphic(GPGraphic.PIXEL, gp);

                view.getPixelGraphicLayer().addGeoPoint(gpg);*/

                /*BufferedImage imgRasterToGeoref = viewRequestManager.drawRasterLayer(new Dimension(wViews, hViews));

                if(imgView != null)

                        setMapParams(imgView, vp);

                if(imgRasterToGeoref != null)

                        setPixelParams(imgRasterToGeoref, lyrToGeoref);

                */

                //A?adimos las ventanas a gvSIG

                PluginServices.getMDIManager().addWindow(view);

                PluginServices.getMDIManager().addWindow(table);

                PluginServices.getMDIManager().addWindow(zoomMap);

                PluginServices.getMDIManager().addWindow(zoomPixel);

                //Inicializamos la extensi?n del componente y dibujamos

                try {

                        viewRequestManager.initRequest(lyrToGeoref.getFullExtent());/*new Rectangle2D.Double(        lyrToGeoref.getFullRasterExtent().getULX(), 

                                                                                                                                        lyrToGeoref.getFullRasterExtent().getULY(), 

                                                                                                                                        lyrToGeoref.getFullRasterExtent().width(), 

                                                                                                                                        lyrToGeoref.getFullRasterExtent().height()));*/

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

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

                        viewRequestManagerZoomPixel.initRequest(lyrToGeoref.getFullExtent());

                        //viewRequestManagerZoomPixel.setDrawParams(null, new Rectangle2D.Double(ext.getX(), ext.getY() - ext.getHeight(), ext.getWidth(), ext.getHeight()));

                        viewRequestManagerZoomPixel.request(ext);

                } catch (InvalidRequestException e1) {

                        RasterToolsUtil.messageBoxError("error_setview_preview", this);

                }

        }

        /**

         * Obtiene y asigna los par?metros para el control de zoom del mapa

         * @param img BufferedImage

         * @param vp ViewPort

         */

        public void setMapParams(BufferedImage img, ViewPort vp) {

                if(zoomMap != null && img != null) {

                        Rectangle2D r = vp.getAdjustedExtent();

                        int w = vp.getImageWidth();

                        zoomMap.setDrawParams(img, r, r.getWidth() / w, new Point2D.Double(r.getCenterX(), r.getCenterY()));

                        zoomMap.getMapGraphicLayer();

                }

        }

        /**

         * Obtiene y asigna los par?metros para el control de zoom del mapa

         * @param img BufferedImage

         * @param vp ViewPort

         */

        public void setPixelParams(BufferedImage img, FLyrRasterSE lyr) {

                if(view != null && zoomPixel != null && img != null && lyr != null) {

                        Rectangle2D extBuf;

                        try {

                                extBuf = viewRequestManager.request(null);

                                viewRequestManager.setDrawParams(img, extBuf);        

                                zoomPixel.setDrawParams(img, extBuf, extBuf.getHeight()/img.getHeight(), new Point2D.Double(extBuf.getCenterX(), extBuf.getCenterY()));

                                zoomPixel.getPixelGraphicLayer();

                        } catch (InvalidRequestException e1) {

                                RasterToolsUtil.messageBoxError("error_setview_preview", this);

                        }

                }

        }

        /**

         * 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) {

                                lyrToGeoref = initDialog.getLayer();

                                if(lyrToGeoref == null) 

                                        RasterToolsUtil.messageBoxError("layer_not_loaded", this);

                                else {

                                        PluginServices.getMDIManager().closeWindow(initDialog);

                                        initWindows();

                                }

                        }

                }

                // 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 = view.getCursorAdjustedWorldCoordinates(zoomPixel.getCanvasWidth(), zoomPixel.getCanvasHeight());

                                viewRequestManagerZoomPixel.request(ext);

                        } catch (InvalidRequestException e1) {

                                RasterToolsUtil.messageBoxError("error_setview_preview", this);

                        }

                }

        }

}