From 4be005617b3b8c3578cca07c5d085d12916f0d1d Mon Sep 17 00:00:00 2001
From: lclc98
Date: Thu, 30 Jun 2016 00:49:21 +1000
Subject: Json format (#2)

* Added new format

* Fixed bug

* Updated Version
---
 .../enigma/bytecode/MethodParameterWriter.java     | 66 ++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 src/main/java/cuchaz/enigma/bytecode/MethodParameterWriter.java

(limited to 'src/main/java/cuchaz/enigma/bytecode/MethodParameterWriter.java')

diff --git a/src/main/java/cuchaz/enigma/bytecode/MethodParameterWriter.java b/src/main/java/cuchaz/enigma/bytecode/MethodParameterWriter.java
new file mode 100644
index 0000000..e53e8e7
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/bytecode/MethodParameterWriter.java
@@ -0,0 +1,66 @@
+/*******************************************************************************
+ * Copyright (c) 2015 Jeff Martin.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the GNU Lesser General Public
+ * License v3.0 which accompanies this distribution, and is available at
+ * http://www.gnu.org/licenses/lgpl.html
+ * <p>
+ * Contributors:
+ * Jeff Martin - initial API and implementation
+ ******************************************************************************/
+package cuchaz.enigma.bytecode;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import cuchaz.enigma.mapping.*;
+import javassist.CtBehavior;
+import javassist.CtClass;
+import javassist.bytecode.CodeAttribute;
+import javassist.bytecode.LocalVariableAttribute;
+
+public class MethodParameterWriter {
+
+    private Translator m_translator;
+
+    public MethodParameterWriter(Translator translator) {
+        m_translator = translator;
+    }
+
+    public void writeMethodArguments(CtClass c) {
+
+        // Procyon will read method arguments from the "MethodParameters" attribute, so write those
+        for (CtBehavior behavior : c.getDeclaredBehaviors()) {
+
+            // if there's a local variable table here, don't write a MethodParameters attribute
+            // let the local variable writer deal with it instead
+            // procyon starts doing really weird things if we give it both attributes
+            CodeAttribute codeAttribute = behavior.getMethodInfo().getCodeAttribute();
+            if (codeAttribute != null && codeAttribute.getAttribute(LocalVariableAttribute.tag) != null) {
+                continue;
+            }
+
+            BehaviorEntry behaviorEntry = EntryFactory.getBehaviorEntry(behavior);
+
+            // get the number of arguments
+            Signature signature = behaviorEntry.getSignature();
+            if (signature == null) {
+                // static initializers have no signatures, or arguments
+                continue;
+            }
+            int numParams = signature.getArgumentTypes().size();
+            if (numParams <= 0) {
+                continue;
+            }
+
+            // get the list of argument names
+            List<String> names = new ArrayList<String>(numParams);
+            for (int i = 0; i < numParams; i++) {
+                names.add(m_translator.translate(new ArgumentEntry(behaviorEntry, i, "")));
+            }
+
+            // save the mappings to the class
+            MethodParametersAttribute.updateClass(behavior.getMethodInfo(), names);
+        }
+    }
+}
-- 
cgit v1.2.3