Режим работы "только обработать очередь".
[pyrungps.git] / render_tiles.py
old mode 100644 (file)
new mode 100755 (executable)
index 5ce33f1..2d81f77
@@ -5,80 +5,121 @@ import sqlite3
 import math
 from pprint import pprint
 
-def queue_render(db,filename):
+def queue_render(db,filename,forced_max_zoom=None):
 
   conn = sqlite3.connect(db)
   cur = conn.cursor()
         
   cur.execute("select minlat,minlon,maxlat,maxlon from tracks where filename=?" , (filename.decode('UTF-8'),))
   minlat,minlon,maxlat,maxlon=cur.fetchone()
-  queue_tiles(db,minlat,minlon,maxlat,maxlon)
+  queue_tiles(db,minlat,minlon,maxlat,maxlon,forced_max_zoom)
+
+def deg2num(lat_deg, lon_deg, zoom):
+  lat_rad = math.radians(lat_deg)
+  n = 2.0 ** zoom
+  xtile = int((lon_deg + 180.0) / 360.0 * n)
+  ytile = int((1.0 - math.log(math.tan(lat_rad) + (1 / math.cos(lat_rad))) / math.pi) / 2.0 * n)
+  return (xtile, ytile)
             
 def queue_tiles(db,minlat,minlon,maxlat,maxlon,forced_max_zoom=None):
 
   conn = sqlite3.connect(db)
   
   # определяем примерный стартовый зум
-  
-  minlatrad = math.radians(minlat)
-  maxlatrad = math.radians(maxlat)
-  
-  minx = (minlon + 180.0)/360.0
-  maxx = (maxlon + 180.0)/360.0
-  
-  miny = (1.0 - math.log(math.tan(minlatrad) + (1 / math.cos(minlatrad))) / math.pi) / 2.0
-  maxy = (1.0 - math.log(math.tan(maxlatrad) + (1 / math.cos(maxlatrad))) / math.pi) / 2.0
 
-  if minx>maxx:
-    minx,maxx = maxx,minx
-    
-  if miny>maxy:
-    miny,maxy = maxy,miny
-  
-  minzoom=9
+  minzoom=8
+
   if forced_max_zoom:
     maxzoom=forced_max_zoom
   else:
-    maxzoom=16  
-  
-  for zoom in range(minzoom,maxzoom+1):
-    
-    n = 2 ** zoom
-    
-    minxt = int(minx * n)
-    minyt = int(miny * n)
-    maxxt = int(maxx * n)
-    maxyt = int(maxy * n)
-    
-    ins = conn.cursor()
-    print zoom,minxt,maxxt,minyt,maxyt
-    ins.execute('insert into render_queue(zoom,minx,maxx,miny,maxy) values(?,?,?,?,?)',(zoom,minxt,maxxt,minyt,maxyt))
-
-    if not forced_max_zoom:
-      if (maxxt-minxt>16) or (maxyt-minyt>12):
-        conn.commit()
+    maxzoom=minzoom
+    while True:
+      minx,miny=deg2num(minlat,minlon,maxzoom)
+      maxx,maxy=deg2num(maxlat,maxlon,maxzoom)
+      
+      print maxzoom,':',minx,'-',maxx,'/',miny,'-',maxy
+      if (maxx-minx>16) or (maxy-miny>12) or (maxzoom==16):
         break
+      else:
+        maxzoom=maxzoom+1   
+  
+  if maxzoom<minzoom:
+    minzoom=maxzoom
+  
+  ins = conn.cursor()
+  print minlat,minlon,maxlat,maxlon,minzoom,maxzoom
+  ins.execute('insert into render_queue(minlat,maxlat,minlon,maxlon,minzoom,maxzoom) values(?,?,?,?,?,?)',(minlat,maxlat,minlon,maxlon,minzoom,maxzoom))
 
   conn.commit()        
 
-def process_queue(db,map):
+def process_queue(db,map,force=False,backend="renderd"):
 
   from os import system
 
   conn = sqlite3.connect(db)
   cur = conn.cursor()
-  cur.execute('select id,zoom,minx,maxx,miny,maxy from render_queue')
+  cur.execute('select id,minlat,maxlat,minlon,maxlon,minzoom,maxzoom from render_queue')
   list=cur.fetchall()
 
   for rec in list:
   
-    id,zoom,minx,maxx,miny,maxy=rec
+    id,minlat,maxlat,minlon,maxlon,minzoom,maxzoom=rec
+    
+    if backend == "tirex" or backend == "default":    
+
+      command = 'map='+map+ \
+        ' z='+str(minzoom)+'-'+str(maxzoom)+ \
+        ' lat='+str(minlat)+','+str(maxlat)+ \
+        ' lon='+str(minlon)+','+str(maxlon)
+
+      if force:
+        command = 'tirex-batch -n 0 --prio=50 '+command
+      else:
+        command = 'tirex-batch -n 0 --prio=50 '+command+' -f not-exists'  
+
+      print command  
+      
+      if system(command)==0:
+        dcur=conn.cursor()
+        dcur.execute('delete from render_queue where id=?',(id,))
+        conn.commit()
+
+    elif backend == "renderd":
+    
+      print minlat,minlon,maxlat,maxlon
+    
+      for zoom in range(minzoom,maxzoom+1):
+        minx,miny=deg2num(minlat,minlon,zoom)
+        maxx,maxy=deg2num(maxlat,maxlon,zoom)
+
+        if minx>maxx:
+          tx=minx
+          maxx=minx
+          minx=tx
+
+        if miny>maxy:
+          ty=miny
+          maxy=miny
+          miny=ty
+
+        print zoom,minx,miny,maxx,maxy
+
+        maps = map.split(',')
+        
+        for map_name in maps:
+
+          command = 'render_list -a -m '+map+ \
+                    ' -z '+str(zoom)+' -Z '+str(zoom)+ \
+                    ' -x '+str(minx)+' -X '+str(maxx)+ \
+                    ' -y '+str(miny)+' -Y '+str(maxy)
+          if force:
+            command = command+ ' --force'
+    
+          print command  
+      
+          if system(command)<>0:
+            return
 
-    command = 'render_list -a -m '+map+ \
-      ' -z '+str(zoom)+' -Z '+str(zoom)+ \
-      ' -x '+str(minx)+' -X '+str(maxx)+ \
-      ' -y '+str(miny)+' -Y '+str(maxy)
-    if system(command)==0:
       dcur=conn.cursor()
       dcur.execute('delete from render_queue where id=?',(id,))
       conn.commit()
@@ -92,12 +133,39 @@ def main():
     help="Data directory", metavar="DIR")
   parser.add_option("-m", "--map", dest="map",
     help="Map name", metavar="MAP")
+  parser.add_option("-z", "--zoom", dest="zoom",
+    help="Maximal zoom (forced), used with coordinates pairs (minlat minlon maxlat maxlon) or filename in arguments", metavar="MAP")
+  parser.add_option("-f", "--force", dest="force", 
+    help="Force tile regeneration (on/off), default off")
+  parser.add_option("-r", "--renderer", dest="renderer", 
+    help="Rendering backend: tirex or renderd")
   (options, args) = parser.parse_args()  
 
   db=options.directory+'/gpx.db'
   map=options.map  
+  zoom=options.zoom
+  if not zoom:
+    zoom=12
+  force=(options.force=='on')
 
-  process_queue(db,map)
+  if options.renderer:
+    renderer=options.renderer
+  else:
+    print "Using default rendering backend..."
+    renderer="default"
+
+  if len(args)==1:
+    filename,=args
+    print "Rendering file: "+filename+"\n"
+    queue_render(db,filename)
+  elif len(args)==4:
+    minlat,minlon,maxlat,maxlon=args
+    print "Rendering region "+minlat+'..'+maxlat+' / '+minlon+'..'+maxlon+"\n"
+    queue_tiles(db,float(minlat),float(minlon),float(maxlat),float(maxlon),int(zoom))
+
+  if map:
+    print "Processing map(s) "+map+"\n"
+    process_queue(db,map,force,renderer)
   
 if __name__ == "__main__":