9846eddd4f46ab75c9b985da63c9053a0c948109
[sbp.git] / src / edu / berkeley / sbp / meta / AnnotationGrammarBindingResolver.java
1 package edu.berkeley.sbp.meta;
2 import edu.berkeley.sbp.util.*;
3 import edu.berkeley.sbp.*;
4 import edu.berkeley.sbp.chr.*;
5 import edu.berkeley.sbp.misc.*;
6 import edu.berkeley.sbp.bind.*;
7 import java.util.*;
8 import java.lang.annotation.*;
9 import java.lang.reflect.*;
10 import java.io.*;
11
12 // FIXME: non-static methods
13 public class AnnotationGrammarBindingResolver extends GrammarBindingResolver {
14
15     private static boolean harsh = true;
16
17     private final Class _cl;
18     private final Class[] _inner;
19     private final Method[] _allMethods;
20
21     public AnnotationGrammarBindingResolver(Class c) {
22         this._cl = c;
23         ArrayList<Class> alc = new ArrayList<Class>();
24         ArrayList<Method> alm = new ArrayList<Method>();
25         add(c, alc, alm);
26         this._inner = (Class[])alc.toArray(new Class[0]);
27         this._allMethods = (Method[])alm.toArray(new Method[0]);
28     }
29
30     public Object repeatTag() { return new Tree.ArrayBuildingTreeFunctor<Object>(); }
31
32     public Sequence tryResolveTag(String tag, String nonTerminalName, Element[] els, Object[] labels, boolean[] drops) {
33         MetaGrammar.Production p = new MetaGrammar.Production(tag, nonTerminalName, els, labels, drops);
34             if (labels != null && labels.length > 0 && "numeric".equals(labels[0]))
35                 System.out.println("here! ");
36         for(Method m : _allMethods)
37             if (new MetaGrammar.Target(m).isCompatible(p))
38                 return new MetaGrammar.Target(m).makeSequence(p);
39         for(Class c : _inner)
40             for(Constructor con : c.getConstructors())
41                 if (new MetaGrammar.Target(con).isCompatible(p))
42                     return new MetaGrammar.Target(con).makeSequence(p);
43         for(Class c : _inner)
44             if (new MetaGrammar.Target(c).isCompatible(p))
45                 return new MetaGrammar.Target(c).makeSequence(p);
46         return null;
47     }
48     public Sequence resolveTag(String tag, String nonTerminalName, Element[] els, Object[] labels, boolean[] drops) {
49         MetaGrammar.Production p = new MetaGrammar.Production(tag, nonTerminalName, els, labels, drops);
50         Sequence ret = tryResolveTag(tag, nonTerminalName, els, labels, drops);
51         if (ret != null) return ret;
52         String message = "could not find a Java method/class/ctor matching tag \""+tag+
53             "\", nonterminal \""+nonTerminalName+"\" with " + els.length + " arguments";
54         if (harsh) {
55             throw new RuntimeException(message);
56         } else {
57             System.err.println(message);
58             return Sequence.rewritingSequence(tag, els, labels, drops);
59         }
60     }
61
62     // helper
63
64     private static void add(Class cl, ArrayList<Class> alc, ArrayList<Method> alm) {
65         for(Method m : cl.getDeclaredMethods())
66             alm.add(m);
67         for(Class c : cl.getDeclaredClasses()) {
68             alc.add(c);
69             add(c, alc, alm);
70         }
71         if (cl.getSuperclass() != Object.class)
72             add(cl.getSuperclass(), alc, alm);
73     }
74
75 }