Weights.java

package umontreal.ssj.latnetbuilder.weights;
 
import java.util.ArrayList;
import java.util.List;

public abstract class Weights<T> {
   protected double defaultWeight = 0.0;
   protected ArrayList<SingletonWeight<T>> weights; // actual weights

   public Weights(List<SingletonWeight<T>> w) {
      weights = new ArrayList<SingletonWeight<T>>(w);
   }

   public Weights() {
      weights = new ArrayList<SingletonWeight<T>>();
   }

   public ArrayList<SingletonWeight<T>> getWeights() {
      return weights;
   }

   public void setDefaultWeight(double dWeight) {
      defaultWeight = dWeight;
   }

   public double getDefaultWeight() {
      return defaultWeight;
   }

   public void add(SingletonWeight<T> singletonWeight) {
      boolean added = false;
      for (SingletonWeight<T> w : weights) {
         if (w.getIndex() == singletonWeight.getIndex() && (!added)) {
            weights.set(weights.indexOf(w), singletonWeight);
            added = true;
         }
      }
      if (!added)
         weights.add(singletonWeight);
   }

   public void add(T index, double weight) {
      add(new SingletonWeight<T>(index, weight));
   }

   public String toString() {
      StringBuffer sb = new StringBuffer("");
      sb.append("Weights [default = " + getDefaultWeight() + "]\n");
 
      sb.append("[");
      for (SingletonWeight<T> w : weights)
         sb.append(w.getWeight() + ",");
      sb.deleteCharAt(sb.length() - 1);
      sb.append("]\n");
      return sb.toString();
   }
 
   /*
    * Methods to be implemented for non-abstract class
    */
   public abstract String toLatNetBuilder();
 
}