git.maemo.org Git - wifihood/blob - wifiscanner/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       border_x , border_y = 40 , 30
  85       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 ) :
  86         self.click_x = event.x
  87         self.click_y = event.y
  88
  89     def release_event ( self, widget, event, *args ) :
  90       min_shift = 50
  91       if self.click_x is not None and self.click_y is not None :
  92         delta_x = int( event.x - self.click_x )
  93         delta_y = int( event.y - self.click_y )
  94         shift = math.sqrt( delta_x * delta_x + delta_y * delta_y )
  95         if shift > min_shift :
  96           self.map.Shift(delta_x, delta_y)
  97       self.click_x , self.click_y = None , None
  98
  99     def on_key_press ( self, widget, event, *args ) :
 100       if event.keyval == gtk.keysyms.Up :
 101           self.map.Up()
 102       elif event.keyval == gtk.keysyms.Down :
 103           self.map.Down()
 104       elif event.keyval == gtk.keysyms.Right :
 105           self.map.Right()
 106       elif event.keyval == gtk.keysyms.Left :
 107           self.map.Left()
 108       else :
 109           print "UNKNOWN",event.keyval
 110
 111     def __init__( self , map_size=(800,480) ) :
 112
 113         self.connect("destroy", self.destroy)
 114
 115         self.set_border_width(10)
 116
 117         self.connect("key-press-event", self.on_key_press)
 118
 119         self.vbox = gtk.VBox(False, 0)
 120
 121         self.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK )
 122         self.connect('button_press_event', self.press_event)
 123         self.connect('button_release_event', self.release_event)
 124
 125         self.config = wifimap.config.Configuration()
 126         self.map = wifimap.view.mapWidget( self.config , map_size )
 127         self.vbox.pack_end( self.map , True , True , 5)
 128
 129         self.size_x , self.size_y = map_size
 130         self.click_x , self.click_y = None , None
 131
 132     def zoomdialog ( self , widget ) :
 133         dialog = ZoomDialog( widget )
 134         dialog.show_all()
 135
 136     def main(self):
 137         gtk.main()
 138
 139 if hildon :
 140
 141     class MapWindow ( AbstractMapWindow , hildon.Window ) :
 142
 143         def __init__(self):
 144             hildon.Window.__init__( self )
 145             AbstractMapWindow.__init__(self)
 146             self.add( self.vbox )
 147             self.create_menu( self.vbox )
 148             self.show_all()
 149
 150         def create_menu ( self , vbox ) :
 151
 152             self.menubar = menubar = hildon.AppMenu()
 153
 154             #zoomlevel = hildon.Button(gtk.HILDON_SIZE_AUTO,
 155             #                          hildon.BUTTON_ARRANGEMENT_VERTICAL,
 156             #                          "Zoom level", None)
 157             #zoomlevel.connect_object( "clicked", self.zoomstack, self.map )
 158             selector = ZoomDialog( self.map )
 159             zoomlevel = hildon.PickerButton(gtk.HILDON_SIZE_AUTO,
 160                                           hildon.BUTTON_ARRANGEMENT_VERTICAL)
 161             zoomlevel.set_title( "Zoom" )
 162             zoomlevel.set_selector( selector )
 163             zoomlevel.connect_object( "value-changed", self.map.ZoomChange , selector )
 164             menubar.append( zoomlevel )
 165
 166             menubar.show_all()
 167             self.set_app_menu( menubar )
 168
 169 else :
 170
 171     class MapWindow ( AbstractMapWindow , gtk.Window ) :
 172
 173         def __init__(self):
 174             gtk.Window.__init__( self , gtk.WINDOW_TOPLEVEL )
 175             AbstractMapWindow.__init__(self)
 176             self.add( self.vbox )
 177             self.create_menu( self.vbox )
 178             self.show_all()
 179
 180             self.resize( self.size_x , self.size_y)
 181
 182         def create_menu ( self , vbox ) :
 183
 184             menubar = gtk.MenuBar()
 185
 186             zoomlevel = gtk.MenuItem( label="Zoom level" )
 187             zoomlevel.connect_object( "activate", self.zoomdialog, self.map )
 188             menubar.append( zoomlevel )
 189
 190             vbox.pack_start(menubar,True,True,5)
 191