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['id'])
  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['id'] = 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     begin
 147       Way.transaction do
 148         t = Time.now
 149         self.timestamp = t
 150         self.save!
 151
 152         tags = self.tags
 153
 154         WayTag.delete_all(['id = ?', self.id])
 155
 156         tags.each do |k,v|
 157           tag = WayTag.new
 158           tag.k = k
 159           tag.v = v
 160           tag.id = self.id
 161           tag.save!
 162         end
 163
 164         nds = self.nds
 165
 166         WayNode.delete_all(['id = ?', self.id])
 167
 168         i = 1
 169         nds.each do |n|
 170           nd = WayNode.new
 171           nd.id = self.id
 172           nd.node_id = n
 173           nd.sequence_id = i
 174           nd.save!
 175           i += 1
 176         end
 177
 178         old_way = OldWay.from_way(self)
 179         old_way.timestamp = t
 180         old_way.save_with_dependencies!
 181       end
 182
 183       return true
 184     rescue => ex
 185       puts ex
 186       return nil
 187     end
 188   end
 189
 190   def preconditions_ok?
 191     return false if self.nds.empty?
 192     self.nds.each do |n|
 193       node = Node.find(:first, :conditions => ["id = ?", n])
 194       unless node and node.visible
 195         return false
 196       end
 197     end
 198     return true
 199   end
 200
 201 end