RoleAnalyzer.java

package kr.ac.snu.selab.soot.analyzer.role;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import kr.ac.snu.selab.soot.analyzer.AbstractAnalyzer;
import kr.ac.snu.selab.soot.analyzer.Analysis;
import kr.ac.snu.selab.soot.analyzer.AnalysisResult;
import kr.ac.snu.selab.soot.core.AbstractProject;
import kr.ac.snu.selab.soot.util.MyUtil;
import kr.ac.snu.selab.soot.util.XMLWriter;
import soot.Hierarchy;
import soot.SootClass;

public class RoleAnalyzer extends AbstractAnalyzer {
	public RoleAnalyzer(AbstractProject project) {
		super(project);
	}

	@Override
	protected void analyze(List<SootClass> classList, Hierarchy hierarchy) {
		Analysis analysis = new Analysis(classList, hierarchy,
				project.isUseSimpleCallGraph());

		Map<String, Integer> analysisFileNameMap = new HashMap<String, Integer>();

		for (AnalysisResult anAnalysisResult : analysis
				.analyzeOverAllAbstractTypes()) {
			if (anAnalysisResult.hasDesignPattern()) {
				String fileName = anAnalysisResult.getAbstractTypeName();
				if (!analysisFileNameMap.containsKey(fileName)) {
					analysisFileNameMap.put(fileName, 0);
				} else {
					int number = analysisFileNameMap.get(fileName) + 1;
					fileName = fileName + String.format("%d", number);
				}

				String outputPath = MyUtil.getPath(outputDirectory, fileName
						+ ".xml");
				XMLWriter writer = new XMLWriter(outputPath);
				anAnalysisResult.writeXML(writer);
				writer.close();
			}
		}

	}

}