X-Git-Url: http://g0dil.de/git?a=blobdiff_plain;f=tools%2Fdrawmodules.py;h=49da65fbf8fb5b504eb12ff10201c85676eaa715;hb=ff4249b4a9c23236b2e56dcd593c91a5c58f7b75;hp=30709c7ef22880798400c4cbe4faf24340fdecdb;hpb=7a961d216978e108ff4d0831f53f7b0f9f8823df;p=senf.git

diff --git a/tools/drawmodules.py b/tools/drawmodules.py
index 30709c7..49da65f 100755
--- a/tools/drawmodules.py
+++ b/tools/drawmodules.py
@@ -3,7 +3,6 @@
 import sys
 
 COLOR_SCHEME = 'pastel19'  # see http://www.graphviz.org/doc/info/colors.html
-SOURCE_SINK = ["ActiveSocketSource", "PassiveSocketSource", "ActiveSocketSink", "PassiveSocketSink"]
 NETEMU = ["InterfaceAnnotater" , "PacketMonitor" , "PacketTypeChecker"]
 mode = "MODULE"
 
@@ -49,13 +48,12 @@ for line in sys.stdin:
 
 for moduleid, (module, cs) in modules.iteritems():
     module = module.split('<',1)[0]
-    if "senf" not in module.split('::',1)[0]:
-        if module.rsplit('::',1)[-1] in NETEMU:
-           color = 5
-        else:
-           color = 6
-    elif module.rsplit('::',1)[-1] in SOURCE_SINK:
+    if module.rsplit('::',1)[-1] in NETEMU:
+        color = 5
+    elif module.endswith("Source") or module.endswith("Sink"):
         color = 1
+    elif "senf" not in module.split('::',1)[0]:
+        color = 6
     else:
         color = 3
     module = module.rsplit('::',1)[-1]