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

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