Support for raw read/write overheads.

We need information about the time it takes to actually
read and write data when we generate task sets for
actual execution. This patch adds these overheads and
support for them.

1 files changed, 20 insertions, 1 deletions

diff --git a/ecrts14/ecrts14.py b/ecrts14/ecrts14.py
index 9f55ea8..bdf9f5e 100755
--- a/ecrts14/ecrts14.py
+++ b/ecrts14/ecrts14.py
@@ -24,7 +24,7 @@ from pprint import pprint
 import traceback
 
 from schedcat.model.tasks import SporadicTask, TaskSystem
-from schedcat.overheads.model import Overheads, CacheDelay, ConsumerOverheads, ProducerOverheads
+from schedcat.overheads.model import Overheads, CacheDelay, ConsumerOverheads, ProducerOverheads, RawOverheads
 
 import schedcat.model.resources as resources
 import schedcat.generator.tasks as tasks
@@ -245,6 +245,23 @@ def get_ovh_dir():
     ovh_dir = os.path.join(parent, 'overheads')
     return ovh_dir
 
+def get_read_overheads(dp, _system):
+    # always assume no polluters
+    polluters = 'False'
+    read_file = '%s/read/dro_hot_host=%s_lvl=mem_polluters=%s_walk=%s_hpages=%s_upages=%s_type=%s.csv' % (dp.host, dp.host, polluters, dp.walk, str(dp.huge_pages), str(dp.uncached), dp.ovh_type)
+    read_file = os.path.join(get_ovh_dir(), read_file)
+    # reuse the CPMD class
+    ro = RawOverheads.from_file(read_file)
+    return ro
+
+def get_write_overheads(dp, _system):
+    # always assume no polluters
+    polluters = 'False'
+    write_file = '%s/write/dwo_cold_host=%s_lvl=mem_polluters=%s_walk=%s_hpages=%s_upages=%s_type=%s.csv' % (dp.host, dp.host, polluters, dp.walk, str(dp.huge_pages), str(dp.uncached), dp.ovh_type)
+    write_file = os.path.join(get_ovh_dir(), write_file)
+    wo = RawOverheads.from_file(write_file)
+    return wo
+
 def get_consumer_overheads(dp, _system):
     co_file = '%s/consumer/dco_host=%s_lvl=mem_polluters=%s_walk=%s_hpages=%s_upages=%s_type=%s.csv' % (dp.host, dp.host, str(dp.polluters), dp.walk, str(dp.huge_pages), str(dp.uncached), dp.ovh_type)
     co_file = os.path.join(get_ovh_dir(), co_file)
@@ -276,6 +293,8 @@ def get_overheads(dp, system = None):
     ovh.cache_affinity_loss.set_max_wss(max_wss)
     ovh.consumer = get_consumer_overheads(dp, system)
     ovh.producer = get_producer_overheads(dp)
+    ovh.read = get_read_overheads(dp, system)
+    ovh.write = get_write_overheads(dp, system)
     return ovh
 
 def process_dp(_dp):