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