updated the plugin to support reading un/weighted feature files

Author: saharmehrpour

.idea/workspace.xml

@@ -12,7 +12,7 @@ class MessageProcessor {
     private static final String[] selectedFragmentKeys = {"filePath", "startOffset", "endOffset", "startLineOffset",
             "lineNumber", "lineText", "text"};
     private static final String[] doiInformationKeys = {"recentVisitedFiles", "recentSearches", "recentVisitedElements"};
-    private static final String[] minedRulesKey = {"minedFrequentItemSets"};
+    private static final String[] minedRulesKey = {"minedFrequentItemSets", "algorithm"};
     private static final String[] elementInfoForForMineRules = {"filePath", "startOffset", "startLineOffset",
             "lineNumber", "text"};
 

ActiveDocumentationPlugin.zip

@@ -116,7 +116,7 @@ void processReceivedMessages(JsonObject messageAsJson) {
                 JsonObject outputContent = LearnDesignRules.analyzeDatabases(projectPath, params, this.directory, alg);
                 // send message
                 sendMessage(MessageProcessor.encodeData(new Object[]{WebSocketConstants.SEND_MINED_DESIGN_RULES,
-                        MessageProcessor.encodeMinedRules(new Object[]{outputContent})
+                        MessageProcessor.encodeMinedRules(new Object[]{outputContent, alg})
                 }).toString());
                 break;
         }

src/MessageProcessor.java

@@ -12,9 +12,19 @@
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 public class LearnDesignRules {
+
+    final static Map<String, String> filePrefix = new HashMap<String, String>() {{
+        put("FPMax", "AttributeEncoding");
+        put("FPClose", "AttributeEncoding");
+        put("CHUI-Miner", "Weighted_AttributeEncoding");
+        put("CHUI-MinerMax", "Weighted_AttributeEncoding");
+    }};
+
     public static JsonObject analyzeDatabases(String projectPath, JsonArray params, String directory, String algorithm) {
 
         String path = projectPath.concat(directory);
@@ -28,14 +38,23 @@ public static JsonObject analyzeDatabases(String projectPath, JsonArray params,
         }
 
         List<String> fileList = new ArrayList<>();
+        List<File> outputFileList = new ArrayList<>();
         File[] listOfFiles = outputDirectory.listFiles();
         if (listOfFiles != null) {
             for (File lof : listOfFiles) {
-                if (lof.isFile() && lof.getName().startsWith("AttributeEncoding")) {
+                if (lof.isFile() && lof.getName().startsWith(filePrefix.get(algorithm))) {
                     fileList.add(lof.getName());
                 }
+                else if (lof.isFile() && lof.getName().startsWith("output_")) {
+                    outputFileList.add(lof);
+                }
             }
         }
+        // delete output files to create new ones
+        for (File file : outputFileList) {
+           file.delete();
+        }
+
         List<String> paramString = new ArrayList<>();
         for (int i=0; i < params.size(); i++) {
             paramString.add(params.get(i).toString());