prerna.reactor.frame.graph.r.NodeDetailsReactor 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.reactor.frame.graph.r;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.apache.logging.log4j.Logger;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
import org.apache.tinkerpop.gremlin.structure.Direction;
import org.apache.tinkerpop.gremlin.structure.Edge;
import org.apache.tinkerpop.gremlin.structure.Vertex;

import prerna.algorithm.api.ITableDataFrame;
import prerna.ds.TinkerFrame;
import prerna.engine.impl.tinker.iGraphUtilities;
import prerna.reactor.frame.r.AbstractRFrameReactor;
import prerna.reactor.frame.r.util.AbstractRJavaTranslator;
import prerna.reactor.task.constant.ConstantTaskCreationHelper;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.sablecc2.om.task.ConstantDataTask;

public class NodeDetailsReactor extends AbstractRFrameReactor {

	private static final String CLASS_NAME = NodeDetailsReactor.class.getName();

	public NodeDetailsReactor() {
		this.keysToGet = new String[]{ReactorKeysEnum.FRAME.getKey(), ReactorKeysEnum.COLUMN.getKey(), ReactorKeysEnum.VALUE.getKey(), ReactorKeysEnum.PANEL.getKey()};
	}
	
	@Override
	public NounMetadata execute() {
		init();
		organizeKeys();
		String[] packages = new String[] { "igraph" };
		this.rJavaTranslator.checkPackages(packages);
		this.rJavaTranslator.executeEmptyR("library(igraph)");
		Logger logger = getLogger(CLASS_NAME);

		ITableDataFrame frame = getFrame();
		if(!(frame instanceof TinkerFrame)) {
			throw new IllegalArgumentException("Frame must be a graph frame type");
		}
		TinkerFrame graph = (TinkerFrame) frame;
		if(!graph.isIGraphSynched()) {
			AbstractRJavaTranslator rJavaTranslator = this.insight.getRJavaTranslator(CLASS_NAME);
			String wd = this.insight.getInsightFolder();
			iGraphUtilities.synchronizeGraphToR(graph, rJavaTranslator, graph.getName(), wd, logger);
		}
		String graphName = graph.getName();
		
		String type = this.keyValue.get(this.keysToGet[1]);
		String instance = this.keyValue.get(this.keysToGet[2]);
		String panelId = this.keyValue.get(this.keysToGet[3]);
		if(panelId == null) {
			panelId = "Temp";
		}
		String uniqueNodeId = type + ":" + instance;
		
		GraphTraversal it = graph.g.traversal().V().has(TinkerFrame.TINKER_ID, uniqueNodeId);
		Vertex v = null;
		if(it.hasNext()) {
			v = it.next();
		}
		
		if(v == null) {
			throw new IllegalArgumentException("Could not find vertex of type = " + type + " and value = " + instance);
		}
		
		List gridData = new ArrayList();
		
		long inE = 0;
		long outE = 0;
		long degree = 0;
		boolean isOrphan = false;
		long eigen = 0;
		
		logger.info("Calculating out edges");
		// count out E
		{
			Iterator subIt = v.edges(Direction.OUT);
			while(subIt.hasNext()) {
				subIt.next();
				outE++;
			}
		}
		logger.info("Calculating in edges");
		// count in E
		{
			Iterator subIt = v.edges(Direction.IN);
			while(subIt.hasNext()) {
				subIt.next();
				inE++;
			}
		}
		logger.info("Calculating degree");
		// get total degree
		degree = inE + outE;
		// determine if orphan
		if(degree == 0) {
			isOrphan = true;
		}
		logger.info("Calculating eigen value");
		// get eigen value
		eigen = graph.eigen(type, instance);
		
		// store these values
		gridData.add(new Object[]{"# in E", inE});
		gridData.add(new Object[]{"# out E", outE});
		gridData.add(new Object[]{"Degree", degree});
		gridData.add(new Object[]{"Is Orphan", isOrphan});
		gridData.add(new Object[]{"Eigen Value", eigen});

		if(graphName != null) {
			logger.info("Calculating articulation points");
			// get the articulation points
			boolean found = false;
			int [] vertices = this.rJavaTranslator.getIntArray("articulation.points(" + graphName + ")");
			for(int vertIndex = 0;vertIndex < vertices.length;  vertIndex++) {
				String thisNodeId = this.rJavaTranslator.getString("vertex_attr(" + graphName + ", \"" + TinkerFrame.TINKER_ID + "\", " + vertices[vertIndex] + ")");
				if(thisNodeId.equals(uniqueNodeId)) {
					found = true;
					gridData.add(new Object[]{"Is Articulation Node", true});
					break;
				}
			}
			
			if(!found) {
				gridData.add(new Object[]{"Is Articulation Node", false});
			}
		}

		ConstantDataTask taskData = ConstantTaskCreationHelper.getGridData(panelId, new String[]{"Metric", "Value"}, gridData);
		// store it in the insight
		if(!panelId.equals("temp")) {
			this.insight.getTaskStore().addTask(taskData);
		}
		
		// return the data
		return new NounMetadata(taskData, PixelDataType.FORMATTED_DATA_SET, PixelOperationType.TASK_DATA);
	}

}