org.refactoringminer.astDiff.utils.Helpers Maven / Gradle / Ivy

Go to download
package org.refactoringminer.astDiff.utils;

import com.github.gumtreediff.tree.Tree;
import com.github.gumtreediff.tree.TreeContext;
import com.github.gumtreediff.utils.Pair;
import gr.uom.java.xmi.diff.UMLAbstractClassDiff;
import gr.uom.java.xmi.diff.UMLModelDiff;
import org.refactoringminer.astDiff.models.ASTDiff;

import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.Map;

/* Created by pourya on 2024-05-22*/
public class Helpers {
    public static String findNameByTree(Map contextMap, Tree t) {
        if (contextMap == null) return null;
        for (Map.Entry stringTreeContextEntry : contextMap.entrySet()) {
            if (stringTreeContextEntry.getValue().getRoot().equals(TreeUtilFunctions.getFinalRoot(t)))
            {
                return stringTreeContextEntry.getKey();
            }
        }
        return null;
    }
    public static Pair findTreeContexts(UMLModelDiff modelDiff, String srcPath, String dstPath) {
        return new Pair<>(modelDiff.getParentModel().getTreeContextMap().get(srcPath),
                modelDiff.getChildModel().getTreeContextMap().get(dstPath));
    }
    public static Pair findTreeContexts(UMLModelDiff modelDiff, UMLAbstractClassDiff classDiff) {
        return findTreeContexts(
                modelDiff,
                classDiff.getOriginalClass().getSourceFile(),
                classDiff.getNextClass().getSourceFile()
        );
    }

    public static Pair findPairOfType(Tree srcOperationNode, Tree dstOperationNode, String searchingType) {
        if (srcOperationNode == null || dstOperationNode == null) return null;
        Tree srcModifier = TreeUtilFunctions.findChildByType(srcOperationNode,searchingType);
        Tree dstModifier = TreeUtilFunctions.findChildByType(dstOperationNode,searchingType);
        if (srcModifier != null && dstModifier != null)
            return new Pair<>(srcModifier, dstModifier);
        return null;
    }

    public static Collection findAppends(Collection diffs, String originalSourceFile, String nextSourceFile) {
        Collection appends = new LinkedHashSet<>();
        for (ASTDiff existing : diffs) {
            if (existing.getSrcPath().equals(originalSourceFile))
                appends.add(existing);
            else if (existing.getDstPath().equals(nextSourceFile))
                appends.add(existing);
        }
        return appends;
    }
}