a7d1ecc9ab4c8312592e9194b5fa10c4f1f77461
[sbp.git] / src / edu / berkeley / sbp / bind / BindingFunctor.java
1 package edu.berkeley.sbp.bind;
2 import edu.berkeley.sbp.*;
3 import edu.berkeley.sbp.util.*;
4 import edu.berkeley.sbp.bind.*;
5 import java.io.*;
6 import java.util.*;
7 import java.lang.reflect.*;
8
9 public class BindingFunctor<T> implements TreeFunctor<T,Object>, ToJava {
10     private Binding _binding;
11     private String _toString;
12     public String toString() {
13         return _toString;
14     }
15     public BindingFunctor(String toString, Binding binding) {
16         this._binding = binding;
17         this._toString = toString;
18     }
19     public void toJava(StringBuffer sb) {
20         sb.append("new BindingFunctor(");
21         sb.append("\"");
22         sb.append(StringUtil.toJavaString(_toString));
23         sb.append("\", ");
24         _binding.toJava(sb);
25         sb.append(")");
26     }
27     public Object invoke(Iterable<Tree<T>> t) {
28         ArrayList ret = new ArrayList();
29         for(Tree tc : t) {
30             if (tc.head() != null && tc.head() instanceof Functor)
31                 ret.add(((TreeFunctor<Object,Object>)tc.head()).invoke(tc.children()));
32             else if (tc.size() == 0)
33                 ret.add(tc.head());
34             else {
35                 System.err.println("FIXME: don't know what to do about " + tc);
36                 ret.add(null);
37             }
38         }
39         Object[] o = (Object[])ret.toArray(new Object[0]);
40         return _binding.invoke(o);
41     }
42 }