X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/12c0f182635ed71b370d608b5b9a614153765cd4..9fd6cbc6c3b06f4b09e3c3339ffb3cc8a68f9bfa:/tools/sg_xml_unit_converter.py

diff --git a/tools/sg_xml_unit_converter.py b/tools/sg_xml_unit_converter.py
index 5b53594932..4fb5390b41 100644
--- a/tools/sg_xml_unit_converter.py
+++ b/tools/sg_xml_unit_converter.py
@@ -1,84 +1,95 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
+# Copyright (c) 2013-2018. The SimGrid Team.
+# All rights reserved.
+
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the license (GNU LGPL) which comes with this package.
+
 # grep -ohrI 'bw=".*"' . | sort | uniq
 
-import sys, fnmatch, os
-from decimal import Decimal
+import sys
+import fnmatch
+import os
+from decimal import Decimal, DecimalException
 import re
 
+
 def to_str(dec):
-  return re.sub(r"(\.\d*?)0*$", r"\1", dec.to_eng_string()).rstrip(".")
-
-def format(xml, formats, attrib):
-  res = []
-  m = re.search(r'%s="(.*?)"'%attrib, xml)
-  while m:
-    b, e = m.span(1)
-    res.append(xml[:b])
-    val = xml[b:e]
-    xml = xml[e:]
-    try:
-      power = Decimal(val)
-      tmp = to_str(power)
-      for p, f in formats:
-        d = power / p
-        if d >= 1.0:
-          tmp = "%s%s"%(to_str(d), f)
-          break
-      res.append(tmp)
-    except:
-      print "Error with:", val
-      res.append(val)
-    m = re.search(r'%s="(.*?)"'%attrib, xml)
-
-  res.append(xml)
-  return "".join(res)
-
-def formats(list):
-  return sorted(((Decimal(i), j) for i,j in list), key=lambda x: x[0], reverse=True)
+    return re.sub(r"(\.\d*?)0*$", r"\1", dec.to_eng_string()).rstrip(".")
+
+
+def convert(xml, formats, attrib):
+    res = []
+    m = re.search(r'%s="(.*?)"' % attrib, xml)
+    while m:
+        b, e = m.span(1)
+        res.append(xml[:b])
+        val = xml[b:e]
+        xml = xml[e:]
+        try:
+            power = Decimal(val)
+            tmp = to_str(power)
+            for p, f in formats:
+                d = power / p
+                if d >= 1.0:
+                    tmp = "%s%s" % (to_str(d), f)
+                    break
+            res.append(tmp)
+        except DecimalException:
+            print "Error with:", val
+            res.append(val)
+        m = re.search(r'%s="(.*?)"' % attrib, xml)
+
+    res.append(xml)
+    return "".join(res)
+
+
+def formats(ll):
+    return sorted(((Decimal(i), j) for i, j in ll), key=lambda x: x[0], reverse=True)
 
 for root, dirnames, filenames in os.walk(sys.argv[1]):
-  for filename in fnmatch.filter(filenames, '*.xml'):
-    print root, dirnames, filename
-    path = os.path.join(root, filename)
-    xml = open(path).read()
-      
-    power_formats = formats([( "1E0",  "f"),
-                             ( "1E3", "kf"),
-                             ( "1E6", "Mf"),
-                             ( "1E9", "Gf"),
-                             ("1E12", "Tf"),
-                             ("1E15", "Pt"),
-                             ("1E18", "Ef"),
-                             ("1E21", "Zf")])
-    xml = format(xml, power_formats, "power")
-    
-    bandwidth_formats = formats([( "1E0",  "Bps"),
-                                 ( "1E3", "kBps"),
-                                 ( "1E6", "MBps"),                     
-                                 ( "1E9", "GBps"),
-                                 ("1E12", "TBps")])
-    xml = format(xml, bandwidth_formats, "bandwidth")
-    xml = format(xml, bandwidth_formats, "bw")
-    xml = format(xml, bandwidth_formats, "bb_bw")
-    xml = format(xml, bandwidth_formats, "bw_in")
-    xml = format(xml, bandwidth_formats, "bw_out")
-    
-    time_formats = formats([(     "1E-12",  "ps"),
-                            (     "1E-9" ,  "ns"),
-                            (     "1E-6" ,  "us"),
-                            (     "1E-3" ,  "ms"),
-                            (     "1E0"  ,   "s"),
-                            (    "60E0"  ,   "m"),
-                            (  "3600E0"  ,   "h"),                     
-                            ( "86400E0"  ,   "d"),
-                            ("604800E0"  ,   "w")])
-    xml = format(xml, time_formats, "latency")
-    xml = format(xml, time_formats, "lat")
-    xml = format(xml, time_formats, "bb_lat")
-    
-    #print xml
-    file = open(path, "w")
-    file.write(xml)
-    file.close()
+    for filename in fnmatch.filter(filenames, '*.xml'):
+        print root, dirnames, filename
+        path = os.path.join(root, filename)
+        xml = open(path).read()
+
+        power_formats = formats([("1E0", "f"),
+                                 ("1E3", "kf"),
+                                 ("1E6", "Mf"),
+                                 ("1E9", "Gf"),
+                                 ("1E12", "Tf"),
+                                 ("1E15", "Pt"),
+                                 ("1E18", "Ef"),
+                                 ("1E21", "Zf")])
+        xml = convert(xml, power_formats, "power")
+
+        bandwidth_formats = formats([("1E0", "Bps"),
+                                     ("1E3", "kBps"),
+                                     ("1E6", "MBps"),
+                                     ("1E9", "GBps"),
+                                     ("1E12", "TBps")])
+        xml = convert(xml, bandwidth_formats, "bandwidth")
+        xml = convert(xml, bandwidth_formats, "bw")
+        xml = convert(xml, bandwidth_formats, "bb_bw")
+        xml = convert(xml, bandwidth_formats, "bw_in")
+        xml = convert(xml, bandwidth_formats, "bw_out")
+
+        time_formats = formats([("1E-12", "ps"),
+                                ("1E-9", "ns"),
+                                ("1E-6", "us"),
+                                ("1E-3", "ms"),
+                                ("1E0", "s"),
+                                ("60E0", "m"),
+                                ("3600E0", "h"),
+                                ("86400E0", "d"),
+                                ("604800E0", "w")])
+        xml = convert(xml, time_formats, "latency")
+        xml = convert(xml, time_formats, "lat")
+        xml = convert(xml, time_formats, "bb_lat")
+
+        # print xml
+        outfile = open(path, "w")
+        outfile.write(xml)
+        outfile.close()