prerna.reactor.frame.r.analytics.RunSimilarityHeatReactor 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.r.analytics;

import java.util.List;
import java.util.Vector;

import prerna.ds.r.RDataTable;
import prerna.ds.r.RSyntaxHelper;
import prerna.reactor.frame.r.AbstractRFrameReactor;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.Utility;

public class RunSimilarityHeatReactor extends AbstractRFrameReactor {

	public RunSimilarityHeatReactor() {
		this.keysToGet = new String[] { ReactorKeysEnum.INSTANCE_KEY.getKey(), ReactorKeysEnum.ATTRIBUTES.getKey(), ReactorKeysEnum.OVERRIDE.getKey() };
	}

	@Override
	public NounMetadata execute() {
		init();
		organizeKeys();
		String[] packages  = new String[] {"data.table", "plyr"};
		this.rJavaTranslator.checkPackages(packages);
		// get Pixel inputs
		RDataTable frame = (RDataTable) this.getFrame();
		String frameName = frame.getName();
		String instanceCol = this.keyValue.get(this.keysToGet[0]);
		List comparisonColumn = getComparisonColumns();
		boolean override = overrideFrame();
		
		// create R syntax to get similarity heat value
		StringBuilder rsb = new StringBuilder();
		String tempFrame = "SimHeatFrame" + Utility.getRandomString(8);
		List temp = getComparisonColumns();
		temp.add(instanceCol);
		// make frame with only used columns
		rsb.append(RSyntaxHelper.getFrameSubset(tempFrame, frameName, temp.toArray()));
		rsb.append(tempFrame + "<-unique(" + tempFrame + ");");
		String mergeBy = RSyntaxHelper.createStringRColVec(comparisonColumn.toArray());
		// combine with self
		rsb.append(tempFrame + " <- merge(" + tempFrame + "," + tempFrame + ", by=" + mergeBy + ", all.x = TRUE, all.y = FALSE, allow.cartesian = TRUE);");
		// remove where systems are the same
		rsb.append(tempFrame + " <- " + tempFrame + "[" + tempFrame + "$" + instanceCol + ".x != " + tempFrame + "$" + instanceCol + ".y,];");
		//# drop opposites i.e. a=b and b=a
		rsb.append(tempFrame + " <- " + tempFrame + "[!duplicated(apply(" + tempFrame + ",1,function(x) paste(sort(x),collapse=''))),];");
		rsb.append(tempFrame + " <- plyr::count(" + tempFrame + ", c('" + instanceCol + ".x', '" + instanceCol + ".y')); ");
		rsb.append(RSyntaxHelper.alterColumnName(tempFrame, instanceCol + ".x", instanceCol + "_1"));
		rsb.append(RSyntaxHelper.alterColumnName(tempFrame, instanceCol + ".y", instanceCol + "_2"));
		rsb.append(RSyntaxHelper.alterColumnName(tempFrame, "freq", "Heat"));
		rsb.append(RSyntaxHelper.asDataTable(tempFrame, tempFrame));
		if (override) {
			rsb.append(RSyntaxHelper.asDataTable(frameName, tempFrame));
			rsb.append("rm(" + tempFrame + ")");
			tempFrame = frameName;
		}
		this.rJavaTranslator.runR(rsb.toString());
		// create new R DataTable from results
		RDataTable returnTable = createNewFrameFromVariable(tempFrame);
		NounMetadata retNoun = new NounMetadata(returnTable, PixelDataType.FRAME);
		retNoun.addAdditionalReturn(
				new NounMetadata("You've successfully completed running similarity heat and generated a new frame", 
						PixelDataType.CONST_STRING, PixelOperationType.SUCCESS));
		// replace existing frame
		if (override) {
			this.insight.setDataMaker(returnTable);
			retNoun = new NounMetadata(returnTable, PixelDataType.FRAME, PixelOperationType.FRAME_DATA_CHANGE,
					PixelOperationType.FRAME_HEADERS_CHANGE);
		}

		return retNoun;
	}
		
	
	private List getComparisonColumns() {
		GenRowStruct grs = this.store.getNoun(this.keysToGet[1]);
		Vector columns = new Vector();
		NounMetadata noun;
		if (grs != null) {
			for (int i = 0; i < grs.size(); i++) {
				noun = grs.getNoun(i);
				if (noun != null) {
					String column = noun.getValue().toString();
					if (column.length() > 0) {
						columns.add(column);
					}
				}
			}
		}
		return columns;
	}
	
	private boolean overrideFrame() {
		GenRowStruct overrideGrs = this.store.getNoun(ReactorKeysEnum.OVERRIDE.getKey());
		if(overrideGrs != null && !overrideGrs.isEmpty()) {
			return (boolean) overrideGrs.get(0);
		}
		// default is to override
		return true;
	}
}