Add parameters for parsing part of the file

Signed-off-by: Vincent Guittot <vincent.guittot@linaro.org>

1 files changed, 23 insertions, 3 deletions

diff --git a/parse_aep.py b/parse_aep.py
index 513c275..040a9c7 100755
--- a/parse_aep.py
+++ b/parse_aep.py
@@ -260,8 +260,10 @@ class AEP_parser:
 
         self.fo.write("\n")
 
-    def parse_AEP(self):
+    def parse_AEP(self, start=0, lenght=0):
     # Parse aep data and calculate the energy consumed
+        begin = 0
+
         label_line = 1
 
         topo = {}
@@ -313,6 +315,17 @@ class AEP_parser:
             # convert text to int and unit to micro-unit
             data = self.parse_text(array, hide)
 
+            # get 1st time stamp
+            if begin <= 0:
+                being = data[0]
+
+            # skip data before start
+            if (data[0]-begin) < start:
+                continue
+
+            # stop after lenght
+            if lenght <> 0 and (data[0]-begin) > (start + lenght):
+                continue
             # add virtual domains
             data = self.add_virtual_data(data, virtual)
 
@@ -360,9 +373,11 @@ if __name__ == '__main__':
     verbose = False
     outfile = ""
     figurefile = ""
+    start = 0
+    lenght = 0
 
     try:
-        opts, args = getopt.getopt(sys.argv[1:], "i:vo:")
+        opts, args = getopt.getopt(sys.argv[1:], "i:vo:s:l:")
     except getopt.GetoptError as err:
         print str(err) # will print something like "option -a not recognized"
         sys.exit(2)
@@ -376,6 +391,11 @@ if __name__ == '__main__':
             parse = True
             outfile = a
 
+        if o == "-s":
+            start = int(float(a)*1000000)
+        if o == "-l":
+            lenght = int(float(a)*1000000)
+
     parser = AEP_parser(infile, outfile, figurefile, verbose, )
-    parser.parse_AEP()
+    parser.parse_AEP(start, lenght)