File: DataVizManager.py

package info (click to toggle)
mayavi 1.5-5
  • links: PTS
  • area: main
  • in suites: lenny
  • size: 2,172 kB
  • ctags: 2,002
  • sloc: python: 17,938; makefile: 53; sh: 27
file content (280 lines) | stat: -rw-r--r-- 10,241 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
"""

This file defines the DataVizManager that controls a list of
ModuleManagers.

This code is distributed under the conditions of the BSD license.  See
LICENSE.txt for details.

Copyright (c) 2001-2002, Prabhu Ramachandran.
"""

__author__ = "Prabhu Ramachandran <prabhu_r@users.sf.net>"
__version__ = "$Revision: 1.6 $"
__date__ = "$Date: 2004/07/06 04:29:30 $"

import ModuleManager, Common, Tkinter

debug = Common.debug

class DataVizManager:

    """ This class controls a list of ModuleManagers."""

    def __init__ (self, data_src, renwin=None):        
        """ The argument 'data_src' refers to a data source instance
        and 'renwin' to a RenderWindow instance."""
        debug ("In DataVizManager::__init__ ()")
        self.data_src = data_src
        self.renwin=renwin
        self.mod_mgr = {}
        self.mm_name = []
        self.cur_mm_name = ""
        self.n_mm = 0        
        # The GUI list box
        self.mm_lst = None
        self.frame = None

    def __del__ (self):
        debug ("In DataVizManager::__del__ ()")

    def clear (self): 
        """This is needed to prevent self-referential loops and is
        usually called before a DataVizManager object is destroyed."""
        debug ("In DataVizManager::clear ()")
        if self.frame:
            self.close_gui ()
        for mm_name in self.mm_name[:]:
            self.del_module_mgr (mm_name)
        del self.data_src

    def config_changed (self): 
        debug ("In DataVizManager::config_changed ()")
        for mm in self.mod_mgr.values ():
            mm.config_changed ()

    def get_data_source (self): 
        debug ("In DataVizManager::get_data_source ()")
        return self.data_src

    def get_render_window (self): 
        debug ("In DataVizManager::get_render_window ()")
        return self.renwin

    def get_current_module_mgr (self): 
        debug ("In DataVizManager::get_current_module_mgr ()")
        if self.cur_mm_name:
            return self.mod_mgr[self.cur_mm_name]
        else:
            return None

    def add_module_mgr (self, mm): 
        debug ("In DataVizManager::add_module_mgr ()")
        self.n_mm = self.n_mm + 1
        k = '%d. %s'%(self.n_mm, mm.__class__.__name__)
        self.mod_mgr[k] = mm
        self.cur_mm_name = k
        self.mm_name.append (k)

    def del_module_mgr (self, key): 
        debug ("In DataVizManager::del_module_mgr ()")
        try:
            self.mod_mgr[key].clear ()
            del self.mod_mgr[key]
        except KeyError:
            msg = "Error: cannot delete ModuleManager with key: %s"%key
            raise KeyError, msg

        indx = self.mm_name.index (key)
        if len (self.mm_name) > 1:
            if indx == 0:
                self.cur_mm_name = self.mm_name[indx + 1]
            else:
                self.cur_mm_name = self.mm_name[indx - 1]
        else:
            self.cur_mm_name = ""

        del self.mm_name[indx]

    def _write_msg (self, file, msg): 
        file.write (msg)
        debug (msg)

    def save_current_module_mgr (self, file): 
        debug ("In DataVizManager::save_current_module_mgr ()")
        self._write_msg (file, "### ModuleManagers ###\n")
        file.write ("1\n")
        self._write_msg (file, "### %s ###\n"%self.cur_mm_name)
        mm = self.mod_mgr[self.cur_mm_name]
        mm.save_config (file)
        self._write_msg (file, "### End of %s ###\n"%self.cur_mm_name)
        self._write_msg (file, "### End of ModuleManagers ###\n")

    def save_config (self, file): 
        debug ("In DataVizManager::save_config ()")
        self._write_msg (file, "### ModuleManagers ###\n")
        file.write ("%d\n"%len (self.mm_name))
        for mm_name in self.mm_name:
            self._write_msg (file, "### %s ###\n"%mm_name)
            mm = self.mod_mgr[mm_name]
            mm.save_config (file)
            self._write_msg (file, "### End of %s ###\n"%mm_name)
        self._write_msg (file, "### End of ModuleManagers ###\n")

    def _read_msg (self, file): 
        debug (file.readline ())

    def load_current_module_mgr (self, file): 
        debug ("In DataVizManager::load_current_module_mgr ()")
        # Get the config from the file
        #self._read_msg (file)
        tmp = file.readline ()
        debug (tmp)
        if tmp != "### ModuleManagers ###\n":
            msg = "Error: Expected data for ModuleManagers not found.\n"\
                  "You have  possibly chosen the wrong file.\n"
            raise ParseException, msg
        n = int (file.readline ())
        # Read the first module manager to the current one.
        self._read_msg (file)
        mm = self.mod_mgr[self.cur_mm_name]
        mm.load_config (file)
        self._read_msg (file)
        # Read the rest separately.
        for i in range (1, n):
            self._read_msg (file)
            self.add_module_mgr_gui ()
            self.mod_mgr[self.cur_mm_name].load_config (file)
            self._read_msg (file)
        self._read_msg (file)

    def load_config (self, file): 
        debug ("In DataVizManager::load_config ()")
        # Get the config from the file
        tmp = file.readline ()
        debug (tmp)
        if tmp != "### ModuleManagers ###\n":
            msg = "Error: Expected data for ModuleManagers not found. "\
                  "You have  possibly chosen the wrong file."
            raise ParseException, msg
        n = int (file.readline ())
        for i in range (n):
            self._read_msg (file)
            self.add_module_mgr_gui ()
            self.mod_mgr[self.cur_mm_name].load_config (file)
            self._read_msg (file)
        self._read_msg (file)

    def create_module_gui (self, master): 
        debug ("In DataVizManager::create_module_gui ()")
        frame = Tkinter.Frame (master)
        frame.pack (side='top', fill='y')
        #label = Tkinter.Label (frame, text="Module Managers")
        #label.grid (row=0, column=0, sticky='ew', columnspan=2)
        scr = Tkinter.Scrollbar (frame, orient='vertical')
        self.mm_lst = Tkinter.Listbox (frame, yscrollcommand=scr.set, 
                                        selectmode='single', height=3,
                                        exportselection=0)
        scr.config (command=self.mm_lst.yview)
        self.mm_lst.grid (row=0, column=0, sticky='ewns')
        scr.grid (row=0, column=1, sticky='ns')
        self.mm_lst.bind ("<Double-Button-1>", self.show_mm)
        
        but_f = Tkinter.Frame (master)
        but_f.pack (side='top', fill='y')
        but1 = Tkinter.Button (but_f, text="Show",
                               command=self.show_mm)
        but1.grid (row=0, column=0, sticky="ew")
        but2 = Tkinter.Button (but_f, text="New",
                               command=self.add_module_mgr_gui)
        but2.grid (row=0, column=1, sticky="ew")        
        but3 = Tkinter.Button (but_f, text="Delete",
                               command=self.del_module_mgr_gui)
        but3.grid (row=0, column=2, sticky="ew")

        lab = "ModuleManager: %s"%self.cur_mm_name
        self.data_label = Tkinter.Label (but_f, text=lab)
        self.data_label.grid (row=1, column=0, columnspan=3, pady=0)

        for mm in self.mm_name:
            self.mm_lst.insert ('end', mm)

        if self.cur_mm_name:
            self.mm_lst.select_clear (0, 'end')
            indx = self.mm_name.index (self.cur_mm_name)
            self.mm_lst.activate (indx)
            self.mm_lst.select_set (indx)

        if self.cur_mm_name:
            self.mm_lst.activate (self.mm_name.index (self.cur_mm_name))
        self.cur_mm_name = ""
        self.show_mm ()
 
    def create_gui (self, master): 
        debug ("In DataVizManager::create_gui ()")
        if not master:
            debug ("In DataVizManager::create_gui: Error: No master widget.")
            return None        
        self.frame = Tkinter.Frame (master, relief='ridge', bd=2)
        self.frame.pack (side='top', fill='y', expand=1)
        self.create_module_gui (self.frame)

    def close_gui (self, event=None): 
        debug ("In DataVizManager::close_gui ()")
        self.frame.destroy ()

    def show_mm (self, event=None): 
        debug ("In DataVizManager::show_mm ()")
        mm_name = self.mm_lst.get ('active')
        if not mm_name:
            return
        if (self.cur_mm_name != mm_name) or (not self.cur_mm_name) :
            if self.cur_mm_name:
                old_mm = self.mod_mgr[self.cur_mm_name]
                old_mm.close_gui ()
            self.cur_mm_name = mm_name
            self.update_label ()
            mm = self.mod_mgr[self.cur_mm_name]
            mm.create_gui (self.frame)

    def update_label (self): 
        debug ("In DataVizManager::update_label ()")
        lab = "ModuleManager: %s"%self.cur_mm_name
        self.data_label.config (text=lab)

    def del_module_mgr_gui (self, event=None): 
        debug ("In DataVizManager::del_module_mgr_gui ()")
        key = self.mm_lst.get ('active')
        self.mm_lst.delete ('active')
        if key:
            if self.cur_mm_name:
                old_mm = self.mod_mgr[self.cur_mm_name]
                old_mm.close_gui ()
            self.del_module_mgr (key)
            if self.cur_mm_name:
                indx = self.mm_name.index (self.cur_mm_name)
                self.mm_lst.select_clear (0, 'end')
                self.mm_lst.activate (indx)
                self.mm_lst.select_set (indx)
                self.cur_mm_name = ""
                self.show_mm ()
            self.update_label ()

    def add_module_mgr_gui (self, event=None): 
        debug ("In DataVizManager::add_module_mgr_gui ()")
        if self.cur_mm_name:
            old_mm = self.mod_mgr[self.cur_mm_name]
            old_mm.close_gui ()
        mm = ModuleManager.ModuleManager (self)
        self.add_module_mgr (mm)
        self.mm_lst.insert ('end', self.cur_mm_name)
        self.mm_lst.select_clear (0, 'end')
        self.mm_lst.activate ('end')
        self.mm_lst.select_set ('end')
        self.cur_mm_name = ""
        self.show_mm ()

    def config_lut (self): 
        debug ("In DataVizManager::config_lut ()")
        self.lut_h.configure (self.frame)