OSDN Git Service

メンバをソートした.
[stigmata/stigmata-plugins.git] / cflib / src / main / java / jp / sourceforge / stigmata / birthmarks / ControlFlowGraphExtractVisitor.java
1 package jp.sourceforge.stigmata.birthmarks;
2
3 import java.util.Iterator;
4 import java.util.LinkedHashMap;
5 import java.util.Map;
6
7 import org.objectweb.asm.ClassAdapter;
8 import org.objectweb.asm.ClassVisitor;
9 import org.objectweb.asm.MethodVisitor;
10 import org.objectweb.asm.tree.MethodNode;
11
12 /**
13  *
14  *
15  * @author tamada
16  */
17 public class ControlFlowGraphExtractVisitor extends ClassAdapter{
18     private Map<String, MethodNode> opcodesMap = new LinkedHashMap<String, MethodNode>();
19
20     public ControlFlowGraphExtractVisitor(ClassVisitor visitor){
21         super(visitor);
22     }
23
24     public Iterator<String> getMethodNames(){
25         return opcodesMap.keySet().iterator();
26     }
27
28     public ControlFlowGraph getGraph(String name){
29         return buildControlFlow(name, opcodesMap.get(name));
30     }
31
32     @Override
33     public MethodVisitor visitMethod(int access, String name, String descriptor, String signature, String[] exceptions){
34         MethodNode node = new MethodNode(access, name, descriptor, signature, exceptions);
35
36         opcodesMap.put(name + descriptor, node);
37
38         return node;
39     }
40
41     private ControlFlowGraph buildControlFlow(String methodName, MethodNode node){
42         return new ControlFlowGraph(methodName, node);
43     }
44 }