Contexts for categories

module/src/main/java/pl/clarin/converter/JSON2XLSX.java

@@ -13,6 +13,7 @@ import java.util.HashMap;
 import org.apache.poi.xssf.usermodel.XSSFRow;
 import org.apache.poi.xssf.usermodel.XSSFSheet;
 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
+import org.json.JSONArray;
 import org.json.JSONException;
 
 import org.json.JSONObject;
@@ -27,7 +28,7 @@ public class JSON2XLSX {
     HashMap<String, Integer> keys = new HashMap<>();
     final String KEY = "__SENTENCES__";
 
-    private void process(String fileIn, JSONObject options, XSSFRow row, int column) throws Exception {
+    private void process(String fileIn, JSONObject options, XSSFRow row, int column, String name) throws Exception {
 
         String content = new String(Files.readAllBytes(Paths.get(fileIn)));
         JSONObject json = new JSONObject(content);
@@ -45,7 +46,7 @@ public class JSON2XLSX {
         }
 
         if (json.has(KEY)) {
-            mergeSentences(json.getJSONObject(KEY));
+            mergeSentences(json.getJSONObject(KEY), name);
         }
 
     }
@@ -73,13 +74,13 @@ public class JSON2XLSX {
             for (String fn : file.list()) {
                 row = sheet.createRow(RowNum);
                 row.createCell(col - 1).setCellValue(fn);
-                process(fileIn + "/" + fn, options, row, col);
+                process(fileIn + "/" + fn, options, row, col, fn);
                 RowNum++;
             }
         } else {
             row = sheet.createRow(RowNum);
 
-            process(fileIn, options, row, col);
+            process(fileIn, options, row, col, "");
         }
         header(row0, col);
 
@@ -93,28 +94,45 @@ public class JSON2XLSX {
     public static void main(String[] args) throws Exception {
 
         new JSON2XLSX().process("test/fog/", "test/fog.xlsx", new JSONObject());
-        new JSON2XLSX().process("test/kat/", "test/kat.xlsx", new JSONObject());
+        new JSON2XLSX().process("test/kat2/", "test/kat2.xlsx", new JSONObject());
     }
 
     private JSONObject allSentences;
 
-    private void mergeSentences(JSONObject json) {
+    private void addFileName(JSONObject json, String name) {
+        for (String key : json.keySet()) {
+            for (String skey : json.getJSONObject(key).keySet()) {
+                JSONArray o = json.getJSONObject(key).getJSONArray(skey);
+                for (int i = 0; i < o.length(); i++) {
+                    o.getJSONObject(i).put("name", name);
+                }
+            }
+        }
+    }
+
+    private void mergeSentences(JSONObject json, String name) {
+        addFileName(json, name);
         for (String key : json.keySet()) {
             if (!allSentences.has(key)) {
                 allSentences.put(key, json.getJSONObject(key));
             } else {
                 for (String skey : json.getJSONObject(key).keySet()) {
                     if (!allSentences.getJSONObject(key).has(skey)) {
-                        allSentences.getJSONObject(key).put(skey, json.getJSONObject(key).getJSONArray(skey));
+                        JSONArray o = json.getJSONObject(key).getJSONArray(skey);    
+                        allSentences.getJSONObject(key).put(skey, o);
                     } else {
-                        for(int i=0;i<json.getJSONObject(key).getJSONArray(skey).length();i++)
-                        allSentences.getJSONObject(key).getJSONArray(skey).put(json.getJSONObject(key).getJSONArray(skey).getString(i));
+                        for (int i = 0; i < json.getJSONObject(key).getJSONArray(skey).length(); i++) {
+                            JSONObject o = json.getJSONObject(key).getJSONArray(skey).getJSONObject(i);
+                            allSentences.getJSONObject(key).getJSONArray(skey).put(o);
+
+                        }
                     }
 
                 }
             }
 
         }
+
     }
 
     private void putAllSentences(XSSFWorkbook workBook) {
@@ -122,14 +140,27 @@ public class JSON2XLSX {
             XSSFSheet sheet = workBook.createSheet(key);
             int col = 2;
             int RowNum = 0;
-            for (String skey : allSentences.getJSONObject(key).keySet()) 
-                for (int  i=0;i<allSentences.getJSONObject(key).getJSONArray(skey).length();i++)
-                {   //System.out.println(allSentences.getJSONObject(key).getJSONArray(skey).get(i));
-                    String value = allSentences.getJSONObject(key).getJSONArray(skey).getString(i);
-                    
+            XSSFRow head = sheet.createRow(RowNum++);
+            head.createCell(col - 2).setCellValue("file name");
+            head.createCell(col - 1).setCellValue("key");
+            head.createCell(col).setCellValue("left context");
+            head.createCell(col + 1).setCellValue("sentence");
+            head.createCell(col + 2).setCellValue("right context");
+            for (String skey : allSentences.getJSONObject(key).keySet()) {
+                for (int i = 0; i < allSentences.getJSONObject(key).getJSONArray(skey).length(); i++) {   //System.out.println(allSentences.getJSONObject(key).getJSONArray(skey).get(i));
+                    JSONObject o = allSentences.getJSONObject(key).getJSONArray(skey).getJSONObject(i);
+                    
+                    String name = o.getString("name");
+                    String s = o.getString("s");
+                    String sl = o.getString("sl");
+                    String sr = o.getString("sr");
                     XSSFRow row = sheet.createRow(RowNum++);
+                    row.createCell(col - 2).setCellValue(name);
                     row.createCell(col - 1).setCellValue(skey);
-                    row.createCell(col).setCellValue(value);
+                    row.createCell(col).setCellValue(sl);
+                    row.createCell(col + 1).setCellValue(s);
+                    row.createCell(col + 2).setCellValue(sr);
+                }
             }
 
         }