prerna.ds.TinkerAlgorithmUtility Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.ds;

import java.util.Comparator;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.tinkerpop.gremlin.process.traversal.P;
import org.apache.tinkerpop.gremlin.process.traversal.Path;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
import org.apache.tinkerpop.gremlin.structure.Vertex;

public final class TinkerAlgorithmUtility {

	protected static final Logger LOGGER = LogManager.getLogger(TinkerAlgorithmUtility.class.getName());
	
	private TinkerAlgorithmUtility() {
		
	}
	
	/**
	 * Return the paths that are cycles in the tinkerframe given a max tinker size
	 * @param tf
	 * @param maxCycleSize
	 * @return
	 */
	public static String runLoopIdentifer(TinkerFrame tf, int cycleSize) {
		GraphTraversal traversal = tf.g.traversal().V().as("a").repeat(__.out().simplePath()).times(cycleSize)
				.where(__.out().as("a")).path().dedup().by(__.unfold().order().by(TinkerFrame.TINKER_ID, new Comparator() {

					@Override
					public int compare(String o1, String o2) {
						return o1.toLowerCase().compareTo(o2.toLowerCase());
					}
					
				}).dedup().fold());
		
		StringBuilder cycles = new StringBuilder();
		while(traversal.hasNext()) {
			Path p = traversal.next();
			int size = p.size();
			Vertex v = p.get(0);
			cycles.append(v.value(TinkerFrame.TINKER_ID) + "");
			for(int stepIdx = 1; stepIdx < size; stepIdx++) {
				v = p.get(stepIdx);
				cycles.append(", " + v.value(TinkerFrame.TINKER_ID));
			}
			cycles.append("\n");
		}
		
		return cycles.toString();
	}
	
	/**
	 * Returns the nodes that are not connected to the provided instance
	 * @param tf
	 * @param type
	 * @param instance
	 * @return
	 */
	public static String runDisconnectedNodesIdentifier(TinkerFrame tf, String type, String instance) {
		GraphTraversal traversal = tf.g.traversal().V()
				.or(
					__.not(__.V().has(TinkerFrame.TINKER_ID, type + ":" + instance).repeat(__.out()).until(__.outE().count().is(0))),
					__.has(TinkerFrame.TINKER_ID, P.neq(type + ":" + instance)).bothE().count().is(0)
				).values(TinkerFrame.TINKER_ID).order();
		
		StringBuilder islandVertices = new StringBuilder();
		while(traversal.hasNext()) {
			islandVertices.append(traversal.next()).append("\n");
		}
		
		return islandVertices.toString();
	}
	
}