Application: gvSIG desktop » gvSIG tools: Vector editing

package org.gvsig.vectorediting.swing.impl;

import java.awt.Image;

import java.awt.event.MouseEvent;

import java.awt.event.MouseWheelEvent;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.MapControlDrawer;

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

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

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

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

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

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

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

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

import org.gvsig.fmap.mapcontrol.tools.Listeners.ToolListener;

public class EditingCompoundBehavior extends CompoundBehavior {

        private enum Mode {

                EDITING, SELECTION

        };

        private Mode mode = Mode.EDITING;

        private IBehavior editing;

        private IBehavior selection;

        private IBehavior zoom;

        public final static int EDITING_INDEX = 0;

        public final static int SELECTION_INDEX = 1;

        public EditingCompoundBehavior(IBehavior editing) {

                super(new Behavior[0]);

                this.editing = editing;

                this.selection = null;

                this.zoom = new MouseWheelBehavior();

        }

        public void addMapBehavior(Behavior mt, boolean draw) {

                if (mt instanceof MouseWheelBehavior) {

                        return;

                }

                throw new UnsupportedOperationException();

        }

        public void removeMapBehavior(Behavior mt) {

                throw new UnsupportedOperationException();

        }

        public boolean containsBehavior(Behavior mt) {

                return (mt == this.editing || mt == this.selection);

        }

        public Behavior getBehavior(int index) {

                switch (index) {

                case EDITING_INDEX:

                        return (Behavior) this.editing;

                case SELECTION_INDEX:

                        return (Behavior) this.selection;

                default:

                        throw new IndexOutOfBoundsException();

                }

        }

        public boolean isDrawnBehavior(int index) {

                switch (mode) {

                case EDITING:

                        return index == EDITING_INDEX;

                case SELECTION:

                        return index == SELECTION_INDEX;

                default:

                        return false;

                }

        }

        public void setDrawnBehavior(int index, boolean draw) {

                switch (index) {

                case EDITING_INDEX:

                        if (draw) {

                                mode = Mode.EDITING;

                        } else {

                                mode = Mode.SELECTION;

                        }

                        break;

                case SELECTION_INDEX:

                        if (draw) {

                                mode = Mode.SELECTION;

                        } else {

                                mode = Mode.EDITING;

                        }

                        break;

                default:

                        throw new IndexOutOfBoundsException();

                }

        }

        public int size() {

                return 2;

        }

        public Image getImageCursor() {

                switch (mode) {

                default:

                case EDITING:

                        return this.editing.getImageCursor();

                case SELECTION:

                        return this.selection.getImageCursor();

                }

        }

        public void mouseClicked(MouseEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mouseClicked(e);

                        break;

                case SELECTION:

                        this.selection.mouseClicked(e);

                }

                this.zoom.mouseClicked(e);

        }

        public void mouseDragged(MouseEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mouseDragged(e);

                        break;

                case SELECTION:

                        this.selection.mouseDragged(e);

                }

                this.zoom.mouseDragged(e);

        }

        public void mouseEntered(MouseEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mouseEntered(e);

                        break;

                case SELECTION:

                        this.selection.mouseEntered(e);

                }

                this.zoom.mouseEntered(e);

        }

        public void mouseExited(MouseEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mouseExited(e);

                        break;

                case SELECTION:

                        this.selection.mouseExited(e);

                }

                this.zoom.mouseExited(e);

        }

        public void mouseMoved(MouseEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mouseMoved(e);

                        break;

                case SELECTION:

                        this.selection.mouseMoved(e);

                }

                this.zoom.mouseMoved(e);

        }

        public void mousePressed(MouseEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mousePressed(e);

                        break;

                case SELECTION:

                        this.selection.mousePressed(e);

                }

                this.zoom.mousePressed(e);

        }

        public void mouseReleased(MouseEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mouseReleased(e);

                        break;

                case SELECTION:

                        this.selection.mouseReleased(e);

                }

                this.zoom.mouseReleased(e);

        }

        public void mouseWheelMoved(MouseWheelEvent e) throws BehaviorException {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.mouseWheelMoved(e);

                        break;

                case SELECTION:

                        this.selection.mouseWheelMoved(e);

                }

                this.zoom.mouseWheelMoved(e);

        }

        public void paintComponent(MapControlDrawer renderer) {

                switch (mode) {

                default:

                case EDITING:

                        this.editing.paintComponent(renderer);

                        break;

                case SELECTION:

                        this.selection.paintComponent(renderer);

                }

                this.zoom.paintComponent(renderer);

        }

        public void setListener(ToolListener listener) {

                if (listener != null) {

                        throw new UnsupportedOperationException(

                                        "CompoundBehavior does not have listeners");

                }

        }

        public ToolListener getListener() {

                return null;

        }

        public void setMapControl(MapControl mc) {

                this.editing.setMapControl(mc);

                this.zoom.setMapControl(mc);

                if (this.selection == null){

                        if (mc !=null) {

//                                StatusBarListener sbl = new StatusBarListener(mc);

                                PointSelectionListener psl = new PointSelectionListener(mc);

                                this.selection = new CompoundBehavior(new Behavior[] {

                                                new PointBehavior(psl) }); //, new MouseMovementBehavior(sbl) });

                                this.selection.setMapControl(mc);

                        }

                } else {

                        this.selection.setMapControl(mc);

                }

        }

}