source/mir/util/GeneratorListFunctions.java

   1 /*
   2  * Copyright (C) 2001, 2002 The Mir-coders group
   3  *
   4  * This file is part of Mir.
   5  *
   6  * Mir is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * Mir is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with Mir; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  * In addition, as a special exception, The Mir-coders gives permission to link
  21  * the code of this program with  any library licensed under the Apache Software License,
  22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
  23  * (or with modified versions of the above that use the same license as the above),
  24  * and distribute linked combinations including the two.  You must obey the
  25  * GNU General Public License in all respects for all of the code used other than
  26  * the above mentioned libraries.  If you modify this file, you may extend this
  27  * exception to your version of the file, but you are not obligated to do so.
  28  * If you do not wish to do so, delete this exception statement from your version.
  29  */
  30 package mir.util;
  31
  32 import java.util.List;
  33 import java.util.Vector;
  34
  35 import mir.generator.Generator;
  36 import mir.generator.GeneratorExc;
  37 import mir.generator.GeneratorFailure;
  38
  39 public class GeneratorListFunctions {
  40   private GeneratorListFunctions() {}
  41
  42   public static class subListFunction implements Generator.GeneratorFunction {
  43     public Object perform(List aParameters) throws GeneratorExc, GeneratorFailure {
  44       try {
  45         int skip;
  46         int maxSize;
  47
  48         if (aParameters.size()>3 || aParameters.size()<2)
  49           throw new GeneratorExc("iteratorSubsetFunction: 2 or 3 parameters expected");
  50         if (aParameters.get(0)==null)
  51           return "";
  52
  53         if (!(aParameters.get(0) instanceof RewindableIterator) && !(aParameters.get(0) instanceof List))
  54           throw new GeneratorExc("iteratorSubsetFunction: first parameter must be a RewindableIterator (not a "+aParameters.get(0).getClass().getName()+")");
  55
  56         skip = StringRoutines.interpretAsInteger(aParameters.get(1));
  57         if (aParameters.size()>=2)
  58           maxSize = StringRoutines.interpretAsInteger(aParameters.get(2));
  59         else
  60           maxSize = -1;
  61
  62
  63         if ((aParameters.get(0) instanceof RewindableIterator))
  64           return new SubsetIterator((RewindableIterator) aParameters.get(0), skip, maxSize);
  65         else {
  66           List list = (List) aParameters.get(0);
  67
  68           if (skip>=list.size())
  69             return new Vector();
  70           if (maxSize<0 || (skip+maxSize)>=list.size())
  71             return list.subList(skip, list.size());
  72           else
  73             return list.subList(skip, skip+maxSize);
  74         }
  75       }
  76       catch (GeneratorExc e) {
  77         throw e;
  78       }
  79       catch (GeneratorFailure e) {
  80         throw e;
  81       }
  82       catch (Throwable e) {
  83         throw new GeneratorFailure(e);
  84       }
  85     };
  86   }
  87
  88   public static class listSizeFunction implements Generator.GeneratorFunction {
  89     public Object perform(List aParameters) throws GeneratorExc, GeneratorFailure {
  90       try {
  91         if (aParameters.size()!=1)
  92           throw new GeneratorExc("listSizeFunction: 1 parameter expected");
  93         if (aParameters.get(0)==null)
  94           return "";
  95
  96         if (!(aParameters.get(0) instanceof RewindableIterator) && !(aParameters.get(0) instanceof List))
  97           throw new GeneratorExc("listSizeFunction: first parameter must be a RewindableIterator (not a "+aParameters.get(0).getClass().getName()+")");
  98
  99         if ((aParameters.get(0) instanceof RewindableIterator)) {
 100           RewindableIterator iterator = (RewindableIterator) aParameters.get(0);
 101           iterator.rewind();
 102           int result=0;
 103           while (iterator.hasNext()) {
 104             result++;
 105             iterator.next();
 106           }
 107
 108           iterator.rewind();
 109           return new Integer(result);
 110         }
 111         else {
 112           List list = (List) aParameters.get(0);
 113           return new Integer(list.size());
 114         }
 115       }
 116       catch (GeneratorExc e) {
 117         throw e;
 118       }
 119       catch (GeneratorFailure e) {
 120         throw e;
 121       }
 122       catch (Throwable e) {
 123         throw new GeneratorFailure(e);
 124       }
 125     };
 126   }
 127 }