Changeset 262

Timestamp:

01/22/11 14:47:17 (10 years ago)

Author:

sc

Message:

synapse/synapse-render_video.py:

• synapse data files loaded into memory

File:

• trunk/synapse/synapse-render_video.py (modified) (6 diffs)

trunk/synapse/synapse-render_video.py

@@ -19 +19 @@
 
 #import Puzzlebox.Synapse.Interface as tgInterface
+
+import sys
 
 from mpl_toolkits.mplot3d import Axes3D
@@ -26 +28 @@
 import numpy
 
+try:
+        import cPickle as pickle
+except:
+        import pickle
+
 #####################################################################
 # Globals
@@ -32 +39 @@
 DEBUG = 1
 
+DEFAULT_INPUT_FILE = 'untitled.synapse'
 DEFAULT_HISTORY_IMAGE_FILE = 'history.png'
 DEFAULT_CURRENT_VALUES_IMAGE_FILE = 'current.png'
 
+DEFAULT_OUTPUT_DIRECTORY = 'output'
+
+DEFAULT_ATTENTION_COLOR = 'red'
+DEFAULT_MEDITATION_COLOR = 'blue'
+
+DEFAULT_HISTORY_WIDTH = 240
+DEFAULT_HISTORY_HEIGHT = 180
+
+DEFAULT_CURRENT_VALUES_WIDTH = 96
+DEFAULT_CURRENT_VALUES_HEIGHT = 720
+
 #####################################################################
 # Functions
 #####################################################################
 
-def render_history(attention=None, \
-                                                        meditation=None ,\
-                                                        width = 240, \
-                                                        height = 180, \
-                                                        attention_color = 'red', \
-                                                        mediation_color = 'blue', \
-                                                        output_file=DEFAULT_HISTORY_IMAGE_FILE):
+def renderHistoryImage(attention=None, \
+                            meditation=None ,\
+                            width = DEFAULT_HISTORY_WIDTH, \
+                            height = DEFAULT_HISTORY_HEIGHT, \
+                            attention_color = DEFAULT_ATTENTION_COLOR, \
+                            mediation_color = DEFAULT_MEDITATION_COLOR, \
+                            output_file=DEFAULT_HISTORY_IMAGE_FILE):
         
         figsize = (width / 100.0, height / 100.0) # @ 100 dpi
@@ -118 +137 @@
 
 
-        ##################################################################
-
-def render_current_values(x=None, y=None ,\
-                                                                        width = 96, \
-                                                                        height = 720, \
-                                                                        x_color = 'red', \
-                                                                        y_color = 'blue', \
-                                                                        output_file=DEFAULT_CURRENT_VALUES_IMAGE_FILE):
+#####################################################################
+
+def renderCurrentValuesImage(x=None, y=None ,\
+                                  width = DEFAULT_CURRENT_VALUES_WIDTH, \
+                                  height = DEFAULT_CURRENT_VALUES_HEIGHT, \
+                                  x_color = DEFAULT_ATTENTION_COLOR, \
+                                  y_color = DEFAULT_MEDITATION_COLOR, \
+                                  output_file=DEFAULT_CURRENT_VALUES_IMAGE_FILE):
         
         figsize = (width / 100.0, height / 100.0) # @ 100 dpi
@@ -178 +197 @@
 
 #####################################################################
+
+def openData(input_file=DEFAULT_INPUT_FILE):
+        
+        file = open(input_file, 'r')
+        data = pickle.load(file)
+        file.close()
+        
+        return(data)
+
+
+#####################################################################
 # Main
 #####################################################################
@@ -183 +213 @@
 if __name__ == '__main__':
 
-        render_history()
-        render_current_values()
-
+        input_file = DEFAULT_INPUT_FILE
+        output_directory = DEFAULT_OUTPUT_DIRECTORY
+
+        for each in sys.argv:
+                if each.startswith("--file="):
+                        input_file = each[ len("--file="): ]
+                if each.startswith("--directory="):
+                        server_port = each[ len("--directory="): ]
+        
+        data = openData(input_file)
+        renderHistoryImage()
+        renderCurrentValuesImage()
+