checkpoint
[sbp.git] / src / edu / berkeley / sbp / misc / Demo.java
index 4e0dc1b..02482fa 100644 (file)
@@ -3,19 +3,320 @@ import edu.berkeley.sbp.util.*;
 import edu.berkeley.sbp.*;
 import edu.berkeley.sbp.chr.*;
 import java.util.*;
+import java.lang.annotation.*;
+import java.lang.reflect.*;
 import java.io.*;
 
 public class Demo {
     public static void main(String[] s) throws Exception {
-        /*
-        Tree<String> gram = new CharParser(MetaGrammar.make()).parse(new FileInputStream(s[0])).expand1();
-        MetaGrammar g = (MetaGrammar)new MetaGrammar().walk(gram);
-        Union meta = g.done();
-        //Tree<String> out = new CharParser(meta).parse(new FileInputStream(s[1])).expand1();
-        Forest<String> out = new CharParser(meta).parse(new FileInputStream(s[1]));
-        GraphViz gv = new GraphViz();
-        out.toGraphViz(gv);
-        gv.dump(new PrintWriter(new OutputStreamWriter(System.out)));
-        */
+        Tree<String> res = new CharParser(MetaGrammar.make()).parse(new FileInputStream(s[0])).expand1();
+        
+        MetaGrammar.Meta m =
+            new ReflectiveMeta(MG.class,
+                               new Class[] {
+                                   MG.Grammar.class,
+                                   MG.NonTerminal.class,
+                                   MG.Range.class,
+                                   MG.El.class,
+                                   MG.Seq.class,
+                                   MG.NonTerminalReference.class,
+                                   MG.StringLiteral.class,
+                                   MG.Epsilon.class,
+                                   MG.Tree.class,
+                                   MG.CharClass.class
+                               });
+        MetaGrammar.Meta.MetaGrammarFile mgf = m.new MetaGrammarFile(res);
+        MetaGrammar.BuildContext bc = new MetaGrammar.BuildContext(mgf);
+        Union meta = mgf.get("s").build(bc);
+
+        System.err.println("parsing " + s[1]);
+        res = new CharParser(meta).parse(new FileInputStream(s[1])).expand1();
+
+        System.out.println(res);
+    }
+
+    public static class ReflectiveMeta extends MetaGrammar.Meta {
+        private final Class _cl;
+        private final Class[] _inner;
+        public ReflectiveMeta(Class c, Class[] inner) {
+            this._cl = c;
+            this._inner = inner;
+        }
+        private boolean match(Method m, String s) { return match(m.getAnnotation(tag.class), null, s); }
+        private boolean match(tag t, Class c, String s) {
+            if (t==null) return false;
+            if (t.value().equals(s)) return true;
+            if (c != null && t.equals("") && c.getSimpleName().equals(s)) return true;
+            return false;
+        }
+        private boolean match(nonterminal t, Class c, String s) {
+            if (t==null) return false;
+            if (t.value().equals(s)) return true;
+            if (c != null && t.equals("") && c.getSimpleName().equals(s)) return true;
+            return false;
+        }
+        private boolean match(Class c, String s, String nonTerminalName) {
+            if (match((tag)c.getAnnotation(tag.class), c, s)) return true;
+            if (match((nonterminal)c.getAnnotation(nonterminal.class), c, nonTerminalName)) return true;
+            return false;
+        }
+        public boolean match(Constructor con, String s, String nonTerminalName) {
+            Class c = con.getDeclaringClass();
+            if (match((tag)con.getAnnotation(tag.class), null, s)) return true;
+            if (match((nonterminal)con.getAnnotation(nonterminal.class), c, s)) return true;
+            return false;
+        }
+        public Sequence resolveTag(String tag, String nonTerminalName, Element[] els, Object[] labels, boolean[] drops) {
+            Production p = new Production(tag, nonTerminalName, els, labels, drops);
+            for(Method m : _cl.getMethods())
+                if (new TargetMethod(m).isCompatible(p))
+                    return Sequence.rewritingSequence(m, els, labels, drops);
+            for(Class c : _inner)
+                for(Constructor con : c.getConstructors())
+                    if (new TargetConstructor(con).isCompatible(p))
+                        return Sequence.rewritingSequence(con, els, labels, drops);
+            for(Class c : _inner)
+                if (new TargetClass(c).isCompatible(p))
+                    return Sequence.rewritingSequence(c, els, labels, drops);
+            throw new RuntimeException("could not find a Java method/class/ctor matching tag \""+tag+"\", nonterminal \""+nonTerminalName+"\"");
+        }
     }
+    /*
+    public static Object makeFlattener(final Method m, final Element[] els, final Object[] labels, final boolean[] drops) {
+        return new Reducer() {
+                public Object reduce(Tree t) {
+                    Object[] o = new Object[m.getParameterTypes()];
+                    int j = 0;
+                    for(int i=0; i<els.length; i++)
+                }
+            };
+    }
+    */
+
+    
+    /**
+     *  Constructors, classes, and methods with this attribute will
+     *  match every production of the nonterminal called "value()"
+     *  that is arg-compatible.  If value() is undefined, then the
+     *  class/constructor/method name is used.
+     */ 
+    @Retention(RetentionPolicy.RUNTIME) public static @interface nonterminal { String value() default ""; }
+
+    /**
+     *  Constructors, classes, and methods with this attribute will
+     *  match every tree tagged with "value()" that is arg-compatible.
+     *  If value() is undefined, then the class/constructor/method
+     *  name is used.
+     */ 
+    @Retention(RetentionPolicy.RUNTIME) public static @interface tag         { String value() default ""; }
+
+    /**
+     *  If any parameter to a method or field in a class has a named
+     *  arg-tag, that parameter/field matches the child of the tree
+     *  which either has that label or else is a reference to a
+     *  nonterminal with the corresponding name.
+     *  
+     *  The remaining non-named arg-tags match the remaining children
+     *  of the tree in sequential order.
+     *
+     *  If any arg-tagged parameters/fields remain, the match fails.
+     *  If there were no arg-tagged parameters-fields, it is as if all
+     *  of them were non-named and arg-tagged.
+     *
+     *  A method/constructor is arg-compatible if all of its arguments
+     *  are arg-compatible.
+     *
+     *  A class is arg-compatible if all of its fields are
+     *  arg-compatible, or if one of its constructors is arg-compatible.
+     *
+     */
+    @Retention(RetentionPolicy.RUNTIME) public static @interface arg         { String value() default ""; }
+
+    public static class Production {
+        public String tag;
+        public String nonTerminal;
+        public Object[] labels;
+        public boolean[] drops;
+        public Element[] elements;
+        public int count = 0;
+        public Production(String tag, String nonTerminal, Element[] elements, Object[] labels, boolean[] drops) {
+            this.tag = tag;
+            this.elements = elements; // FIXME: use this
+            this.nonTerminal = nonTerminal;
+            this.labels = labels;
+            this.drops = drops;
+            for(int i=0; i<drops.length; i++)
+                if (!drops[i])
+                    count++;
+        }
+    }
+
+    public static abstract class Target {
+        public abstract String getName();
+        public abstract tag getTag();
+        public abstract nonterminal getNonTerminal();
+        public abstract boolean buildSequence(Production p);
+        public boolean isCompatible(Production p) {
+            tag t = getTag();
+            if (t != null &&
+                (t.value().equals(p.tag) ||
+                 (t.value().equals("") && p.tag.equals(getName()))))
+                return buildSequence(p);
+
+            nonterminal n = getNonTerminal();
+            if (n != null &&
+                (n.value().equals(p.nonTerminal) ||
+                 (n.value().equals("") && p.nonTerminal.equals(getName()))))
+                return buildSequence(p);
+
+            return false;
+        }
+        public int[] buildSequence(Production p, String[] names, arg[] argtags) {
+            int argTagged = 0;
+            for(int i=0; i<argtags.length; i++)
+                if (argtags[i] != null)
+                    argTagged++;
+
+            // FIXME: can be smarter here
+            if (names.length==p.count) {
+                int[] ret = new int[p.count];
+                for(int i=0; i<p.count; i++) ret[i] = i;
+                return ret;
+            } else if (argTagged==p.count) {
+                int[] ret = new int[argTagged];
+                int j = 0;
+                for(int i=0; i<argtags.length; i++)
+                    if (argtags[i]!=null)
+                        ret[j++] = i;
+                return ret;
+            } else {
+                return null;
+            }
+        }
+    }
+    public static class TargetClass extends Target {
+        public final Class _class;
+        public TargetClass(Class _class) { this._class = _class; }
+        public String getName() { return _class.getSimpleName(); }
+        public tag getTag() { return (tag)_class.getAnnotation(tag.class); }
+        public nonterminal getNonTerminal() { return (nonterminal)_class.getAnnotation(nonterminal.class); }
+        public boolean buildSequence(Production p) {
+            Field[]  f       = _class.getDeclaredFields();
+            String[] names   = new String[f.length];
+            arg[]    argtags = new arg[f.length];
+            for(int i=0; i<f.length; i++) {
+                names[i]   = f[i].getName();
+                argtags[i] = f[i].getAnnotation(arg.class);
+            }
+            int[] ret = buildSequence(p, names, argtags);
+            if (ret!=null) return true;
+            for(Constructor c : _class.getConstructors())
+                if (new TargetConstructor(c).buildSequence(p))
+                    return true;
+            return false;
+        }
+    }
+    public static class TargetConstructor extends Target {
+        public final Constructor _ctor;
+        public TargetConstructor(Constructor _ctor) { this._ctor = _ctor; }
+        public String getName() { return _ctor.getName(); }
+        public tag getTag() { return (tag)_ctor.getAnnotation(tag.class); }
+        public nonterminal getNonTerminal() { return (nonterminal)_ctor.getAnnotation(nonterminal.class); }
+        public boolean buildSequence(Production p) {
+            Annotation[][] annotations = _ctor.getParameterAnnotations();
+            int len = annotations.length;
+            int ofs = 0;
+            String name = _ctor.getDeclaringClass().getName();
+            if (name.indexOf('$') > name.lastIndexOf('.')) {
+                len--;
+                ofs++;
+            }
+            String[] names   = new String[len];
+            arg[]    argtags = new arg[len];
+            for(int i=0; i<names.length; i++)
+                for(Annotation a : annotations[i+ofs])
+                    if (a instanceof arg)
+                        argtags[i+ofs] = (arg)a;
+            int[] ret = buildSequence(p, names, argtags);
+            if (ret!=null) return true;
+            return false;
+        }
+    }
+    public static class TargetMethod extends Target {
+        public final Method _method;
+        public TargetMethod(Method _method) { this._method = _method; }
+        public String getName() { return _method.getName(); }
+        public tag getTag() { return (tag)_method.getAnnotation(tag.class); }
+        public nonterminal getNonTerminal() { return (nonterminal)_method.getAnnotation(nonterminal.class); }
+        public boolean buildSequence(Production p) {
+            Annotation[][] annotations = _method.getParameterAnnotations();
+            String[] names   = new String[annotations.length];
+            arg[]    argtags = new arg[annotations.length];
+            for(int i=0; i<names.length; i++)
+                for(Annotation a : annotations[i])
+                    if (a instanceof arg)
+                        argtags[i] = (arg)a;
+            int[] ret = buildSequence(p, names, argtags);
+            if (ret!=null) return true;
+            return false;
+        }
+    }
+
+    public class MG {
+        public @tag void grammar(Grammar g) { }
+        public @nonterminal void Grammar(NonTerminal[] n) { }
+        public @nonterminal class Grammar {
+            public @arg("NonTerminal") NonTerminal[] nonterminals;
+        }
+        public @nonterminal class NonTerminal {
+            public @arg("Word") String  name;
+            public @arg("RHS")  Seq[][] sequences;
+        }
+
+        public @tag void range(char c) { }
+        public @nonterminal class Range {
+            public @tag("range") Range(char only) { first = only; last = only; }
+            public @arg char first;
+            public @arg char last;
+        }
+        public class El { }
+        public abstract @nonterminal("Sequence") class Seq { }
+        public @tag("&")   Seq and(Sequence s,    Element[] elements) { return null; }
+        public @tag("&~")  Seq andnot(Sequence s, Element[] elements) { return null; }
+        public @tag        Seq ps(Element[] elements) { return null; }
+        public @tag("::")  Seq tag(String tagname, Seq seq) { return null; }
+        public @tag(":")   void colon(String s, Element e) { }
+        public @tag(")")   void close(String foo) { }
+        public @tag("/")   Seq slash(Seq s, Element e) { return null; }
+        public @tag("->")  Seq arrow(Seq s, Element e) { return null; }
+
+        public @tag("nonTerminal") class NonTerminalReference { public @arg String nonTerminal; }
+        public @tag("literal")     class StringLiteral        { public @arg String string; }
+        public @tag("()")          class Epsilon              { }
+        public @tag("{")           class Tree                 { @arg Seq body; }
+        public @tag("[")           class CharClass            { public CharClass(Range[] ranges) { } }
+
+        public @tag("++")  void plusmax(El e) { }
+        public @tag("+")   void plus(El e) { }
+        public @tag("++/") void plusmaxfollow(El e, El sep) { }
+        public @tag("+/")  void plusfollow(El e, El sep) { }
+        public @tag("**")  void starmax(El e) { }
+        public @tag("*")   void star(El e) { }
+        public @tag("**/") void starmaxfollow(El e, El sep) { }
+        public @tag("*/")  void starfollow(El e, El sep) { }
+        public @tag("!")   void bang(El e) { }
+        public @tag("?")   void question(El e) { }
+        public @tag("^")   void caret(String s) { }
+        public @tag("~")   void tilde(El e) { }
+        public @tag("^^")  void doublecaret(El e) { }
+        public @tag("(")   void subexpression(Seq[][] rhs) { }
+
+        public @nonterminal("Word")    String word(String s) { return null; }
+        public @nonterminal("Quoted")  String quoted(String s) { return null; }
+        public @nonterminal("escaped") String c(char c) { return null; }
+        public @tag("\"\"")            String emptystring() { return null; }
+        public @tag("\r")              String lf() { return null; }
+        public @tag("\n")              String cr() { return null; }
+    }
+
 }