Adding time constraints into the scripts.

[pingpong.git] / Code / Projects / SmartPlugDetector / build.gradle

diff --git a/Code/Projects/SmartPlugDetector/build.gradle b/Code/Projects/SmartPlugDetector/build.gradle
index 72c6b713e36eb4f85f9d2c85110b5f4c7bb052fa..2027ed3fe448d1aeaee7377fb7bd8fcb559a7ac0 100644 (file)
--- a/Code/Projects/SmartPlugDetector/build.gradle
+++ b/Code/Projects/SmartPlugDetector/build.gradle
@@ -4,9 +4,17 @@ version '1.0-SNAPSHOT'
 apply plugin: 'java'
 apply plugin: 'application'
 
+// Increase max memory
+applicationDefaultJvmArgs = ["-Xmx300g"]
+
 sourceCompatibility = 1.8
 
-mainClassName = "edu.uci.iotproject.Main"
+//mainClassName = "edu.uci.iotproject.Main"
+//mainClassName = "edu.uci.iotproject.detection.SignatureDetector"
+//mainClassName = "edu.uci.iotproject.detection.layer2.Layer2SignatureDetector"
+//mainClassName = "edu.uci.iotproject.evaluation.DetectionResultsAnalyzer"
+mainClassName = System.getProperty("mainClass")
+
 
 repositories {
     mavenCentral()
@@ -26,4 +34,7 @@ dependencies {
 
     // Apache Commons Math for clustering
     compile 'org.apache.commons:commons-math3:3.6.1'
-}
+
+    // JGraphT: Java Graph library
+    compile 'org.jgrapht:jgrapht-core:1.2.0'
+}
\ No newline at end of file