model: add parameter for not adding default cubes

1 files changed, 8 insertions, 1 deletions

diff --git a/rumba/elements/topology.py b/rumba/elements/topology.py
index 0ff1957..f717a5a 100644
--- a/rumba/elements/topology.py
+++ b/rumba/elements/topology.py
@@ -202,17 +202,24 @@ class NormalDIF(DIF):
     """
     Normal DIF.
     """
-    def __init__(self, name, members=None, policy=None):
+    def __init__(self,
+                 name,
+                 members=None,
+                 policy=None,
+                 add_default_qos_cubes=True):
         """
         :param name: The name of the DIF.
         :param members: The list of members.
         :param policy: Policies of the normal DIF.
+        :param add_default_qos_cubes: should the prototype-dependant default
+                                      QoS cubes be added to this DIF?
         """
         DIF.__init__(self, name, members)
         if policy is None:
             policy = Policy(self)
         self.policy = policy
         self.qos_cubes = []
+        self.add_default_qos_cubes = add_default_qos_cubes
         self._last_cube_id = 0
 
     def add_policy(self, comp, pol, **params):