OperateSelectionCommand.java

package edu.isi.karma.controller.command.worksheet.selection;

import edu.isi.karma.controller.command.CommandException;
import edu.isi.karma.controller.command.CommandType;
import edu.isi.karma.controller.command.WorksheetSelectionCommand;
import edu.isi.karma.controller.command.selection.LargeSelection;
import edu.isi.karma.controller.command.selection.LargeSelection.Operation;
import edu.isi.karma.controller.command.selection.MiniSelection;
import edu.isi.karma.controller.command.selection.Selection;
import edu.isi.karma.controller.command.selection.SelectionManager;
import edu.isi.karma.controller.command.selection.SuperSelection;
import edu.isi.karma.controller.update.ErrorUpdate;
import edu.isi.karma.controller.update.UpdateContainer;
import edu.isi.karma.controller.update.WorksheetSuperSelectionListUpdate;
import edu.isi.karma.controller.update.WorksheetUpdateFactory;
import edu.isi.karma.rep.HNode;
import edu.isi.karma.rep.HTable;
import edu.isi.karma.rep.RepFactory;
import edu.isi.karma.rep.Worksheet;
import edu.isi.karma.rep.Workspace;

public class OperateSelectionCommand extends WorksheetSelectionCommand {

	private String hNodeId;
	private String pythonCode;
	private String operation;
	private Selection previousSelection;
	private boolean onError;
	public OperateSelectionCommand(String id, String model, String worksheetId, String selectionId, 
			String hNodeId, String operation, 
			String pythonCode, boolean onError) {
		super(id, model, worksheetId, selectionId);
		this.hNodeId = hNodeId;
		this.pythonCode = pythonCode;
		this.operation = operation;
		this.onError = onError;
		addTag(CommandTag.Selection);
	}

	@Override
	public String getCommandName() {
		return this.getClass().getSimpleName();
	}

	@Override
	public String getTitle() {
		return "Operate Selection";
	}

	@Override
	public String getDescription() {
		return operation;
	}

	@Override
	public CommandType getCommandType() {
		return CommandType.undoable;
	}

	@Override
	public UpdateContainer doIt(Workspace workspace) throws CommandException {
		inputColumns.clear();
		outputColumns.clear();
		Worksheet worksheet = workspace.getWorksheet(worksheetId);
		RepFactory factory = workspace.getFactory();
		SuperSelection superSel = this.getSuperSelection(worksheet);
		HTable hTable = factory.getHTable(factory.getHNode(hNodeId).getHTableId());
		Selection currentSel = superSel.getSelection(hTable.getId());
		Selection anotherSel = null;
		if (!operation.equalsIgnoreCase(Operation.Invert.name())) {
			anotherSel = new MiniSelection(workspace, worksheetId, hTable.getId(), factory.getNewId("SEL"), superSel.getName(), pythonCode, onError);
			worksheet.getSelectionManager().addSelection(anotherSel);
		}
		if (currentSel == null && operation.equalsIgnoreCase(Operation.Invert.name()) ) {
			return getErrorUpdate("No defined Selection");
		}
		if (currentSel == null) {
			currentSel = new MiniSelection(workspace, worksheetId, hTable.getId(), factory.getNewId("SEL"), superSel.getName(), SelectionManager.defaultCode, onError);
			worksheet.getSelectionManager().addSelection(currentSel);
		}
		try {
			Operation operation = Operation.valueOf(Operation.class, this.operation);
			Selection t = new LargeSelection(workspace, worksheetId, hTable.getId(), factory.getNewId("SEL"), superSel.getName(), currentSel, anotherSel, operation);
			worksheet.getSelectionManager().addSelection(t);
			outputColumns.addAll(t.getInputColumns());
			previousSelection = superSel.getSelection(t.getHTableId());
			if (previousSelection != null)
				superSel.removeSelection(previousSelection);
			superSel.addSelection(t);
			
		}catch (Exception e) {
			return getErrorUpdate("The operation is undefined");
		}
		WorksheetUpdateFactory.detectSelectionStatusChange(worksheetId, workspace, this);
		if(!this.isExecutedInBatch()) {
			UpdateContainer uc = WorksheetUpdateFactory.createWorksheetHierarchicalAndCleaningResultsUpdates(worksheetId, superSel, workspace.getContextId());
			return uc;
		} else {
			return new UpdateContainer();
		}
	}

	@Override
	public UpdateContainer undoIt(Workspace workspace) {
		inputColumns.clear();
		outputColumns.clear();
		Worksheet worksheet = workspace.getWorksheet(worksheetId);
		SuperSelection superSel = getSuperSelection(worksheet);
		HNode hNode = workspace.getFactory().getHNode(hNodeId);
		Selection currentSel = superSel.getSelection(hNode.getHTableId());
		if (previousSelection != null) {
			superSel.addSelection(previousSelection);
			outputColumns.addAll(previousSelection.getInputColumns());
		}
		if (currentSel != null) {
			worksheet.getSelectionManager().removeSelection(currentSel);
			superSel.removeSelection(currentSel);
		}
		WorksheetUpdateFactory.detectSelectionStatusChange(worksheetId, workspace, this);
		UpdateContainer uc = WorksheetUpdateFactory.createWorksheetHierarchicalAndCleaningResultsUpdates(worksheetId, superSel, workspace.getContextId());	
		uc.add(new WorksheetSuperSelectionListUpdate(worksheetId));
		return uc;
	}
	
	public UpdateContainer getErrorUpdate(String msg) {
		return new UpdateContainer(new ErrorUpdate(msg));
	}
	
	public String getHNodeId() {
		return hNodeId;
	}

}