vcs.maemo.org Git - wifihood/blob - wifiscanner/wifimap/replay.py

   1
   2 import config
   3 import db
   4
   5 import gobject
   6
   7 import os
   8
   9 class ReplayScanner ( gobject.GObject ) :
  10
  11     def __init__ ( self , ifname="wlan0" ) :
  12         gobject.GObject.__init__( self )
  13         self.scan_timeout = 0
  14         conf = config.Configuration( 'scanner' )
  15         self.db = db.database( os.path.join( conf.homedir , "wifireplay.db" ) )
  16
  17         # Values specific to replaying
  18         self._file = os.path.join( conf.homedir , "wiscan_gui.info" )
  19         self._fd = None
  20         self._current = None
  21         self._speed = 10
  22
  23         # Values to be set by GPSDevice changed events
  24         self.info = None, 0, 0, None, None, None, None
  25         self.satellites = None
  26         self.cells = None
  27         self.ngps = 0
  28
  29         # Values to be set by wireless scans
  30         self.scanlist = {}
  31         self.tstamp = 0
  32         self.nscan = 0
  33         self.nfp = 0
  34
  35         # Values from the Scanner object
  36         self.newap = 0
  37         self.newaps = False
  38         self.aps = {}
  39
  40     def start ( self , timeout=5000 , writelog=False ) :
  41         self.scan_timeout = timeout
  42         self.db.open()
  43         self._fd = open( self._file )
  44         self._current = self._fd.readline().split()
  45
  46     def stop ( self ) :
  47         self.scan_timeout = 0
  48         self.db.close()
  49         self._fd.close()
  50
  51     def scan ( self ) :
  52
  53         if not self.scan_timeout :
  54             return
  55
  56         next = self._fd.readline().split()
  57         if not next :
  58             return
  59         delta = float(next[0]) - float(self._current[0])
  60         gobject.timeout_add( int(1000 / self._speed * delta) , self.scan )
  61
  62         self.info = [ self._current[1] ]
  63         self.info.extend( map( lambda x : float(x) , self._current[2:8] ) )
  64         if self.info[0] == "FIX" :
  65             self.ngps += 1
  66
  67         self.nscan +=1
  68         self.scanlist.clear()
  69         self.tstamp = float(self._current[0])
  70         for n in range(10, len(self._current), 2) :
  71             self.nfp += 1
  72             self.scanlist[ self._current[n] ] = int(self._current[n+1])
  73
  74         newap = 0
  75         self.newaps = False
  76         for mac,max_rss in self.scanlist.iteritems() :
  77             stored = self.db.db.execute( "SELECT rss, n, lat, lon FROM ap WHERE mac='%s'" % mac ).fetchone()
  78             if stored :
  79                 if not self.aps.has_key( mac ) :
  80                     self.aps[ mac ] = stored[1:]
  81                 if stored[0] > max_rss :
  82                     max_rss = stored[0]
  83                 self.db.update( mac , max_rss , self.tstamp , self.info[4:] )
  84             else :
  85                 newap += 1
  86                 if not self.aps.has_key( mac ) :
  87                     self.aps[ mac ] = [ 0 , 0 , 0 ]
  88                 self.db.add( mac , max_rss , self.tstamp , self.info[4:] )
  89             self.aps[0] += 1
  90             self.aps[1] += self.info[4]
  91             self.aps[2] += self.info[5]
  92         if newap :
  93             self.newap += newap
  94             self.newaps = True
  95
  96         self._current = next
  97
  98     def report ( self ) :
  99         return "%d gps\t%d scan\t%d fp\t%d ap\t%d total ap" % ( self.ngps , self.nscan , self.nfp , self.newap , self.db.nrows() )
 100
 101
 102 gobject.type_register(ReplayScanner)
 103
 104 if __name__ == "__main__" :
 105     loop = gobject.MainLoop()
 106     sample = ReplayScanner()
 107     def show_scan(wifiscanner):
 108         gobject.timeout_add( 5000 , show_scan , sample )
 109         print "scan results %s" % wifiscanner.report()
 110         print "  tstamp %s" % wifiscanner.tstamp
 111         c = 0
 112         for k,v in wifiscanner.scanlist.iteritems() :
 113            c += 1
 114            print "    %s %s" % ( k , v )
 115            if c > 5 :
 116                print "    ..."
 117                break
 118         print
 119     sample.start()
 120     sample.scan()
 121     gobject.timeout_add( 5100 , show_scan , sample )
 122     loop.run()
 123     sample.stop()
 124