osrm-backend/Rakefile

require 'OSM/StreamParser'
require 'socket'
require 'digest/sha1'
require 'cucumber/rake/task'

Cucumber::Rake::Task.new do |t|
  t.cucumber_opts = %w{--format pretty}
end

areas = {
  :kbh => { :country => 'denmark', :bbox => 'top=55.6972 left=12.5222 right=12.624 bottom=55.6376' },
  :frd => { :country => 'denmark', :bbox => 'top=55.7007 left=12.4765 bottom=55.6576 right=12.5698' },
  :regh => { :country => 'denmark', :bbox => 'top=56.164 left=11.792 bottom=55.403 right=12.731' },
  :dk => { :country => 'denmark', :bbox => nil },
  :skaane => { :counry => 'sweden', :bbox => 'top=56.55 left=12.4 bottom=55.3 right=14.6' }
}



osm_data_area_name = ARGV[1] ? ARGV[1].to_s.to_sym : :kbh
raise "Unknown data area." unless areas[osm_data_area_name]
osm_data_country = areas[osm_data_area_name][:country]
osm_data_area_bbox = areas[osm_data_area_name][:bbox]


task osm_data_area_name.to_sym {}   #define empty task to prevent rake from whining. will break if area has same name as a task


def each_process name, &block
  process_list = `ps -o pid -o state -o ucomm`
  process_list.scan /(\d+)\s+([^\s]*)\s+(#{name})/ do |pid,state|
  yield pid.to_i, state.strip if ['I','R','S','T'].include? state[0]
end
end

def up?
  find_pid('osrm-routed') != nil
end

def find_pid name
  each_process(name) { |pid,state| return pid.to_i }
  return nil
end

def wait_for_shutdown name
  timeout = 10
  (timeout*10).times do
    return if find_pid(name) == nil
    sleep 0.1
  end
  raise "*** Could not terminate #{name}."
end

def write_server_ini osm_file
  s=<<-EOF
Threads = 1
IP = 0.0.0.0
Port = 5000

hsgrData=data/#{osm_file}.osrm.hsgr
nodesData=data/#{osm_file}.osrm.nodes
ramIndex=data/#{osm_file}.osrm.ramIndex
fileIndex=data/#{osm_file}.osrm.fileIndex
namesData=data/#{osm_file}.osrm.names
EOF
  File.open( 'server.ini', 'w') {|f| f.write( s ) }
end


desc "Rebuild and run tests"
task :default => [:build, :cucumber]

desc "Build using Scons"
task :build do
  system "scons"
end

desc "Download OSM data"
task :download do
  puts "Downloading..."
  raise "Error while downloading data." unless system "curl http://download.geofabrik.de/osm/europe/#{osm_data_country}.osm.pbf -o sandbox/data/#{osm_data_country}.osm.pbf"
  if osm_data_area_bbox
    puts "Cropping and converting to protobuffer..."
    raise "Error while cropping data." unless system "osmosis --read-pbf file=sandbox/data/#{osm_data_country}.osm.pbf --bounding-box #{osm_data_area_bbox} --write-pbf file=sandbox/data/#{osm_data_area_name}.osm.pbf omitmetadata=true"
  end
end

desc "Crop OSM data"
task :crop do
  if osm_data_area_bbox
    raise "Error while cropping data." unless system "osmosis --read-pbf file=sandbox/data/#{osm_data_country}.osm.pbf --bounding-box #{osm_data_area_bbox} --write-pbf file=sandbox/data/#{osm_data_area_name}.osm.pbf omitmetadata=true"
  end
end

desc "Reprocess OSM data"
task :process do
  Dir.chdir "sandbox" do
    raise "Error while extracting data." unless system "../osrm-extract data/#{osm_data_area_name}.osm.pbf"
    puts
    raise "Error while preparing data." unless system "../osrm-prepare data/#{osm_data_area_name}.osrm #{osm_data_area_name}.osrm.restrictions"
    puts
  end
end

desc "Delete preprocessing files"
task :clean do
  File.delete *Dir.glob('sandbox/data/*.osrm')
  File.delete *Dir.glob('sandbox/data/*.osrm.*')
end

desc "Run all test"
task :test do
  system "cucumber"
  puts
end

desc "Run the routing server in the terminal. Press Ctrl-C to stop."
task :run do
  Dir.chdir "sandbox" do
    write_server_ini osm_data_area_name
    system "../osrm-routed"
  end
end

desc "Launch the routing server in the background. Use rake:down to stop it."
task :up do
  Dir.chdir "sandbox" do
    abort("Already up.") if up?
    write_server_ini osm_data_area_name
    pipe = IO.popen('../osrm-routed 1>>osrm-routed.log 2>>osrm-routed.log')
    timeout = 5
    (timeout*10).times do
      begin
        socket = TCPSocket.new('localhost', 5000)
        socket.puts 'ping'
      rescue Errno::ECONNREFUSED
        sleep 0.1
      end
    end
  end
end

desc "Stop the routing server."
task :down do
  pid = find_pid 'osrm-routed'
  abort("Already down.") unless pid
  Process.kill 'TERM', pid
end

desc "Kill the routing server(s)."
task :kill do
  each_process('osrm-routed') { |pid,state| Process.kill 'KILL', pid }
  each_process('osrm-prepare') { |pid,state| Process.kill 'KILL', pid }
  each_process('osrm-extract') { |pid,state| Process.kill 'KILL', pid }
  wait_for_shutdown 'osrm-routed'
  wait_for_shutdown 'osrm-prepare'
  wait_for_shutdown 'osrm-extract'  
end

desc "Get PID of routing server(s)."
task :pid do
  each_process 'osrm-routed' do |pid,state|
    puts "#{pid}\t#{state}"
  end
end