WeightsComparable.java

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

 
public abstract class WeightsComparable<T extends Comparable<T>> extends Weights<T> {
 
   protected ArrayList<SingletonWeightComparable<T>> weights; // actual weights
   boolean sorted = false;

   public WeightsComparable(List<SingletonWeightComparable<T>> w) {
      weights = new ArrayList<SingletonWeightComparable<T>>(w);
   }

   public WeightsComparable() {
      weights = new ArrayList<SingletonWeightComparable<T>>();
   }

   // TODO: changing return type with getWeights() does not work. do we really need
   // this?
   public ArrayList<SingletonWeightComparable<T>> getComparableWeights() {
      return weights;
   }

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

   public void sort() {
      Collections.sort(weights);
      sorted = true;
   }

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