Application: gvSIG desktop » gvSIG tools: Vector editing

/*

 * Copyright 2014 DiSiD Technologies S.L.L. All rights reserved.

 * 

 * Project  : DiSiD org.gvsig.vectorediting.lib.impl 

 * SVN Id   : $Id$

 */

package org.gvsig.vectorediting.swing.impl;

import java.awt.Color;

import java.awt.Graphics;

import java.awt.Image;

import java.awt.event.MouseEvent;

import java.awt.geom.Point2D;

import java.awt.image.BufferedImage;

import java.util.HashMap;

import java.util.Map;

import org.cresques.cts.IProjection;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.ui.mdiFrame.MainFrame;

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

import org.gvsig.app.project.documents.view.toolListeners.StatusBarListener;

import org.gvsig.fmap.geom.Geometry;

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

import org.gvsig.fmap.mapcontext.MapContext;

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.FLayers;

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

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

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

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

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

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.MapControlDrawer;

import org.gvsig.fmap.mapcontrol.MapControlLocator;

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

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

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

import org.gvsig.utils.console.JConsole;

import org.gvsig.utils.console.ResponseListener;

import org.gvsig.utils.console.jedit.JEditTextArea;

import org.gvsig.vectorediting.lib.api.EditingLocator;

import org.gvsig.vectorediting.lib.api.EditingManager;

import org.gvsig.vectorediting.lib.api.EditingService;

import org.gvsig.vectorediting.lib.api.EditingServiceParameter;

import org.gvsig.vectorediting.lib.api.EditingServiceParameter.TYPE;

import org.gvsig.vectorediting.swing.api.EditingBehavior;

public class DefaultEditingBehavior extends Behavior implements EditingBehavior {

  private Map<FLyrVect, EditingService> serviceRegistration;

  private MapControl mapControl;

  private FLyrVect currentLayer;

  private EditingService activeService;

  private EditingServiceParameter currentParam;

  private Point adjustedPoint;

  private StatusBarListener sbl = null;

  private static final Image imageCursor = new BufferedImage(32, 32,

      BufferedImage.TYPE_INT_ARGB);

  static {

    Graphics g = imageCursor.getGraphics();

    int size1 = 15;

    int x = 16;

    int y = 16;

    g.setColor(Color.MAGENTA);

    g.drawLine((x - size1), (y), (x + size1), (y));

    g.drawLine((x), (y - size1), (x), (y + size1));

    g.drawRect((x - 6), (y - 6), 12, 12);

    g.drawRect((x - 3), (y - 3), 6, 6);

  }

  private LayerListener layerListener = new LayerListener() {

    public void activationChanged(LayerEvent e) {

      DefaultViewPanel view = getActiveView();

      FLayer layer = e.getSource();

      if (layer instanceof FLyrVect) {

        if (layer.isActive() && layer.isEditing()) {

          view.showConsole();

          getMapControl().setTool("VectorEditing");

          // cleanBehavior();

        }

        else if (!layer.isEditing() && layer.isActive()) {

          changeCurrentLayer((FLyrVect) layer);

          view.hideConsole();

          getMapControl().setTool("zoomIn");

          PluginServices.getMainFrame().setSelectedTool("ZOOM_IN");

        }

        changeCurrentLayer((FLyrVect) layer);

      }

    }

    public void drawValueChanged(LayerEvent e) {}

    public void editionChanged(LayerEvent e) {}

    public void nameChanged(LayerEvent e) {}

    public void visibilityChanged(LayerEvent e) {}

  };

  public void cleanBehavior() {

    // activeService = null;

    // currentParam = null;

    DefaultViewPanel view = getActiveView();

    view.getConsolePanel().addText("\nSeleccione un herramienta\n",

        JConsole.MESSAGE);

//    MainFrame main = PluginServices.getMainFrame();

//    main.setSelectedTool("");  We can define a default tool. I.E Zoom in.

  }

  public DefaultEditingBehavior(MapControl mapControl) {

    // TODO Auto-generated constructor stub

    if (mapControl != null) {

      this.mapControl = mapControl;

      serviceRegistration = new HashMap<FLyrVect, EditingService>();

      sbl = new StatusBarListener(mapControl);

      initMapControlListeners(mapControl);

      FLayers layers = mapControl.getMapContext().getLayers();

      for (int i = 0; i < layers.getLayersCount(); i++) {

        if (layers.getLayer(i) instanceof FLyrVect

            && layers.getLayer(i).isActive())

          changeCurrentLayer((FLyrVect) layers.getLayer(i));

      }

    }

  }

  public void activateTool(String name) {

    EditingManager manager = EditingLocator.getManager();

    if (currentLayer != null) {

      EditingService service = serviceRegistration.get(currentLayer);

      if (service == null) {

        service = (EditingService) manager

            .getEditingService(name, currentLayer.getFeatureStore());

        serviceRegistration.put(currentLayer, service);

      }

      if (service != null) {

        this.activeService = service;

        service.start();

        getNextParameter();

      }

    }

  }

  private void askQuestion(EditingServiceParameter param) {

    DefaultViewPanel viewActive = getActiveView();

    if (viewActive != null) {

      viewActive.getConsolePanel().addText(

          "\n" + "#" + param.getDescription() + " > ", JConsole.MESSAGE);

    }

  }

  private DefaultViewPanel getActiveView() {

    org.gvsig.andami.ui.mdiManager.IWindow activeWindow = PluginServices

        .getMDIManager().getActiveWindow();

    if (activeWindow instanceof DefaultViewPanel) {

      return (DefaultViewPanel) activeWindow;

    }

    return null;

  }

  @Override

  public ToolListener getListener() {

    return new ToolListener() {

      /**

       * 

       */

      public boolean cancelDrawing() {

        return false;

      }

      /**

       * 

       */

      public Image getImageCursor() {

        return imageCursor;

      }

    };

  }

  private void getNextParameter() {

    currentParam = activeService.next();

    if (currentParam == null) {

      activeService.finish();

      mapControl.rePaintDirtyLayers();

      activeService.start();

      getNextParameter();

    }

    else {

      askQuestion(currentParam);

      setCaretPosition(getActiveView());

    }

  }

  private void initMapControlListeners(MapControl mapControl) {

    DefaultViewPanel view = getActiveView();

    view.addConsoleListener("VectorEditingConsole", new ResponseListener() {

      public void acceptResponse(String response) {

        try {

          textEntered(response);

        }

        catch (RuntimeException e) {

          showErrorConsole(e);

          getNextParameter();

        }

      }

      private void showErrorConsole(RuntimeException e) {

        DefaultViewPanel view = getActiveView();

        view.getConsolePanel().addText(e.getMessage(), JConsole.ERROR);

      }

    });

    MapContext context = mapControl.getMapContext();

    FLayers layers = context.getLayers();

    layers.addLayerListener(layerListener);

    layers.addLayerCollectionListener(new LayerCollectionListener() {

      public void layerAdded(LayerCollectionEvent e) {

        FLayers layers2 = e.getLayers();

        for (int i = 0; i < layers2.getLayersCount(); i++) {

          FLayer layer = layers2.getLayer(i);

          if (layer instanceof FLyrVect) {

            ((FLyrVect) layer).addLayerListener(layerListener);

          }

        }

      }

      public void layerAdding(LayerCollectionEvent e)

          throws CancelationException {}

      public void layerMoved(LayerPositionEvent e) {}

      public void layerMoving(LayerPositionEvent e) throws CancelationException {}

      public void layerRemoved(LayerCollectionEvent e) {

        FLayers layers2 = e.getLayers();

        for (int i = 0; i < layers2.getLayersCount(); i++) {

          FLayer layer = layers2.getLayer(i);

          if (layer instanceof FLyrVect) {

            ((FLyrVect) layer).removeLayerListener(layerListener);

          }

        }

      }

      public void layerRemoving(LayerCollectionEvent e)

          throws CancelationException {}

      public void visibilityChanged(LayerCollectionEvent e)

          throws CancelationException {}

    });

  }

  public void mouseClicked(MouseEvent e) throws BehaviorException {

    // TODO Auto-generated method stub

    if (activeService != null) {

      if (currentParam != null) {

        switch (currentParam.getType()) {

          case POSITION:

            ViewPort vp = mapControl.getViewPort();

            Point point;

            point = vp.convertToMapPoint(e.getX(), e.getY());

            this.activeService.value(point);

          case LIST_POSITIONS:

          case OPTION:

          case SELECTION:

          case GEOMETRY:

          case VALUE:

        }

        getNextParameter();

      }

    }

  }

  public void mouseEntered(MouseEvent e) throws BehaviorException {

    // TODO Auto-generated method stub

  }

  public void mouseMoved(MouseEvent e) throws BehaviorException {

    ViewPort vp = mapControl.getViewPort();

    adjustedPoint = vp.convertToMapPoint(e.getX(),e.getY());

    showCoords(e.getPoint());

    getMapControl().repaint();

  }

  public void mousePressed(MouseEvent e) throws BehaviorException {

    // TODO Auto-generated method stub

  }

  public void mouseReleased(MouseEvent e) throws BehaviorException {

    // TODO Auto-generated method stub

  }

  public void paintComponent(MapControlDrawer mapControlDrawer) {

    super.paintComponent(mapControlDrawer);

    if (activeService == null || adjustedPoint == null) {

      return;

    }

    Geometry helperGeo = activeService.draw(adjustedPoint);

    if (helperGeo != null) {

      mapControl.getMapControlDrawer().draw(helperGeo,

          MapControlLocator.getMapControlManager().getAxisReferenceSymbol());

    }

  }

  private void setCaretPosition(DefaultViewPanel view) {

    JEditTextArea jeta = view.getConsolePanel().getTxt();

    jeta.requestFocusInWindow();

    jeta.setCaretPosition(jeta.getText().length());

  }

  private void changeCurrentLayer(FLyrVect layer) {

    this.currentLayer = layer;

    this.activeService = serviceRegistration.get(layer);

    if (activeService != null) {

      getNextParameter();

      MainFrame main = PluginServices.getMainFrame();

      main.setSelectedTool(activeService.getName());

    }

  }

  private void showCoords(Point2D point) {

    String[] axisText = new String[2];

    axisText[0] = "X = ";

    axisText[1] = "Y = ";

    ViewPort vp = mapControl.getMapContext().getViewPort();

    IProjection iProj = vp.getProjection();

    Point p = vp.convertToMapPoint(point);

    Point2D p2D = new Point2D.Double(p.getX(), p.getY());

    sbl.setFractionDigits(p2D);

    axisText = sbl.setCoorDisplayText(axisText);

    MainFrame mF = PluginServices.getMainFrame();

    if (mF != null) {

      mF.getStatusBar().setMessage(

          "units",

          PluginServices.getText(this,

              MapContext.getDistanceNames()[vp.getDistanceUnits()]));

      mF.getStatusBar().setControlValue("scale",

          String.valueOf(mapControl.getMapContext().getScaleView()));

      mF.getStatusBar().setMessage("projection", iProj.getAbrev());

      String[] coords = sbl.getCoords(p2D);

      mF.getStatusBar().setMessage("x", axisText[0] + coords[0]);

      mF.getStatusBar().setMessage("y", axisText[1] + coords[1]);

    }

  }

  private void textEntered(String response) {

    if (response == null) {

      cleanBehavior();

    }

    else {

      TYPE type = currentParam.getType();

      try {

        switch (type) {

          case POSITION:

            String[] numbers = new String[1];

            numbers[0] = response;

            numbers = response.split(",");

            if (numbers.length == 2) {

              if (numbers[0].startsWith("(") && numbers[1].endsWith(")\n")) { // CCS

                numbers[0] = numbers[0].replace("(", "");

                numbers[1] = numbers[1].replace(")\n", "");

              }

              double[] values = new double[] { Double.parseDouble(numbers[0]),

                  Double.parseDouble(numbers[1]) };

              Point2D point = new Point2D.Double(values[0], values[1]);

              this.activeService.value(point);

            }

            else {

              throw new RuntimeException("Datos inv?lidos: " + response);

            }

          case LIST_POSITIONS:

          case OPTION:

          case SELECTION:

          case GEOMETRY:

          case VALUE:

        }

        getNextParameter();

      }

      catch (Exception e) {

        throw new RuntimeException("Datos inv?lidos: " + response);

      }

    }

  }

}