Perform some cleanup
[wifihood] / wifiview.py
index 83f148f..b2d942c 100755 (executable)
@@ -6,62 +6,26 @@ try :
 except :
     hildon = False
 
-import urllib2
 import math
 
 import os
 
-import wifimap.config , wifimap.db
+import wifimap.config
 
-class mapWidget ( gtk.Image ) :
+import wifimap.view
 
-  def __init__ ( self , config ) :
+class mapWidget ( wifimap.view.AbstractmapWidget , gtk.Image ) :
 
-    self.conf = config
-    gtk.Image.__init__(self)
+  def __init__ ( self , config , map_size=(800,480) ) :
+    wifimap.view.AbstractmapWidget.__init__( self , config , map_size )
 
-    # Maximum width should be 800, but actually gets reduced
-    self.win_x , self.win_y = 800 , 480
-    self.tile_size = 256
+    gtk.Image.__init__(self)
 
     p = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, True, 8, self.win_x, self.win_y)
     self.set_from_pixbuf(p)
 
-    self.reftile_x , self.refpix_x = self.lon2tilex( self.conf.lon , self.conf.zoom )
-    self.reftile_y , self.refpix_y = self.lat2tiley( self.conf.lat , self.conf.zoom )
-
     self.composeMap()
 
-  def lon2tilex ( self , lon , zoom ) :
-    number = math.modf( ( lon + 180 ) / 360 * 2 ** zoom )
-    return int( number[1] ) , int( self.tile_size * number[0] )
-
-  def lat2tiley ( self , lat , zoom ) :
-    lat = lat * math.pi / 180
-    number = math.modf( ( 1 - math.log( math.tan( lat ) + 1 / math.cos( lat ) ) / math.pi ) / 2 * 2 ** zoom )
-    return int( number[1] ) , int( self.tile_size * number[0] )
-
-  def tilex2lon ( self , ( tilex , pixx ) , zoom ) :
-    tilex = float(tilex)
-    pixx = float(pixx)
-    return ( tilex + pixx/self.tile_size ) / 2.0 ** zoom * 360.0 - 180.0
-
-  def tiley2lat ( self , ( tiley , pixy ) , zoom ) :
-    tiley = float(tiley)
-    pixy = float(pixy)
-    tiley = math.pi * ( 1 - 2 * ( tiley + pixy/self.tile_size ) / 2.0 ** zoom )
-    return math.degrees( math.atan( math.sinh( tiley ) ) )
-
-  def gps2pix ( self , ( lat , lon ) , ( center_x , center_y ) ) :
-
-    x_pos = self.lon2tilex( lon , self.conf.zoom )
-    y_pos = self.lat2tiley( lat , self.conf.zoom )
-
-    dest_x = self.tile_size * ( x_pos[0] - self.reftile_x ) + center_x + x_pos[1]
-    dest_y = self.tile_size * ( y_pos[0] - self.reftile_y ) + center_y + y_pos[1]
-
-    return dest_x , dest_y
-
   def composeMap( self ) :
     center_x , center_y = self.win_x / 2 , self.win_y / 2
 
@@ -116,6 +80,16 @@ class mapWidget ( gtk.Image ) :
             pixbuf.copy_area( init_x, init_y, size_x, size_y, self.get_pixbuf(), dest_x , dest_y )
         del(pixbuf)
 
+      self.draw_paths
+
+  def draw_paths( self ) :
+
+    center_x , center_y = self.win_x / 2 , self.win_y / 2
+
+    # To get the central pixel in the window center, we must shift to the tile origin
+    center_x -= self.refpix_x
+    center_y -= self.refpix_y
+
     pixmap,mask = self.get_pixbuf().render_pixmap_and_mask()
     red = pixmap.new_gc()
     red.foreground = pixmap.get_colormap().alloc_color("red")
@@ -143,89 +117,30 @@ class mapWidget ( gtk.Image ) :
 
     self.get_pixbuf().get_from_drawable( pixmap , pixmap.get_colormap() , 0, 0 , 0 , 0 , self.win_x, self.win_y )
 
+  def plot_APs( self ) :
 
-  def tilename ( self , x , y , zoom ) :
-    file = self.tile2file( self.reftile_x + x , self.reftile_y + y , zoom )
-    try :
-      os.stat(file)
-    except :
-    #  if mapDownload :
-      if False :
-        try :
-          # useful members : response.code, response.headers
-          response = urllib2.urlopen( "http://tile.openstreetmap.org/%s/%s/%s.png" % ( zoom , x , y ) )
-          if response.geturl() == "http://tile.openstreetmap.org/11/0/0.png" :
-              return None
-          fd = open( file , 'w' )
-          fd.write( response.read() )
-          fd.close()
-        except :
-          return None
-      else :
-        return None
-    return file
-
-  def tile2file( self , tilex , tiley , zoom ) :
-    rootdir = "%s/%s/%s" % ( self.conf.mapsdir , self.conf.mapclass , zoom )
-    if not os.path.isdir( rootdir ) :
-      os.mkdir(rootdir)
-    rootsubdir = "%s/%s" % ( rootdir , tilex )
-    if not os.path.isdir( rootsubdir ) :
-      os.mkdir(rootsubdir)
-    return "%s/%s.png" % ( rootsubdir , tiley )
-
-  def Shift( self , dx , dy ) :
-    self.hide()
-
-    tile_x , tile_y = ( self.refpix_x - dx ) / self.tile_size , ( self.refpix_y - dy ) / self.tile_size
-    self.reftile_x += tile_x
-    self.reftile_y += tile_y
-
-    self.refpix_x -= dx + self.tile_size * tile_x
-    self.refpix_y -= dy + self.tile_size * tile_y
+    center_x , center_y = self.win_x / 2 , self.win_y / 2
 
-    self.composeMap()
-    self.show()
-
-  def ZoomChange ( self , selector ) :
-    model = selector.get_model(0)
-    active = selector.get_active(0)
-    value = model.get( model.get_iter(active) , 0 )
-    self.SetZoom( int(value[0]) )
-
-  def SetZoom( self , zoom ) :
-    self.hide()
-    lat = self.tiley2lat( ( self.reftile_y , self.refpix_y ) , self.conf.zoom )
-    lon = self.tilex2lon( ( self.reftile_x , self.refpix_x ) , self.conf.zoom )
-    self.reftile_x , self.refpix_x = self.lon2tilex( lon , zoom )
-    self.reftile_y , self.refpix_y = self.lat2tiley( lat , zoom )
-    self.conf.zoom = zoom
-    self.composeMap()
-    self.show()
+    # To get the central pixel in the window center, we must shift to the tile origin
+    center_x -= self.refpix_x
+    center_y -= self.refpix_y
 
-  def Up( self ) :
-    self.hide()
-    self.reftile_y -= 1
-    self.composeMap()
-    self.show()
+    pixmap,mask = self.get_pixbuf().render_pixmap_and_mask()
+    blue = pixmap.new_gc()
+    blue.foreground = pixmap.get_colormap().alloc_color("blue")
 
-  def Down( self ) :
-    self.hide()
-    self.reftile_y += 1
-    self.composeMap()
-    self.show()
+    db = wifimap.db.database( os.path.join( self.conf.homedir , self.conf.dbname ) )
+    db.open()
+    # NOTE : Intervals for query are just educated guesses to fit in window
+    lat , lon = self.conf.lat , self.conf.lon
+    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 ) ) :
+        if ap[3] > 1 :
+            dest_x , dest_y = self.gps2pix( ( ap[4]/ap[3] , ap[5]/ap[3] ) , ( center_x , center_y ) )
+            pixmap.draw_rectangle(blue, True , dest_x , dest_y , 3 , 3 )
+    db.close()
 
-  def Right( self ) :
-    self.hide()
-    self.reftile_x += 1
-    self.composeMap()
-    self.show()
+    self.get_pixbuf().get_from_drawable( pixmap , pixmap.get_colormap() , 0, 0 , 0 , 0 , self.win_x, self.win_y )
 
-  def Left( self ) :
-    self.hide()
-    self.reftile_x -= 1
-    self.composeMap()
-    self.show()
 
 if hildon :
 
@@ -344,7 +259,7 @@ class AbstractMapWindow:
       else :
           print "UNKNOWN",event.keyval
 
-    def __init__(self):
+    def __init__( self , map_size=(800,480) ) :
 
         self.connect("destroy", self.destroy)
 
@@ -352,8 +267,7 @@ class AbstractMapWindow:
 
         self.connect("key-press-event", self.on_key_press)
 
-        vbox = gtk.VBox(False, 0)
-        self.add( vbox )
+        self.vbox = gtk.VBox(False, 0)
 
         # To get explicit GDK_BUTTON_PRESS instead of paired GDK_LEAVE_NOTIFY & GDK_ENTER_NOTIFY
 #        self.add_events(gtk.gdk.BUTTON_MOTION_MASK | gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.POINTER_MOTION_MASK)
@@ -364,15 +278,10 @@ class AbstractMapWindow:
         self.connect('button_release_event', self.release_event)
         #
         self.config = wifimap.config.Configuration()
-        self.map = mapWidget( self.config )
-        vbox.pack_end( self.map , True , True , 5)
-
-        self.create_menu( vbox )
+        self.map = mapWidget( self.config , map_size )
+        self.vbox.pack_end( self.map , True , True , 5)
 
-        # and the window
-        self.show_all()
-
-        self.size_x , self.size_y = 800 , 480
+        self.size_x , self.size_y = map_size
         self.click_x , self.click_y = None , None
 
     def zoomdialog ( self , widget ) :
@@ -389,6 +298,9 @@ if hildon :
         def __init__(self):
             hildon.Window.__init__( self )
             AbstractMapWindow.__init__(self)
+            self.add( self.vbox )
+            self.create_menu( self.vbox )
+            self.show_all()
 
         def create_menu ( self , vbox ) :
 
@@ -416,6 +328,10 @@ else :
         def __init__(self):
             gtk.Window.__init__( self , gtk.WINDOW_TOPLEVEL )
             AbstractMapWindow.__init__(self)
+            self.add( self.vbox )
+            self.create_menu( self.vbox )
+            self.show_all()
+
             self.resize( self.size_x , self.size_y)
 
         def create_menu ( self , vbox ) :