1 files changed, 75 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/mapping/LocalVariableDefEntry.java b/src/main/java/cuchaz/enigma/mapping/LocalVariableDefEntry.java
new file mode 100644
index 0000000..cc677c5
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/mapping/LocalVariableDefEntry.java
@@ -0,0 +1,75 @@
+package cuchaz.enigma.mapping;
+
+import com.google.common.base.Preconditions;
+import cuchaz.enigma.utils.Utils;
+
+/**
+ * TypeDescriptor...
+ * Created by Thog
+ * 19/10/2016
+ */
+public class LocalVariableDefEntry extends LocalVariableEntry {
+
+        protected final MethodDefEntry ownerEntry;
+        protected final TypeDescriptor desc;
+
+        public LocalVariableDefEntry(MethodDefEntry ownerEntry, int index, String name, TypeDescriptor desc) {
+                super(ownerEntry, index, name);
+                Preconditions.checkNotNull(desc, "Variable desc cannot be null");
+
+                this.ownerEntry = ownerEntry;
+                this.desc = desc;
+        }
+
+        public LocalVariableDefEntry(MethodDefEntry ownerEntry, int index, String name) {
+                super(ownerEntry, index, name);
+
+                this.ownerEntry = ownerEntry;
+
+                int namedIndex = getNamedIndex();
+                if (namedIndex < 0) {
+                        this.desc = TypeDescriptor.of(ownerEntry.getOwnerClassEntry().getName());
+                } else {
+                        this.desc = ownerEntry.getDesc().getArgumentDescs().get(namedIndex);
+                }
+        }
+
+        @Override
+        public MethodDefEntry getOwnerEntry() {
+                return this.ownerEntry;
+        }
+
+        public TypeDescriptor getDesc() {
+                return desc;
+        }
+
+        public int getNamedIndex() {
+                // If we're not static, "this" is bound to index 0
+                int indexOffset = ownerEntry.getAccess().isStatic() ? 0 : 1;
+                return index - indexOffset;
+        }
+
+        @Override
+        public LocalVariableDefEntry updateOwnership(ClassEntry classEntry) {
+                return new LocalVariableDefEntry(ownerEntry.updateOwnership(classEntry), index, name, desc);
+        }
+
+        @Override
+        public int hashCode() {
+                return Utils.combineHashesOrdered(this.ownerEntry, this.desc.hashCode(), this.name.hashCode(), Integer.hashCode(this.index));
+        }
+
+        @Override
+        public boolean equals(Object other) {
+                return other instanceof LocalVariableDefEntry && equals((LocalVariableDefEntry) other);
+        }
+
+        public boolean equals(LocalVariableDefEntry other) {
+                return this.ownerEntry.equals(other.ownerEntry) && this.desc.equals(other.desc) && this.name.equals(other.name) && this.index == other.index;
+        }
+
+        @Override
+        public String toString() {
+                return this.ownerEntry + "(" + this.index + ":" + this.name + ":" + this.desc + ")";
+        }
+}