MapSquareIterator.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.model.mapmodel;

import java.awt.Point;
import java.util.Iterator;
import java.util.NoSuchElementException;
import net.sf.gridarta.model.archetype.Archetype;
import net.sf.gridarta.model.gameobject.GameObject;
import net.sf.gridarta.model.maparchobject.MapArchObject;
import net.sf.gridarta.utils.Size2D;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public class MapSquareIterator<G extends GameObject<G, A, R>, A extends MapArchObject<A>, R extends Archetype<G, A, R>> implements Iterator<MapSquare<G, A, R>> {

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

    private final int direction;

    private final int mapWidth;

    private final int mapHeight;

    private int point;

    private int remainingMapSquares;

    public MapSquareIterator(@NotNull final MapModel<G, A, R> mapModel, @Nullable final Point start, final int direction, final boolean skipFirst) {
        if (direction != -1 && direction != +1) {
            throw new IllegalArgumentException();
        }
        this.mapModel = mapModel;
        this.direction = direction;
        final Size2D mapSize = mapModel.getMapArchObject().getMapSize();
        mapWidth = mapSize.getWidth();
        mapHeight = mapSize.getHeight();
        if (start != null) {
            point = start.x + start.y * mapWidth;
        } else if (direction > 0) {
            point = 0;
        } else {
            point = mapWidth * mapHeight - 1;
        }
        remainingMapSquares = mapWidth * mapHeight;
        if (skipFirst) {
            nextMapSquare();
        }
    }

    @Override
    public void remove() {
        throw new UnsupportedOperationException();
    }

    @Override
    public boolean hasNext() {
        return remainingMapSquares > 0;
    }

    @Override
    public MapSquare<G, A, R> next() {
        if (remainingMapSquares <= 0) {
            throw new NoSuchElementException();
        }
        remainingMapSquares--;

        final MapSquare<G, A, R> square = mapModel.getMapSquare(new Point(point % mapWidth, point / mapWidth));
        nextMapSquare();
        return square;
    }

    private void nextMapSquare() {
        point += direction;
        if (point < 0) {
            point += mapWidth * mapHeight;
        } else if (point >= mapWidth * mapHeight) {
            point -= mapWidth * mapHeight;
        }
    }

}