com.tinkerpop.blueprints.util.VerticesFromEdgesIterable Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of blueprints-core Show documentation

Core interfaces and utilities for Blueprints

There is a newer version: 2.6.0

package com.tinkerpop.blueprints.util;

import com.tinkerpop.blueprints.Direction;
import com.tinkerpop.blueprints.Edge;
import com.tinkerpop.blueprints.Vertex;

import java.util.Iterator;

/**
 * VerticesFromEdgesIterable is a helper class that returns vertices that meet the direction/label criteria of the incident edges.
 *
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public class VerticesFromEdgesIterable implements Iterable {

    private final Iterable iterable;
    private final Direction direction;
    private final Vertex vertex;

    public VerticesFromEdgesIterable(final Vertex vertex, final Direction direction, final String... labels) {
        this.direction = direction;
        this.vertex = vertex;
        this.iterable = vertex.getEdges(direction, labels);
    }

    public Iterator iterator() {
        return new Iterator() {
            final Iterator itty = iterable.iterator();

            public void remove() {
                this.itty.remove();
            }

            public boolean hasNext() {
                return this.itty.hasNext();
            }

            public Vertex next() {
                if (direction.equals(Direction.OUT)) {
                    return this.itty.next().getVertex(Direction.IN);
                } else if (direction.equals(Direction.IN)) {
                    return this.itty.next().getVertex(Direction.OUT);
                } else {
                    final Edge edge = this.itty.next();
                    if (edge.getVertex(Direction.IN).equals(vertex))
                        return edge.getVertex(Direction.OUT);
                    else
                        return edge.getVertex(Direction.IN);
                }
            }
        };
    }
}