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