Add feature set selector

Author: oshando

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/SwanPipeline.java

@@ -49,8 +49,8 @@ public void run() throws IOException, InterruptedException {
         testDataset.load(dataset.getMethods());
 
         //Initialize and populate features
-        FeaturesHandler featuresHandler = new FeaturesHandler(dataset, testDataset, options);
-        featuresHandler.createFeatures();
+        FeatureSetSelector featureSetSelector = new FeatureSetSelector();
+        IFeatureSet featureSet = featureSetSelector.select(dataset, testDataset, options);
 
         //Train and evaluate model for SRM and CWE categories
         ModelEvaluator modelEvaluator = new ModelEvaluator(featuresHandler, options, testDataset.getMethods());

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/features/FeatureSetSelector.java

@@ -0,0 +1,25 @@
+package de.fraunhofer.iem.swan.features;
+
+import de.fraunhofer.iem.swan.cli.SwanOptions;
+import de.fraunhofer.iem.swan.features.code.soot.SourceFileLoader;
+import de.fraunhofer.iem.swan.io.dataset.SrmList;
+import de.fraunhofer.iem.swan.model.ModelEvaluator;
+
+public class FeatureSetSelector {
+
+    public IFeatureSet select(SrmList trainData, SourceFileLoader testData, SwanOptions options) {
+
+        switch (ModelEvaluator.Mode.valueOf(options.getLearningMode().toUpperCase())) {
+
+            case MEKA:
+                MekaFeatureSet mekaFeatureSet = new MekaFeatureSet(trainData, testData, options);
+                mekaFeatureSet.createFeatures();
+                return mekaFeatureSet;
+            case WEKA:
+                WekaFeatureSet wekaFeatureSet = new WekaFeatureSet(trainData, testData, options);
+                wekaFeatureSet.createFeatures();
+                return wekaFeatureSet;
+        }
+        return null;
+    }
+}

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/features/IFeatureSet.java

@@ -0,0 +1,16 @@
+package de.fraunhofer.iem.swan.features;
+
+import de.fraunhofer.iem.swan.data.Category;
+import de.fraunhofer.iem.swan.data.Method;
+import weka.core.Attribute;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
+
+public interface IFeatureSet {
+
+     void createFeatures();
+
+     ArrayList<Attribute> createAttributes(Set<Category> categories, Set<Method> methods, List<FeatureSet.Type> featureSets);
+    }