Changeset 238 for tbroadcast

Timestamp:

Aug 26, 2006, 2:48:07 PM (18 years ago)

Author:

garonne

Message:

check cycles

File:

• tbroadcast/v2/python/tbroadcast.py (modified) (6 diffs)

tbroadcast/v2/python/tbroadcast.py

@@ -25 +25 @@
         self.pool     = ThreadPool(num_workers=num_workers)
         self.packages = {}  
+        self.check_cycles()
         self.instanciate_packages ()      
 
+
+    def check_cycles (self):
+        cmd = 'cmt show cycles'
+        cycle_found = False
+        status, output = commands.getstatusoutput (cmd)
+        if status != 0:
+            print output
+            sys.exit(-1)    
+        lines = string.split(output, '\n')
+        for line in lines:            
+            if line [0] != '#':            
+                if not cycle_found:
+                    cycle_found = True
+                    print "# Error: cycles found, not possible to execute broadcast with threads. See the followings packages:"
+                print line
+        if cycle_found:
+            sys.exit(-1)
+    
     def instanciate_packages(self):
         # We create the schedule of the work units
-        print '# first, we initialize the DAG (takes a long time... huk should be improved asap) '
+        print '# First, we initialize the DAG by parsing cmt show uses (takes a certain time... huk should be improved asap) '
         cmd = 'cmt show uses'
         status, output = commands.getstatusoutput (cmd)
@@ -45 +64 @@
                 if name not in self.packages[current_package]['dependencies']:
                    self.packages[current_package]['dependencies'].append (name)                
-                   print '\n#', indice,':: add package', name
+                   #print '\n#', indice,':: add package', name
                    indice = indice + 1
                 if not self.packages.has_key (name):
@@ -56 +75 @@
                                if string.split (ligne)[2] not in self.packages[name]['dependencies']:
                                    self.packages[name]['dependencies'].append (string.split (ligne)[2])
-                                   print "# add dependency", string.split (ligne)[2], ' to package ',name
+                                   #print "# add dependency", string.split (ligne)[2], ' to package ',name
                            else:
                                 found = False                                                                                                   
@@ -97 +116 @@
                         sys.exit(-1) 
                     self.packages[result[0]]['path'] = full_path
-        print '# really takes a long time... '
+        print '# really takes a certain time ... '
 
     def get_current_package(self):    
@@ -150 +169 @@
                 result.append(key)
         return result
+
+    def is_work_units (self):
+        result = list ()
+        for key in self.packages.keys():
+            if self.is_work_unit_waiting(key) :
+                return True
+        return False        
 
     def suppress_work_unit (self, name):
@@ -168 +194 @@
 
     def execute (self, command):
-        self.print_dependencies ()
+        #self.print_dependencies ()
         packages = self.get_next_work_units()
         if len(packages) !=0:
             print '\n# Execute parallel actions within ', packages                      
-        for package in packages:
-            self.add_work_unit (package, command)
+            for package in packages:
+                self.add_work_unit (package, command)
 
     def execute_all(self,command):
         self.execute (command)
         self.wait()
+        while self.is_work_units(): 
+            self.wait()   
         
     def wait (self):