1 files changed, 96 insertions, 0 deletions
diff --git a/src/main/java/com/sk89q/warmroast/StackTraceNode.java b/src/main/java/com/sk89q/warmroast/StackTraceNode.java
new file mode 100644
index 0000000..1857cd1
--- /dev/null
+++ b/src/main/java/com/sk89q/warmroast/StackTraceNode.java
@@ -0,0 +1,96 @@
+/*
+ * WarmRoast
+ * Copyright (C) 2013 Albert Pham <http://www.sk89q.com>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package com.sk89q.warmroast;
+
+import java.util.List;
+
+public class StackTraceNode extends StackNode {
+    
+    private final String className;
+    private final String methodName;
+
+    public StackTraceNode(String className, String methodName) {
+        super(className + "." + methodName + "()");
+        this.className = className;
+        this.methodName = methodName;
+    }
+
+    public String getClassName() {
+        return className;
+    }
+
+    public String getMethodName() {
+        return methodName;
+    }
+    
+    @Override
+    public String getNameHtml(McpMapping mapping) {
+        ClassMapping classMapping = mapping.mapClass(getClassName());
+        if (classMapping != null) {
+            String className = "<span class=\"matched\" title=\"" + 
+                    escapeHtml(getClassName()) + "\">" +
+                    escapeHtml(classMapping.getActual()) + "</span>";
+            
+            List<String> actualMethods = classMapping.mapMethod(getMethodName());
+            if (actualMethods.size() == 0) {
+                return className + "." + escapeHtml(getMethodName()) + "()";
+            } else if (actualMethods.size() == 1) {
+                return className + 
+                        ".<span class=\"matched\" title=\"" + 
+                        escapeHtml(getMethodName()) + "\">" + 
+                        escapeHtml(actualMethods.get(0)) + "</span>()";
+            } else {
+                StringBuilder builder = new StringBuilder();
+                boolean first = true;
+                for (String m : actualMethods) {
+                    if (!first) {
+                        builder.append(" ");
+                    }
+                    builder.append(m);
+                    first = false;
+                }
+                return className + 
+                        ".<span class=\"multiple-matches\" title=\"" + 
+                            builder.toString() + "\">" + escapeHtml(getMethodName()) + "</span>()";
+            }
+        } else {
+            String actualMethod = mapping.fromMethodId(getMethodName());
+            if (actualMethod == null) {
+                return escapeHtml(getClassName()) + "." + escapeHtml(getMethodName()) + "()";
+            } else {
+                return className + 
+                        ".<span class=\"matched\" title=\"" + 
+                        escapeHtml(getMethodName()) + "\">" + 
+                        escapeHtml(actualMethod) + "</span>()";
+            }
+        }
+    }
+
+    @Override
+    public int compareTo(StackNode o) {
+        if (getTotalTime() == o.getTotalTime()) {
+            return 0;
+        } else if (getTotalTime()> o.getTotalTime()) {
+            return -1;
+        } else {
+            return 1;
+        }
+    }
+
+}