SplitLineByLineTool.java

/**
 * OrbisGIS is a java GIS application dedicated to research in GIScience.
 * OrbisGIS is developed by the GIS group of the DECIDE team of the 
 * Lab-STICC CNRS laboratory, see <http://www.lab-sticc.fr/>.
 *
 * The GIS group of the DECIDE team is located at :
 *
 * Laboratoire Lab-STICC – CNRS UMR 6285
 * Equipe DECIDE
 * UNIVERSITÉ DE BRETAGNE-SUD
 * Institut Universitaire de Technologie de Vannes
 * 8, Rue Montaigne - BP 561 56017 Vannes Cedex
 * 
 * OrbisGIS is distributed under GPL 3 license.
 *
 * Copyright (C) 2007-2014 CNRS (IRSTV FR CNRS 2488)
 * Copyright (C) 2015-2017 CNRS (Lab-STICC UMR CNRS 6285)
 *
 * This file is part of OrbisGIS.
 *
 * OrbisGIS is free software: you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the Free Software
 * Foundation, either version 3 of the License, or (at your option) any later
 * version.
 *
 * OrbisGIS is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 * A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * OrbisGIS. If not, see <http://www.gnu.org/licenses/>.
 *
 * For more information, please consult: <http://www.orbisgis.org/>
 * or contact directly:
 * info_at_ orbisgis.org
 */
package org.orbisgis.view.map.tools;

import com.vividsolutions.jts.geom.*;
import java.util.Observable;
import org.gdms.data.DataSource;
import org.gdms.data.types.*;
import org.gdms.driver.DriverException;
import org.gdms.geometryUtils.GeometryEdit;
import org.orbisgis.coremap.layerModel.MapContext;
import org.orbisgis.view.icons.OrbisGISIcon;
import org.orbisgis.view.map.tool.Handler;
import org.orbisgis.view.map.tool.ToolManager;
import org.orbisgis.view.map.tool.TransitionException;
import javax.swing.*;
import org.gdms.data.values.Value;
import org.gdms.data.values.ValueFactory;

/**
 * Split a linestring or a multilinestring using drawing a line.
 */
public class SplitLineByLineTool extends AbstractLineTool {

    @Override
    public void update(Observable o, Object arg) {
        //PlugInContext.checkTool(this);
    }

    @Override
    public boolean isEnabled(MapContext vc, ToolManager tm) {
        return ToolUtilities.geometryTypeIs(vc,
                TypeFactory.createType(Type.LINESTRING),
                TypeFactory.createType(Type.MULTILINESTRING))
                && ToolUtilities.isActiveLayerEditable(vc) && ToolUtilities.isSelectionEqualsTo(vc, 1);
    }

    @Override
    public boolean isVisible(MapContext vc, ToolManager tm) {
        return isEnabled(vc, tm);
    }

    @Override
    public String getName() {
        return i18n.tr("Split line by line");
    }

    @Override
    public String getTooltip() {
        return i18n.tr("Split a line by drawing a line");
    }

    @Override
    public ImageIcon getImageIcon() {
        return OrbisGISIcon.getIcon("edition/splitlinebyline");
    }

    @Override
    protected void lineDone(LineString ls, MapContext vc, ToolManager tm) throws TransitionException {
        Handler handler = tm.getCurrentHandlers().get(0);
        DataSource sds = vc.getActiveLayer().getDataSource();
        try {
            Geometry geom = sds.getGeometry(handler.getGeometryIndex());
            if (ToolUtilities.geometryTypeIs(mc, TypeFactory.createType(Type.MULTILINESTRING))) {
                Geometry lines = GeometryEdit.splitMultiLineStringWithLine((MultiLineString) geom, ls);
                if (!lines.equals(geom)) {
                    sds.setGeometry(handler.getGeometryIndex(), lines);
                }
            } else if (ToolUtilities.geometryTypeIs(mc, TypeFactory.createType(Type.LINESTRING))) {
                Geometry lines = GeometryEdit.splitLineStringWithLine((LineString) geom, ls);
                if (!lines.equals(geom)) {
                    sds.deleteRow(handler.getGeometryIndex());
                    Value[] row = sds.getRow(handler.getGeometryIndex());
                    for (int i = 0; i < lines.getNumGeometries(); i++) {
                        row[sds.getSpatialFieldIndex()] = ValueFactory.createValue(lines.getGeometryN(i));
                        sds.insertFilledRow(row);
                    }
                    sds.setGeometry(handler.getGeometryIndex(), lines);
                }
            }

        } catch (DriverException e) {
            throw new TransitionException(i18n.tr("Cannot split the line"), e);
        }
    }
}