src/edu/berkeley/sbp/meta/ArrayBuildingTreeFunctor.java

   1 package edu.berkeley.sbp.meta;
   2
   3 import edu.berkeley.sbp.util.*;
   4 import edu.berkeley.sbp.*;
   5 import edu.berkeley.sbp.chr.*;
   6 import edu.berkeley.sbp.misc.*;
   7 import edu.berkeley.sbp.bind.*;
   8 import java.util.*;
   9 import java.lang.annotation.*;
  10 import java.lang.reflect.*;
  11 import java.io.*;
  12
  13 public class ArrayBuildingTreeFunctor<T> implements TreeFunctor<T,T[]>, ToJava {
  14
  15     public void toJava(StringBuffer sb) { sb.append("new ArrayBuildingTreeFunctor()"); }
  16     public String toString() { return ""; }
  17
  18     public T[] invoke(Iterable<Tree<T>> t) {
  19         ArrayList ret = new ArrayList();
  20         for(Tree tc : t) {
  21             if (tc.head() != null && tc.head() instanceof Functor)
  22                 ret.add(((Functor<Iterable<Tree>,Object>)tc.head()).invoke(tc.children()));
  23             else if (tc.numChildren() == 0)
  24                 ret.add(tc.head());
  25             else {
  26                 System.err.println("FIXME: don't know what to do about " + tc);
  27                 ret.add(null);
  28             }
  29         }
  30         return (T[])ret.toArray(new Object[0]);
  31     }
  32 }