1 files changed, 75 insertions, 0 deletions
diff --git a/plugins/coverage-common/src/com/intellij/coverage/view/PercentageCoverageColumnInfo.java b/plugins/coverage-common/src/com/intellij/coverage/view/PercentageCoverageColumnInfo.java
new file mode 100644
index 000000000000..52de0da33d10
--- /dev/null
+++ b/plugins/coverage-common/src/com/intellij/coverage/view/PercentageCoverageColumnInfo.java
@@ -0,0 +1,75 @@
+package com.intellij.coverage.view;
+
+import com.intellij.coverage.CoverageEngine;
+import com.intellij.coverage.CoverageSuitesBundle;
+import com.intellij.ide.util.treeView.AbstractTreeNode;
+import com.intellij.ide.util.treeView.NodeDescriptor;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.Comparing;
+import com.intellij.util.ui.ColumnInfo;
+
+import java.util.Comparator;
+
+/**
+* User: anna
+* Date: 1/9/12
+*/
+class PercentageCoverageColumnInfo extends ColumnInfo<NodeDescriptor, String> {
+  private final int myColumnIdx;
+  private final Comparator<NodeDescriptor> myComparator;
+  private final CoverageSuitesBundle mySuitesBundle;
+  private final CoverageViewManager.StateBean myStateBean;
+
+  PercentageCoverageColumnInfo(int columnIdx,
+                               String name,
+                               final CoverageSuitesBundle suitesBundle,
+                               CoverageViewManager.StateBean stateBean) {
+    super(name);
+    this.myColumnIdx = columnIdx;
+    myComparator = new Comparator<NodeDescriptor>() {
+      @Override
+      public int compare(NodeDescriptor o1, NodeDescriptor o2) {
+        final String val1 = valueOf(o1);
+        final String val2 = valueOf(o2);
+        if (val1 != null && val2 != null) {
+          final int percentageIndex1 = val1.indexOf('%');
+          final int percentageIndex2 = val2.indexOf('%');
+          if (percentageIndex1 > -1 && percentageIndex2 >-1) {
+            final String percentage1 = val1.substring(0, percentageIndex1);
+            final String percentage2 = val2.substring(0, percentageIndex2);
+            final int compare = Comparing.compare(Integer.parseInt(percentage1), Integer.parseInt(percentage2));
+            if (compare == 0) {
+              final int total1 = val1.indexOf('/');
+              final int total2 = val2.indexOf('/');
+              if (total1 > -1 && total2 > -1) {
+                final int r1 = val1.indexOf(')', total1);
+                final int r2 = val2.indexOf(')', total2);
+                if (r1 > -1 && r2 > -1) {
+                  return Integer.parseInt(val2.substring(total2 + 1, r2)) - Integer.parseInt(val1.substring(total1 + 1, r1)) ;
+                }
+              }
+            }
+            return compare;
+          }
+          if (percentageIndex1 > -1) return 1;
+          if (percentageIndex2 > -1) return -1;
+        }
+        return Comparing.compare(val1, val2);
+      }
+    };
+    mySuitesBundle = suitesBundle;
+    myStateBean = stateBean;
+  }
+
+  @Override
+  public String valueOf(NodeDescriptor node) {
+    final CoverageEngine coverageEngine = mySuitesBundle.getCoverageEngine();
+    final Project project = node.getProject();
+    return coverageEngine.createCoverageViewExtension(project, mySuitesBundle, myStateBean).getPercentage(myColumnIdx, (AbstractTreeNode)node);
+  }
+
+  @Override
+  public Comparator<NodeDescriptor> getComparator() {
+    return myComparator;
+  }
+}