average.java

A framework written in Java for implementing high-level and dynamic languages, compiling them into J

// Copyright (c) 2002  Per M.A. Bothner.
// This is free software;  for terms and warranty disclaimer see ./COPYING.

package gnu.xquery.util;
import gnu.mapping.*;
import gnu.lists.*;
import gnu.math.*;
import gnu.kawa.functions.AddOp;

public class Average extends Procedure1
{
  public static final Average avg = new Average("avg");

  public Average (String name)
  {
    super(name);
  }

  public Object apply1(Object arg)
    throws Throwable
  {
    Object sum = Values.empty;
    int count = 0;
    if (arg instanceof Values)
      {
	TreeList tlist = (TreeList) arg;
	int index = 0;
	for (;;)
	  {
	    Object next = tlist.getPosNext(index);
	    if (next == Sequence.eofValue)
	      break;
	    count++;
            sum = sum == Values.empty ? next : ArithOp.add.apply2(sum, next);
	    index = tlist.nextPos(index);
	  }
      }
    else
      {
	count = 1;
	sum = arg;
      }
    if (sum == Values.empty)
      return sum;
    return sum = ArithOp.div.apply2(sum, IntNum.make(count));
  }
}