]> git.openstreetmap.org Git - rails.git/blob - app/models/segment.rb
use render proc to write out results, see if that makes things faster
[rails.git] / app / models / segment.rb
1 class Segment < ActiveRecord::Base
2   require 'xml/libxml'
3   set_table_name 'current_segments'
4
5   validates_presence_of :user_id, :timestamp
6   validates_inclusion_of :visible, :in => [ true, false ]
7   validates_numericality_of :node_a, :node_b
8
9   has_many :old_segments, :foreign_key => :id
10   belongs_to :user
11
12   # using belongs_to :foreign_key = 'node_*', since if use has_one :foreign_key = 'id', segment preconditions? fails checking for segment id in node table
13   belongs_to :from_node, :class_name => 'Node', :foreign_key => 'node_a'
14   belongs_to :to_node, :class_name => 'Node', :foreign_key => 'node_b'
15
16   def self.from_xml(xml, create=false)
17     begin
18       p = XML::Parser.new
19       p.string = xml
20       doc = p.parse
21
22       segment = Segment.new
23
24       doc.find('//osm/segment').each do |pt|
25         segment.node_a = pt['from'].to_i
26         segment.node_b = pt['to'].to_i
27
28         unless create
29           if pt['id'] != '0'
30             segment.id = pt['id'].to_i
31           end
32         end
33
34         segment.visible = true
35
36         if create
37           segment.timestamp = Time.now
38         else
39           if pt['timestamp']
40             segment.timestamp = Time.parse(pt['timestamp'])
41           end
42         end
43
44         tags = []
45
46         pt.find('tag').each do |tag|
47           tags << [tag['k'],tag['v']]
48         end
49
50         tags = tags.collect { |k,v| "#{k}=#{v}" }.join(';')
51         tags = '' if tags.nil?
52
53         segment.tags = tags
54       end
55     rescue
56       segment = nil
57     end
58
59     return segment
60   end
61
62   def save_with_history
63     begin
64       Segment.transaction do
65         self.timestamp = Time.now
66         self.save!
67         old_segment = OldSegment.from_segment(self)
68         old_segment.save!
69       end
70
71       return true
72     rescue
73       return nil
74     end
75   end
76
77   def to_xml
78     doc = OSM::API.new.get_xml_doc
79     doc.root << to_xml_node()
80     return doc
81   end
82
83   def to_xml_node(user_display_name_cache  = nil)
84     el1 = XML::Node.new 'segment'
85     el1['id'] = self.id.to_s
86     el1['from'] = self.node_a.to_s
87     el1['to'] = self.node_b.to_s
88
89     user_display_name_cache = {} if user_display_name_cache.nil?
90
91     if user_display_name_cache and user_display_name_cache[self.user_id]
92       # use the cache if available
93     elsif self.user.data_public?
94       user_display_name_cache[self.user_id] = self.user.display_name
95     else
96       user_display_name_cache[self.user_id] = nil
97     end
98
99     el1['user'] = user_display_name_cache[self.user_id] unless user_display_name_cache[self.user_id].nil?
100
101     Segment.split_tags(el1, self.tags)
102     el1['visible'] = self.visible.to_s
103     el1['timestamp'] = self.timestamp.xmlschema
104     return el1
105   end
106
107   def self.split_tags(el, tags)
108     tags.split(';').each do |tag|
109       parts = tag.split('=')
110       key = ''
111       val = ''
112       key = parts[0].strip unless parts[0].nil?
113       val = parts[1].strip unless parts[1].nil?
114       if key != '' && val != ''
115         el2 = XML::Node.new('tag')
116         el2['k'] = key.to_s
117         el2['v'] = val.to_s
118         el << el2
119       end
120     end
121   end
122
123   def preconditions_ok?
124     from_node and from_node.visible and to_node and to_node.visible
125   end
126
127 end