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.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.gui.beans.incrementabletask.IIncrementable;

import org.gvsig.gui.beans.incrementabletask.IncrementableEvent;

import org.gvsig.gui.beans.incrementabletask.IncrementableListener;

import org.gvsig.gui.beans.incrementabletask.IncrementableTask;

import org.gvsig.i18n.Messages;

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

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

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

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

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.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.layer.GPGraphic;

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

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.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 implements IIncrementable, IncrementableListener {

        public static final int               UNDEFINED                      = -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;

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

        private ViewRasterRequestManager      viewRasterRequestManager       = null;  

        private ViewMapRequestManager         viewMapRequestManager          = null;

        private GeorefControlPanel            georefControlPanel             = null;

        private TableWrapper                  table                          = null;

        private MapControl                    mapCtrl                        = null;

        private GeoreferencingOptionsDataModel 

                                              dataModel                      = null;

        private SynchronizedViews             syncViews                      = null;

        private ViewDialogListener            listener                       = null;

        private GeoreferencingProcessActions  processActions                 = null;

        private ApplicationControlsListener   appControlsListener            = null;

        private PointListListener             pointListListener              = null;

        public static boolean                 enableTableEvent               = true;

        protected IncrementableTask           incrementableLauncherTask      = null;

        private int                           percentLauncher                = 0;

        /**

         * Crea la ventana de IncrementableTask

         */

        public IncrementableTask getIncrementableTask() {

                if (incrementableLauncherTask == null) {

                        incrementableLauncherTask = new IncrementableTask(this);

                        incrementableLauncherTask.addIncrementableListener(this);

                }

                return incrementableLauncherTask;

        }

        public void initialize() {

                fManager = GeoreferencingFrameworkLocator.getManager();

                sManager = GeoreferencingSwingLocator.getSwingManager();

                listener = new ViewDialogListener(this);

                processActions = new GeoreferencingProcessActions();

                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, 380, Messages.getText("georeferencing"));

        }

        /**

         * Gets the Mapcontrol for the view selected by the user 

         * @return

         */

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

        }

        /**

         * Gets the Mapcontrol for the first view found. This is useful without reference cartography.

         * When the application ends and the layer is loaded, the layer generated will be loaded 

         * in the first view found. 

         * @return

         */

        public MapControl getFirstMapCtrl() {

                if(mapCtrl == null) {

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

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

                                if(windowList[i] instanceof IView) {

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

                createAppControlsListener();

                createPointListListener();

                initCartography();

                /*getIncrementableTask().showWindow();

                getIncrementableTask().start();

                SwingUtilities.invokeLater(new Runnable() {

                        public void run() {

                                initCartography();

                        }

                });*/

        }

        /**

         * Builds the raster view

         * @return

         */

        public GeoreferencingView getViewRaster() {

                if(viewRaster == null) {

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

                        viewRasterRequestManager = new ViewRasterRequestManager(viewRaster, dataModel.getDataStore());

                        viewRaster.setMinxMaxyUL(false);

                        viewRaster.setViewListener(listener);

                        viewRaster.setRequestListener(viewRasterRequestManager);

                        viewRaster.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

                }

                return viewRaster;

        }

        /**

         * Builds the map view

         * @return

         */

        public GeoreferencingView getViewMap() {

                if(viewMap == null) {

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

                        viewMap.setMinxMaxyUL(true);

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

                        viewMap.setViewListener(listener);

                        viewMap.setRequestListener(viewMapRequestManager);

                        viewMap.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

                }

                return viewMap;

        }

        /**

         * Builds the zoom for the map view

         * @return

         */

        public GeoreferencingView getZoomMap() {

                if(zoomMap == null) {

                        zoomMap = sManager.createZoom(true, false);

                        zoomMap.setMinxMaxyUL(true);

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

                        zoomMap.setRequestListener(zoomMapRequestManager);

                        zoomMap.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

                }

                return zoomMap;

        }

        /**

         * Builds the zoom for the raster view

         * @return

         */

        public GeoreferencingView getZoomRaster() {

                if(zoomRaster == null) {

                        zoomRaster = sManager.createZoom(true, false);

                        zoomRasterRequestManager = new ViewRasterRequestManager(zoomRaster, dataModel.getDataStore());

                        zoomRaster.setMinxMaxyUL(false);

                        zoomRaster.setRequestListener(zoomRasterRequestManager);

                        zoomRaster.setTool(GeoreferencingSwingLibrary.SELECT_POINT_TOOL, listener);

                }

                return zoomRaster;

        }

        public TableWrapper getTableWrapper() {

                if(table == null) {

                        table = new TableWrapper(

                                        getDataModel(), 

                                        getGeorefControlPanel());

                }

                return table;

        }

        public void createAppControlsListener() {

                if(appControlsListener == null) {

                        appControlsListener = new ApplicationControlsListener(getTableWrapper(), this);

                        getTableWrapper().getTable().addListener(appControlsListener);

                }

        }

        public void createPointListListener() {

                if(pointListListener == null) {

                        pointListListener = new PointListListener(this);

                }

        }

        public void buildNewPointList() {

                GeoPointList geoPointList = getDataModel().getDataStore().getGeoPointList();

                long id = System.currentTimeMillis();

                if(geoPointList == null)

                        return;

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

                        GeoPoint gp = geoPointList.get(i);

                        gp.setNumber(i);

                        GPGraphic graphicGCP = new GPGraphic(gp);

                        graphicGCP.setId(id);                

                        getDataModel().getGCPList().add(graphicGCP);

                        getTableWrapper().getTable().addRow(new Object[] {

                                        true, 

                                        gp.getNumber(),

                                        gp.getMapPoint().getX(),

                                        gp.getMapPoint().getY(),

                                        gp.getPixelPoint().getX(),

                                        gp.getPixelPoint().getY(),

                                        gp.getErrorX(),

                                        gp.getErrorY(),

                                        gp.getRms(),

                                        id});

                        id += 1;

                        if(i == geoPointList.size() - 1)

                                getDataModel().getGCPList().setSelectedPoint(i);

                }

                getTableWrapper().updateErrors();

                getSyncViews().redrawPoints();

        }

        /**

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

                getDataModel().setType(WITHOUT_MAP);

                viewRaster = getViewRaster();

                zoomRaster = getZoomRaster();

                PagedTable pt = getTableWrapper().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, getGeorefControlPanel(), pt, listener);

                try {

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

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

                } catch (Exception e1) {

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

                        fManager.closeAllWindows();

                }

                buildNewPointList();

        }

        /**

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

                getDataModel().setType(WITH_MAP);

                viewRaster = getViewRaster();

                zoomRaster = getZoomRaster();

                viewMap = getViewMap();

                zoomMap = getZoomMap();

                PagedTable pt = getTableWrapper().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, getGeorefControlPanel(), pt, listener);

                getDataModel().setType(WITH_MAP);

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

                        processActions.setViewMapRequest(viewMapRequestManager);

                } catch (Exception e1) {

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

                        fManager.closeAllWindows();

                }

                buildNewPointList();

        }

        /**

         * Initializes views with the cartography

         * @throws InvalidRequestException

         */

        private void initCartography() {

                try {

                        percentLauncher = 10;

                        if(getDataModel().getType() == WITH_MAP) {

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

                                percentLauncher = 25;

                                //Inicializamos el zoom de mapa

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

                                zoomMapRequestManager.initRequest(ext);

                        }

                        percentLauncher = 50;

                        //Inicializamos la vista de raster

                        viewRasterRequestManager.initRequest(dataModel.getRasterExtent());

                        percentLauncher = 75;

                        //Inicializamos el zoom de raster

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

                        zoomRasterRequestManager.initRequest(dataModel.getRasterExtent());

                        zoomRasterRequestManager.request(ext);

                        percentLauncher = 100;

                } catch (InvalidRequestException e1) {

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

                        fManager.closeAllWindows();

                } /*finally {

                        getIncrementableTask().processFinalize();

                }*/

        }

        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 GeoreferencingProcessActions getProcessActions() {

                return processActions;

        }

        public GeoreferencingOptionsDataModel getDataModel() {

                return GeoreferencingSwingLocator.getSwingManager().getDataModel();

        }

        //***********************************************

        //Launcher process

        public void actionCanceled(IncrementableEvent e) {

        }

        public void actionResumed(IncrementableEvent e) {

        }

        public void actionSuspended(IncrementableEvent e) {

        }

        public String getLabel() {

                return Messages.getText("loading_georef_windows");

        }

        public String getLog() {

                return Messages.getText("loading_georef_windows");

        }

        public int getPercent() {

                return percentLauncher;

        }

        public String getTitle() {

                return Messages.getText("loading");

        }

        public boolean isCancelable() {

                return false;

        }

        public boolean isPausable() {

                return false;

        }

}