Add toolkit option

Author: rohithsh

swan-assist/src/main/java/de/fraunhofer/iem/swan/assist/actions/RunSwanAnalysisImpl.java

@@ -85,7 +85,7 @@ public void run(@NotNull ProgressIndicator indicator) {
                 SwanOptions options = new CliRunner().initializeOptions();
                 options.setTestDataDir(parameters.get(Constants.SOURCE_DIRECTORY));
                 options.setOutputDir( parameters.get(Constants.OUTPUT_DIRECTORY));
-                options.setToolkit("meka");
+                options.setToolkit(parameters.get(Constants.TOOLKIT).toLowerCase());
                 options.setPhase("predict");
                 options.setTrainDataDir("");
 

swan-assist/src/main/java/de/fraunhofer/iem/swan/assist/ui/dialog/RunAnalysisDialog.form

@@ -148,6 +148,46 @@
           </component>
         </children>
       </grid>
+      <grid id="7ce0b" binding="toolkitPanel" layout-manager="GridLayoutManager" row-count="1" column-count="3" same-size-horizontally="false" same-size-vertically="false" hgap="-1" vgap="-1">
+        <margin top="0" left="0" bottom="0" right="0"/>
+        <constraints>
+          <grid row="2" column="0" row-span="1" col-span="1" vsize-policy="3" hsize-policy="3" anchor="0" fill="3" indent="0" use-parent-layout="false"/>
+        </constraints>
+        <properties/>
+        <border type="none"/>
+        <children>
+          <component id="c3485" class="javax.swing.JLabel">
+            <constraints>
+              <grid row="0" column="0" row-span="1" col-span="1" vsize-policy="0" hsize-policy="0" anchor="8" fill="0" indent="0" use-parent-layout="false"/>
+            </constraints>
+            <properties>
+              <text resource-bundle="dialog_messages" key="Launcher.Toolkit"/>
+            </properties>
+          </component>
+          <component id="d7cdd" class="javax.swing.JRadioButton" binding="WEKARadioButton" default-binding="true">
+            <constraints>
+              <grid row="0" column="1" row-span="1" col-span="1" vsize-policy="0" hsize-policy="3" anchor="8" fill="0" indent="0" use-parent-layout="false"/>
+            </constraints>
+            <properties>
+              <text resource-bundle="dialog_messages" key="Launcher.wekaRadioButton"/>
+            </properties>
+          </component>
+          <component id="7d7a5" class="javax.swing.JRadioButton" binding="MEKARadioButton" default-binding="true">
+            <constraints>
+              <grid row="0" column="2" row-span="1" col-span="1" vsize-policy="0" hsize-policy="3" anchor="8" fill="0" indent="0" use-parent-layout="false"/>
+            </constraints>
+            <properties>
+              <text resource-bundle="dialog_messages" key="Launcher.mekaRadioButton"/>
+            </properties>
+          </component>
+        </children>
+      </grid>
     </children>
   </grid>
+  <buttonGroups>
+    <group name="toolkit">
+      <member id="d7cdd"/>
+      <member id="7d7a5"/>
+    </group>
+  </buttonGroups>
 </form>

swan-assist/src/main/java/de/fraunhofer/iem/swan/assist/ui/dialog/RunAnalysisDialog.java

@@ -45,6 +45,9 @@ public class RunAnalysisDialog extends DialogWrapper {
     private JButton trainButton;
     private JPanel trainingPanel;
     private JCheckBox trainingPathCheckbox;
+    private JRadioButton WEKARadioButton;
+    private JRadioButton MEKARadioButton;
+    private JPanel toolkitPanel;
     private HashMap<String, String> parameters = new HashMap<String, String>();
     private ResourceBundle resourceBundle;
     private Properties config;
@@ -223,6 +226,11 @@ private void setParameters() {
         //set the configuration file to default
         parameters.put(Constants.CONFIGURATION_FILE, config.getProperty("swan_default_param_value"));
 
+        if(!MEKARadioButton.isSelected()){
+            parameters.put(Constants.TOOLKIT, MEKARadioButton.getText());
+        }else{
+            parameters.put(Constants.TOOLKIT, WEKARadioButton.getText());
+        }
         parameters.put(Constants.SOURCE_DIRECTORY, sourceDirTextbox.getText());
         parameters.put(Constants.OUTPUT_DIRECTORY, outputDir.getText());
         parameters.put(Constants.OUTPUT_LOG, config.getProperty("log_suffix"));

swan-assist/src/main/java/de/fraunhofer/iem/swan/assist/util/Constants.java

@@ -32,6 +32,7 @@ public class Constants {
     public final static String ANALYSIS_RESULT = PLUGIN_ID + "analysisResult";
     public final static String TRAIN_FILE_SUGGESTED = PLUGIN_ID + "file_suggested_methods";
 
+    public final static String TOOLKIT = "toolkit";
     public final static String DEFAULT_TRAINING_PATH = PLUGIN_ID + "defaultTrainingPath";
     public final static String PROJECT_CONFIGURATION_FILE = PLUGIN_ID + "defaultConfigurationFile";
 

swan-assist/src/main/resources/dialog_messages.properties

@@ -116,4 +116,8 @@ CWE601.FullName= URL Redirection to Untrusted Site ('Open Redirect')
 CWE862.FullName= Missing Authorization
 CWE863.FullName= Incorrect Authorization
 
+Launcher.Toolkit=Toolkit
+Launcher.mekaRadioButton=MEKA
+Launcher.wekaRadioButton=WEKA
+