org.snapscript.tree.variable.Variable Maven / Gradle / Ivy

Go to download
package org.snapscript.tree.variable;

import java.util.concurrent.atomic.AtomicInteger;

import org.snapscript.core.Compilation;
import org.snapscript.core.Context;
import org.snapscript.core.Evaluation;
import org.snapscript.core.constraint.Constraint;
import org.snapscript.core.convert.proxy.ProxyWrapper;
import org.snapscript.core.error.ErrorHandler;
import org.snapscript.core.module.Module;
import org.snapscript.core.module.Path;
import org.snapscript.core.scope.Scope;
import org.snapscript.core.scope.index.Index;
import org.snapscript.core.scope.index.LocalScopeFinder;
import org.snapscript.core.variable.Value;
import org.snapscript.core.variable.bind.VariableBinder;
import org.snapscript.tree.NameReference;

public class Variable implements Compilation {
   
   private final NameReference reference;
   
   public Variable(Evaluation identifier) {
      this.reference = new NameReference(identifier);
   }
   
   @Override
   public Evaluation compile(Module module, Path path, int line) throws Exception {
      Scope scope = module.getScope();
      Context context = module.getContext();
      ErrorHandler handler = context.getHandler();
      ProxyWrapper wrapper = context.getWrapper();
      String name = reference.getName(scope);
      
      return new CompileResult(handler, wrapper, name);
   }
   
   private static class CompileResult extends Evaluation {
   
      private final LocalScopeFinder finder;
      private final VariableBinder binder;
      private final AtomicInteger offset;
      private final String name;
      
      public CompileResult(ErrorHandler handler, ProxyWrapper wrapper, String name) {
         this.binder = new VariableBinder(handler, wrapper, name);
         this.finder = new LocalScopeFinder();
         this.offset = new AtomicInteger(-1);
         this.name = name;
      }
   
      @Override
      public void define(Scope scope) throws Exception{
         Index index = scope.getIndex();
         int depth = index.get(name);
   
         offset.set(depth);
      }
      
      @Override
      public Constraint compile(Scope scope, Constraint left) throws Exception{
         int depth = offset.get();
         Value value = finder.find(scope, name, depth);
         
         if(value == null) {
            return binder.compile(scope);
         }
         return value;
      } 
      
      @Override
      public Value evaluate(Scope scope, Object left) throws Exception{
         int depth = offset.get();
         Value value = finder.find(scope, name, depth);
         
         if(value == null) {
            return binder.bind(scope);
         }
         return value;
      } 
   }
}