bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/types/TVar.java

   1 package org.simantics.scl.compiler.types;
   2
   3 import java.util.ArrayList;
   4
   5 import org.simantics.scl.compiler.environment.Environment;
   6 import org.simantics.scl.compiler.internal.types.HashCodeUtils;
   7 import org.simantics.scl.compiler.internal.types.TypeHashCodeContext;
   8 import org.simantics.scl.compiler.internal.types.ast.TVarAst;
   9 import org.simantics.scl.compiler.internal.types.ast.TypeAst;
  10 import org.simantics.scl.compiler.types.exceptions.KindUnificationException;
  11 import org.simantics.scl.compiler.types.kinds.KMetaVar;
  12 import org.simantics.scl.compiler.types.kinds.Kind;
  13 import org.simantics.scl.compiler.types.kinds.Kinds;
  14 import org.simantics.scl.compiler.types.util.Polarity;
  15 import org.simantics.scl.compiler.types.util.TypeUnparsingContext;
  16
  17 import gnu.trove.map.hash.THashMap;
  18 import gnu.trove.map.hash.TObjectIntHashMap;
  19 import gnu.trove.set.hash.THashSet;
  20
  21
  22 public final class TVar extends Type {
  23     public static final TVar[] EMPTY_ARRAY = new TVar[0];
  24
  25     private Kind kind;
  26
  27     TVar(Kind kind) {
  28         this.kind = kind;
  29     }
  30
  31     public Kind getKind() {
  32         if(kind instanceof KMetaVar)
  33             kind = Kinds.canonical(kind);
  34         return kind;
  35     }
  36
  37     @Override
  38     public Type replace(TVar var, Type replacement) {
  39         if(this == var)
  40             return replacement;
  41         else
  42             return this;
  43     }
  44
  45     @Override
  46     public TypeAst toTypeAst(TypeUnparsingContext context) {
  47         return new TVarAst(context.getName(this));
  48     }
  49
  50     @Override
  51     public boolean equals(Object obj) {
  52         while(obj instanceof TMetaVar) {
  53             TMetaVar metaVar = (TMetaVar)obj;
  54             if(metaVar.ref == null)
  55                 return false;
  56             else
  57                 obj = metaVar.ref;
  58         }
  59         return this == obj;
  60     }
  61
  62     @Override
  63     public int hashCode() {
  64         return System.identityHashCode(this);
  65     }
  66
  67     @Override
  68     public void updateHashCode(TypeHashCodeContext context) {
  69         TObjectIntHashMap<TVar> varHashCode = context.getVarHashCode();
  70         if(varHashCode != null && varHashCode.containsKey(this))
  71             context.append(varHashCode.get(this));
  72         else
  73             context.append(System.identityHashCode(this));
  74     }
  75
  76     @Override
  77     public void collectFreeVars(ArrayList<TVar> vars) {
  78         if(!vars.contains(this))
  79             vars.add(this);
  80     }
  81
  82     @Override
  83     public void collectMetaVars(ArrayList<TMetaVar> vars) {
  84     }
  85
  86     @Override
  87     public void collectMetaVars(THashSet<TMetaVar> vars) {
  88     }
  89
  90     @Override
  91     public void collectEffectMetaVars(ArrayList<TMetaVar> vars) {
  92     }
  93
  94     @Override
  95     public boolean isGround() {
  96         return false;
  97     }
  98
  99         public Kind inferKind(Environment context) throws KindUnificationException {
 100             if(kind == null)
 101                 kind = Kinds.metaVar();
 102         return kind;
 103     }
 104
 105     @Override
 106     public boolean containsMetaVars() {
 107         return false;
 108     }
 109
 110     @Override
 111     public void toName(TypeUnparsingContext context, StringBuilder b) {
 112         b.append(context.getName(this));
 113     }
 114
 115     @Override
 116     public int getClassId() {
 117         return VAR_ID;
 118     }
 119
 120     @Override
 121     public boolean contains(TMetaVar other) {
 122         return false;
 123     }
 124
 125     @Override
 126     public Type convertMetaVarsToVars() {
 127         return this;
 128     }
 129
 130     @Override
 131     public void addPolarity(Polarity polarity) {
 132     }
 133
 134     @Override
 135     public Type head() {
 136         return this;
 137     }
 138
 139     @Override
 140     public Type copySkeleton(THashMap<TMetaVar, TMetaVar> metaVarMap) {
 141         return this;
 142     }
 143
 144     @Override
 145     public int hashCode(int hash) {
 146         return HashCodeUtils.update(hash, System.identityHashCode(this));
 147     }
 148
 149     @Override
 150     public int hashCode(int hash, TVar[] boundVars) {
 151         for(int i=0;i<boundVars.length;++i)
 152             if(boundVars[i] == this) {
 153                 hash = HashCodeUtils.updateWithPreprocessedValue(hash, BOUND_VAR_HASH);
 154                 return HashCodeUtils.update(hash, i);
 155             }
 156         return HashCodeUtils.update(hash, System.identityHashCode(this));
 157     }
 158
 159     @Override
 160     public boolean equalsCanonical(Type other) {
 161         return this == other;
 162     }
 163
 164     @Override
 165     public Kind getKind(Environment context) {
 166         return kind;
 167     }
 168 }