af5afb7c90690c8437fe1c6ee43f118de5d024c5
[rails.git] / app / controllers / swf_controller.rb
1 class SwfController < ApplicationController
2   skip_before_filter :verify_authenticity_token
3   before_filter :check_api_readable
4
5   # to log:
6   # RAILS_DEFAULT_LOGGER.error("Args: #{args[0]}, #{args[1]}, #{args[2]}, #{args[3]}")
7   # $log.puts Time.new.to_s+','+Time.new.usec.to_s+": started GPS script"
8   # http://localhost:3000/api/0.4/swf/trackpoints?xmin=-2.32402605810577&xmax=-2.18386309423859&ymin=52.1546608755772&ymax=52.2272777906895&baselong=-2.25325793066437&basey=61.3948537948532&masterscale=5825.4222222222
9
10   # ====================================================================
11   # Public methods
12
13   # ---- trackpoints    compile SWF of trackpoints
14
15   def trackpoints
16     # - Initialise
17
18     baselong = params['baselong'].to_f
19     basey = params['basey'].to_f
20     masterscale = params['masterscale'].to_f
21
22     bbox = BoundingBox.new(params['xmin'], params['ymin'],
23                            params['xmax'], params['ymax'])
24     start = params['start'].to_i
25
26     # - Begin movie
27
28     bounds_left = 0
29     bounds_right = 320 * 20
30     bounds_bottom = 0
31     bounds_top = 240 * 20
32
33     m = ''
34     m += swf_record(9, 255.chr + 155.chr + 155.chr)                     # Background
35     absx = 0
36     absy = 0
37     xl = yb = 9999999
38     xr = yt = -9999999
39
40     # - Send SQL for GPS tracks
41
42     b = ''
43     lasttime = 0
44     lasttrack = lastfile = '-1'
45
46     if params['token']
47       user = User.authenticate(:token => params[:token])
48       sql = "SELECT gps_points.latitude*0.0000001 AS lat,gps_points.longitude*0.0000001 AS lon,gpx_files.id AS fileid," +                      "      EXTRACT(EPOCH FROM gps_points.timestamp) AS ts, gps_points.trackid AS trackid " +                            " FROM gpx_files,gps_points " +                         "WHERE gpx_files.id=gpx_id " +                          "  AND gpx_files.user_id=#{user.id} " +                         "  AND " + OSM.sql_for_area(bbox, "gps_points.") +                      "  AND (gps_points.timestamp IS NOT NULL) " +                           "ORDER BY fileid DESC,ts " +                            "LIMIT 10000 OFFSET #{start}"
49     else
50       sql = "SELECT latitude*0.0000001 AS lat,longitude*0.0000001 AS lon,gpx_id AS fileid," +                        "      EXTRACT(EPOCH FROM timestamp) AS ts, gps_points.trackid AS trackid " +                               " FROM gps_points " +                           "WHERE " + OSM.sql_for_area(bbox, "gps_points.") +                              "  AND (gps_points.timestamp IS NOT NULL) " +                           "ORDER BY fileid DESC,ts " +                            "LIMIT 10000 OFFSET #{start}"
51     end
52     gpslist = ActiveRecord::Base.connection.select_all sql
53
54     # - Draw GPS trace lines
55
56     r = start_shape
57     gpslist.each do |row|
58       xs = (long2coord(row['lon'].to_f, baselong, masterscale) * 20).floor
59       ys = (lat2coord(row['lat'].to_f, basey, masterscale) * 20).floor
60       xl = [xs, xl].min
61       xr = [xs, xr].max
62       yb = [ys, yb].min
63       yt = [ys, yt].max
64       if row['ts'].to_i - lasttime > 180 || row['fileid'] != lastfile || row['trackid'] != lasttrack # or row['ts'].to_i==lasttime
65         b += start_and_move(xs, ys, '01')
66         absx = xs.floor
67         absy = ys.floor
68       end
69       b += draw_to(absx, absy, xs, ys)
70       absx = xs.floor
71       absy = ys.floor
72       lasttime = row['ts'].to_i
73       lastfile = row['fileid']
74       lasttrack = row['trackid']
75       r += [b.slice!(0...80)].pack("B*") while b.length > 80
76     end
77
78     #   (Unwayed segments removed)
79
80     # - Write shape
81
82     b += end_shape
83     r += [b].pack("B*")
84     m += swf_record(2, pack_u16(1) + pack_rect(xl, xr, yb, yt) + r)
85     m += swf_record(4, pack_u16(1) + pack_u16(1))
86
87     # - Create Flash header and write to browser
88
89     m += swf_record(1, '')                                                                      # Show frame
90     m += swf_record(0, '')                                                                      # End
91
92     m = pack_rect(bounds_left, bounds_right, bounds_bottom, bounds_top) + 0.chr + 12.chr + pack_u16(1) + m
93     m = 'FWS' + 6.chr + pack_u32(m.length + 8) + m
94
95     render :text => m, :content_type => "application/x-shockwave-flash"
96   end
97
98   private
99
100   # =======================================================================
101   # SWF functions
102
103   # -----------------------------------------------------------------------
104   # Line-drawing
105
106   def start_shape
107     s = 0.chr                                           # No fill styles
108     s += 2.chr                                          # Two line styles
109     s += pack_u16(0) + 0.chr + 255.chr + 255.chr        # Width 5, RGB #00FFFF
110     s += pack_u16(0) + 255.chr + 0.chr + 255.chr        # Width 5, RGB #FF00FF
111     s += 34.chr                                                                         # 2 fill, 2 line index bits
112     s
113   end
114
115   def end_shape
116     '000000'
117   end
118
119   def start_and_move(x, y, col)
120     d = '001001'                                        # Line style change, moveTo
121     l = [length_sb(x), length_sb(y)].max
122     d += sprintf("%05b%0#{l}b%0#{l}b", l, x, y)
123     d += col                                            # Select line style
124     d
125   end
126
127   def draw_to(absx, absy, x, y)
128     dx = x - absx
129     dy = y - absy
130
131     # Split the line up if there's anything>16383, because
132     # that would overflow the 4 bits allowed for length
133     mstep = [dx.abs / 16383, dy.abs / 16383, 1].max.ceil
134     xstep = dx / mstep
135     ystep = dy / mstep
136     d = ''
137     1.upto(mstep).each do
138       d += draw_section(x, y, x + xstep, y + ystep)
139       x += xstep
140       y += ystep
141     end
142     d
143   end
144
145   def draw_section(x1, y1, x2, y2)
146     d = '11'                                                                                    # TypeFlag, EdgeFlag
147     dx = x2 - x1
148     dy = y2 - y1
149     l = [length_sb(dx), length_sb(dy)].max
150     d += sprintf("%04b", l - 2)
151     d += '1'                                                                                    # GeneralLine
152     d += sprintf("%0#{l}b%0#{l}b", dx, dy)
153     d
154   end
155
156   # -----------------------------------------------------------------------
157   # Specific data types
158
159   # SWF data block type
160
161   def swf_record(id, r)
162     if r.length > 62
163       # Long header: tag id, 0x3F, length
164       return pack_u16((id << 6) + 0x3F) + pack_u32(r.length) + r
165     else
166       # Short header: tag id, length
167       return pack_u16((id << 6) + r.length) + r
168     end
169   end
170
171   # SWF RECT type
172
173   def pack_rect(a, b, c, d)
174     l = [length_sb(a),
175          length_sb(b),
176          length_sb(c),
177          length_sb(d)].max
178     # create binary string (00111001 etc.) - 5-byte length, then bbox
179     n = sprintf("%05b%0#{l}b%0#{l}b%0#{l}b%0#{l}b", l, a, b, c, d)
180     # pack into byte string
181     [n].pack("B*")
182   end
183
184   # -----------------------------------------------------------------------
185   # Generic pack functions
186
187   def pack_u16(n)
188     [n.floor].pack("v")
189   end
190
191   def pack_u32(n)
192     [n.floor].pack("V")
193   end
194
195   # Find number of bits required to store arbitrary-length binary
196
197   def length_sb(n)
198     Math.frexp(n + (n == 0 ? 1 : 0))[1] + 1
199   end
200
201   # ====================================================================
202   # Co-ordinate conversion
203   # (this is duplicated from amf_controller, should probably share)
204
205   def lat2coord(a, basey, masterscale)
206     -(lat2y(a) - basey) * masterscale
207   end
208
209   def long2coord(a, baselong, masterscale)
210     (a - baselong) * masterscale
211   end
212
213   def lat2y(a)
214     180 / Math::PI * Math.log(Math.tan(Math::PI / 4 + a * (Math::PI / 180) / 2))
215   end
216
217   def sqlescape(a)
218     a.gsub("'", "''").gsub(92.chr, 92.chr + 92.chr)
219   end
220 end