Changeset 730 for trunk/src/common.py

Timestamp:

03/12/10 10:14:48 (2 years ago)

Author:

pierre

Message:

Add a method to save Positions into a file. Polish and test the new Connector2 module, by adding a connectors_benchmark script in the unittest that allow to plot graphs of the connections for all the connectors and according to all the parameters. Need to play a bit with, but everything seems to be OK. Andrew, maybe please have a small look, and as soon as you consider that the progress bar, the connections are ok for you, you can commit the new connectors2 module and erase the old one.

Files:

• trunk/src/common.py (modified) (1 diff)

trunk/src/common.py

@@ -898 +898 @@
         raise NotImplementedError()
     
+    def savePositions(self, filename, gather=True, compatible_output=True):
+        """
+        Save positions to file. The output format is id x y z
+        """
+        fmt = "%s\t%s\t%s\t%s\n" % ("%d", "%g", "%g", "%g")
+        lines = []
+        for cell in self.all():  
+            x,y,z = cell.position
+            line = fmt  % (cell, x, y, z)
+            lines.append(line)
+        if rank() == 0:
+            f = open(filename, 'w')
+            f.writelines(lines)
+            f.close()
+    
 # ==============================================================================