com.tinkerpop.gremlin.process.SimpleTraverser Maven / Gradle / Ivy

Go to download
package com.tinkerpop.gremlin.process;


import com.tinkerpop.gremlin.structure.Edge;
import com.tinkerpop.gremlin.structure.Property;
import com.tinkerpop.gremlin.structure.Vertex;
import com.tinkerpop.gremlin.structure.util.detached.DetachedEdge;
import com.tinkerpop.gremlin.structure.util.detached.DetachedProperty;
import com.tinkerpop.gremlin.structure.util.detached.DetachedVertex;

/**
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public class SimpleTraverser implements Traverser {

    private static final String PATH_ERROR_MESSAGE = "Path tracking is not supported by this Traverser: " + SimpleTraverser.class;

    protected T t;
    protected String future = NO_FUTURE;
    protected int loops = 0;

    protected SimpleTraverser() {

    }

    public SimpleTraverser(final T t) {
        this.t = t;
    }

    public T get() {
        return this.t;
    }

    public void set(final T t) {
        this.t = t;
    }

    public String getFuture() {
        return this.future;
    }

    public void setFuture(final String as) {
        this.future = as;
    }

    public boolean hasPath() {
        return false;
    }

    public Path getPath() {
        throw new IllegalStateException(PATH_ERROR_MESSAGE);
    }

    public void setPath(final Path path) {
        throw new IllegalStateException(PATH_ERROR_MESSAGE);
    }

    public int getLoops() {
        return this.loops;
    }

    public void incrLoops() {
        this.loops++;
    }

    public void resetLoops() {
        this.loops = 0;
    }

    public  SimpleTraverser makeChild(final String as, final R r) {
        final SimpleTraverser traverser = new SimpleTraverser<>(r);
        traverser.future = this.future;
        traverser.loops = this.loops;
        return traverser;
    }

    public SimpleTraverser makeSibling() {
        final SimpleTraverser traverser = new SimpleTraverser<>(this.t);
        traverser.future = this.future;
        traverser.loops = this.loops;
        return traverser;
    }

    public String toString() {
        return t.toString();
    }

    public int hashCode() {
        return this.t.hashCode() + this.getFuture().hashCode() + this.getLoops();
    }

    public boolean equals(final Object object) {
        return object instanceof SimpleTraverser && this.t.equals(((SimpleTraverser) object).get());
        /*return object instanceof SimpleTraverser &&
                ((SimpleTraverser) object).get().equals(this.t) &&
                ((SimpleTraverser) object).getFuture().equals(this.getFuture()) &&
                ((SimpleTraverser) object).getLoops() == this.getLoops();*/

    }

    public SimpleTraverser deflate() {
        if (this.t instanceof Vertex) {
            this.t = (T) DetachedVertex.detach((Vertex) this.t);
        } else if (this.t instanceof Edge) {
            this.t = (T) DetachedEdge.detach((Edge) this.t);
        } else if (this.t instanceof Property) {
            this.t = (T) DetachedProperty.detach((Property) this.t);
        }
        return this;
    }

    public SimpleTraverser inflate(final Vertex vertex) {
        if (this.t instanceof DetachedVertex) {
            this.t = (T) ((DetachedVertex) this.t).attach(vertex);
        } else if (this.t instanceof DetachedEdge) {
            this.t = (T) ((DetachedEdge) this.t).attach(vertex);
        } else if (this.t instanceof DetachedProperty) {
            this.t = (T) ((DetachedProperty) this.t).attach(vertex);
        }
        return this;
    }
}