443ae807337cb2c3b33909860e6882da9f2a6308
[pyrungps.git] / render_tiles.py
1 #!/usr/bin/env python
2 # coding: UTF-8
3
4 import sqlite3
5 import math
6 from pprint import pprint
7
8 def queue_render(db,filename,forced_max_zoom=None):
9
10   conn = sqlite3.connect(db)
11   conn.text_factory = str
12   cur = conn.cursor()
13         
14   cur.execute("select minlat,minlon,maxlat,maxlon from tracks where filename=?" , (filename.decode('UTF-8'),))
15   minlat,minlon,maxlat,maxlon=cur.fetchone()
16   queue_tiles(db,minlat,minlon,maxlat,maxlon,forced_max_zoom)
17
18 def deg2num(lat_deg, lon_deg, zoom):
19   lat_rad = math.radians(lat_deg)
20   n = 2.0 ** zoom
21   xtile = int((lon_deg + 180.0) / 360.0 * n)
22   ytile = int((1.0 - math.log(math.tan(lat_rad) + (1 / math.cos(lat_rad))) / math.pi) / 2.0 * n)
23   return (xtile, ytile)
24             
25 def queue_tiles(db,minlat,minlon,maxlat,maxlon,forced_max_zoom=None):
26
27   conn = sqlite3.connect(db)
28   
29   # определяем примерный стартовый зум
30
31   minzoom=8
32
33   if forced_max_zoom:
34     maxzoom=forced_max_zoom
35   else:
36     maxzoom=minzoom
37     while True:
38       minx,miny=deg2num(minlat,minlon,maxzoom)
39       maxx,maxy=deg2num(maxlat,maxlon,maxzoom)
40       
41       print maxzoom,':',minx,'-',maxx,'/',miny,'-',maxy
42       if (maxx-minx>16) or (maxy-miny>12) or (maxzoom==16):
43         break
44       else:
45         maxzoom=maxzoom+1   
46   
47   if maxzoom<minzoom:
48     minzoom=maxzoom
49   
50   ins = conn.cursor()
51   print minlat,minlon,maxlat,maxlon,minzoom,maxzoom
52   ins.execute('insert into render_queue(minlat,maxlat,minlon,maxlon,minzoom,maxzoom) values(?,?,?,?,?,?)',(minlat,maxlat,minlon,maxlon,minzoom,maxzoom))
53
54   conn.commit()        
55
56 def process_queue(db,map,force=False,backend="renderd"):
57
58   from os import system
59
60   conn = sqlite3.connect(db)
61   cur = conn.cursor()
62   cur.execute('select id,minlat,maxlat,minlon,maxlon,minzoom,maxzoom from render_queue')
63   list=cur.fetchall()
64
65   for rec in list:
66   
67     id,minlat,maxlat,minlon,maxlon,minzoom,maxzoom=rec
68     
69     if backend == "tirex" or backend == "default":    
70
71       command = 'map='+map+ \
72         ' z='+str(minzoom)+'-'+str(maxzoom)+ \
73         ' lat='+str(minlat)+','+str(maxlat)+ \
74         ' lon='+str(minlon)+','+str(maxlon)
75
76       if force:
77         command = 'tirex-batch -n 0 --prio=50 '+command
78       else:
79         command = 'tirex-batch -n 0 --prio=50 '+command+' -f not-exists'  
80
81       print command  
82       
83       if system(command)==0:
84         dcur=conn.cursor()
85         dcur.execute('delete from render_queue where id=?',(id,))
86         conn.commit()
87
88     elif backend == "renderd":
89     
90       print minlat,minlon,maxlat,maxlon
91     
92       for zoom in range(minzoom,maxzoom+1):
93         minx,miny=deg2num(minlat,minlon,zoom)
94         maxx,maxy=deg2num(maxlat,maxlon,zoom)
95
96         if minx>maxx:
97           tx=minx
98           maxx=minx
99           minx=tx
100
101         if miny>maxy:
102           ty=miny
103           maxy=miny
104           miny=ty
105
106         print zoom,minx,miny,maxx,maxy
107
108         maps = map.split(',')
109         
110         for map_name in maps:
111
112           command = 'render_list -a -m '+map+ \
113                     ' -z '+str(zoom)+' -Z '+str(zoom)+ \
114                     ' -x '+str(minx)+' -X '+str(maxx)+ \
115                     ' -y '+str(miny)+' -Y '+str(maxy)
116           if force:
117             command = command+ ' --force'
118     
119           print command  
120       
121           if system(command)<>0:
122             return
123
124       dcur=conn.cursor()
125       dcur.execute('delete from render_queue where id=?',(id,))
126       conn.commit()
127
128 def main():
129
130   from optparse import OptionParser
131   
132   parser = OptionParser()
133   parser.add_option("-d", "--data", dest="directory",
134     help="Data directory", metavar="DIR")
135   parser.add_option("-m", "--map", dest="map",
136     help="Map name", metavar="MAP")
137   parser.add_option("-z", "--zoom", dest="zoom",
138     help="Maximal zoom (forced), used with coordinates pairs (minlat minlon maxlat maxlon) or filename in arguments", metavar="MAP")
139   parser.add_option("-f", "--force", dest="force", 
140     help="Force tile regeneration (on/off), default off")
141   parser.add_option("-r", "--renderer", dest="renderer", 
142     help="Rendering backend: tirex or renderd")
143   (options, args) = parser.parse_args()  
144
145   db=options.directory+'/gpx.db'
146   map=options.map  
147   zoom=options.zoom
148   if not zoom:
149     zoom=12
150   force=(options.force=='on')
151
152   if options.renderer:
153     renderer=options.renderer
154   else:
155     print "Using default rendering backend..."
156     renderer="default"
157
158   if len(args)==1:
159     filename,=args
160     print "Rendering file: "+filename+"\n"
161     queue_render(db,filename)
162   elif len(args)==4:
163     minlat,minlon,maxlat,maxlon=args
164     print "Rendering region "+minlat+'..'+maxlat+' / '+minlon+'..'+maxlon+"\n"
165     queue_tiles(db,float(minlat),float(minlon),float(maxlat),float(maxlon),int(zoom))
166
167   if map:
168     print "Processing map(s) "+map+"\n"
169     process_queue(db,map,force,renderer)
170   
171 if __name__ == "__main__":
172
173   main()          
174