Application: gvSIG desktop » gvSIG tools: Vector editing

import java.util.ArrayList;

import java.util.List;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.project.documents.view.ViewDocument;

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

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

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

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

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

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

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

import org.gvsig.vectorediting.swing.impl.DefaultEditingBehavior;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class EditingExtension extends Extension {

  private static final Logger logger = LoggerFactory

      .getLogger(DefaultEditingBehavior.class);

  private List<String> serviceNames;

  public void execute(String actionCommand) {

    DefaultViewPanel view = getActiveView();

    EditingSwingManager swingManager = EditingSwingLocator.getSwingManager();

    if (view != null) {

      FLyrVect layer = getActiveLayer(view);

      EditingContext editingContext = swingManager.getEditingContext(view

          .getMapControl());

      if ("start-editing".equals(actionCommand)) {

      else if ("end-editing".equals(actionCommand)) {

        if (layer != null && layer.isEditing()) {

          editingContext.endEdition(layer);

          ApplicationLocator.getManager().refreshMenusAndToolBars();

        }

      }

      else if (serviceNames.contains(actionCommand)) {

        editingContext.activateService(actionCommand);

      }

    buildServiceNames();

  private void buildServiceNames() {

    serviceNames = new ArrayList<String>();

    serviceNames.add("insert-point");

    serviceNames.add("insert-circle-cr");

    serviceNames.add("insert-circumference-cr");

    serviceNames.add("insert-polyline");

    serviceNames.add("insert-polygon");

    serviceNames.add("modify-symmetry");

    serviceNames.add("modify-internal-polygon");

  }

    IconThemeHelper.registerIcon("action", "insert-point", this);

    IconThemeHelper.registerIcon("action", "insert-circle-cr", this);

    IconThemeHelper.registerIcon("action", "insert-circumference-cr", this);

    IconThemeHelper.registerIcon("action", "insert-polyline", this);

    IconThemeHelper.registerIcon("action", "insert-polygon", this);

    IconThemeHelper.registerIcon("action", "modify-symmetry", this);

    IconThemeHelper.registerIcon("action", "modify-internal-polygon", this);

    return true;

  }

  public boolean isVisible() {

    return true;

  }

  @Override

  public boolean isVisible(String action) {

    DefaultViewPanel view = getActiveView();

    FLyrVect activeLayer = getActiveLayer(view);

    if ("start-editing".equals(action)) {

      return (view != null && activeLayer != null && !activeLayer.isEditing());

    }

    else if ("end-editing".equals(action) || serviceNames.contains(action)) {

      return (view != null && activeLayer != null && activeLayer.isEditing());

    }

    return false;

  @Override

  public boolean isEnabled(String action) {

    EditingManager manager = EditingLocator.getManager();

    if ("start-editing".equals(action)) {

      return (canBeEdited(activeLayer));

    }

    else if ("end-editing".equals(action)) {

      return activeLayer.isEditing();

    }

    else if(serviceNames.contains(action)){

      EditingServiceInfo serviceInfo = null;

      try {

        serviceInfo = manager.getServiceInfo(action);

        return serviceInfo.isCompatibleWith(activeLayer.getFeatureStore());

      }

      catch (ServiceInformationException e) {

        logger.error("Can't get service information of " + action, e);

      }

    }

    return false;

  public boolean canQueryByAction() {

    return true;

  }

  private DefaultViewPanel getActiveView() {

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

        .getMDIManager().getActiveWindow();

    if (activeWindow instanceof DefaultViewPanel) {

      return (DefaultViewPanel) activeWindow;

    }

    return null;

  }

  private boolean canBeEdited(FLyrVect layer) {

    if (layer != null) {

      boolean isWritable = layer.isWritable();

      boolean isNotTransformed = layer.getFeatureStore().getTransforms()

          .isEmpty();

      return isWritable && isNotTransformed && !layer.isEditing();

    }

    return false;

  }

  private FLyrVect getActiveLayer(DefaultViewPanel vista) {

    if (vista != null) {

      ViewDocument viewDocument = vista.getViewDocument();

      FLayer[] actives = viewDocument.getMapContext().getLayers().getActives();

      if (actives.length == 1 && actives[0] instanceof FLyrVect) {

        return (FLyrVect) actives[0];

      }

    }

    return null;

  }