Changeset 262


Ignore:
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:
1 edited

Legend:

Unmodified
Added
Removed
  • trunk/synapse/synapse-render_video.py

    r261 r262  
    1919 
    2020#import Puzzlebox.Synapse.Interface as tgInterface 
     21 
     22import sys 
    2123 
    2224from mpl_toolkits.mplot3d import Axes3D 
     
    2628import numpy 
    2729 
     30try: 
     31        import cPickle as pickle 
     32except: 
     33        import pickle 
     34 
    2835##################################################################### 
    2936# Globals 
     
    3239DEBUG = 1 
    3340 
     41DEFAULT_INPUT_FILE = 'untitled.synapse' 
    3442DEFAULT_HISTORY_IMAGE_FILE = 'history.png' 
    3543DEFAULT_CURRENT_VALUES_IMAGE_FILE = 'current.png' 
    3644 
     45DEFAULT_OUTPUT_DIRECTORY = 'output' 
     46 
     47DEFAULT_ATTENTION_COLOR = 'red' 
     48DEFAULT_MEDITATION_COLOR = 'blue' 
     49 
     50DEFAULT_HISTORY_WIDTH = 240 
     51DEFAULT_HISTORY_HEIGHT = 180 
     52 
     53DEFAULT_CURRENT_VALUES_WIDTH = 96 
     54DEFAULT_CURRENT_VALUES_HEIGHT = 720 
     55 
    3756##################################################################### 
    3857# Functions 
    3958##################################################################### 
    4059 
    41 def render_history(attention=None, \ 
    42                                                         meditation=None ,\ 
    43                                                         width = 240, \ 
    44                                                         height = 180, \ 
    45                                                         attention_color = 'red', \ 
    46                                                         mediation_color = 'blue', \ 
    47                                                         output_file=DEFAULT_HISTORY_IMAGE_FILE): 
     60def renderHistoryImage(attention=None, \ 
     61                            meditation=None ,\ 
     62                            width = DEFAULT_HISTORY_WIDTH, \ 
     63                            height = DEFAULT_HISTORY_HEIGHT, \ 
     64                            attention_color = DEFAULT_ATTENTION_COLOR, \ 
     65                            mediation_color = DEFAULT_MEDITATION_COLOR, \ 
     66                            output_file=DEFAULT_HISTORY_IMAGE_FILE): 
    4867         
    4968        figsize = (width / 100.0, height / 100.0) # @ 100 dpi 
     
    118137 
    119138 
    120         ################################################################## 
    121  
    122 def render_current_values(x=None, y=None ,\ 
    123                                                                         width = 96, \ 
    124                                                                         height = 720, \ 
    125                                                                         x_color = 'red', \ 
    126                                                                         y_color = 'blue', \ 
    127                                                                         output_file=DEFAULT_CURRENT_VALUES_IMAGE_FILE): 
     139##################################################################### 
     140 
     141def renderCurrentValuesImage(x=None, y=None ,\ 
     142                                  width = DEFAULT_CURRENT_VALUES_WIDTH, \ 
     143                                  height = DEFAULT_CURRENT_VALUES_HEIGHT, \ 
     144                                  x_color = DEFAULT_ATTENTION_COLOR, \ 
     145                                  y_color = DEFAULT_MEDITATION_COLOR, \ 
     146                                  output_file=DEFAULT_CURRENT_VALUES_IMAGE_FILE): 
    128147         
    129148        figsize = (width / 100.0, height / 100.0) # @ 100 dpi 
     
    178197 
    179198##################################################################### 
     199 
     200def openData(input_file=DEFAULT_INPUT_FILE): 
     201         
     202        file = open(input_file, 'r') 
     203        data = pickle.load(file) 
     204        file.close() 
     205         
     206        return(data) 
     207 
     208 
     209##################################################################### 
    180210# Main 
    181211##################################################################### 
     
    183213if __name__ == '__main__': 
    184214 
    185         render_history() 
    186         render_current_values() 
    187  
     215        input_file = DEFAULT_INPUT_FILE 
     216        output_directory = DEFAULT_OUTPUT_DIRECTORY 
     217 
     218        for each in sys.argv: 
     219                if each.startswith("--file="): 
     220                        input_file = each[ len("--file="): ] 
     221                if each.startswith("--directory="): 
     222                        server_port = each[ len("--directory="): ] 
     223         
     224        data = openData(input_file) 
     225        renderHistoryImage() 
     226        renderCurrentValuesImage() 
     227 
Note: See TracChangeset for help on using the changeset viewer.