Application: gvSIG desktop » gvSIG tools: Vector editing

/*

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

 *

 * Project  : DiSiD org.gvsig.vectorediting.app.mainplugin

 * SVN Id   : $Id$

 */

package org.gvsig.vectorediting.app.mainplugin.extensions;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

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

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

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

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

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

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

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

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);

  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)) {

        if (canBeEdited(layer)) {

          MapControl mapControl = view.getMapControl();

          StatusBarListener sbl = new StatusBarListener(mapControl);

          editingContext.beginEdition(layer,

              new Behavior[] { new MouseMovementBehavior(sbl) });

          ApplicationLocator.getManager().refreshMenusAndToolBars();

        }

      }

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

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

          editingContext.endEdition(layer);

          ApplicationLocator.getManager().refreshMenusAndToolBars();

        }

      } else {

        editingContext.activateService(actionCommand);

      }

    }

  }

  public void initialize() {

    registerIcons();

  }

  private void registerIcons() {

    IconThemeHelper.registerIcon("action", "vector-editing", this);

    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);

  }

  public boolean isEnabled() {

    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 {

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

    }

  }

  @Override

  public boolean isEnabled(String action) {

    EditingManager manager = EditingLocator.getManager();

    DefaultViewPanel vista = getActiveView();

    FLyrVect activeLayer = getActiveLayer(vista);

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

      return (canBeEdited(activeLayer));

    }

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

      return activeLayer.isEditing();

    } else {

      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;

  }

}