CopyBuffer.java

Go to the documentation of this file.

/*
 * Gridarta MMORPG map editor for Crossfire, Daimonin and similar games.
 * Copyright (C) 2000-2015 The Gridarta Developers.
 *
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

package net.sf.gridarta.gui.copybuffer;

import java.awt.Point;
import java.awt.Rectangle;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import net.sf.gridarta.gui.map.mapview.MapView;
import net.sf.gridarta.model.archetype.Archetype;
import net.sf.gridarta.model.baseobject.BaseObject;
import net.sf.gridarta.model.gameobject.GameObject;
import net.sf.gridarta.model.gameobject.GameObjectFactory;
import net.sf.gridarta.model.maparchobject.MapArchObject;
import net.sf.gridarta.model.maparchobject.MapArchObjectFactory;
import net.sf.gridarta.model.mapmodel.InsertionModeSet;
import net.sf.gridarta.model.mapmodel.MapModel;
import net.sf.gridarta.model.mapmodel.MapModelFactory;
import net.sf.gridarta.model.mapmodel.MapModelListener;
import net.sf.gridarta.model.mapmodel.MapSquare;
import net.sf.gridarta.model.mapviewsettings.MapViewSettings;
import net.sf.gridarta.utils.MathUtils;
import net.sf.gridarta.utils.Size2D;
import org.jetbrains.annotations.NotNull;

public class CopyBuffer<G extends GameObject<G, A, R>, A extends MapArchObject<A>, R extends Archetype<G, A, R>> {

    @NotNull
    private final MapViewSettings mapViewSettings;

    @NotNull
    private final MapModel<G, A, R> mapModel;

    @NotNull
    private final GameObjectFactory<G, A, R> gameObjectFactory;

    @NotNull
    private final InsertionModeSet<G, A, R> insertionModeSet;

    public CopyBuffer(@NotNull final MapViewSettings mapViewSettings, @NotNull final GameObjectFactory<G, A, R> gameObjectFactory, @NotNull final MapArchObjectFactory<A> mapArchObjectFactory, @NotNull final MapModelFactory<G, A, R> mapModelFactory, @NotNull final InsertionModeSet<G, A, R> insertionModeSet) {
        this.mapViewSettings = mapViewSettings;
        final A mapArchObject = mapArchObjectFactory.newMapArchObject(false);
        mapArchObject.setMapName("cb");
        mapModel = mapModelFactory.newMapModel(mapArchObject);
        this.gameObjectFactory = gameObjectFactory;
        this.insertionModeSet = insertionModeSet;
    }

    public void addMapModelListener(@NotNull final MapModelListener<G, A, R> listener) {
        mapModel.addMapModelListener(listener);
    }

    public void removeMapModelListener(@NotNull final MapModelListener<G, A, R> listener) {
        mapModel.removeMapModelListener(listener);
    }

    public void clear(@NotNull final MapView<G, A, R> mapView, @NotNull final Rectangle selectedRec) {
        copyNCut(mapView, selectedRec, CopyMode.DO_CLEAR);
    }

    public void cut(@NotNull final MapView<G, A, R> mapView, @NotNull final Rectangle selectedRec) {
        copyNCut(mapView, selectedRec, CopyMode.DO_CUT);
    }

    public void copy(@NotNull final MapView<G, A, R> mapView, @NotNull final Rectangle selectedRec) {
        copyNCut(mapView, selectedRec, CopyMode.DO_COPY);
    }

    private void copyNCut(@NotNull final MapView<G, A, R> mapView, @NotNull final Rectangle selectedRec, @NotNull final CopyMode copyMode) {
        mapModel.beginTransaction("Cut / Clear");
        try {
            copyMode.prepare(mapModel, new Size2D(selectedRec.width, selectedRec.height));

            final MapModel<G, A, R> mapModel2 = mapView.getMapControl().getMapModel();
            mapModel2.beginTransaction("Cut / Clear"); // TODO: I18N/L10N
            try {
                final Collection<G> gameObjectsToDelete = new HashSet<>();
                final Point pos = new Point();
                for (final MapSquare<G, A, R> square : mapView.getSelectedSquares()) {
                    square.getMapLocation(pos, -selectedRec.x, -selectedRec.y);
                    for (final G gameObject : square) {
                        copyMode.process(mapModel, gameObject, mapViewSettings.isEditType(gameObject), gameObjectsToDelete, pos, gameObjectFactory, insertionModeSet);
                    }
                }

                for (final GameObject<G, A, R> gameObject : gameObjectsToDelete) {
                    gameObject.remove();
                }
            } finally {
                mapModel2.endTransaction();
            }
        } finally {
            mapModel.endTransaction();
        }
    }

    public void paste(@NotNull final MapView<G, A, R> mapView, @NotNull final Point startLocation) {
        final MapModel<G, A, R> mapModel2 = mapView.getMapControl().getMapModel();
        final MapArchObject<A> mapArchObject = mapModel2.getMapArchObject();
        mapModel2.beginTransaction("Paste"); // TODO: I18N/L10N
        try {
            final Point pos = new Point();
            for (final MapSquare<G, A, R> square : mapModel) {
                square.getMapLocation(pos);
                pos.translate(startLocation.x, startLocation.y);
                if (mapArchObject.isPointValid(pos)) {
                    for (final BaseObject<G, A, R, ?> gameObject : square) {
                        if (!gameObject.isMulti()) {
                            mapModel2.insertBaseObject(gameObject, pos, true, false, insertionModeSet.getTopmostInsertionMode());
                        }
                    }
                }
            }

            for (final MapSquare<G, A, R> square : mapModel) {
                square.getMapLocation(pos);
                pos.translate(startLocation.x, startLocation.y);
                if (mapArchObject.isPointValid(pos)) {
                    for (final BaseObject<G, A, R, ?> gameObject : square) {
                        if (gameObject.isMulti()) {
                            mapModel2.insertBaseObject(gameObject, pos, true, false, insertionModeSet.getTopmostInsertionMode());
                        }
                    }
                }
            }
        } finally {
            mapModel2.endTransaction();
        }
    }

    public void pasteTiled(@NotNull final MapView<G, A, R> mapView, @NotNull final Iterable<MapSquare<G, A, R>> selectedSquares, @NotNull final Point origin) {
        final Point sourcePoint = new Point();
        final Point destinationPoint = new Point();
        final MapModel<G, A, R> mapModel2 = mapView.getMapControl().getMapModel();
        final Size2D mapModelSize = mapModel.getMapArchObject().getMapSize();
        mapModel2.beginTransaction("Paste"); // TODO: I18N/L10N
        try {
            for (final MapSquare<G, A, R> destinationMapSquare : selectedSquares) {
                sourcePoint.x = MathUtils.mod(destinationMapSquare.getMapX() - origin.x, mapModelSize.getWidth());
                sourcePoint.y = MathUtils.mod(destinationMapSquare.getMapY() - origin.y, mapModelSize.getHeight());
                final MapSquare<G, A, R> sourceMapSquare = mapModel.getMapSquare(sourcePoint);
                for (final BaseObject<G, A, R, ?> gameObject : sourceMapSquare) {
                    if (!gameObject.isMulti()) {
                        destinationPoint.x = destinationMapSquare.getMapX();
                        destinationPoint.y = destinationMapSquare.getMapY();
                        mapModel2.insertBaseObject(gameObject, destinationPoint, true, false, insertionModeSet.getTopmostInsertionMode());
                    }
                }
            }

            for (final MapSquare<G, A, R> destinationMapSquare : selectedSquares) {
                sourcePoint.x = MathUtils.mod(destinationMapSquare.getMapX() - origin.x, mapModelSize.getWidth());
                sourcePoint.y = MathUtils.mod(destinationMapSquare.getMapY() - origin.y, mapModelSize.getHeight());
                final MapSquare<G, A, R> sourceMapSquare = mapModel.getMapSquare(sourcePoint);
                for (final BaseObject<G, A, R, ?> gameObject : sourceMapSquare) {
                    if (gameObject.isMulti()) {
                        destinationPoint.x = destinationMapSquare.getMapX();
                        destinationPoint.y = destinationMapSquare.getMapY();
                        mapModel2.insertBaseObject(gameObject, destinationPoint, true, false, insertionModeSet.getTopmostInsertionMode());
                    }
                }
            }
        } finally {
            mapModel2.endTransaction();
        }
    }

    @NotNull
    public List<G> getAllGameObjects() {
        return mapModel.getAllGameObjects();
    }

    public boolean isEmpty() {
        return mapModel.isEmpty();
    }

}