git.maemo.org Git - wifihood/blob - view.py

   1 #!/usr/bin/python
   2
   3 import gtk
   4 import gobject
   5
   6 import urllib2
   7 import math
   8
   9 import os
  10
  11 class background_map ( gtk.gdk.Pixmap ) :
  12
  13     def __init__ ( self , map_size , tileloader ) :
  14         bordersize = 1
  15
  16         self.tileloader = tileloader
  17
  18         # Values for minimun fit without border
  19         center = map( lambda x :     int( math.ceil( x / float(tileloader.tilesize) ) / 2 )     , map_size )
  20         size = map( lambda x : 2 * x + 1 , center )
  21
  22         self.center = map( lambda x : x + bordersize , center )
  23         self.size = map( lambda x : x + 2 * bordersize , size )
  24         pixsize = map( lambda x : x * tileloader.tilesize , self.size )
  25
  26         # FIXME : seems that reproducing the previous behaviour requires an extra subtraction of 128 to vpor[1]
  27         #         when moving to non-integer viewports, the shift turns and adition, and changes to vpor[0]
  28         self.__vport_base = bordersize * tileloader.tilesize , bordersize * tileloader.tilesize
  29
  30         gtk.gdk.Pixmap.__init__( self , None , pixsize[0] , pixsize[1] , 24 )
  31
  32         self.fill = map( lambda x : False , range( self.size[0] * self.size[1] ) )
  33
  34         self.loadtiles()
  35
  36     def index ( self , x , y ) :
  37         return x + y * self.size[0]
  38
  39     def get_viewport ( self ) :
  40         refpix = self.tileloader.get_refpix()
  41         return self.__vport_base[0] + refpix[0] , self.__vport_base[1] + refpix[1]
  42
  43     def loadtiles ( self ) :
  44
  45         for x in range(self.size[0]) :
  46             for y in range(self.size[1]) :
  47
  48               if not self.fill[ self.index(x,y) ] :
  49                 pixbuf = self.tileloader.get_tile( (x-self.center[0],y-self.center[1]) )
  50                 if pixbuf :
  51                     self.fill[ self.index(x,y) ] = True
  52                 else :
  53                     pixbuf = self.tileloader.emptytile()
  54
  55                 dest_x = self.tileloader.tilesize * x
  56                 dest_y = self.tileloader.tilesize * y
  57                 self.draw_pixbuf( None , pixbuf , 0 , 0 , dest_x , dest_y )
  58
  59
  60     def do_change_viewport ( self , dx , dy ) :
  61         dx , dy = self.tileloader.do_change_refpix( dx , dy )
  62         if dx or dy :
  63             self.do_change_reftile( dx , dy )
  64
  65     def do_change_reftile( self , dx , dy ) :
  66         self.tileloader.do_change_reftile( dx , dy )
  67
  68         pixsize = self.get_size()
  69         pixbuf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, False, 8, pixsize[0] , pixsize[1] )
  70         pixbuf.get_from_drawable( self , self.get_colormap() , 0 , 0 , 0 , 0 , pixsize[0] , pixsize[1] )
  71
  72         # width , source , destination
  73         x_vals = [ pixsize[0] , 0 , 0 ]
  74         y_vals = [ pixsize[1] , 0 , 0 ]
  75
  76         if dx :
  77             x_vals[0] -= abs(dx) * self.tileloader.tilesize
  78             x_vals[cmp(dx,0)] = abs(dx) * self.tileloader.tilesize
  79             if dx > 0 :
  80               for x in range(1,1+dx) :
  81                 for y in range(self.size[1]) : self.fill[ self.index(self.size[0]-x,y) ] = False
  82             if dx < 0 :
  83               for x in range(1,1-dx) :
  84                 for y in range(self.size[1]) : self.fill[ self.index(x,y) ] = False
  85         if dy :
  86             y_vals[0] -= abs(dy) * self.tileloader.tilesize
  87             y_vals[cmp(dy,0)] = abs(dy) * self.tileloader.tilesize
  88             if dy > 0 :
  89               for y in range(1,1+dy) :
  90                 for x in range(self.size[0]) : self.fill[ self.index(x,self.size[1]-y) ] = False
  91             if dy < 0 :
  92               for y in range(1,1-dy) :
  93                 for x in range(self.size[0]) : self.fill[ self.index(x,y) ] = False
  94
  95         self.draw_pixbuf( None , pixbuf , x_vals[1] , y_vals[1] , x_vals[-1] , y_vals[-1] , x_vals[0] , y_vals[0] )
  96         self.loadtiles()
  97
  98
  99 class tile_loader :
 100
 101     def __init__ ( self , conf ) :
 102         self.tilesize = 256
 103         self.rootdir = "%s/%s/%s" % ( conf.mapsdir , conf.mapclass , conf.zoom )
 104         self.__reftile , self.__refpix = self.get_reference( conf )
 105
 106     def do_change_refpix ( self , dx , dy ) :
 107         self.__refpix[0] += dx
 108         self.__refpix[1] += dy
 109         tileshift = self.__refpix[0] / self.tilesize , self.__refpix[1] / self.tilesize
 110         self.__refpix[0] %= self.tilesize
 111         self.__refpix[1] %= self.tilesize
 112         return tileshift
 113
 114     def do_change_reftile( self , dx , dy ) :
 115         self.__reftile[0] += dx
 116         self.__reftile[1] += dy
 117
 118     def get_refpix ( self ) :
 119         return self.__refpix
 120
 121     def get_reference ( self , conf ) :
 122         tilex = self.lon2tilex( conf.lon , conf.zoom )
 123         tiley = self.lat2tiley( conf.lat , conf.zoom )
 124         tile = tilex[1] , tiley[1]
 125         pix = tilex[0] , tiley[0]
 126         return map( int , tile ) , map( lambda x : int( self.tilesize * x ) , pix )
 127
 128     def lon2tilex ( self , lon , zoom ) :
 129         return math.modf( ( lon + 180 ) / 360 * 2 ** zoom )
 130
 131     def lat2tiley ( self , lat , zoom ) :
 132         lat = lat * math.pi / 180
 133         return math.modf( ( 1 - math.log( math.tan( lat ) + 1 / math.cos( lat ) ) / math.pi ) / 2 * 2 ** zoom )
 134
 135     def get_tile ( self , tile ) :
 136         file = self.tilepath( self.__reftile[0] + tile[0] , self.__reftile[1] + tile[1] )
 137         try :
 138             os.stat(file)
 139             return gtk.gdk.pixbuf_new_from_file( file )
 140         except :
 141             try :
 142                 # useful members : response.code, response.headers
 143                 response = urllib2.urlopen( "http://tile.openstreetmap.org/%s/%s/%s.png" % ( zoom , x , y ) )
 144                 if response.geturl() != "http://tile.openstreetmap.org/11/0/0.png" :
 145                     fd = open( file , 'w' )
 146                     fd.write( response.read() )
 147                     fd.close()
 148                     # FIXME : can this actually produce a gobject.GError exception ?
 149                     return gtk.gdk.pixbuf_new_from_file( file )
 150             except :
 151                 pass
 152
 153         return None
 154
 155     def tilepath( self , tilex , tiley ) :
 156       return "%s/%s/%s.png" % ( self.rootdir , tilex , tiley )
 157
 158     def emptytile( self ) :
 159         pixbuf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, False, 8, self.tilesize, self.tilesize )
 160         pixbuf.fill( 0x00000000 )
 161         return pixbuf
 162
 163
 164 class AbstractmapWidget :
 165
 166   def __init__ ( self , config , map_size ) :
 167
 168     self.conf = config
 169
 170     # Maximum width should be 800, but actually gets reduced
 171     self.win_x , self.win_y = map_size
 172     self.tile_size = 256
 173
 174     self.reftile_x , self.refpix_x = self.lon2tilex( self.conf.lon , self.conf.zoom )
 175     self.reftile_y , self.refpix_y = self.lat2tiley( self.conf.lat , self.conf.zoom )
 176
 177   def recenter ( self , latlon ) :
 178
 179         center = self.gps2pix( latlon , self.center() )
 180         pixel = self.gps2pix( (self.conf.lat,self.conf.lon) , self.center() )
 181
 182         distance = math.sqrt( (pixel[0]-center[0])**2 + (pixel[1]-center[1])**2 )
 183
 184         # FIXME : instead of hardcoded, should depend on the actual display size
 185         if distance > 150 :
 186             self.conf.set_latlon( latlon )
 187
 188             self.reftile_x , self.refpix_x = self.lon2tilex( self.conf.lon , self.conf.zoom )
 189             self.reftile_y , self.refpix_y = self.lat2tiley( self.conf.lat , self.conf.zoom )
 190
 191             self.composeMap()
 192
 193   def tilex2lon ( self , ( tilex , pixx ) , zoom ) :
 194         tilex = float(tilex)
 195         pixx = float(pixx)
 196         return ( tilex + pixx/self.tile_size ) / 2.0 ** zoom * 360.0 - 180.0
 197
 198   def tiley2lat ( self , ( tiley , pixy ) , zoom ) :
 199         tiley = float(tiley)
 200         pixy = float(pixy)
 201         tiley = math.pi * ( 1 - 2 * ( tiley + pixy/self.tile_size ) / 2.0 ** zoom )
 202         return math.degrees( math.atan( math.sinh( tiley ) ) )
 203
 204   def SetZoom( self , zoom ) :
 205         self.hide()
 206         lat = self.tiley2lat( ( self.reftile_y , self.refpix_y ) , self.conf.zoom )
 207         lon = self.tilex2lon( ( self.reftile_x , self.refpix_x ) , self.conf.zoom )
 208         self.reftile_x , self.refpix_x = self.lon2tilex( lon , zoom )
 209         self.reftile_y , self.refpix_y = self.lat2tiley( lat , zoom )
 210         self.conf.set_zoom( zoom )
 211         self.composeMap()
 212         self.show()
 213
 214   def lon2tilex ( self , lon , zoom ) :
 215     number = math.modf( ( lon + 180 ) / 360 * 2 ** zoom )
 216     return int( number[1] ) , int( self.tile_size * number[0] )
 217
 218   def lat2tiley ( self , lat , zoom ) :
 219     lat = lat * math.pi / 180
 220     number = math.modf( ( 1 - math.log( math.tan( lat ) + 1 / math.cos( lat ) ) / math.pi ) / 2 * 2 ** zoom )
 221     return int( number[1] ) , int( self.tile_size * number[0] )
 222
 223   def gps2pix ( self , ( lat , lon ) , ( center_x , center_y ) ) :
 224
 225     x_pos = self.lon2tilex( lon , self.conf.zoom )
 226     y_pos = self.lat2tiley( lat , self.conf.zoom )
 227
 228     dest_x = self.tile_size * ( x_pos[0] - self.reftile_x ) + center_x + x_pos[1]
 229     dest_y = self.tile_size * ( y_pos[0] - self.reftile_y ) + center_y + y_pos[1]
 230
 231     return dest_x , dest_y
 232
 233   def tilename ( self , x , y , zoom ) :
 234     file = self.tile2file( self.reftile_x + x , self.reftile_y + y , zoom )
 235     try :
 236       os.stat(file)
 237     except :
 238     #  if mapDownload :
 239       if False :
 240         try :
 241           # useful members : response.code, response.headers
 242           response = urllib2.urlopen( "http://tile.openstreetmap.org/%s/%s/%s.png" % ( zoom , x , y ) )
 243           if response.geturl() == "http://tile.openstreetmap.org/11/0/0.png" :
 244               return None
 245           fd = open( file , 'w' )
 246           fd.write( response.read() )
 247           fd.close()
 248         except :
 249           return None
 250       else :
 251         return None
 252     return file
 253
 254   def tile2file( self , tilex , tiley , zoom ) :
 255     rootdir = "%s/%s/%s" % ( self.conf.mapsdir , self.conf.mapclass , zoom )
 256     if not os.path.isdir( rootdir ) :
 257       os.mkdir(rootdir)
 258     rootsubdir = "%s/%s" % ( rootdir , tilex )
 259     if not os.path.isdir( rootsubdir ) :
 260       os.mkdir(rootsubdir)
 261     return "%s/%s.png" % ( rootsubdir , tiley )
 262
 263 class interactiveMapWidget :
 264
 265   def Shift( self , dx , dy ) :
 266         self.mapwidget._bg.do_change_viewport( dx , dy )
 267         self.mapwidget.update_background()
 268
 269   def Up( self ) :
 270         self.mapwidget._bg.do_change_reftile( 0 , -1 )
 271         self.mapwidget.update_background()
 272
 273   def Down( self ) :
 274         self.mapwidget._bg.do_change_reftile( 0 , +1 )
 275         self.mapwidget.update_background()
 276
 277   def Right( self ) :
 278         self.mapwidget._bg.do_change_reftile( +1 , 0 )
 279         self.mapwidget.update_background()
 280
 281   def Left( self ) :
 282         self.mapwidget._bg.do_change_reftile( -1 , 0 )
 283         self.mapwidget.update_background()
 284
 285
 286 class simpleMapWidget ( AbstractmapWidget , gtk.Image ) :
 287
 288     def __init__ ( self , config , map_size=(800,480) ) :
 289         AbstractmapWidget.__init__( self , config , map_size )
 290
 291         gtk.Image.__init__(self)
 292
 293         self._bg = background_map( map_size , tile_loader( config ) )
 294
 295         self.update_background()
 296
 297     def update_background( self ) :
 298         vport = self._bg.get_viewport()
 299         p = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, True, 8, self.win_x , self.win_y )
 300         p.get_from_drawable( self._bg , self._bg.get_colormap() , vport[0] , vport[1] , 0 , 0 , self.win_x , self.win_y )
 301         self.set_from_pixbuf(p)
 302
 303     def composeMap( self ) :
 304         center_x , center_y = self.center()
 305
 306         # Ranges should be long enough as to fill the screen
 307         # Maybe they should be decided based on self.win_x, self.win_y
 308         for i in range(-3,4) :
 309             for j in range(-3,4) :
 310                 file = self.tilename( i , j , self.conf.zoom )
 311                 if file is None :
 312                     pixbuf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, False, 8, self.tile_size, self.tile_size )
 313                     pixbuf.fill( 0x00000000 )
 314                 else :
 315                     try :
 316                         pixbuf = gtk.gdk.pixbuf_new_from_file( file )
 317                     except gobject.GError , ex :
 318                         print "Corrupted file %s" % ( file )
 319                         os.unlink( file )
 320                         #file = self.tilename( self.reftile_x + i , self.reftile_y + j , self.conf.zoom )
 321                         file = self.tilename( i , j , self.conf.zoom )
 322                         try :
 323                             pixbuf = gtk.gdk.pixbuf_new_from_file( file )
 324                         except :
 325                             print "Total failure for tile for %s,%s" % ( self.reftile_x + i , self.reftile_y + j )
 326                             pixbuf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, False, 8, self.tile_size, self.tile_size )
 327
 328                 dest_x = self.tile_size * i + center_x
 329                 dest_y = self.tile_size * j + center_y
 330
 331                 init_x = 0
 332                 size_x = self.tile_size
 333                 if dest_x < 0 :
 334                    init_x = abs(dest_x)
 335                    size_x = self.tile_size + dest_x
 336                    dest_x = 0
 337                 if dest_x + self.tile_size > self.win_x :
 338                    size_x = self.win_x - dest_x
 339
 340                 init_y = 0
 341                 size_y = self.tile_size
 342                 if dest_y < 0 :
 343                    init_y = abs(dest_y)
 344                    size_y = self.tile_size + dest_y
 345                    dest_y = 0
 346                 if dest_y + self.tile_size > self.win_y :
 347                    size_y = self.win_y - dest_y
 348
 349                 if ( size_x > 0 and size_y > 0 ) and ( init_x < self.tile_size and init_y < self.tile_size ) :
 350                     pixbuf.copy_area( init_x, init_y, size_x, size_y, self.get_pixbuf(), dest_x , dest_y )
 351                 del(pixbuf)
 352
 353 #        self.draw_paths()
 354 #        self.plot_APs()
 355
 356     def center( self ) :
 357
 358         center_x , center_y = self.win_x / 2 , self.win_y / 2
 359
 360         # To get the central pixel in the window center, we must shift to the tile origin
 361         center_x -= self.refpix_x
 362         center_y -= self.refpix_y
 363
 364         return center_x , center_y
 365
 366     def plot( self , pixmap , coords , colorname , radius=3 ) :
 367
 368         center_x , center_y = self.center()
 369
 370         gc = pixmap.new_gc()
 371         gc.foreground = pixmap.get_colormap().alloc_color( colorname )
 372
 373         dest_x , dest_y = self.gps2pix( coords , ( center_x , center_y ) )
 374         pixmap.draw_rectangle(gc, True , dest_x , dest_y , radius , radius )
 375
 376     def draw_paths( self ) :
 377
 378         pixmap,mask = self.get_pixbuf().render_pixmap_and_mask()
 379
 380         filename = "data/wiscan_gui.info.old"
 381         fd = open( filename )
 382         for line in fd.readlines() :
 383             values = line.split()
 384             if values[1] == "FIX" :
 385                 self.plot( pixmap , ( float(values[5]) , float(values[6]) ) , "red" )
 386         fd.close()
 387
 388         self.get_pixbuf().get_from_drawable( pixmap , pixmap.get_colormap() , 0, 0 , 0 , 0 , self.win_x, self.win_y )
 389
 390     def plot_APs( self ) :
 391
 392         pixmap,mask = self.get_pixbuf().render_pixmap_and_mask()
 393
 394         db = wifimap.db.database( os.path.join( self.conf.homedir , self.conf.dbname ) )
 395         db.open()
 396         # NOTE : Intervals for query are just educated guesses to fit in window
 397         lat , lon = self.conf.lat , self.conf.lon
 398         for ap in db.db.execute( "SELECT * FROM ap where lat/n>%f and lat/n<%f and lon/n>%f and lon/n<%f" % ( lat - 0.003 , lat + 0.003 , lon - 0.007 , lon + 0.007 ) ) :
 399             if ap[3] > 1 :
 400                 self.plot( pixmap , ( ap[4]/ap[3] , ap[5]/ap[3] ) , "blue" )
 401         db.close()
 402
 403         self.get_pixbuf().get_from_drawable( pixmap , pixmap.get_colormap() , 0, 0 , 0 , 0 , self.win_x, self.win_y )
 404
 405 class mapWidget ( gtk.EventBox , interactiveMapWidget ) :
 406
 407     def __init__ ( self , config ) :
 408         gtk.EventBox.__init__( self )
 409         self.mapwidget = simpleMapWidget( config )
 410         self.add( self.mapwidget )
 411
 412         self.click_x , self.click_y = None , None
 413         self.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK )
 414         self.connect_object('button_press_event', self.press_event, self.mapwidget)
 415         self.connect_object('button_release_event', self.release_event, self)
 416
 417     def press_event ( self , widget , event ) :
 418         self.click_x , self.click_y = event.x , event.y
 419
 420     def release_event ( self , widget, event ) :
 421         # NOTE : we use origin-current for deltas because the map center moves in the opposite direction
 422         delta_x = int( self.click_x - event.x )
 423         delta_y = int( self.click_y - event.y )
 424         widget.Shift(delta_x, delta_y)
 425         self.click_x , self.click_y = None , None
 426
 427 if __name__ == "__main__" :
 428
 429     class StaticConfiguration :
 430
 431         def __init__ ( self , type=None ) :
 432             self._type = type
 433
 434             self.homedir , self.dbname = None , None
 435             self.mapsdir , self.mapclass = "/boot/home/localuser/Maps" , "OpenStreetMap I"
 436
 437             self.store_log , self.use_mapper , self.store_gps = None , None , None
 438
 439             self.lat , self.lon = 40.416 , -3.683
 440             self.zoom = 15
 441
 442     def on_key_press ( widget, event , map ) :
 443         if event.keyval == gtk.keysyms.Up :
 444             map.Up()
 445         elif event.keyval == gtk.keysyms.Down :
 446             map.Down()
 447         elif event.keyval == gtk.keysyms.Right :
 448             map.Right()
 449         elif event.keyval == gtk.keysyms.Left :
 450             map.Left()
 451         else :
 452             print "UNKNOWN",event.keyval
 453
 454     config = StaticConfiguration()
 455     mapwidget = mapWidget( config )
 456     window = gtk.Window()
 457     window.connect("destroy", gtk.main_quit )
 458     window.connect("key-press-event", on_key_press, mapwidget )
 459     window.add( mapwidget )
 460     window.show_all()
 461     gtk.main()
 462