app/models/way.rb

   1 class Way < ActiveRecord::Base
   2   require 'xml/libxml'
   3
   4   belongs_to :user
   5
   6   has_many :way_nodes, :foreign_key => 'id', :order => 'sequence_id'
   7   has_many :way_tags, :foreign_key => 'id'
   8
   9   has_many :old_ways, :foreign_key => 'id', :order => 'version'
  10
  11   set_table_name 'current_ways'
  12
  13   def self.from_xml(xml, create=false)
  14     begin
  15       p = XML::Parser.new
  16       p.string = xml
  17       doc = p.parse
  18
  19       way = Way.new
  20
  21       doc.find('//osm/way').each do |pt|
  22         if !create and pt['id'] != '0'
  23           way.id = pt['id'].to_i
  24         end
  25
  26         if create
  27           way.timestamp = Time.now
  28           way.visible = true
  29         else
  30           if pt['timestamp']
  31             way.timestamp = Time.parse(pt['timestamp'])
  32           end
  33         end
  34
  35         pt.find('tag').each do |tag|
  36           way.add_tag_keyval(tag['k'], tag['v'])
  37         end
  38
  39         pt.find('nd').each do |nd|
  40           way.add_nd_num(nd['ref'])
  41         end
  42       end
  43     rescue
  44       way = nil
  45     end
  46
  47     return way
  48   end
  49
  50   def to_xml
  51     doc = OSM::API.new.get_xml_doc
  52     doc.root << to_xml_node()
  53     return doc
  54   end
  55
  56   def to_xml_node(visible_nodes = nil, user_display_name_cache = nil)
  57     el1 = XML::Node.new 'way'
  58     el1['id'] = self.id.to_s
  59     el1['visible'] = self.visible.to_s
  60     el1['timestamp'] = self.timestamp.xmlschema
  61
  62     user_display_name_cache = {} if user_display_name_cache.nil?
  63
  64     if user_display_name_cache and user_display_name_cache.key?(self.user_id)
  65       # use the cache if available
  66     elsif self.user.data_public?
  67       user_display_name_cache[self.user_id] = self.user.display_name
  68     else
  69       user_display_name_cache[self.user_id] = nil
  70     end
  71
  72     el1['user'] = user_display_name_cache[self.user_id] unless user_display_name_cache[self.user_id].nil?
  73
  74     # make sure nodes are output in sequence_id order
  75     ordered_nodes = []
  76     self.way_nodes.each do |nd|
  77       if visible_nodes
  78         # if there is a list of visible nodes then use that to weed out deleted nodes
  79         if visible_nodes[nd.node_id]
  80           ordered_nodes[nd.sequence_id] = nd.node_id.to_s
  81         end
  82       else
  83         # otherwise, manually go to the db to check things
  84         if nd.node.visible? and nd.node.visible?
  85           ordered_nodes[nd.sequence_id] = nd.node_id.to_s
  86         end
  87       end
  88     end
  89
  90     ordered_nodes.each do |nd_id|
  91       if nd_id and nd_id != '0'
  92         e = XML::Node.new 'nd'
  93         e['ref'] = nd_id
  94         el1 << e
  95       end
  96     end
  97
  98     self.way_tags.each do |tag|
  99       e = XML::Node.new 'tag'
 100       e['k'] = tag.k
 101       e['v'] = tag.v
 102       el1 << e
 103     end
 104     return el1
 105   end
 106
 107   def nds
 108     unless @nds
 109         @nds = Array.new
 110         self.way_nodes.each do |nd|
 111             @nds += [nd.node_id]
 112         end
 113     end
 114     @nds
 115   end
 116
 117   def tags
 118     unless @tags
 119         @tags = Hash.new
 120         self.way_tags.each do |tag|
 121             @tags[tag.k] = tag.v
 122         end
 123     end
 124     @tags
 125   end
 126
 127   def nds=(s)
 128     @nds = s
 129   end
 130
 131   def tags=(t)
 132     @tags = t
 133   end
 134
 135   def add_nd_num(n)
 136     @nds = Array.new unless @nds
 137     @nds << n.to_i
 138   end
 139
 140   def add_tag_keyval(k, v)
 141     @tags = Hash.new unless @tags
 142     @tags[k] = v
 143   end
 144
 145   def save_with_history!
 146     t = Time.now
 147
 148     Way.transaction do
 149       self.timestamp = t
 150       self.save!
 151     end
 152
 153     WayTag.transaction do
 154       tags = self.tags
 155
 156       WayTag.delete_all(['id = ?', self.id])
 157
 158       tags.each do |k,v|
 159         tag = WayTag.new
 160         tag.k = k
 161         tag.v = v
 162         tag.id = self.id
 163         tag.save!
 164       end
 165     end
 166
 167     WayNode.transaction do
 168       nds = self.nds
 169
 170       WayNode.delete_all(['id = ?', self.id])
 171
 172       i = 1
 173       nds.each do |n|
 174         nd = WayNode.new
 175         nd.id = self.id
 176         nd.node_id = n
 177         nd.sequence_id = i
 178         nd.save!
 179         i += 1
 180       end
 181     end
 182
 183     old_way = OldWay.from_way(self)
 184     old_way.timestamp = t
 185     old_way.save_with_dependencies!
 186   end
 187
 188   def preconditions_ok?
 189     return false if self.nds.empty?
 190     self.nds.each do |n|
 191       node = Node.find(:first, :conditions => ["id = ?", n])
 192       unless node and node.visible
 193         return false
 194       end
 195     end
 196     return true
 197   end
 198
 199 end