change package name. jp.naist.se.stigmata -> jp.sourceforge.stigmata

[stigmata/stigmata.git] / src / main / java / jp / sourceforge / stigmata / printer / csv / ComparisonResultSetCsvPrinter.java

diff --git a/src/main/java/jp/sourceforge/stigmata/printer/csv/ComparisonResultSetCsvPrinter.java b/src/main/java/jp/sourceforge/stigmata/printer/csv/ComparisonResultSetCsvPrinter.java
new file mode 100644 (file)
index 0000000..281d7a8
--- /dev/null
+++ b/src/main/java/jp/sourceforge/stigmata/printer/csv/ComparisonResultSetCsvPrinter.java
@@ -0,0 +1,85 @@
+package jp.sourceforge.stigmata.printer.csv;
+
+/*
+ * $Id$
+ */
+
+import java.io.PrintWriter;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+
+import jp.sourceforge.stigmata.ComparisonPair;
+import jp.sourceforge.stigmata.ComparisonResultSet;
+import jp.sourceforge.stigmata.printer.AbstractComparisonResultSetPrinter;
+import jp.sourceforge.stigmata.result.CertainPairComparisonResultSet;
+
+/**
+ * 
+ * 
+ *
+ * @author Haruaki TAMADA
+ * @version $Revision$ $Date$
+ */
+public class ComparisonResultSetCsvPrinter extends AbstractComparisonResultSetPrinter{
+    public ComparisonResultSetCsvPrinter(){
+    }
+
+    @Override
+    public void printResult(PrintWriter out, ComparisonResultSet resultset){
+        if(resultset instanceof CertainPairComparisonResultSet){
+            printResultImpl(out, (CertainPairComparisonResultSet)resultset);
+        }
+        else{
+            printResultImpl(out, resultset);
+        }
+    }
+
+    private void printResultImpl(PrintWriter out, CertainPairComparisonResultSet resultset){
+        for(ComparisonPair pair: resultset){
+            out.print(pair.getTarget1().getName());
+            out.print(",");
+            out.print(pair.getTarget1().getName());
+            out.print(",");
+            out.println(pair.calculateSimilarity());
+        }
+    }
+
+    private void printResultImpl(PrintWriter out, ComparisonResultSet resultset){
+        Map<String, Map<String, Double>> map = new LinkedHashMap<String, Map<String, Double>>();
+        List<String> names = new ArrayList<String>();
+
+        for(ComparisonPair pair: resultset){
+            Map<String, Double> val = map.get(pair.getTarget1().getName());
+            if(val == null){
+                val = new HashMap<String, Double>();
+            }
+            val.put(pair.getTarget2().getName(), new Double(pair.calculateSimilarity()));
+            if(!names.contains(pair.getTarget2().getName())){
+                names.add(pair.getTarget2().getName());
+            }
+            map.put(pair.getTarget1().getName(), val);
+        }
+
+        for(String name: names){
+            out.print(",");
+            out.print(name);
+        }
+        out.println();
+        for(Map.Entry<String, Map<String, Double>> entry: map.entrySet()){
+            out.print(entry.getKey());
+            Map<String, Double> element = entry.getValue();
+            for(String name: names){
+                out.print(",");
+                Double v = element.get(name);
+                if(v != null){
+                    out.print(v.doubleValue());
+                }
+            }
+            out.println();
+        }
+        out.flush();
+    }
+}