1 files changed, 9 insertions, 0 deletions

diff --git a/gen/generator.py b/gen/generator.py
index 8b3a189..e49606d 100644
--- a/gen/generator.py
+++ b/gen/generator.py
@@ -152,6 +152,7 @@ class Generator(object):
         rates_arg = []
         etoe_arg = []
         exec_arg = []
+        discount_arg = []
         cluster_arg = []
         clustersz_arg = []
         wss_arg = []
@@ -164,6 +165,7 @@ class Generator(object):
             cluster_arg_t = []
             graph_desc_arg_t = []
             exec_arg_t = []
+            discount_arg_t = []
             rates_arg_t = []
             wss_arg_t = []
             split_arg_t = []
@@ -174,6 +176,9 @@ class Generator(object):
                 cluster_arg_t.append('node_' + str(n.id) + ':' + str(n.task.partition))
                 cost_str = format(n.task.cost/1000.0, '.4f').rstrip('0').rstrip('.')
                 exec_arg_t.append('node_' + str(n.id) + ':' + cost_str)
+                if n.task.cost_discount > 10:
+                    discount_str = format(n.task.cost_discount/1000.0, '.4f').rstrip('0').rstrip('.')
+                    discount_arg_t.append('node_' + str(n.id) + ':' + discount_str)
                 if n.task.split != 1:
                     split_arg_t.append('node_' + str(n.id) + ':' + str(n.task.split))
                 if n.isSrc == True:
@@ -193,12 +198,14 @@ class Generator(object):
             cluster_arg_t = ','.join(cluster_arg_t)
             graph_desc_arg_t = ','.join(graph_desc_arg_t)
             exec_arg_t = ','.join(exec_arg_t)
+            discount_arg_t = ','.join(discount_arg_t)
             wss_arg_t = ','.join(wss_arg_t)
             split_arg_t = ','.join(split_arg_t)
             rates_arg_t = ','.join(rates_arg_t)
 
             cluster_arg.append(cluster_arg_t)
             exec_arg.append(exec_arg_t)
+            discount_arg.append(discount_arg_t)
             graph_desc_arg.append(graph_desc_arg_t)
             wss_arg.append(wss_arg_t)
             split_arg.append(split_arg_t)
@@ -223,6 +230,8 @@ class Generator(object):
             pgm_args_t = '';
             pgm_args_t += '--wait --cluster ' + cluster_arg[i] + ' --clusterSize ' + clustersz_arg[i]
             pgm_args_t += ' --graph ' + graph_desc_arg[i] + ' --rates ' + rates_arg[i] + ' --execution ' + exec_arg[i]
+            if len(discount_arg[i]) != 0:
+                pgm_args_t += ' --discount ' + discount_arg[i]
             if len(split_arg[i]) != 0:
                 pgm_args_t += ' --split ' + split_arg[i]
             if len(wss_arg[i]) != 0: