Application: gvSIG desktop » gvSIG tools: Vector editing

/*

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

 *

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

 * SVN Id   : $Id$

 */

package org.gvsig.vectorediting.swing.impl;

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.Font;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import javax.swing.JLabel;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import org.gvsig.editing.EditingNotification;

import org.gvsig.editing.EditingNotificationManager;

import org.gvsig.fmap.dal.feature.FeatureStore;

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

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.MapControlLocator;

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.service.AbstractManager;

import org.gvsig.tools.service.Service;

import org.gvsig.tools.service.ServiceException;

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

import org.gvsig.vectorediting.lib.api.exceptions.CreateEditingBehaviorException;

import org.gvsig.vectorediting.lib.api.exceptions.EndEditingException;

import org.gvsig.vectorediting.lib.api.exceptions.StartEditingException;

import org.gvsig.vectorediting.lib.api.exceptions.VectorEditingException;

import org.gvsig.vectorediting.lib.impl.DefaultEditingProviderManager;

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultEditingSwingManager extends AbstractManager implements

    EditingSwingManager {

  private static final Logger logger = LoggerFactory

      .getLogger(EditingManager.class);

  private I18nManager i18nManager = ToolsLocator.getI18nManager();

  private final int CANCEL = 0;

  private final int SAVE_CHANGES = 1;

  private final int DISCARD_CHANGES = 2;

  private final int EXPORT_LAYER = 3;

  public DefaultEditingSwingManager() {

    super(new DefaultEditingProviderManager());

  }

  public void activateService(String name, MapControl mapControl) {

    if (mapControl != null && mapControl.hasTool("VectorEditing")) {

      CompoundBehavior editingCompoundBehavior = getEditingCompoundBehavior(mapControl);

      mapControl.setTool("VectorEditing");

      EditingBehavior editingBehavior = (EditingBehavior) editingCompoundBehavior.getBehavior(EditingCompoundBehavior.EDITING_INDEX);

      editingCompoundBehavior.setDrawnBehavior(EditingCompoundBehavior.EDITING_INDEX, true);

      editingBehavior.activateService(name);

    }

  }

  public void beginEdition(FLyrVect layer, MapControl mapControl, Behavior[] additionalBehaviors) {

    EditingNotificationManager editingNotificationManager = MapControlLocator

        .getEditingNotificationManager();

    EditingNotification notification = editingNotificationManager

        .notifyObservers(this, EditingNotification.BEFORE_ENTER_EDITING_STORE,

            null, layer);

    if (notification.isCanceled()) {

      logger.info("Edit layer '" + layer.getName()

          + "' canceled by somme observer.", new StartEditingException(

          "Edit layer '" + layer.getName() + "' canceled.", null));

    }

    try {

                addBehaviors(mapControl, additionalBehaviors);

        } catch (CreateEditingBehaviorException e1) {

                logger.error(e1.getMessage(), e1);

        }

    try {

      layer.getFeatureStore().edit();

    }

    catch (Exception e) {

        logger.error("Can't set featureStore in edit mode",

                new VectorEditingException(e));    }

    // layer.getFeatureStore().addObserver(view);

    layer.getFeatureStore().addObserver(mapControl);

    editingNotificationManager.notifyObservers(this,

        EditingNotification.AFTER_ENTER_EDITING_STORE, null, layer);

  }

  /**

   * @param layer

   * @param mapControl

   * @throws EndEditingException

   */

  private void discardChanges(FLyrVect layer, MapControl mapControl)

      throws EndEditingException {

    FeatureStore featureStore = layer.getFeatureStore();

    try {

      featureStore.cancelEditing();

      EditingCompoundBehavior editingCompoundBehavior = getEditingCompoundBehavior(mapControl);

      EditingBehavior editingBehavior = (EditingBehavior) editingCompoundBehavior.getBehavior(EditingCompoundBehavior.EDITING_INDEX);

      editingBehavior.cleanBehavior();

      editingBehavior.hideConsole();

    }

    catch (Exception e) {

      throw new EndEditingException(e);

    }

    featureStore.deleteObserver(mapControl);

  }

  /**

   * @param layer

   * @param mapControl

   * @param option

   */

  private void doAction(FLyrVect layer, MapControl mapControl, int option) {

    switch (option) {

      case SAVE_CHANGES:

        try {

          saveChanges(layer, mapControl);

        }

        catch (VectorEditingException e) {

          logger.error("Changes can not be saved in " + layer.getName(), e);

        }

        break;

      case DISCARD_CHANGES:

        try {

          discardChanges(layer, mapControl);

        }

        catch (VectorEditingException e) {

          logger.error("Changes can not be discared in " + layer.getName(), e);

        }

        break;

      case EXPORT_LAYER:

        try {

          exportLayer(layer, mapControl);

        }

        catch (VectorEditingException e) {

          logger.error("Changes can not be imported", e);

        }

        break;

      case CANCEL:

        break;

    }

  }

  public void endEdition(FLyrVect layer, MapControl mapControl) {

    if (layer.isEditing()) {

      EditingNotificationManager editingNotificationManager = MapControlLocator

          .getEditingNotificationManager();

      EditingNotification notification = editingNotificationManager

          .notifyObservers(this, EditingNotification.BEFORE_EXIT_EDITING_STORE,

              null, layer);

      if (notification.isCanceled()) {

        logger.info("Stop edit layer '" + layer.getName()

            + "' canceled by somme observer.", new EndEditingException(

            "Stop edit layer '" + layer.getName() + "' canceled.", null));

      }

      mapControl.getCanceldraw().setCanceled(true);

      int option;

      if (layer.isWritable()) {

        option = showPanelSaveOrDiscard(mapControl, layer.getName());

      }

      else {

        option = showPanelExportOrDiscard(mapControl, layer.getName());

      }

      doAction(layer, mapControl, option);

      editingNotificationManager.notifyObservers(this,

          EditingNotification.AFTER_EXIT_EDITING_STORE, null, layer);

    }

  }

  private void exportLayer(FLyrVect layer, MapControl mapCotrol)

      throws EndEditingException {

    // TODO Export layer

  }

  /**

   * @param translatedQuestion

   * @param translatedFirstTag

   * @param translatedFirstDescription

   * @return

   */

  private JPanel getExplanationPanel(String translatedQuestion,

                                     String translatedFirstTag,

                                     String translatedFirstDescription) {

    JPanel resp = new JPanel(new BorderLayout(10, 10));

    JLabel topLabel = new JLabel(translatedQuestion);

    JPanel mainPanel = new JPanel(new GridBagLayout());

    GridBagConstraints cc = new GridBagConstraints();

    cc.gridx = 0;

    cc.gridy = 0;

    cc.anchor = GridBagConstraints.WEST;

    cc.insets = new Insets(3, 6, 3, 6);

    Font boldf = mainPanel.getFont().deriveFont(Font.BOLD);

    JLabel lbl = new JLabel(translatedFirstTag);

    lbl.setFont(boldf);

    mainPanel.add(lbl, cc);

    cc.gridx = 1;

    mainPanel.add(new JLabel(translatedFirstDescription), cc);

    cc.gridx = 0;

    cc.gridy = 1;

    lbl = new JLabel(i18nManager.getTranslation("discard"));

    lbl.setFont(boldf);

    mainPanel.add(lbl, cc);

    cc.gridx = 1;

    mainPanel

        .add(

            new JLabel(i18nManager.getTranslation("discard_and_loose_changes")),

            cc);

    cc.gridx = 0;

    cc.gridy = 2;

    lbl = new JLabel(i18nManager.getTranslation("continue"));

    lbl.setFont(boldf);

    mainPanel.add(lbl, cc);

    cc.gridx = 1;

    mainPanel.add(

        new JLabel(i18nManager

            .getTranslation("do_not_save_yet_stay_in_editing_mode")), cc);

    resp.add(mainPanel, BorderLayout.CENTER);

    resp.add(topLabel, BorderLayout.NORTH);

    return resp;

  }

  public Service getService(DynObject parameters) throws ServiceException {

    // TODO Auto-generated method stub

    return null;

  }

  /**

   * @param mapControl

 * @throws CreateEditingBehaviorException

   */

  private void addBehaviors(MapControl mapControl, Behavior[] additionalBehavior) throws CreateEditingBehaviorException {

    EditingBehavior editingBehavior;

    EditingCompoundBehavior editingCompoundBehavior;

    if (!mapControl.hasTool("VectorEditing")) {

      editingBehavior = new DefaultEditingBehavior(mapControl);

      editingCompoundBehavior = new EditingCompoundBehavior(editingBehavior);

      ((DefaultEditingBehavior)editingBehavior).setCompoundBehavior(editingCompoundBehavior);

      if(additionalBehavior != null){

              Behavior[] behaviors = new Behavior[additionalBehavior.length+1];

              behaviors[0] = editingCompoundBehavior;

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

                      behaviors[i+1]= additionalBehavior[i];

              }

              mapControl.addBehavior("VectorEditing",behaviors);

      } else {

              mapControl.addBehavior("VectorEditing",editingCompoundBehavior);

      }

    }

    else {

            editingCompoundBehavior = getEditingCompoundBehavior(mapControl);

            editingBehavior = (EditingBehavior) editingCompoundBehavior.getBehavior(EditingCompoundBehavior.EDITING_INDEX);

            ((DefaultEditingBehavior)editingBehavior).setCompoundBehavior(editingCompoundBehavior);

            editingBehavior.cleanBehavior();

    }

    mapControl.setTool("VectorEditing");

    editingBehavior.showConsole();

  }

  /**

   * @param layer

   * @param mapControl

   * @throws EndEditingException

   */

  private void saveChanges(FLyrVect layer, MapControl mapControl)

      throws EndEditingException {

    FeatureStore featureStore = layer.getFeatureStore();

    try {

      featureStore.finishEditing();

      EditingBehavior editingBehavior;

    EditingCompoundBehavior editingCompoundBehavior = getEditingCompoundBehavior(mapControl);

          editingBehavior = (EditingBehavior) editingCompoundBehavior.getBehavior(EditingCompoundBehavior.EDITING_INDEX);

    editingBehavior.cleanBehavior();

    editingBehavior.hideConsole();

    }

    catch (Exception e) {

      throw new EndEditingException(e);

    }

    featureStore.deleteObserver(mapControl);

  }

private EditingCompoundBehavior getEditingCompoundBehavior(MapControl mapControl) {

        EditingCompoundBehavior editingCompoundBehavior;

          CompoundBehavior compoundBehavior = (CompoundBehavior) mapControl.getMapTool("VectorEditing");

          if(compoundBehavior instanceof EditingCompoundBehavior){

                  editingCompoundBehavior = (EditingCompoundBehavior) compoundBehavior;

          } else {

                  editingCompoundBehavior = (EditingCompoundBehavior) compoundBehavior.getBehavior(0);

          }

        return editingCompoundBehavior;

}

  /**

   * @param name

   * @return

   */

  private int showPanelExportOrDiscard(Component root, String name) {

    Object[] options = { i18nManager.getTranslation("export"),

        "       " + i18nManager.getTranslation("discard") + "       ",

        i18nManager.getTranslation("continue") };

    String question = i18nManager

        .getTranslation("no_existe_writer_para_este_formato_de_capa_o_no_tiene_permisos_de_escritura_los_datos_que_desea_hacer");

    String firstLabel = i18nManager.getTranslation("export");

    String firstDesc = i18nManager.getTranslation("export_to_another_format");

    JPanel explanation_panel = getExplanationPanel(question, firstLabel,

        firstDesc);

    int resp = JOptionPane.showOptionDialog(

        root, explanation_panel,

        i18nManager.getTranslation("stop_edition"),

        JOptionPane.YES_NO_CANCEL_OPTION, JOptionPane.QUESTION_MESSAGE, null,

        options, options[2]);

    if (resp == JOptionPane.NO_OPTION) {

      return DISCARD_CHANGES;

    }

    else if (resp == JOptionPane.YES_OPTION) {

      return EXPORT_LAYER;

    }

    return CANCEL;

  }

  /**

   * @param layerName

   * @return

   */

  private int showPanelSaveOrDiscard(Component root, String layerName) {

    Object[] options = { i18nManager.getTranslation("_Guardar"),

        "       " + i18nManager.getTranslation("_Descartar") + "       ",

        i18nManager.getTranslation("_Continuar") };

    String question = i18nManager

        .getTranslation("realmente_desea_guardar_la_capa");

    question = question + " '" + layerName + "'?";

    String firstLabel = i18nManager.getTranslation("guardar");

    String firstDesc = i18nManager.getTranslation("save_changes_performed");

    JPanel explanation_panel = getExplanationPanel(question, firstLabel,

        firstDesc);

    int resp = JOptionPane.showOptionDialog(

        root, explanation_panel,

        i18nManager.getTranslation("stop_edition"),

        JOptionPane.YES_NO_CANCEL_OPTION, JOptionPane.QUESTION_MESSAGE, null,

        options, options[2]);

    if (resp == JOptionPane.YES_OPTION) {

      return SAVE_CHANGES;

    }

    else if (resp == JOptionPane.NO_OPTION) {

      return DISCARD_CHANGES;

    }

    return CANCEL;

  }

}