package wcanalysis.heuristic.model;

import gov.nasa.jpf.symbc.numeric.PathCondition;
import gov.nasa.jpf.vm.ElementInfo;
import gov.nasa.jpf.vm.ThreadInfo;
import gov.nasa.jpf.vm.VM;

/**
 * @author Kasper Luckow
 * Simple statebuilder for tracking memory allocations along paths
 */
public final class HeapAllocState extends State {
  public final static class HeapAllocStateBuilder extends StateBuilderAdapter {

    private long heapAlloced = 0;
    
    //For now, we only keep this guy for statistics purposes...
    private long allocNum = 0;
    
    private HeapAllocStateBuilder(long memAlloced, long allocNum) {
      this.heapAlloced = memAlloced;
      this.allocNum = allocNum;
    }
    
    @Override
    public void handleObjectCreated(VM vm, ThreadInfo ti, ElementInfo ei) {
      this.heapAlloced += ei.getHeapSize();
      this.allocNum++;
    }
    
    @Override
    public StateBuilder copy() {
      return new HeapAllocStateBuilder(heapAlloced, allocNum);
    }
    
    @Override
    public State build(PathCondition resultingPC) {
      return new HeapAllocState(heapAlloced, allocNum, resultingPC);
    }
  }
  
  private final long heapAlloced;
  private final long allocNum;
  
  private HeapAllocState(long heapAlloced, long allocNum, PathCondition pc) {
    super(pc);
    this.heapAlloced = heapAlloced;
    this.allocNum = allocNum;
  }

  @Override
  public String getCSVHeader() {
    return "memAlloced,allocNum";
  }

  @Override
  public String getCSV() {
    return String.valueOf(heapAlloced) + "," + String.valueOf(allocNum);
  }

  @Override
  public int compareTo(State o) {
    if(!(o instanceof HeapAllocState)) {
      throw new IllegalStateException("Expected state fo type " + HeapAllocState.class.getName());
    }
    
    //The notion of worst case is ONLY determined by the total size of allocations along the path; not the number of allocs
    HeapAllocState other = (HeapAllocState)o;
    return this.heapAlloced < other.heapAlloced ? -1:
      this.heapAlloced > other.heapAlloced ? 1 : 0;
  }

  public long getHeapAlloced() {
    return this.heapAlloced;
  }
  
  @Override
  public double getWC() {
    return this.getHeapAlloced();
  }
}

@@ -439,28 +439,24 @@
 lloced = 0;%0A
-    
 %0A    //For n
@@ -540,24 +540,68 @@
 m = 0;%0A    %0A
+    public HeapAllocStateBuilder() %7B %7D%0A    %0A
     private 

Add def ctor to heap allocation cost model