OrderDependentWeights.java

package umontreal.ssj.latnetbuilder.weights;
 
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;

public class OrderDependentWeights extends WeightsComparable<Integer> {

   protected String fileDir = "";
   protected String fileName = "";

   public OrderDependentWeights(ArrayList<SingletonWeightComparable<Integer>> weightList) {
      super(weightList);
   }

   public OrderDependentWeights() {
      super();
   }

   public void setFileDir(String dir) {
      fileDir = dir;
   }

   public String getFileDir() {
      return fileDir;
   }

   public void setFileName(String name) {
      fileName = name;
   }

   public String getFileName() {
      return fileName;
   }

   public void write() throws IOException {
      FileWriter file = new FileWriter(fileDir + fileName);
      StringBuffer sb = new StringBuffer("");
      for (SingletonWeightComparable<Integer> w : weights)
         sb.append("order " + w.getIndex() + ":\t" + w.getWeight() + "\n");
      sb.append("default:\t" + defaultWeight);
      file.write(sb.toString());
      file.close();
   }

   String printBody() {
      if (!sorted)
         sort();
      StringBuffer sb = new StringBuffer("");
      if (weights.size() > 0) {
         int index = 1;
         for (SingletonWeightComparable<Integer> w : weights) {
            while (index < w.getIndex()) {
               sb.append(getDefaultWeight() + ",");
               index++;
            }
            sb.append(w.getWeight() + ",");
            index++;
         }
         sb.deleteCharAt(sb.length() - 1);
      }
      return sb.toString();
 
   }

   @Override
   public String toString() {
      StringBuffer sb = new StringBuffer("");
      sb.append("Order dependent weights [default = " + getDefaultWeight() + "]:\n");
      if (weights.size() > 0)
         sb.append("[");
      sb.append(printBody());
      return sb.toString() + (weights.size() > 0 ? "]" : "");
   }

   public String toLatNetBuilder() {
      StringBuffer sb = new StringBuffer("");
      sb.append("order-dependent:" + getDefaultWeight());
      if (weights.size() > 0)
         sb.append(":");
      sb.append(printBody());
 
      return sb.toString();
   }
 
}