Add CLI options for source file directories

Author: oshando

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/cli/CliRunner.java

@@ -17,6 +17,12 @@ public class CliRunner implements Callable<Integer> {
     @CommandLine.Option(names = {"-train", "--train-data"}, description = {"Path of training JARs or class files"})
     private String trainDataDir = "";
 
+    @CommandLine.Option(names = {"-test-source", "--test-data-source"}, description = {"Path of test source files"})
+    private String testDataSourceDir = "";
+
+    @CommandLine.Option(names = {"-train-source", "--train-data-source"}, description = {"Path of training source files"})
+    private String trainDataDirSource = "";
+
     @CommandLine.Option(names = {"-d", "--dataset"}, description = {"Path to JSON dataset file"})
     private String datasetJson = "/dataset/swan-dataset.json";
 
@@ -73,6 +79,8 @@ public SwanOptions initializeOptions(){
                 phase);
         options.setPredictionThreshold(predictionThreshold);
         options.setInstances(arffInstancesFiles);
+        options.setTrainDataSourceDir(trainDataDirSource);
+        options.setTestDataSourceDir(testDataSourceDir);
 
         return options;
     }

swan-pipeline/src/main/java/de/fraunhofer/iem/swan/cli/SwanOptions.java

@@ -12,6 +12,8 @@ public class SwanOptions {
 
     private String testDataDir;
     private String trainDataDir;
+    private String testDataSourceDir;
+    private String trainDataSourceDir;
     private String datasetJson;
     private String outputDir;
     private List<String> featureSet;
@@ -126,8 +128,8 @@ public void setCweClasses(List<String> cweClasses) {
         this.cweClasses = cweClasses;
     }
 
-    public List<String> getAllClasses(){
-        return Stream.of(srmClasses,cweClasses).flatMap(Collection::stream).collect(Collectors.toList());
+    public List<String> getAllClasses() {
+        return Stream.of(srmClasses, cweClasses).flatMap(Collection::stream).collect(Collectors.toList());
     }
 
     public boolean isExportArffData() {
@@ -186,6 +188,22 @@ public void setInstances(List<String> instancesArff) {
         this.arffInstancesFiles = instancesArff;
     }
 
+    public String getTestDataSourceDir() {
+        return testDataSourceDir;
+    }
+
+    public void setTestDataSourceDir(String testDataSourceDir) {
+        this.testDataSourceDir = testDataSourceDir;
+    }
+
+    public String getTrainDataSourceDir() {
+        return trainDataSourceDir;
+    }
+
+    public void setTrainDataSourceDir(String trainDataSourceDir) {
+        this.trainDataSourceDir = trainDataSourceDir;
+    }
+
     @Override
     public String toString() {
         return "SwanOptions{" +