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.georeferencing.app.georeferencingclient;

import java.awt.geom.Rectangle2D;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JPanel;

import org.gvsig.andami.PluginServices;

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

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.documents.Document;

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.app.project.documents.view.gui.IView;

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

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

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

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.i18n.Messages;

import org.gvsig.raster.georeferencing.app.georeferencingclient.gui.CheckBoxColumnEditor;

import org.gvsig.raster.georeferencing.app.georeferencingclient.gui.CheckBoxColumnRenderer;

import org.gvsig.raster.georeferencing.app.georeferencingclient.gui.GCPModel;

import org.gvsig.raster.georeferencing.app.georeferencingclient.listener.ApplicationControlsListener;

import org.gvsig.raster.georeferencing.app.georeferencingclient.listener.LauncherDialogListener;

import org.gvsig.raster.georeferencing.app.georeferencingclient.listener.ViewDialogListener;

import org.gvsig.raster.georeferencing.app.georeferencingclient.listener.ViewMapRequestManager;

import org.gvsig.raster.georeferencing.app.georeferencingclient.listener.ViewRasterRequestManager;

import org.gvsig.raster.georeferencing.framework.GeoreferencingFrameworkLocator;

import org.gvsig.raster.georeferencing.framework.GeoreferencingFrameworkManager;

import org.gvsig.raster.georeferencing.framework.LayoutEnvironment;

import org.gvsig.raster.georeferencing.swing.GeoreferencingLauncher;

import org.gvsig.raster.georeferencing.swing.GeoreferencingOptionsDataModel;

import org.gvsig.raster.georeferencing.swing.GeoreferencingSwingLibrary;

import org.gvsig.raster.georeferencing.swing.GeoreferencingSwingLocator;

import org.gvsig.raster.georeferencing.swing.GeoreferencingSwingManager;

import org.gvsig.raster.georeferencing.swing.control.GeorefControlPanel;

import org.gvsig.raster.georeferencing.swing.exception.InvalidRequestException;

import org.gvsig.raster.georeferencing.swing.impl.GeoreferencingSwingImplLibrary;

import org.gvsig.raster.georeferencing.swing.impl.tool.SelectPointTool;

import org.gvsig.raster.georeferencing.swing.view.GeoreferencingView;

import org.gvsig.raster.georeferencing.swing.view.IViewRequestManager;

import org.gvsig.raster.georeferencing.swing.view.SynchronizedViews;

import org.gvsig.raster.swing.RasterSwingLocator;

import org.gvsig.raster.swing.RasterSwingManager;

import org.gvsig.raster.swing.pagedtable.ModelLoader;

import org.gvsig.raster.swing.pagedtable.PagedTable;

/**

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

        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;

        public static int                     DEFAULT_DEGREE                 = 3;

        public static int                     ZOOM_RATIO                     = 5;

        private GeoreferencingFrameworkManager 

                                              fManager                       = null;

        private GeoreferencingSwingManager    sManager                       = null;

        private RasterDataStore               storeToGeoref                  = null;

        private String                        selectedView                   = null;

        //Dialogos

        private GeoreferencingLauncher        launcher                       = null;

        private GeoreferencingView            viewRaster                     = null;

        private GeoreferencingView            viewMap                        = null;

        private GeoreferencingView            zoomMap                        = null;

        private GeoreferencingView            zoomRaster                     = null;

        private ViewRasterRequestManager      zoomRasterRequestManager       = null;  

        private ViewMapRequestManager         zoomMapRequestManager          = null;

        //Herramientas de selecci?n

        private SelectPointTool               viewRasterSelectPointTool      = null;

        private SelectPointTool               viewMapSelectPointTool         = null;

        private SelectPointTool               zoomRasterSelectPointTool      = null;

        private SelectPointTool               zoomMapSelectPointTool         = null;

        private GeorefControlPanel            georefControlPanel             = null;

        private PagedTable                    table                          = null;

        private ApplicationControlsListener   buttonsListener                = null;

        private MapControl                    mapCtrl                        = null;

        private GeoreferencingOptionsDataModel 

                                              dataModel                      = null;

        private GeoPointList                  geoPointList                   = null;

        private SynchronizedViews             syncViews                      = null;

        private ViewDialogListener            listener                       = null;

        private GeoreferencingProcessActions  processActions                 = null;

        //private LayersPointManager            layersPointManager             = null;

        class GeoPointListImpl extends ArrayList<GeoPoint> implements GeoPointList {

                private static final long serialVersionUID = 1L;

        }

        public void initialize() {

                fManager = GeoreferencingFrameworkLocator.getManager();

                sManager = GeoreferencingSwingLocator.getSwingManager();

                listener = new ViewDialogListener(this);

                processActions = new GeoreferencingProcessActions();

                //getLayerPointManager();        

                //getViewsManager();

                launcher();

        }

        /**

         * Launcher of the initial dialog

         */

        private void launcher() {

                ApplicationManager application = ApplicationLocator.getManager();

                List<Document> docs = application.getProjectManager().getCurrentProject().getDocuments();

                List<String> viewList = new ArrayList<String>();

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

                        Document doc = docs.get(i);

                        if(doc instanceof ViewDocument) {

                                viewList.add(doc.getName());

                        }

                }

                launcher = sManager.createWindowLauncher(viewList, DEFAULT_DEGREE);

                LauncherDialogListener listener = new LauncherDialogListener(this, launcher);

                launcher.setButtonsListener(listener);

                fManager.showWindow(launcher.getComponent(), 400, 420, Messages.getText("georeferencing"));

        }

        /**

         * Obtiene el gestor para capas de puntos de la aplicaci?n,

         * @return LayersPointManager

         */

        /*        if(layersPointManager == null)

                        layersPointManager = new LayersPointManager(this);

                return layersPointManager;

        }*/

        /**

         * Obtiene el gestor para las vistas de la aplicaci?n,

         * @return ViewsManager

         */

        /*public ViewsManager getViewsManager() {

                if(viewsManager == null)

                        viewsManager = new ViewsManager();

                return viewsManager;

        }*/

        public void setSelectedView(String selectedView) {

                this.selectedView = selectedView;

        }

        /**

         * Obtiene la lista de capas de la vista de gvSIG

         * @return

         */

        /*public FLayers getLayers() {

                MapControl mapControl = null;

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

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

                        if(windowList[i] instanceof IView) 

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

                }

                if(mapControl != null)

                        return mapControl.getMapContext().getLayers();

                return null;

        }*/

        private MapControl getMapCtrl() {

                if(mapCtrl == null) {

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

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

                                if(windowList[i] instanceof IView) {

                                        String name = ((IView)windowList[i]).getViewDocument().getName();

                                        if(name.equals(dataModel.getSelectedView())) {

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

                                        }

                                }

                        }                        

                }

                return mapCtrl;

        }

        public void endLauncherDialog(GeoreferencingOptionsDataModel dataModel) {

                this.dataModel = dataModel;

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

                        initWindowsWithMap(dataModel);

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

                        initWindowsWithoutMap(dataModel);

        }

        /**

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

         * de ventanas con cartograf?a de referencia.

         */

        private void initWindowsWithoutMap(GeoreferencingOptionsDataModel dataModel) {

                RasterDataStore store = dataModel.getDataStore();

                getOptionsDataModel().setType(WITHOUT_MAP);

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

                viewRaster = sManager.createView(true, false, true, listener);

                ViewRasterRequestManager viewRasterRequestManager = new ViewRasterRequestManager(viewRaster, store);

                viewRaster.setViewListener(listener);

                viewRaster.setRequestListener(viewRasterRequestManager);

                viewRaster.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

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

                zoomRaster = sManager.createZoom(true, false);

                zoomRasterRequestManager = new ViewRasterRequestManager(zoomRaster, store);

                zoomRaster.setRequestListener(zoomRasterRequestManager);

                zoomRaster.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

                PagedTable pt = getTable();

                //Creamos el layout

                LayoutEnvironment layoutEnv = fManager.createLayoutEnvironment();

                layoutEnv.setTable((JPanel)pt.getComponent());

                layoutEnv.setViewRasterToGeoref(viewRaster.getComponent());

                layoutEnv.setViewZoomRasterToGeoref(zoomRaster.getComponent());

                layoutEnv.setControlBar(getGeorefControlPanel().getComponent());

                fManager.createLayout(layoutEnv);

                GeoreferencingView[] v = new GeoreferencingView[]{null, viewRaster, null, zoomRaster};

                IViewRequestManager[] rm = new IViewRequestManager[]{null, viewRasterRequestManager, null, zoomRasterRequestManager};

                syncViews = sManager.createSyncViews(v, rm, getGeoPointList(), pt, listener);

//                

//                viewsManager.setViews(viewRaster, null, zoomRaster, null);

//                viewsManager.setRequestsManager(viewRasterRequestManager, null, zoomRasterRequestManager, null);

//                viewsManager.setTablePanel(table);

//                

                try {

                        viewRaster.setCursorSize(zoomRaster.getCanvasWidth() / ZOOM_RATIO, zoomRaster.getCanvasHeight() / ZOOM_RATIO);

                        viewRaster.setCursorPosition(viewRaster.getCanvasWidth() / 2, viewRaster.getCanvasHeight() / 2);

                        //Inicializamos la vista de raster

                        viewRasterRequestManager.initRequest(dataModel.getDataStore().getExtent().toRectangle2D());

                        //Inicializamos el zoom de raster

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

                        zoomRasterRequestManager.initRequest(dataModel.getDataStore().getExtent().toRectangle2D());

                        zoomRasterRequestManager.request(ext);

                } catch (InvalidRequestException e1) {

                        GeoreferencingSwingImplLibrary.messageBoxError("error_setview_preview", null, e1);

                        fManager.closeAllWindows();

                }catch (Exception e1) {

                        GeoreferencingSwingImplLibrary.messageBoxError("error_setview_preview", null, e1);

                        fManager.closeAllWindows();

                }

        }

        public PagedTable getTable() {

                if(table == null) {

                        RasterSwingManager manager = RasterSwingLocator.getSwingManager();

                        String[] columnNames = {

                                        "-", 

                                        Messages.getText("num"), 

                                        Messages.getText("worldx"), 

                                        Messages.getText("worldy"), 

                                        Messages.getText("imagex"), 

                                        Messages.getText("imagey"), 

                                        "Error X", "Error Y", "RMS"};

                        int[] columnSizes = {40, 37, -1, -1, -1, -1, -1, -1, -1};

                        GCPModel model = new GCPModel(columnNames);

                        ModelLoader loader = manager.createModelLoader(model);

                        CheckBoxColumnRenderer render = new CheckBoxColumnRenderer(null);

                        CheckBoxColumnEditor editor = new CheckBoxColumnEditor();

                        loader.setRenderForColumn(0, render);

                        loader.setCellEditorForColumn(0, editor);

                        loader.setColumnNames(columnNames);

                        loader.setColumnWidths(columnSizes);

                        table = manager.createPagedTable(loader);

                }

                return 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(GeoreferencingOptionsDataModel dataModel) {

                RasterDataStore store = dataModel.getDataStore();

                getOptionsDataModel().setType(WITH_MAP);

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

                viewRaster = sManager.createView(true, false, true, listener);

                ViewRasterRequestManager viewRasterRequestManager = new ViewRasterRequestManager(viewRaster, store);

                viewRaster.setViewListener(listener);

                viewRaster.setRequestListener(viewRasterRequestManager);

                viewRaster.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

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

                zoomRaster = sManager.createZoom(true, false);

                zoomRasterRequestManager = new ViewRasterRequestManager(zoomRaster, store);

                zoomRaster.setRequestListener(zoomRasterRequestManager);

                zoomRaster.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

                //Inicializaci?n de la ventana con la vista del terreno

                viewMap = sManager.createView(true, false, false, listener);

                ViewMapRequestManager viewMapRequestManager = new ViewMapRequestManager(viewMap, getMapCtrl());

                viewMap.setViewListener(listener);

                viewMap.setRequestListener(viewMapRequestManager);

                viewMap.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

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

                zoomMap = sManager.createZoom(true, false);

                zoomMapRequestManager = new ViewMapRequestManager(zoomMap, getMapCtrl());

                zoomMap.setRequestListener(zoomMapRequestManager);

                zoomMap.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

                PagedTable pt = getTable();

                //Creamos el layout

                LayoutEnvironment layoutEnv = fManager.createLayoutEnvironment();

                layoutEnv.setTable((JPanel)pt.getComponent());

                layoutEnv.setViewRasterToGeoref(viewRaster.getComponent());

                layoutEnv.setViewZoomRasterToGeoref(zoomRaster.getComponent());

                layoutEnv.setViewBaseCartography(viewMap.getComponent());

                layoutEnv.setViewZoomBaseCartography(zoomMap.getComponent());

                layoutEnv.setControlBar(getGeorefControlPanel().getComponent());

                fManager.createLayout(layoutEnv);

                GeoreferencingView[] v = new GeoreferencingView[]{viewMap, viewRaster, zoomMap, zoomRaster};

                IViewRequestManager[] rm = new IViewRequestManager[]{viewMapRequestManager, viewRasterRequestManager, zoomMapRequestManager, zoomRasterRequestManager};

                syncViews = sManager.createSyncViews(v, rm, getGeoPointList(), pt, listener);

                getOptionsDataModel().setType(WITH_MAP);

//                

//                viewsManager.setViews(viewRaster, viewMap, zoomRaster, zoomMap);

//                viewsManager.setRequestsManager(viewRasterRequestManager, viewMapRequestManager, zoomRasterRequestManager, zoomMapRequestManager);

//                viewsManager.setTablePanel(table);

//                

                try {

                        viewMap.setCursorSize(zoomMap.getCanvasWidth() / ZOOM_RATIO, zoomMap.getCanvasHeight() / ZOOM_RATIO);

                        viewRaster.setCursorSize(zoomRaster.getCanvasWidth() / ZOOM_RATIO, zoomRaster.getCanvasHeight() / ZOOM_RATIO);

                        viewMap.setCursorPosition(viewMap.getCanvasWidth() / 2, viewMap.getCanvasHeight() / 2);

                        viewRaster.setCursorPosition(viewRaster.getCanvasWidth() / 2, viewRaster.getCanvasHeight() / 2);

                        if(getMapCtrl().getMapContext().getLayers().getFullEnvelope() == null)

                                GeoreferencingSwingImplLibrary.messageBoxError("error_setview_preview", null);

                        //Inicializamos la vista de mapa

                        Envelope e = getMapCtrl().getMapContext().getLayers().getFullEnvelope();

                        Rectangle2D r = new Rectangle2D.Double(e.getMinimum(0), e.getMinimum(1), e.getLength(0), e.getLength(1));

                        viewMapRequestManager.initRequest(r);

                        //Inicializamos la vista de raster

                        viewRasterRequestManager.initRequest(store.getExtent().toRectangle2D());

                        //Inicializamos el zoom de mapa

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

                        zoomMapRequestManager.initRequest(ext);

                        //Inicializamos el zoom de raster

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

                        zoomRasterRequestManager.initRequest(store.getExtent().toRectangle2D());

                        zoomRasterRequestManager.request(ext);

//                        layersPointManager.createGCPPIxelLayer(viewRaster, viewRasterRequestManager, zoomRasterRequestManager);

//                        layersPointManager.createGCPMapLayer(viewMap, viewMapRequestManager, zoomMapRequestManager);

//                        

                        processActions.setViewMapRequest(viewMapRequestManager);

                } catch (InvalidRequestException e1) {

                        GeoreferencingSwingImplLibrary.messageBoxError("error_setview_preview", this, e1);

                        fManager.closeAllWindows();

                }catch (Exception e1) {

                        GeoreferencingSwingImplLibrary.messageBoxError("error_setview_preview", this, e1);

                        fManager.closeAllWindows();

                }

        }

        /**

         * A?ade una capa en la vista del mapa para previsualizar el resultado

         * @param lyr

         */

        /*public void addTestRasterLayer(FLyrRaster lyr) {

                viewsManager.addTestRasterLayer(lyr, 

                                getOptions().getAlgorithm(), 

                                getOptions().getInterpolationMethod(), 

                                getOptions().getDegree(), 

                                getOptions().getOutFile());

        }*/

        /**

         * Elimina la capa de test de la vista de mapa

         * @throws InvalidRequestException 

         */

        /*public void removeTestRasterLayer() {

                viewsManager.removeTestRasterLayer();

        }*/

        /**

         * Obtiene la ?ltima capa procesada con los puntos de  control

         * @return FLyrRaterSE

         */

        /*public FLyrRaster getLastTestLayer() {

                return viewsManager.getLastTestLayer();

        }*/

        /**

         * Gets the georeferencing options data model

         * @return

         */

        public GeoreferencingOptionsDataModel getOptionsDataModel() {

                GeoreferencingSwingManager manager = GeoreferencingSwingLocator.getSwingManager();

                return manager.getDataModel();

                //options = manager.createWindowOptions(Georeferencing.DEFAULT_DEGREE, this);

        }

        public GeoreferencingView getViewRaster() {

                return viewRaster;

        }

        public GeoreferencingView getViewMap() {

                return viewMap;

        }

        public GeoreferencingView getZoomMap() {

                return zoomMap;

        }

        public GeoreferencingView getZoomRaster() {

                return zoomRaster;

        }

        public ViewRasterRequestManager getZoomRasterRequestManager() {

                return zoomRasterRequestManager;

        }

        public ViewMapRequestManager getZoomMapRequestManager() {

                return zoomMapRequestManager;

        }

        public GeorefControlPanel getGeorefControlPanel() {

                if(georefControlPanel == null) {

                        georefControlPanel = sManager.createGeorefControlPanel();

                }

                return georefControlPanel;

        }

        public SynchronizedViews getSyncViews() {

                return syncViews;

        }

        public GeoPointList getGeoPointList() {

                if(geoPointList == null)

                        geoPointList = new GeoPointListImpl();

                return geoPointList;

        }

        /**

         * Obtiene la capa a georreferenciar

         * @return FLyrRaster

         */

        public RasterDataStore getLayer() {

                return launcher.getDataModel().getDataStore();

        }

}