Application: gvSIG desktop

package org.gvsig.fmap.mapcontrol.swing.pickercontroller.impl;

import org.gvsig.tools.swing.api.documentfilters.DoubleDocumentFilter;

import java.awt.Dimension;

import java.awt.Image;

import java.awt.event.ActionEvent;

import java.net.URL;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JTextField;

import javax.swing.JToggleButton;

import javax.swing.SwingConstants;

import javax.swing.text.JTextComponent;

import org.apache.commons.lang.StringUtils;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.Geometry.DIMENSIONS;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.GeometryUtils;

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

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

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

import org.gvsig.fmap.mapcontrol.tools.Events.EnvelopeEvent;

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

import org.gvsig.tools.IllegalValueException;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.CoercionException;

import org.gvsig.tools.dataTypes.DataType;

import org.gvsig.tools.i18n.I18nManager;

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

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

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

import org.gvsig.tools.swing.api.pickercontroller.AbstractPickerController;

import org.gvsig.tools.swing.icontheme.IconTheme;

import org.gvsig.tools.swing.icontheme.IconThemeManager;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class EnvelopePickerControllerImpl extends AbstractPickerController<Envelope>{

        public static void selfRegister() {

            IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

            IconTheme theme = iconThemeManager.getActive();

            URL url = ScalePickerControllerImpl.class.getClassLoader().getResource("images/picker-envelope-cursor-capture.png");

            if( url != null ) {

                ImageIcon icon = new ImageIcon(url);

                theme.registerDefault("mapcontrol", "picker", "picker-envelope", "picker-envelope-cursor-capture", icon, url);

            }

            url = ScalePickerControllerImpl.class.getClassLoader().getResource("images/picker-envelope-capture.png");

            if( url != null ) {

                ImageIcon icon = new ImageIcon(url);

                theme.registerDefault("mapcontrol", "picker", "picker-envelope", "picker-envelope-capture", icon, url);

            }

            url = ScalePickerControllerImpl.class.getClassLoader().getResource("images/picker-envelope-from-mapcontrol.png");

            if( url != null ) {

                ImageIcon icon = new ImageIcon(url);

                theme.registerDefault("mapcontrol", "picker", "picker-envelope", "picker-envelope-from-mapcontrol", icon, url);

            }

            ToolsSwingUtils.registerSubgroupIconScreenshot(

                    EnvelopePickerControllerImpl.class, 

                    "picker",

                    "picker-envelope",

                    "/images/screenshots/picker-envelope1.png"

            );    

        }

    private class CaptureEnvelopeListener implements RectangleListener  {

        @Override

        public void rectangle(EnvelopeEvent event) throws BehaviorException {

            set(event.getWorldCoordRect());

        }

        @Override

        public Image getImageCursor() {

            return captureCursor.getImage();

        }

        @Override

        public boolean cancelDrawing() {

            return false;

        }

    }

    private final MapControl mapControl;

    private final JTextField txtUpperLeftX;

    private final JTextField txtUpperLeftY;

    private final JTextField txtLowerRightX;

    private final JTextField txtLowerRightY;

    private final JTextComponent txtEnvelope;

    private final JButton btnMapControlEnvelope;

    private final JToggleButton btnCapture;

    private ImageIcon captureCursor;    

    private RectangleBehavior captureTool;

    private String previosTool;

    public EnvelopePickerControllerImpl(

            MapControl mapControl,

            JTextField txtUpperLeftX, 

            JTextField txtUpperLeftY, 

            JTextField txtLowerRightX, 

            JTextField txtLowerRightY, 

            JButton btnMapControlEnvelope, 

            final JToggleButton btnCapture

        ) {

        this.mapControl = mapControl;

        this.txtEnvelope = null;

        this.txtUpperLeftX = txtUpperLeftX;

        this.txtUpperLeftY = txtUpperLeftY;

        this.txtLowerRightX = txtLowerRightX;

        this.txtLowerRightY = txtLowerRightY;

        this.btnMapControlEnvelope = btnMapControlEnvelope;

        this.btnCapture = btnCapture;

        this.initComponents();      

    }

    public EnvelopePickerControllerImpl(

            MapControl mapControl,

            JTextComponent txtEnvelope, 

            JButton btnMapControlEnvelope, 

            final JToggleButton btnCapture

        ) {

        this.mapControl = mapControl;

        this.txtEnvelope = txtEnvelope;

        this.txtUpperLeftX = new JTextField();

        this.txtUpperLeftY = new JTextField();

        this.txtLowerRightX = new JTextField();

        this.txtLowerRightY = new JTextField();

        this.btnMapControlEnvelope = btnMapControlEnvelope;

        this.btnCapture = btnCapture;

        this.initComponents();

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        I18nManager i18n = ToolsLocator.getI18nManager();

        this.captureCursor = new ImageIcon(this.getIcon("picker-envelope-cursor-capture").getImage());

        if( this.btnCapture!=null ) {

            this.btnCapture.setIcon(this.getIcon("picker-envelope-capture"));

            this.btnCapture.setToolTipText(i18n.getTranslation("_Capture_an_area_in_the_view"));

            if( this.btnCapture.getText().equals("...") ) {

                this.btnCapture.setText("");

            }

            this.btnCapture.addActionListener((ActionEvent e) -> {

                doCaptureEnvelope(btnCapture.isSelected());

            });

            this.btnCapture.setEnabled(this.mapControl!=null);

        }

        if( this.btnMapControlEnvelope!=null ) {

            this.btnMapControlEnvelope.setIcon(this.getIcon("picker-envelope-from-mapcontrol"));

            this.btnMapControlEnvelope.setToolTipText(i18n.getTranslation("_Capture_the_view_area"));

            if( this.btnMapControlEnvelope.getText().equals("...") ) {

                this.btnMapControlEnvelope.setText("");

            }

            this.btnMapControlEnvelope.addActionListener((ActionEvent e) -> {

                doSetEnvelopeFromMapControlEnvelope();

            });

            this.btnMapControlEnvelope.setEnabled(this.mapControl!=null);

        }

        if( this.mapControl!=null ) {

            this.captureTool = new RectangleBehavior(new CaptureEnvelopeListener());

            this.mapControl.addBehavior("picker-envelope-capture", captureTool);

            this.previosTool = this.mapControl.getCurrentTool();

        }

        if( this.txtEnvelope==null ) {

            toolsSwingManager.addClearButton(this.txtUpperLeftX);

            toolsSwingManager.addClearButton(this.txtUpperLeftY);

            toolsSwingManager.addClearButton(this.txtLowerRightX);

            toolsSwingManager.addClearButton(this.txtLowerRightY);

            toolsSwingManager.setDefaultPopupMenu(this.txtUpperLeftX);

            toolsSwingManager.setDefaultPopupMenu(this.txtUpperLeftY);

            toolsSwingManager.setDefaultPopupMenu(this.txtLowerRightX);

            toolsSwingManager.setDefaultPopupMenu(this.txtLowerRightY);

            this.txtUpperLeftX.setHorizontalAlignment(SwingConstants.RIGHT);

            this.txtUpperLeftY.setHorizontalAlignment(SwingConstants.RIGHT);

            this.txtLowerRightX.setHorizontalAlignment(SwingConstants.RIGHT);

            this.txtLowerRightY.setHorizontalAlignment(SwingConstants.RIGHT);

            this.txtUpperLeftX.setText("#########,00000");

            Dimension dim = this.txtUpperLeftX.getPreferredSize();

            this.txtUpperLeftX.setText("");

            this.txtUpperLeftX.setPreferredSize(dim);

            this.txtUpperLeftY.setPreferredSize(dim);

            this.txtLowerRightX.setPreferredSize(dim);

            this.txtLowerRightY.setPreferredSize(dim);

            DoubleDocumentFilter.install(this.txtUpperLeftX);

            DoubleDocumentFilter.install(this.txtUpperLeftY);

            DoubleDocumentFilter.install(this.txtLowerRightX);

            DoubleDocumentFilter.install(this.txtLowerRightY);

        } else {

            toolsSwingManager.addClearButton(this.txtEnvelope);

            toolsSwingManager.setDefaultPopupMenu(this.txtEnvelope);

//            this.txtEnvelope.setEditable(false);

        }

    }

    protected void doSetEnvelopeFromMapControlEnvelope() {

        if( !this.isEditable() || mapControl==null ) {

            return;

        }

        try {

            Envelope env = this.mapControl.getViewPort().getEnvelope();

            if( env == null ) {

                return;

            }

            Envelope theEnvelope = (Envelope) env.clone();

            this.set(theEnvelope);

        } catch (Exception ex) {

            LOG.warn("Can't get envelope from mapcontrol.", ex);

        }

    }

    protected void doCaptureEnvelope(boolean enabled) {

        if( !this.isEditable() || mapControl==null ) {

            return;

        }

        Envelope env = this.mapControl.getViewPort().getEnvelope();

        if( env == null ) {

            return;

        }

        if( enabled ) {

            this.previosTool = this.mapControl.getCurrentTool();

            this.mapControl.setTool("picker-envelope-capture");

        } else {

            this.mapControl.setTool(this.previosTool);

        }

    }

    protected String getEnvelopeAsString(Envelope value) {

        String s = String.format(

            "%+f, %+f, %+f, %+f",

            value.getMinimum(Geometry.DIMENSIONS.X), 

            value.getMaximum(Geometry.DIMENSIONS.Y),

            value.getMaximum(Geometry.DIMENSIONS.X), 

            value.getMinimum(Geometry.DIMENSIONS.Y)

        );

        return s;

    }

    @Override

    public boolean isValid() {

        if( this.txtEnvelope==null ) {

            if( DoubleDocumentFilter.isEmpty(this.txtUpperLeftX) 

                    && DoubleDocumentFilter.isEmpty(this.txtUpperLeftY)

                    && DoubleDocumentFilter.isEmpty(this.txtLowerRightX)

                    && DoubleDocumentFilter.isEmpty(this.txtLowerRightY) ) {

                return true;

            }

            if( DoubleDocumentFilter.isEmpty(this.txtUpperLeftX) 

                    || DoubleDocumentFilter.isEmpty(this.txtUpperLeftY)

                    || DoubleDocumentFilter.isEmpty(this.txtLowerRightX)

                    || DoubleDocumentFilter.isEmpty(this.txtLowerRightY) ) {

                return false;

            }

            if( !DoubleDocumentFilter.isValid(this.txtUpperLeftX) ) {

                return false;

            }

            if( !DoubleDocumentFilter.isValid(this.txtUpperLeftY) ) {

                return false;

            }

            if( !DoubleDocumentFilter.isValid(this.txtLowerRightX) ) {

                return false;

            }

            if( !DoubleDocumentFilter.isValid(this.txtLowerRightY) ) {

                return false;

            }

            return true;

        } else {

            if( StringUtils.isBlank(this.txtEnvelope.getText()) ) {

                return true;

            }

            try {

                GeometryUtils.createFrom(this.txtEnvelope.getText());

            } catch(Exception ex) {

                return false;

            }

            return true;

        }

    }

    @Override

    public boolean isEmpty() {

        if( this.txtEnvelope==null ) {

            if( DoubleDocumentFilter.isEmpty(this.txtUpperLeftX) 

                    && DoubleDocumentFilter.isEmpty(this.txtUpperLeftY)

                    && DoubleDocumentFilter.isEmpty(this.txtLowerRightX)

                    && DoubleDocumentFilter.isEmpty(this.txtLowerRightY) ) {

                return true;

            }

            return false;

        } else {

            return StringUtils.isBlank(this.txtEnvelope.getText());

        }

    }

    @Override

    public Envelope get() {

        if( this.isEmpty() ) {

            return null;

        }

        if( !this.isValid() ) {

            throw new IllegalValueException();

        }

        Envelope envelope = null;

        if( this.txtEnvelope==null ) {

            try {

                double upperLeftX = DoubleDocumentFilter.getValue(this.txtUpperLeftX);

                double upperLeftY = DoubleDocumentFilter.getValue(this.txtUpperLeftY);

                double lowerRightX = DoubleDocumentFilter.getValue(this.txtLowerRightX);

                double lowerRightY = DoubleDocumentFilter.getValue(this.txtLowerRightY);

                GeometryManager geomManager = GeometryLocator.getGeometryManager();

                envelope = geomManager.createEnvelope(Geometry.SUBTYPES.GEOM2D);

                Point upperLeft = geomManager.createPoint(upperLeftX, upperLeftY, Geometry.SUBTYPES.GEOM2D);

                Point lowerRight = geomManager.createPoint(lowerRightX, lowerRightY, Geometry.SUBTYPES.GEOM2D);

                envelope.setUpperCorner(upperLeft);

                envelope.setLowerCorner(lowerRight);

            } catch (Exception ex) {

                return null;

            }

        } else {

            try {

                Geometry geom = GeometryUtils.createFrom(this.txtEnvelope.getText());

                envelope = geom.getEnvelope();

            } catch(Exception ex) {

                return null;

            }

        }

        if( this.mapControl!=null ) {

            envelope.setProjection(this.mapControl.getProjection());

        }

        return envelope;

    }

    @Override

    public void set(Envelope envelope) {

        if( envelope == null ) {

            this.clear();

            return;

        }

        if( envelope.isCollapsed() ) {

            this.clear();

            return;

        }

        if( this.txtEnvelope==null ) {

            this.txtUpperLeftX.setText(String.format("%+f", envelope.getMinimum(DIMENSIONS.X)));

            this.txtUpperLeftY.setText(String.format("%+f", envelope.getMaximum(DIMENSIONS.Y)));

            this.txtLowerRightX.setText(String.format("%+f", envelope.getMaximum(DIMENSIONS.X)));

            this.txtLowerRightY.setText(String.format("%+f", envelope.getMinimum(DIMENSIONS.Y)));

        } else {

            this.txtEnvelope.setText(envelope.getBox2D().convertToWKTQuietly());

        }

        fireChangeEvent();

    }

    @Override

    public void coerceAndSet(Object value) {

        if( value == null ) {

            this.clear();

            return;

        }

        DataType dataType = ToolsLocator.getDataTypesManager()

                .get(DataTypes.ENVELOPE);

        try {

            this.set((Envelope) dataType.coerce(value));

        } catch (CoercionException ex) {

            LOG.warn("Can't set value.", ex);

        }

    }

    @Override

    public void setEnabled(boolean enabled) {

        if( this.txtEnvelope==null ) {

            this.txtUpperLeftX.setEnabled(enabled);

            this.txtUpperLeftY.setEnabled(enabled);

            this.txtLowerRightX.setEnabled(enabled);

            this.txtLowerRightY.setEnabled(enabled);

        } else {

            this.txtEnvelope.setEnabled(enabled);

        }

        if( this.mapControl == null ) {

            this.btnCapture.setEnabled(false);

            this.btnMapControlEnvelope.setEnabled(false);

        } else {

            this.btnCapture.setEnabled(enabled);

            this.btnMapControlEnvelope.setEnabled(enabled);

        }

    }

    @Override

    public boolean isEnabled() {

        if( this.txtEnvelope!=null ) {

            return this.txtEnvelope.isEnabled();

        }

        return true;

    }

    @Override

    public void setEditable(boolean editable) {

        super.setEditable(editable);

        if( this.txtEnvelope==null ) {

            this.txtUpperLeftX.setEditable(editable);

            this.txtUpperLeftY.setEditable(editable);

            this.txtLowerRightX.setEditable(editable);

            this.txtLowerRightY.setEditable(editable);

        } else {

            this.txtEnvelope.setEditable(editable);

        }

    }

    public void clear() {

        if( this.txtEnvelope==null ) {

            this.txtUpperLeftX.setText("");

            this.txtUpperLeftY.setText("");

            this.txtLowerRightX.setText("");

            this.txtLowerRightY.setText("");

        } else {

            this.txtEnvelope.setText("");

        }

        fireChangeEvent();

    }

}