prerna.reactor.frame.CollectNewColReactor 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;

import prerna.algorithm.api.ITableDataFrame;
import prerna.ds.nativeframe.NativeFrame;
import prerna.ds.py.PandasFrame;
import prerna.ds.r.RDataTable;
import prerna.query.querystruct.SelectQueryStruct;
import prerna.reactor.task.TaskBuilderReactor;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.sablecc2.om.task.BasicIteratorTask;

public class CollectNewColReactor extends TaskBuilderReactor {
	
	public CollectNewColReactor() {
		this.keysToGet = new String[] { ReactorKeysEnum.QUERY_STRUCT.getKey() };
	}
	
	public NounMetadata execute() {
		// based on the frame type we will create the formula
		if(! ((this.task=getTask()) instanceof BasicIteratorTask)) {
			throw new IllegalArgumentException("Can only add a new column using a basic query on a frame");
		}
		SelectQueryStruct qs = ((BasicIteratorTask) this.task).getQueryStruct();
		ITableDataFrame frame = qs.getFrame();
		TaskBuilderReactor reactor = null;
		if(frame instanceof RDataTable) {
			reactor = new prerna.reactor.frame.r.RCollectNewColReactor();
		} else if(frame instanceof PandasFrame) {
			reactor = new prerna.reactor.frame.py.PyCollectNewColReactor();
		} else if(frame instanceof NativeFrame) {
			reactor = new prerna.reactor.frame.nativeframe.NativeCollectNewColReactor();
		} else {
			throw new IllegalArgumentException("Can only add a new column using an R or Pandas frame");
		}
		
		// set the task in the store directly instead of double executing
		this.store.makeNoun(PixelDataType.TASK.getKey()).add(new NounMetadata(this.task, PixelDataType.TASK));
		// pass the references/values
		// return the execution result
		reactor.In();
		reactor.setInsight(this.insight);
		reactor.setNounStore(this.store);
		return reactor.execute();
	}

	@Override
	protected void buildTask() {
		// do nothing
	}
}