import edu.neu.ccs.demeterf.ID;
import edu.neu.ccs.demeterf.IDfa;
import edu.neu.ccs.demeterf.Traversal;
import edu.neu.ccs.demeterf.control.EdgeControl;
import edu.neu.ccs.demeter.Ident;

class FL2SL extends ID{
    static Env<String> empty = new Env<String>();
    static Traversal trav = 
        new Traversal(new FL2SL(),
                      EdgeControl.builtIns(Field.class, Formal.class, Method.class));
    static Env<String> convert(List c){ return trav.traverse(c); }
    static Env<String> convert(MethodDefs d){ return trav.traverse(d.mthds); }

    Env<String> apply(Formal f){ return empty.push(""+f.id); }
    Env<String> apply(Field f){ return empty.push(""+f.id); }
    Env<String> apply(ClassDecl c){ return empty.push(""+c.id); }
    Env<String> apply(Method m){ return empty.push(""+m.id); }
    Env<String> combine(ConsList c, Env<String> f, Env<String> r){ return r.push(f); }
    Env<String> combine(EmptyList e){ return empty; }
}

class RewriteEx extends RuntimeException{ RewriteEx(String s){ super(s); } }

public class Rewrite extends IDfa{
    static Traversal trav = new Traversal(new Rewrite(),
                                          EdgeControl.builtIns(Ident.class));
    static <T> T rewrite(T t){ return trav.<T>traverse(t, new Env<String>().push("a")); }
    
    Env<String> update(ClassDecl cl, Env<String> env){
        return FL2SL.convert(cl.flds);
    }
    Env<String> update(Method m, Env<String> env){
        return env.push("this").push(FL2SL.convert(m.getArgs()));
    }
    Env<String> update(Constr c, Env<String> env){
        return env.push(FL2SL.convert(c.getArgs()));
    }
    Env<String> update(RevDefRest r, Env<String> env){ return env.push(""+r.id); }
    
    VarExp apply(SymExp s, Env<String> env){
        return new AddrExp(env.lookup(""+s.id));
    }
}