Application: gvSIG desktop » gvSIG 3D

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 gvSIG Association

 *

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

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.view3d.main;

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.Dimension;

import java.awt.event.MouseEvent;

import java.awt.event.MouseListener;

import javax.swing.JButton;

import javax.swing.JFrame;

import javax.swing.JMenu;

import javax.swing.JMenuBar;

import javax.swing.JMenuItem;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JToolBar;

import javax.swing.JTree;

import javax.swing.SwingUtilities;

import javax.swing.WindowConstants;

import javax.swing.tree.DefaultMutableTreeNode;

import javax.swing.tree.DefaultTreeModel;

import javax.swing.tree.TreePath;

import javax.swing.tree.TreeSelectionModel;

import org.cresques.cts.IProjection;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.layers.CancelationException;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.LayerCollectionEvent;

import org.gvsig.fmap.mapcontext.layers.LayerCollectionListener;

import org.gvsig.fmap.mapcontext.layers.LayerPositionEvent;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.MapControlCreationException;

import org.gvsig.fmap.mapcontrol.MapControlLocator;

import org.gvsig.fmap.mapcontrol.MapControlManager;

import org.gvsig.fmap.mapcontrol.tools.PanListenerImpl;

import org.gvsig.fmap.mapcontrol.tools.PointSelectionListener;

import org.gvsig.fmap.mapcontrol.tools.ZoomInListenerImpl;

import org.gvsig.fmap.mapcontrol.tools.ZoomOutListenerImpl;

import org.gvsig.fmap.mapcontrol.tools.Behavior.Behavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.MoveBehavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.PointBehavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.RectangleBehavior;

import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

import org.gvsig.view3d.swing.api.View3DSwingLocator;

import org.gvsig.view3d.swing.api.View3DSwingManager;

/**

 * @author <a href="mailto:lmarques@disid.com">Lluis Marques</a>

 *

 */

public class Main {

    private static final Logger logger = LoggerFactory.getLogger(Main.class);

    final View3DSwingManager view3DManager;

    final MapControlManager mapControlManager;

    final DataManager dataManager;

    final MapContextManager mapContextManager;

    final String DEFAULT_CRS_CODE = "EPSG:4326";

    final PaletteActions paletteActions;

    IProjection defaultProjection;

    MapControl mapControl;

    JFrame mainFrame;

    JTree tree;

    DefaultTreeModel treeModel;

    DefaultMutableTreeNode rootNode;

    MouseListener treeMouseListener;

    public Main() {

        view3DManager = View3DSwingLocator.getManager();

        mapControlManager = MapControlLocator.getMapControlManager();

        dataManager = DALLocator.getDataManager();

        mapContextManager = MapContextLocator.getMapContextManager();

        defaultProjection = CRSFactory.getCRS(DEFAULT_CRS_CODE);

        paletteActions = new PaletteActions(this);

        MapContextLocator

            .registerDefaultOrderManager(DummieLayerOrderManager.class);

    }

    public static void main(String[] args) {

        new DefaultLibrariesInitializer().fullInitialize();

        Main main = new Main();

        main.doMain();

    }

    private void doMain() {

        try {

            MapContext mapContext =

                new MapContext(new ViewPort(defaultProjection));

            mapControl = mapControlManager.createJMapControlPanel(mapContext);

            addCollecionListener();

            mapControl.addBehavior("zoom", new Behavior[] {

                new RectangleBehavior(new ZoomInListenerImpl(mapControl)),

                new PointBehavior(new ZoomOutListenerImpl(mapControl),

                    Behavior.BUTTON_RIGHT) });

            mapControl.addBehavior("pan", new MoveBehavior(new PanListenerImpl(

                mapControl)));

            mapControl.addBehavior("pointSelection",

                new Behavior[] { new PointBehavior(new PointSelectionListener(

                    mapControl)) });

            IProjection defaultProjection = CRSFactory.getCRS(DEFAULT_CRS_CODE);

            mapControl.getViewPort().setProjection(defaultProjection);

            mapControl.setTool("pan");

        } catch (MapControlCreationException e) {

            logger.error("Can't create mapControl", e);

        }

        // Create JFrame to show data

        mainFrame = new JFrame("View 3D test app");

        mainFrame.setDefaultCloseOperation(WindowConstants.EXIT_ON_CLOSE);

        mainFrame.setPreferredSize(new Dimension(1055, 680));

        mainFrame.add(mapControl, BorderLayout.CENTER);

        // Create menu bar

        createMenu();

        // Create tools bar

        createToolBar();

        // Create simple TOC

        createTOC();

        // Display the window.

        mainFrame.pack();

        mainFrame.setLocation(100, 0);

        mainFrame.setVisible(true);

    }

    private void addCollecionListener() {

        mapControl.getMapContext().getLayers()

            .addLayerCollectionListener(new LayerCollectionListener() {

                public void visibilityChanged(LayerCollectionEvent e)

                    throws CancelationException {

                }

                public void layerRemoving(LayerCollectionEvent e)

                    throws CancelationException {

                }

                public void layerRemoved(LayerCollectionEvent e) {

                }

                public void layerMoving(LayerPositionEvent e)

                    throws CancelationException {

                }

                public void layerMoved(LayerPositionEvent e) {

                }

                public void layerAdding(LayerCollectionEvent e)

                    throws CancelationException {

                }

                public void layerAdded(LayerCollectionEvent e) {

                    FLayer layer = e.getAffectedLayer();

                    DefaultMutableTreeNode layerNode =

                        new DefaultMutableTreeNode(layer.getName());

                    getRootNode().add(layerNode);

                    getTreeModel().reload(getRootNode());

                    // Expand nodes

                    getTree().expandRow(0);

                }

            });

    }

    private void createMenu() {

        // Create the menu bar.

        JMenuBar menuBar = new JMenuBar();

        // Build the menu.

        JMenu menuFile = new JMenu("File");

        menuFile.add(new JMenuItem(paletteActions.getAction("exit")));

        JMenu menuLayer = new JMenu("Layer");

        menuLayer.add(new JMenuItem(paletteActions

            .getAction("add-vectorial-layer")));

        menuLayer.add(new JMenuItem(paletteActions

            .getAction("add-raster-layer")));

        menuLayer.add(new JMenuItem(paletteActions.getAction("remove-layer")));

        JMenu menu3D = new JMenu("3D");

        menu3D.add(new JMenuItem(paletteActions

            .getAction("create-sphere-view-3d")));

        menu3D.add(new JMenuItem(paletteActions

            .getAction("create-flat-view-3d")));

        menu3D.add(new JMenuItem(paletteActions

            .getAction("create-reload-view-3d")));

        menu3D.add(new JMenuItem(paletteActions

            .getAction("create-synchronize-view-3d")));

        menuBar.add(menuFile);

        menuBar.add(menuLayer);

        menuBar.add(menu3D);

        mainFrame.setJMenuBar(menuBar);

    }

    private void createToolBar() {

        JToolBar toolBar = new JToolBar();

        toolBar

            .add(new JButton(paletteActions.getAction("add-vectorial-layer")));

        toolBar.add(new JButton(paletteActions.getAction("add-raster-layer")));

        toolBar.add(new JButton(paletteActions.getAction("remove-layer")));

        toolBar.add(new JButton(paletteActions

            .getAction("create-sphere-view-3d")));

        toolBar

            .add(new JButton(paletteActions.getAction("create-flat-view-3d")));

        toolBar.add(new JButton(paletteActions

            .getAction("create-reload-view-3d")));

        toolBar.add(new JButton(paletteActions

            .getAction("create-synchronize-view-3d")));

        mainFrame.add(toolBar, BorderLayout.PAGE_START);

    }

    private void createTOC() {

        JPanel toc = new JPanel();

        toc.setBackground(Color.WHITE);

        rootNode = getRootNode();

        treeModel = getTreeModel();

        tree = getTree();

        toc.add(tree);

        mainFrame.add(toc, BorderLayout.WEST);

    }

    private DefaultMutableTreeNode getRootNode() {

        if (rootNode == null) {

            rootNode = new DefaultMutableTreeNode("Layers");

        }

        return rootNode;

    }

    protected DefaultTreeModel getTreeModel() {

        if (treeModel == null) {

            treeModel = new DefaultTreeModel(getRootNode());

        }

        return treeModel;

    }

    protected JTree getTree() {

        if (tree == null) {

            tree = new JTree(getTreeModel());

            tree.getSelectionModel().setSelectionMode(

                TreeSelectionModel.SINGLE_TREE_SELECTION);

            tree.addMouseListener(getTreeMouseListener());

        }

        return tree;

    }

    private MouseListener getTreeMouseListener() {

        if (treeMouseListener == null) {

            treeMouseListener = new MouseListener() {

                public void mouseReleased(MouseEvent e) {

                }

                public void mousePressed(MouseEvent e) {

                    int selRow = tree.getRowForLocation(e.getX(), e.getY());

                    TreePath selPath =

                        tree.getPathForLocation(e.getX(), e.getY());

                    if (selRow != -1) {

                        if (e.getClickCount() == 2

                            && SwingUtilities.isLeftMouseButton(e)) {

                            DefaultMutableTreeNode selectedNode =

                                (DefaultMutableTreeNode) selPath

                                    .getLastPathComponent();

                            if(selectedNode == getRootNode()){

                                return;

                            }

                            // TODO Open 3D properties dialog to set layer load

                            // mode.

                            ThreadSafeDialogsManager tsdm =

                                ToolsSwingLocator.getThreadSafeDialogsManager();

                            tsdm.messageDialog(

                                "Open " + selectedNode.toString()

                                    + " 3D properties"

                                    + " dialog to set layer load mode",

                                "3D properties",

                                JOptionPane.INFORMATION_MESSAGE);

                        }

                    }

                }

                public void mouseExited(MouseEvent e) {

                }

                public void mouseEntered(MouseEvent e) {

                }

                public void mouseClicked(MouseEvent e) {

                }

            };

        }

        return treeMouseListener;

    }

    public MapControl getMapControl() {

        return this.mapControl;

    }

    public MapContext getMapContext() {

        return this.mapControl.getMapContext();

    }

    public IProjection getDefaultProjection() {

        return defaultProjection;

    }

}