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   # Find a way given it's ID, and in a single SQL call also grab its nodes
  51   #
  52   # You can't pull in all the tags too unless we put a sequence_id on the way_tags table and have a multipart key
  53   def self.find_eager(id)
  54     way = Way.find(id, :include => {:way_nodes => :node})
  55   end
  56
  57   # Find a way given it's ID, and in a single SQL call also grab its nodes and tags
  58   def to_xml
  59     doc = OSM::API.new.get_xml_doc
  60     doc.root << to_xml_node()
  61     return doc
  62   end
  63
  64   def to_xml_node(visible_nodes = nil, user_display_name_cache = nil)
  65     el1 = XML::Node.new 'way'
  66     el1['id'] = self.id.to_s
  67     el1['visible'] = self.visible.to_s
  68     el1['timestamp'] = self.timestamp.xmlschema
  69
  70     user_display_name_cache = {} if user_display_name_cache.nil?
  71
  72     if user_display_name_cache and user_display_name_cache.key?(self.user_id)
  73       # use the cache if available
  74     elsif self.user.data_public?
  75       user_display_name_cache[self.user_id] = self.user.display_name
  76     else
  77       user_display_name_cache[self.user_id] = nil
  78     end
  79
  80     el1['user'] = user_display_name_cache[self.user_id] unless user_display_name_cache[self.user_id].nil?
  81
  82     # make sure nodes are output in sequence_id order
  83     ordered_nodes = []
  84     self.way_nodes.each do |nd|
  85       if visible_nodes
  86         # if there is a list of visible nodes then use that to weed out deleted nodes
  87         if visible_nodes[nd.node_id]
  88           ordered_nodes[nd.sequence_id] = nd.node_id.to_s
  89         end
  90       else
  91         # otherwise, manually go to the db to check things
  92         if nd.node.visible? and nd.node.visible?
  93           ordered_nodes[nd.sequence_id] = nd.node_id.to_s
  94         end
  95       end
  96     end
  97
  98     ordered_nodes.each do |nd_id|
  99       if nd_id and nd_id != '0'
 100         e = XML::Node.new 'nd'
 101         e['ref'] = nd_id
 102         el1 << e
 103       end
 104     end
 105
 106     self.way_tags.each do |tag|
 107       e = XML::Node.new 'tag'
 108       e['k'] = tag.k
 109       e['v'] = tag.v
 110       el1 << e
 111     end
 112     return el1
 113   end
 114
 115   def nds
 116     unless @nds
 117       @nds = Array.new
 118       self.way_nodes.each do |nd|
 119         @nds += [nd.node_id]
 120       end
 121     end
 122     @nds
 123   end
 124
 125   def tags
 126     unless @tags
 127       @tags = {}
 128       self.way_tags.each do |tag|
 129         @tags[tag.k] = tag.v
 130       end
 131     end
 132     @tags
 133   end
 134
 135   def nds=(s)
 136     @nds = s
 137   end
 138
 139   def tags=(t)
 140     @tags = t
 141   end
 142
 143   def add_nd_num(n)
 144     @nds = Array.new unless @nds
 145     @nds << n.to_i
 146   end
 147
 148   def add_tag_keyval(k, v)
 149     @tags = Hash.new unless @tags
 150     @tags[k] = v
 151   end
 152
 153   def save_with_history!
 154     t = Time.now
 155
 156     Way.transaction do
 157       self.timestamp = t
 158       self.save!
 159     end
 160
 161     WayTag.transaction do
 162       tags = self.tags
 163
 164       WayTag.delete_all(['id = ?', self.id])
 165
 166       tags.each do |k,v|
 167         tag = WayTag.new
 168         tag.k = k
 169         tag.v = v
 170         tag.id = self.id
 171         tag.save!
 172       end
 173     end
 174
 175     WayNode.transaction do
 176       nds = self.nds
 177
 178       WayNode.delete_all(['id = ?', self.id])
 179
 180       sequence = 1
 181       nds.each do |n|
 182         nd = WayNode.new
 183         nd.id = [self.id, sequence]
 184         nd.node_id = n
 185         nd.save!
 186         sequence += 1
 187       end
 188     end
 189
 190     old_way = OldWay.from_way(self)
 191     old_way.timestamp = t
 192     old_way.save_with_dependencies!
 193   end
 194
 195   def preconditions_ok?
 196     return false if self.nds.empty?
 197     self.nds.each do |n|
 198       node = Node.find(:first, :conditions => ["id = ?", n])
 199       unless node and node.visible
 200         return false
 201       end
 202     end
 203     return true
 204   end
 205
 206 end