From 35ed8aeb915cf963f10f3b06bd81b879e80a7310 Mon Sep 17 00:00:00 2001
From: "Gustavo J. A. M. Carneiro" <gjc@inescporto.pt>
Date: Wed, 6 Jun 2012 18:42:13 +0100
Subject: [PATCH] Automatically add the visualizer module to programs running
 with waf --run --vis

---
 wscript | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/wscript b/wscript
index 09a33a215..71a8455c9 100644
--- a/wscript
+++ b/wscript
@@ -823,6 +823,9 @@ def build(bld):
         # nothing more; this greatly speeds up compilation when all you
         # want to do is run a test program.
         Options.options.targets += ',' + os.path.basename(program_name)
+        if getattr(Options.options, "visualize", False):
+            program_obj = wutils.find_program(program_name, bld.env)
+            program_obj.use.append('ns3-visualizer')
         for gen in bld.all_task_gen:
             if type(gen).__name__ in ['ns3header_taskgen', 'ns3moduleheader_taskgen']:
                 gen.post()