Add machine learning feature weights

Author: oshando

swan_core/pom.xml

@@ -6,7 +6,7 @@
 	<groupId>de.upb.cs.swt</groupId>
 	<artifactId>swan_core</artifactId>
 	<packaging>jar</packaging>
-	<version>1.2.0</version>
+	<version>1.2.1</version>
 
 	<name>SWAN Weakness Detector</name>
 	<description>SWAN is a machine-learning approach for detection of methods of interest for security in Java libraries.</description>

swan_core/src/main/java/de/fraunhofer/iem/swan/FeatureHandler.java

@@ -14,7 +14,7 @@
 import soot.SootMethod;
 import soot.options.Options;
 
-public abstract class AbstractSootFeature implements IFeature {
+public abstract class AbstractSootFeature extends WeightedFeature implements IFeature {
 
   private static boolean SOOT_INITIALIZED = false;
 

swan_core/src/main/java/de/fraunhofer/iem/swan/features/AbstractSootFeature.java

@@ -8,7 +8,7 @@
  * 
  * @author Siegfried Rasthofer
  */
-public class BaseNameOfClassPackageName implements IFeature {
+public class BaseNameOfClassPackageName extends WeightedFeature implements IFeature {
   private final String baseNameOfClassPackageName;
 
   public BaseNameOfClassPackageName(String baseNameOfClassPackageName) {

swan_core/src/main/java/de/fraunhofer/iem/swan/features/BaseNameOfClassPackageName.java

@@ -9,7 +9,7 @@
  * @author Lisa Nguyen Quang Do
  *
  */
-public class IsImplicitMethod implements IFeature {
+public class IsImplicitMethod extends WeightedFeature implements IFeature {
 
   @Override
   public Type applies(Method method) {
@@ -21,4 +21,13 @@ public String toString() {
     return "<Implicit method>";
   }
 
+  @Override
+  public void setWeight(int weight) {
+    super.setWeight(weight);
+  }
+
+  @Override
+  public int getWeight() {
+    return super.getWeight();
+  }
 }

swan_core/src/main/java/de/fraunhofer/iem/swan/features/IsImplicitMethod.java

@@ -11,7 +11,7 @@
  * @author Siegfried Rasthofer
  *
  */
-public class MethodAnonymousClassFeature implements IFeature {
+public class MethodAnonymousClassFeature extends WeightedFeature implements IFeature {
 
   private final boolean anonymousClass;
 

swan_core/src/main/java/de/fraunhofer/iem/swan/features/MethodAnonymousClassFeature.java

@@ -3,7 +3,7 @@
 import de.fraunhofer.iem.swan.IFeature;
 import de.fraunhofer.iem.swan.data.Method;
 
-public class MethodClassConcreteNameFeature implements IFeature {
+public class MethodClassConcreteNameFeature extends WeightedFeature implements IFeature {
 
   private final String className;
 

swan_core/src/main/java/de/fraunhofer/iem/swan/features/MethodClassConcreteNameFeature.java

@@ -3,7 +3,7 @@
 import de.fraunhofer.iem.swan.IFeature;
 import de.fraunhofer.iem.swan.data.Method;
 
-public class MethodClassContainsNameFeature implements IFeature {
+public class MethodClassContainsNameFeature extends WeightedFeature implements IFeature {
 
   private final String partOfName;
 

swan_core/src/main/java/de/fraunhofer/iem/swan/features/MethodClassContainsNameFeature.java

@@ -3,7 +3,7 @@
 import de.fraunhofer.iem.swan.IFeature;
 import de.fraunhofer.iem.swan.data.Method;
 
-public class MethodClassEndsWithNameFeature implements IFeature {
+public class MethodClassEndsWithNameFeature extends WeightedFeature implements IFeature {
 
   private final String partOfName;
 

swan_core/src/main/java/de/fraunhofer/iem/swan/features/MethodClassEndsWithNameFeature.java

@@ -3,7 +3,7 @@
 import de.fraunhofer.iem.swan.IFeature;
 import de.fraunhofer.iem.swan.data.Method;
 
-public class MethodHasParametersFeature implements IFeature {
+public class MethodHasParametersFeature extends WeightedFeature implements IFeature {
 
   public MethodHasParametersFeature() {}