All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.snapscript.core.stack.StackTrace Maven / Gradle / Ivy

Go to download 

package org.snapscript.core.stack;

import java.util.Iterator;
import java.util.concurrent.atomic.AtomicInteger;

import org.snapscript.common.ArrayStack;
import org.snapscript.common.Stack;
import org.snapscript.core.function.Function;
import org.snapscript.core.trace.Trace;

public class StackTrace implements Iterable  {
   
   private final Stack functions;
   private final Stack stack;
   private final AtomicInteger depth;
   private final int threshold;

   public StackTrace() {
      this(40);
   }
   
   public StackTrace(int threshold) {
      this.functions = new ArrayStack();
      this.stack = new ArrayStack();
      this.depth = new AtomicInteger();
      this.threshold = threshold;
   }
   
   public Function current() {
      return functions.peek();
   }
   
   @Override
   public Iterator iterator(){
      return stack.iterator();
   }
   
   public void before(Trace trace) {
      stack.push(trace);
   }
   
   public void before(Function function) {
      int size = depth.get();
      
      if(size + 1 > threshold) {
         throw new StackOverflowException("Stack overflow for " + function);   
      }
      depth.getAndIncrement();
      functions.push(function);
      stack.push(function);
   }
   
   public void after(Trace trace) { // remove from stack
      while(!stack.isEmpty()) {
         Object next = stack.pop();
         
         if(next == trace) {
            break;
         }
         Function context = functions.peek();
         
         if(next == context) {
            functions.pop();
            depth.getAndDecrement();
         } 
      }
   }
   
   public void after(Function function) { 
      while(!stack.isEmpty()) {
         Object next = stack.pop();
         
         if(next == function) {
            functions.pop();
            depth.getAndDecrement();
            break;
         }
         Function context = functions.peek();
         
         if(next == context) {
            functions.pop();
            depth.getAndDecrement();
         }
      }
   }
   
   public void clear() {
      stack.clear();
   }
} 















© 2015 - 2024 Weber Informatics LLC | Privacy Policy