REFACTORING : move mapWidget into the wifimap module
[wifihood] / wifiview.py
1
2 import gtk
3 try :
4     import hildon
5 except :
6     hildon = False
7
8 import math
9
10 import wifimap.config
11
12 import wifimap.view
13
14
15 if hildon :
16
17     class ZoomDialog ( hildon.TouchSelector ) :
18
19         def __init__ ( self , widget ) :
20             hildon.TouchSelector.__init__( self )
21
22             zooms = gtk.ListStore(str)
23
24             active = index = 0
25             for zoom in range(8,19) :
26                 iter = zooms.append()
27                 zooms.set( iter , 0 , "%2d" % zoom )
28                 if zoom == widget.conf.zoom :
29                     active = index
30                 index += 1
31
32             column = self.append_text_column( zooms , True )
33             #renderer = gtk.CellRendererText()
34             #column = self.append_column( zooms , renderer )
35             #column.set_property('text-column', 0)
36
37             # NOTE : with text=True, we must use 1 instead of 0
38             self.set_active( 0 , active )
39
40 else :
41
42     class ZoomDialog ( gtk.Dialog ) :
43
44         def __init__ ( self , widget ) :
45             gtk.Dialog.__init__( self , "Select zoom level",
46                                  None,
47                                  gtk.DIALOG_MODAL,
48                                  ( gtk.STOCK_OK, gtk.RESPONSE_ACCEPT,
49                                    gtk.STOCK_CANCEL, gtk.RESPONSE_REJECT
50                                    )
51                                  )
52
53             zooms = gtk.ListStore(int)
54             combo = gtk.ComboBox( zooms )
55
56             for zoom in range(8,19) :
57                 iter = zooms.append()
58                 zooms.set( iter , 0 , zoom )
59                 if zoom == widget.conf.zoom :
60                     combo.set_active_iter( iter )
61
62             cell = gtk.CellRendererText()
63             combo.pack_start(cell, True)
64             combo.add_attribute(cell, 'text', 0)
65
66             self.vbox.pack_start(combo , True, True, 0)
67
68             self.connect_object( "response", self.response , combo , widget )
69
70         def response ( self , combo , response  , widget ) :
71             if response == gtk.RESPONSE_ACCEPT :
72                 item = combo.get_active_iter()
73                 model = combo.get_model()
74                 widget.SetZoom( model.get(item,0)[0] )
75             self.destroy()
76
77
78 class AbstractMapWindow:
79
80     def destroy(self, widget, data=None):
81         gtk.main_quit()
82
83     def press_event ( self, widget, event, *args ) :
84       # FIXME : Set only if far enough from borders
85       border_x = 40
86       border_y = 30
87       print "press  ",event.get_coords(),event.get_root_coords()
88       if event.x > border_x and event.y > border_y and event.x < ( self.size_x - border_x ) and event.y < ( self.size_y - border_y ) :
89         self.click_x = event.x
90         self.click_y = event.y
91
92     def release_event ( self, widget, event, *args ) :
93       min_shift = 50
94       print "unpress",event.get_coords(),event.get_root_coords()
95       if self.click_x is not None and self.click_y is not None :
96         delta_x = int( event.x - self.click_x )
97         delta_y = int( event.y - self.click_y )
98         shift = math.sqrt( delta_x * delta_x + delta_y * delta_y )
99         if shift > min_shift :
100           self.map.Shift(delta_x, delta_y)
101         #  if delta_x > 100 :
102         #    self.map.Left()
103         #  elif delta_x < -100 :
104         #    self.map.Right()
105         #  elif delta_y > 100 :
106         #    self.map.Up()
107         #  elif delta_y < -100 :
108         #    self.map.Down()
109       self.click_x , self.click_y = None , None
110
111     def screen_event ( self, widget, event, *args ) :
112       print "REDIOS",event
113       print "      ",widget
114       print "      ",args
115
116
117     def on_button_press ( self, widget, event, *args ) :
118       print "HOLA",event
119
120     def on_key_press ( self, widget, event, *args ) :
121       if event.keyval == gtk.keysyms.Up :
122           self.map.Up()
123       elif event.keyval == gtk.keysyms.Down :
124           self.map.Down()
125       elif event.keyval == gtk.keysyms.Right :
126           self.map.Right()
127       elif event.keyval == gtk.keysyms.Left :
128           self.map.Left()
129       else :
130           print "UNKNOWN",event.keyval
131
132     def __init__( self , map_size=(800,480) ) :
133
134         self.connect("destroy", self.destroy)
135
136         self.set_border_width(10)
137
138         self.connect("key-press-event", self.on_key_press)
139
140         self.vbox = gtk.VBox(False, 0)
141
142         # To get explicit GDK_BUTTON_PRESS instead of paired GDK_LEAVE_NOTIFY & GDK_ENTER_NOTIFY
143 #        self.add_events(gtk.gdk.BUTTON_MOTION_MASK | gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.POINTER_MOTION_MASK)
144         self.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK )
145         #
146 #        self.connect('motion_notify_event', self.screen_event)
147         self.connect('button_press_event', self.press_event)
148         self.connect('button_release_event', self.release_event)
149         #
150         self.config = wifimap.config.Configuration()
151         self.map = wifimap.view.mapWidget( self.config , map_size )
152         self.vbox.pack_end( self.map , True , True , 5)
153
154         self.size_x , self.size_y = map_size
155         self.click_x , self.click_y = None , None
156
157     def zoomdialog ( self , widget ) :
158         dialog = ZoomDialog( widget )
159         dialog.show_all()
160
161     def main(self):
162         gtk.main()
163
164 if hildon :
165
166     class MapWindow ( AbstractMapWindow , hildon.Window ) :
167
168         def __init__(self):
169             hildon.Window.__init__( self )
170             AbstractMapWindow.__init__(self)
171             self.add( self.vbox )
172             self.create_menu( self.vbox )
173             self.show_all()
174
175         def create_menu ( self , vbox ) :
176
177             self.menubar = menubar = hildon.AppMenu()
178
179             #zoomlevel = hildon.Button(gtk.HILDON_SIZE_AUTO,
180             #                          hildon.BUTTON_ARRANGEMENT_VERTICAL,
181             #                          "Zoom level", None)
182             #zoomlevel.connect_object( "clicked", self.zoomstack, self.map )
183             selector = ZoomDialog( self.map )
184             zoomlevel = hildon.PickerButton(gtk.HILDON_SIZE_AUTO,
185                                           hildon.BUTTON_ARRANGEMENT_VERTICAL)
186             zoomlevel.set_title( "Zoom" )
187             zoomlevel.set_selector( selector )
188             zoomlevel.connect_object( "value-changed", self.map.ZoomChange , selector )
189             menubar.append( zoomlevel )
190
191             menubar.show_all()
192             self.set_app_menu( menubar )
193
194 else :
195
196     class MapWindow ( AbstractMapWindow , gtk.Window ) :
197
198         def __init__(self):
199             gtk.Window.__init__( self , gtk.WINDOW_TOPLEVEL )
200             AbstractMapWindow.__init__(self)
201             self.add( self.vbox )
202             self.create_menu( self.vbox )
203             self.show_all()
204
205             self.resize( self.size_x , self.size_y)
206
207         def create_menu ( self , vbox ) :
208
209             menubar = gtk.MenuBar()
210
211             zoomlevel = gtk.MenuItem( label="Zoom level" )
212             zoomlevel.connect_object( "activate", self.zoomdialog, self.map )
213             menubar.append( zoomlevel )
214
215             vbox.pack_start(menubar,True,True,5)
216