]> git.openstreetmap.org Git - rails.git/blob - lib/potlatch.rb
Update to rails 3.0.4 and composite_primary_keys 3.1.1
[rails.git] / lib / potlatch.rb
1 # The Potlatch module provides helper functions for potlatch and its communication with the server
2 module Potlatch
3
4   # The AMF class is a set of helper functions for encoding and decoding AMF.
5   class AMF
6     
7     # Return two-byte integer
8     def self.getint(s) 
9       s.getc*256+s.getc
10     end
11
12     # Return four-byte long
13     def self.getlong(s) 
14       ((s.getc*256+s.getc)*256+s.getc)*256+s.getc
15     end
16
17     # Return string with two-byte length 
18     def self.getstring(s) 
19       len=s.getc*256+s.getc
20       s.read(len)
21     end
22
23     # Return eight-byte double-precision float 
24     def self.getdouble(s) 
25       a=s.read(8).unpack('G')                   # G big-endian, E little-endian
26       a[0]
27     end
28
29     # Return numeric array
30     def self.getarray(s) 
31       len=getlong(s)
32       arr=[]
33       for i in (0..len-1)
34         arr[i]=getvalue(s)
35       end
36       arr
37     end
38
39     # Return object/hash 
40     def self.getobject(s) 
41       arr={}
42       while (key=getstring(s))
43         if (key=='') then break end
44         arr[key]=getvalue(s)
45       end
46       s.getc            # skip the 9 'end of object' value
47       arr
48     end
49
50     # Parse and get value
51     def self.getvalue(s) 
52       case s.getc
53       when 0;   return getdouble(s)                     # number
54       when 1;   return s.getc                           # boolean
55       when 2;   return getstring(s)                     # string
56       when 3;   return getobject(s)                     # object/hash
57       when 5;   return nil                                      # null
58       when 6;   return nil                                      # undefined
59       when 8;   s.read(4)                                       # mixedArray
60         return getobject(s)                     #  |
61       when 10;return getarray(s)                        # array
62       else;     return nil                                      # error
63       end
64     end
65
66     # Envelope data into AMF writeable form
67     def self.putdata(index,n) 
68       d =encodestring(index+"/onResult")
69       d+=encodestring("null")
70       d+=[-1].pack("N")
71       d+=encodevalue(n)
72     end
73
74     # Pack variables as AMF
75     def self.encodevalue(n) 
76       case n.class.to_s
77       when 'Array'
78         a=10.chr+encodelong(n.length)
79         n.each do |b|
80           a+=encodevalue(b)
81         end
82         a
83       when 'Hash'
84         a=3.chr
85         n.each do |k,v|
86           a+=encodestring(k.to_s)+encodevalue(v)
87         end
88         a+0.chr+0.chr+9.chr
89       when 'String'
90         2.chr+encodestring(n)
91       when 'Bignum','Fixnum','Float'
92         0.chr+encodedouble(n)
93       when 'NilClass'
94         5.chr
95           when 'TrueClass'
96         0.chr+encodedouble(1)
97           when 'FalseClass'
98         0.chr+encodedouble(0)
99       else
100         Rails.logger.error("Unexpected Ruby type for AMF conversion: "+n.class.to_s)
101       end
102     end
103
104     # Encode string with two-byte length
105     def self.encodestring(n) 
106       a,b=n.size.divmod(256)
107       a.chr+b.chr+n
108     end
109
110     # Encode number as eight-byte double precision float 
111     def self.encodedouble(n) 
112       [n].pack('G')
113     end
114
115     # Encode number as four-byte long
116     def self.encodelong(n) 
117       [n].pack('N')
118     end
119
120   end
121
122
123   # The Potlatch class is a helper for Potlatch
124   class Potlatch
125
126     # ----- getpresets
127     #             in:   none
128     #             does: reads tag preset menus, colours, and autocomplete config files
129     #         out:  [0] presets, [1] presetmenus, [2] presetnames,
130     #                           [3] colours, [4] casing, [5] areas, [6] autotags
131     #                           (all hashes)
132     def self.get_presets
133       Rails.logger.info("  Message: getpresets")
134
135       # Read preset menus
136       presets={}
137       presetmenus={}; presetmenus['point']=[]; presetmenus['way']=[]; presetmenus['POI']=[]
138       presetnames={}; presetnames['point']={}; presetnames['way']={}; presetnames['POI']={}
139       presettype=''
140       presetcategory=''
141       # StringIO.open(txt) do |file|
142       File.open("#{Rails.root}/config/potlatch/presets.txt") do |file|
143         file.each_line {|line|
144           t=line.chomp
145           if (t=~/(\w+)\/(\w+)/) then
146             presettype=$1
147             presetcategory=$2
148             presetmenus[presettype].push(presetcategory)
149             presetnames[presettype][presetcategory]=["(no preset)"]
150           elsif (t=~/^([\w\s]+):\s?(.+)$/) then
151             pre=$1; kv=$2
152             presetnames[presettype][presetcategory].push(pre)
153             presets[pre]={}
154             kv.split(',').each {|a|
155               if (a=~/^(.+)=(.*)$/) then presets[pre][$1]=$2 end
156             }
157           end
158         }
159       end
160
161       # Read colours/styling
162       colours={}; casing={}; areas={}
163       File.open("#{Rails.root}/config/potlatch/colours.txt") do |file|
164         file.each_line {|line|
165           t=line.chomp
166           if (t=~/(\w+)\s+([^\s]+)\s+([^\s]+)\s+([^\s]+)/) then
167             tag=$1
168             if ($2!='-') then colours[tag]=$2.hex end
169             if ($3!='-') then casing[tag]=$3.hex end
170             if ($4!='-') then areas[tag]=$4.hex end
171           end
172         }
173       end
174
175       # Read relations colours/styling
176       relcolours={}; relalphas={}; relwidths={}
177       File.open("#{Rails.root}/config/potlatch/relation_colours.txt") do |file|
178         file.each_line {|line|
179           t=line.chomp
180           if (t=~/(\w+)\s+([^\s]+)\s+([^\s]+)\s+([^\s]+)/) then
181             tag=$1
182             if ($2!='-') then relcolours[tag]=$2.hex end
183             if ($3!='-') then relalphas[tag]=$3.to_i end
184             if ($4!='-') then relwidths[tag]=$4.to_i end
185           end
186         }
187       end
188
189       # Read POI presets
190       icon_list=[]; icon_tags={};
191       File.open("#{Rails.root}/config/potlatch/icon_presets.txt") do |file|
192         file.each_line {|line|
193           (icon,tags)=line.chomp.split("\t")
194           icon_list.push(icon)
195           icon_tags[icon]=Hash[*tags.scan(/([^;=]+)=([^;=]+)/).flatten]
196         }
197       end
198       icon_list.reverse!
199       
200       # Read auto-complete
201       autotags={}; autotags['point']={}; autotags['way']={}; autotags['POI']={};
202       File.open("#{Rails.root}/config/potlatch/autocomplete.txt") do |file|
203         file.each_line {|line|
204           t=line.chomp
205           if (t=~/^([\w:]+)\/(\w+)\s+(.+)$/) then
206             tag=$1; type=$2; values=$3
207             if values=='-' then autotags[type][tag]=[]
208             else autotags[type][tag]=values.split(',').sort.reverse end
209           end
210         }
211       end
212
213       [presets,presetmenus,presetnames,colours,casing,areas,autotags,relcolours,relalphas,relwidths,icon_list,{},icon_tags]
214     end
215   end
216
217 end
218