prerna.reactor.algorithms.RAlgReactor 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.algorithms;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import org.apache.logging.log4j.Logger;

import prerna.reactor.frame.r.AbstractRFrameReactor;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.Constants;
import prerna.util.DIHelper;
import prerna.util.Utility;

public class RAlgReactor extends AbstractRFrameReactor {

	private static final String CLASS_NAME = RAlgReactor.class.getName();
	
	public RAlgReactor() {
		this.keysToGet = new String[]{"filename", "function"};
	}
	
	@Override
	public NounMetadata execute() {
		Logger logger = getLogger(CLASS_NAME);
		init();

		Map fInputs = getFunctionInputs();
		String fileName = fInputs.get(this.keysToGet[0]);
		String functionName = fInputs.get(this.keysToGet[1]);

		String fileLoc = DIHelper.getInstance().getProperty(Constants.BASE_FOLDER);
		fileLoc = fileLoc.replace("\\", "/") + "/R/UserScripts/" + fileName;
		
		// loop through and generate the syntax
		StringBuilder script = new StringBuilder();
		script.append("source(\"").append(fileLoc).append("\");");
		script.append(functionName).append("(");
		int counter = 0;
		for(String key : fInputs.keySet()) {
			if(key.equals(this.keysToGet[0]) || key.equalsIgnoreCase(this.keysToGet[1])) {
				continue;
			}
			
			String inputValue = fInputs.get(key);
			if(counter == 0) {
				script.append(key).append("=").append(inputValue);
			} else {
				script.append(",").append(key).append("=").append(inputValue);
			}
			counter++;
		}
		script.append(");");
		
		String scriptStr = script.toString();
		logger.info("Running script : " + Utility.cleanLogString(scriptStr));
		this.rJavaTranslator.runR(scriptStr);
		
		return new NounMetadata(scriptStr, PixelDataType.CONST_STRING);
	}
	
	/**
	 * Merging all the inputs into R specific syntax
	 * Will generate appropriate vector syntax vs. scalar syntax for inputs
	 * NOTE ::: Cannot have the variable name "all" since its already used by the reactor
	 * @return
	 */
	private Map getFunctionInputs() {
		Map inputs = new HashMap();
		
		Set keys = this.store.getNounKeys();
		for(String key : keys) {
			if(key.equals("all")) {
				// ignore the all key
				continue;
			} else if(key.equals(this.keysToGet[0]) || key.equals(this.keysToGet[1])) {
				// this is a string but we do not want quotes around it
				inputs.put(key, this.store.getNoun(key).get(0).toString());
			} else {
				// other input to process
				GenRowStruct grs = this.store.getNoun(key);
				int size = grs.size();
				if(size == 1) {
					NounMetadata noun = grs.getNoun(0);
					if(noun.getNounType() == PixelDataType.CONST_STRING) {
						// we have a string
						// wrap it in quotes
						inputs.put(key, "\"" + noun.getValue() + "\"");
						
					} else if(noun.getNounType() == PixelDataType.CONST_INT || noun.getNounType() == PixelDataType.CONST_DECIMAL) {
						// we have a number
						// just stringify it without adding quotes
						inputs.put(key, noun.getValue() + "");
					}
				} else {
					// right now, assume it is a string array
					StringBuilder script = new StringBuilder("c(");
					NounMetadata noun = grs.getNoun(0);
					script.append("\"" + noun.getValue() + "\"");
					for(int i = 1; i < size; i++) {
						noun = grs.getNoun(1);
						script.append(", \"" + noun.getValue() + "\"");
					}
					script.append(")");
					// add the column vec
					inputs.put(key, script.toString());
				}
			}
		}
		
		return inputs;
	}
}