prerna.ds.TinkerFrameStatRoutine Maven / Gradle / Ivy
The newest version!
//package prerna.ds;
//
//import java.util.ArrayList;
//import java.util.Arrays;
//import java.util.HashMap;
//import java.util.HashSet;
//import java.util.List;
//import java.util.Map;
//import java.util.Set;
//
//import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
//import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
//
//import prerna.algorithm.api.IAnalyticTransformationRoutine;
//import prerna.algorithm.api.ITableDataFrame;
//import prerna.ds.h2.H2Frame;
//import prerna.om.SEMOSSParam;
//import prerna.rdf.query.builder.GremlinBuilder;
//
//public class TinkerFrameStatRoutine implements IAnalyticTransformationRoutine {
//
// private static final String PRIMARY_SELECTOR = "primary_selector";
// private static final String SECONDARY_SELECTOR = "secondary_selector";
//
// private static final String COUNT = "COUNT";
// private static final String AVERAGE = "AVERAGE";
// private static final String MIN = "MIN";
// private static final String MAX = "MAX";
// private static final String SUM = "SUM";
//
// private Map functionMap;
// private String newColumn;
//
// public TinkerFrameStatRoutine() {
// functionMap = new HashMap(0);
// }
//
// @Override
// public String getName() {
// return "TinkerFrameStatRoutine";
// }
//
// @Override
// public String getResultDescription() {
// return "";
// }
//
// @Override
// public void setSelectedOptions(Map selected) {
// functionMap = selected;
// }
//
// @Override
// public List getOptions() {
// return null;
// }
//
// @Override
// public String getDefaultViz() {
// return null;
// }
//
// @Override
// public Map getResultMetadata() {
// return null;
// }
//
// @Override
// public ITableDataFrame runAlgorithm(ITableDataFrame... data) throws RuntimeException {
// // TODO Auto-generated method stub
// //grab first, cast to tinker
// ITableDataFrame table = data[0];
//
// //gather the necessary parameters
// String valueColumn = this.functionMap.get("name").toString();
// String mathType = this.functionMap.get("math").toString();
// String newColumnHeader = this.functionMap.get("calcName").toString();
// String[] columnHeaders = ((List)this.functionMap.get("GroupBy")).toArray(new String[0]);
//
// if(table instanceof H2Frame) {
// H2Frame h2Frame = (H2Frame)table;
//
// //calculate group by and add to tinker
// try {
// addStatColumnToTinker(h2Frame, columnHeaders, mathType, newColumnHeader, valueColumn);
// } catch(ClassCastException e) {
// throw new ClassCastException("Error with computation. Please make sure aggregation values are non-empty and numerical.");
// }
// this.newColumn = newColumnHeader;
// }
//
// else if(table instanceof TinkerFrame) {
// TinkerFrame tinkerGraph = (TinkerFrame)table;
//
// //calculate group by and add to tinker
// try {
// addStatColumnToTinker(tinkerGraph, columnHeaders, mathType, newColumnHeader, valueColumn);
// } catch(ClassCastException e) {
// throw new ClassCastException("Error with computation. Please make sure aggregation values are non-empty and numerical.");
// }
// this.newColumn = newColumnHeader;
//
// } else {
// throw new IllegalArgumentException("Cannot run stat routine on instance of"+table.getClass().getName());
// }
//
//
// return null;
// }
//
// /**
// *
// * @param tinker
// * @param columnHeader - the column to join on
// * @param mathType - type of operation to do
// * @param newColumnName - name of the new column
// * @param valueColumn - the column to do calculations on
// */
// private void addStatColumnToTinker(H2Frame tinker, String[] columnHeader, String mathType, String newColumnName, String valueColumn) {
// Map dataType = new HashMap<>(1);
// dataType.put(newColumnName, "DOUBLE");
// tinker.connectTypes(columnHeader[0], newColumnName, dataType);
// tinker.applyGroupBy(columnHeader, newColumnName, valueColumn, mathType);
// }
//
// /**
// *
// * @param tinker
// * @param columnHeader - the column to join on
// * @param mathType - type of operation to do
// * @param newColumnName - name of the new column
// * @param valueColumn - the column to do calculations on
// */
// private void addStatColumnToTinker(TinkerFrame tinker, String[] columnHeader, String mathType, String newColumnName, String valueColumn) throws ClassCastException{
//
// GremlinBuilder builder = GremlinBuilder.prepareGenericBuilder(tinker.getSelectors(), tinker.g, ((TinkerMetaData)tinker.metaData).g, null);
// GraphTraversal statIterator = (GraphTraversal)builder.executeScript();
//
// boolean singleColumn = columnHeader.length == 1;
//
// if(singleColumn) {
// statIterator = statIterator.group().by(__.select(columnHeader[0]).values(TinkerFrame.TINKER_NAME)).as(PRIMARY_SELECTOR);
// } else if(columnHeader.length == 2) {
// statIterator = statIterator.group().by(__.select(columnHeader[0], columnHeader[1]).by(TinkerFrame.TINKER_NAME)).as(PRIMARY_SELECTOR);
// } else {
// String[] restOfHeaders = Arrays.copyOfRange(columnHeader, 2, columnHeader.length-1);
// statIterator = statIterator.group().by(__.select(columnHeader[0], columnHeader[1], restOfHeaders).by(TinkerFrame.TINKER_NAME)).as(PRIMARY_SELECTOR);
// }
//
// switch(mathType.toUpperCase()) {
// case AVERAGE : {
// statIterator = statIterator.by(__.select(valueColumn).values(TinkerFrame.TINKER_NAME).mean()).as(SECONDARY_SELECTOR);break;
// }
// case MIN : {
// statIterator = statIterator.by(__.select(valueColumn).values(TinkerFrame.TINKER_NAME).min()).as(SECONDARY_SELECTOR);break;
// }
// case MAX : {
// statIterator = statIterator.by(__.select(valueColumn).values(TinkerFrame.TINKER_NAME).max()).as(SECONDARY_SELECTOR);break;
// }
// case SUM : {
// statIterator = statIterator.by(__.select(valueColumn).values(TinkerFrame.TINKER_NAME).sum()).as(SECONDARY_SELECTOR);break;
// }
// case COUNT : {
// statIterator = statIterator.by(__.select(valueColumn).values(TinkerFrame.TINKER_NAME).count()).as(SECONDARY_SELECTOR); break;
// }
// default : {
// statIterator = statIterator.by(__.select(valueColumn).values(TinkerFrame.TINKER_NAME).mean()).as(SECONDARY_SELECTOR);break;
// }
// }
//
// statIterator = statIterator.select(PRIMARY_SELECTOR);
//
// try {
// if(statIterator.hasNext()) {
//
// //the result of the graph traversal
// Map> resultMap;
// Map groupByMap;
//
// groupByMap = (Map)statIterator.next();
//
// //create the edgehash associated with the new result map
// Map> newEdgeHash = new HashMap>(1);
// Set edgeSet = new HashSet(1);
// edgeSet.add(newColumnName);
//
// Map logicalToUnique = new HashMap();
// logicalToUnique.put(newColumnName, newColumnName);
//
// if(singleColumn) {
//
// for(String column : columnHeader) {
// newEdgeHash.put(column, edgeSet);
// logicalToUnique.put(column, column);
// }
// Map dataType = new HashMap<>(1);
// dataType.put(newColumnName, "DOUBLE");
// tinker.mergeEdgeHash(newEdgeHash, dataType);
//
// String cHeader = columnHeader[0];
// for(Object key : groupByMap.keySet()) {
//
// Map newRow = new HashMap(2);
// newRow.put(cHeader, key);
// newRow.put(newColumnName, groupByMap.get(key));
//
// tinker.addRelationship(newRow, newEdgeHash, logicalToUnique);
// }
//
// // add appropriate blanks
// List addedCols = new ArrayList();
// addedCols.add(newColumnName);
// tinker.insertBlanks(columnHeader[0], addedCols);
// } else {
//
// Map dataType = new HashMap<>(1);
// dataType.put(newColumnName, "DOUBLE");
// tinker.connectTypes(columnHeader, newColumnName, dataType);
//
// String primKeyName = TinkerMetaHelper.getPrimaryKey(columnHeader);
// for(Object key : groupByMap.keySet()) {
// Map cleanRow = new HashMap(columnHeader.length+1);
// Map mapKey = (Map)key;
//
// Object[] values = new Object[columnHeader.length];
// for(int i = 0; i < columnHeader.length; i++) {
// values[i] = mapKey.get(columnHeader[i]);
// cleanRow.put(columnHeader[i], values[i]);
// }
//
// cleanRow.put(newColumnName, groupByMap.get(key));
// cleanRow.put(primKeyName, TinkerMetaHelper.getPrimaryKey(values));
//
// tinker.addRelationship(cleanRow);
// }
//
// // add appropriate blanks
// List addedCols = new ArrayList();
// addedCols.add(primKeyName);
// for(String column : columnHeader) {
// tinker.insertBlanks(column, addedCols);
// }
// addedCols.clear();
// addedCols.add(newColumnName);
// tinker.insertBlanks(primKeyName, addedCols);
//
// }
// }
// } catch(ClassCastException e) {
// throw new ClassCastException(e.getMessage());
// }
// }
//
// @Override
// public List getChangedColumns() {
// List changedColumns = new ArrayList(1);
// changedColumns.add(this.newColumn);
// return changedColumns;
// }
//
//}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy