small fixes here and there

[mir.git] / source / mir / util / GeneratorListFunctions.java

diff --git a/source/mir/util/GeneratorListFunctions.java b/source/mir/util/GeneratorListFunctions.java
index 039d202..dcb8614 100755 (executable)
--- a/source/mir/util/GeneratorListFunctions.java
+++ b/source/mir/util/GeneratorListFunctions.java
@@ -18,13 +18,13 @@
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  *
  * In addition, as a special exception, The Mir-coders gives permission to link
- * the code of this program with  any library licensed under the Apache Software License, 
- * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library 
- * (or with modified versions of the above that use the same license as the above), 
- * and distribute linked combinations including the two.  You must obey the 
- * GNU General Public License in all respects for all of the code used other than 
- * the above mentioned libraries.  If you modify this file, you may extend this 
- * exception to your version of the file, but you are not obligated to do so.  
+ * the code of this program with  any library licensed under the Apache Software License,
+ * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
+ * (or with modified versions of the above that use the same license as the above),
+ * and distribute linked combinations including the two.  You must obey the
+ * GNU General Public License in all respects for all of the code used other than
+ * the above mentioned libraries.  If you modify this file, you may extend this
+ * exception to your version of the file, but you are not obligated to do so.
  * If you do not wish to do so, delete this exception statement from your version.
  */
 package mir.util;
@@ -84,4 +84,44 @@ public class GeneratorListFunctions {
       }
     };
   }
+
+  public static class listSizeFunction implements Generator.GeneratorFunction {
+    public Object perform(List aParameters) throws GeneratorExc, GeneratorFailure {
+      try {
+        if (aParameters.size()!=1)
+          throw new GeneratorExc("listSizeFunction: 1 parameter expected");
+        if (aParameters.get(0)==null)
+          return "";
+
+        if (!(aParameters.get(0) instanceof RewindableIterator) && !(aParameters.get(0) instanceof List))
+          throw new GeneratorExc("listSizeFunction: first parameter must be a RewindableIterator (not a "+aParameters.get(0).getClass().getName()+")");
+
+        if ((aParameters.get(0) instanceof RewindableIterator)) {
+          RewindableIterator iterator = (RewindableIterator) aParameters.get(0);
+          iterator.rewind();
+          int result=0;
+          while (iterator.hasNext()) {
+            result++;
+            iterator.next();
+          }
+
+          iterator.rewind();
+          return new Integer(result);
+        }
+        else {
+          List list = (List) aParameters.get(0);
+          return new Integer(list.size());
+        }
+      }
+      catch (GeneratorExc e) {
+        throw e;
+      }
+      catch (GeneratorFailure e) {
+        throw e;
+      }
+      catch (Throwable e) {
+        throw new GeneratorFailure(e);
+      }
+    };
+  }
 }