gumtree.spoon.diff.ActionClassifier Maven / Gradle / Ivy

package gumtree.spoon.diff;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import com.github.gumtreediff.actions.model.Action;
import com.github.gumtreediff.actions.model.Delete;
import com.github.gumtreediff.actions.model.Insert;
import com.github.gumtreediff.actions.model.Move;
import com.github.gumtreediff.actions.model.Update;
import com.github.gumtreediff.matchers.Mapping;
import com.github.gumtreediff.matchers.MappingStore;
import com.github.gumtreediff.tree.ITree;

import gumtree.spoon.builder.SpoonGumTreeBuilder;
import gumtree.spoon.diff.operations.DeleteOperation;
import gumtree.spoon.diff.operations.InsertOperation;
import gumtree.spoon.diff.operations.MoveOperation;
import gumtree.spoon.diff.operations.Operation;

/**
 * Action Classifier
 *
 * @author Matias Martinez, [email protected]
 */
public class ActionClassifier {
	// /
	// ROOT CLASSIFIER
	// /
	private List srcUpdTrees = new ArrayList<>();
	private List dstUpdTrees = new ArrayList<>();
	private List srcMvTrees = new ArrayList<>();
	private List dstMvTrees = new ArrayList<>();
	private List srcDelTrees = new ArrayList<>();
	private List dstAddTrees = new ArrayList<>();
	private Map originalActionsSrc = new HashMap<>();
	private Map originalActionsDst = new HashMap<>();

	public ActionClassifier(Set rawMappings, List actions) {
		clean();
		MappingStore mappings = new MappingStore(rawMappings);
		for (Action action : actions) {
			final ITree original = action.getNode();
			if (action instanceof Delete) {
				srcDelTrees.add(original);
				originalActionsSrc.put(original, action);
			} else if (action instanceof Insert) {
				dstAddTrees.add(original);
				originalActionsDst.put(original, action);
			} else if (action instanceof Update) {
				ITree dest = mappings.getDst(original);
				original.setMetadata(SpoonGumTreeBuilder.SPOON_OBJECT_DEST,
						dest.getMetadata(SpoonGumTreeBuilder.SPOON_OBJECT));
				srcUpdTrees.add(original);
				dstUpdTrees.add(dest);
				originalActionsSrc.put(original, action);
			} else if (action instanceof Move) {
				ITree dest = mappings.getDst(original);
				original.setMetadata(SpoonGumTreeBuilder.SPOON_OBJECT_DEST,
						dest.getMetadata(SpoonGumTreeBuilder.SPOON_OBJECT));
				srcMvTrees.add(original);
				dstMvTrees.add(dest);
				originalActionsDst.put(dest, action);
			}
		}
	}

	/**
	 * This method retrieves ONLY the ROOT actions
	 */
	public List getRootActions() {
		final List rootActions = srcUpdTrees.stream().map(t -> originalActionsSrc.get(t))
				.collect(Collectors.toList());

		rootActions.addAll(srcDelTrees.stream() //
				.filter(t -> !srcDelTrees.contains(t.getParent()) && !srcUpdTrees.contains(t.getParent())) //
				.map(t -> originalActionsSrc.get(t)) //
				.collect(Collectors.toList()));

		rootActions.addAll(dstAddTrees.stream() //
				.filter(t -> !dstAddTrees.contains(t.getParent()) && !dstUpdTrees.contains(t.getParent())) //
				.map(t -> originalActionsDst.get(t)) //
				.collect(Collectors.toList()));

		rootActions.addAll(dstMvTrees.stream() //
				.filter(t -> !dstMvTrees.contains(t.getParent())) //
				.map(t -> originalActionsDst.get(t)) //
				.collect(Collectors.toList()));

		rootActions.removeAll(Collections.singleton(null));
		return rootActions;
	}

	private void clean() {
		srcUpdTrees.clear();
		dstUpdTrees.clear();
		srcMvTrees.clear();
		dstMvTrees.clear();
		srcDelTrees.clear();
		dstAddTrees.clear();
		originalActionsSrc.clear();
		originalActionsDst.clear();
	}

	public static List replaceMoveFromAll(Diff editScript) {
		return replaceMove(editScript.getMappingsComp(), editScript.getAllOperations(), true);
	}

	public static List replaceMoveFromRoots(Diff editScript) {
		return replaceMove(editScript.getMappingsComp(), editScript.getRootOperations(), false);
	}

	/**
	 * replaces moves by Insert/Delete operations
	 * 
	 * @param mapping
	 * @param ops
	 * @return
	 */
	public static List replaceMove(MappingStore mapping, List ops, boolean all) {
		List newOps = new ArrayList<>();

		List dels = ops.stream().filter(e -> e instanceof DeleteOperation).map(e -> e.getAction().getNode())
				.collect(Collectors.toList());
		List inss = ops.stream().filter(e -> e instanceof InsertOperation).map(e -> e.getAction().getNode())
				.collect(Collectors.toList());

		for (Operation operation : ops) {

			if (operation instanceof MoveOperation) {

				MoveOperation movOp = (MoveOperation) operation;
				ITree node = movOp.getAction().getNode();

				// Create the delete

				Delete deleteAction = new Delete(node);

				DeleteOperation delOp = new DeleteOperation(deleteAction);

				// add to the final list
				if (all || !inParent(dels, node.getParent())) {
					newOps.add(delOp);
				}

				// Now the insert
				ITree dstNode = mapping.getDst(node);
				ITree parentInAction = movOp.getAction().getParent();
				ITree parent = (mapping.hasSrc(parentInAction)) ? mapping.getDst(parentInAction) : parentInAction;
				int pos = movOp.getPosition();

				Insert insertAc = new Insert(dstNode, parent, pos);

				InsertOperation insertOp = new InsertOperation(insertAc);

				if (all || !inParent(inss, parentInAction)) {
					newOps.add(insertOp);
				}

			} else {
				newOps.add(operation);
			}

		}

		return newOps;
	}

	public static boolean inParent(List trees, ITree parent) {
		if (parent == null) {
			return false;
		}
		if (trees.contains(parent))
			return true;
		else
			return inParent(trees, parent.getParent());

	}
}