require 'rubygems'
require 'pp'
require 'yaml'
class Array
 def to_yaml_style
  :inline
 end
end
Dir.chdir("output")
 
def getTimePoints()
        $time_points = []
        $time_points_sources = Hash.new([])
        Dir.glob("*.yml") { |file|
                timetable = YAML::load_file(file)
                $time_points = $time_points | timetable["time_points"]
                timetable["time_points"].each do |timepoint| 
                        $time_points_sources[timepoint] = $time_points_sources[timepoint] | [ file ]
                end
        }
end
 
getTimePoints()
#pp $time_points.sort!
#pp $time_points_sources.sort
 
time_point_corrections = {"North Lynehamham" => "North Lyneham",
  "Woden Bus Station Platform 10)" => "Woden Bus Station (Platform 10)",
  "Saint AndrewsVillage Hughes"=>"Saint Andrews Village Hughes",
 "Flemmington Road / Sandford St"=>"Flemington Road / Sandford St",
 "City Interchange"=>"City Bus Station",
 "City Interchange (Platform 9)"=>"City Bus Station (Platform 9)",
 "Bridbabella Gardens Nursing Home"=>"Brindabella Gardens Nursing Home",
 "Bridbabella GardensNursing Home"=> "Brindabella Gardens Nursing Home",
 "BrindabellaBusiness Park"=> "Brindabella Business Park",
 "NarrabundahTerminus"=>"Narrabundah Terminus",
 "Railway StationKingston"=>"Railway Station Kingston",
 "Saint AndrewsVillage Hughes"=>"Saint Andrews Village Hughes",
 "Dickson ShopsAntill Street"=>"Dickson Shops",
 "Cohen St Bus Station (Platform 3)"=>"Cohen Street Bus Station (Platform 3)",
 "Cohen St Bus Station (Platform 6)"=>"Cohen Street Bus Station (Platform 6)",
 "Newcastle Streetafter Isa Street"=>"Newcastle Street after Isa Street",
 "William Web / Ginninderra Drive"=>"William Webb / Ginninderra Drive",
 "Procor / Mead"=>"Proctor / Mead",
"Fyshwick DirectFactory Outlet"=>"Fyshwick Direct Factory Outlet",
"Dickson ShopsAntill Street"=>"Dickson Shops/Antill St",
"Yarrulumla Shops"=>"Yarralumla Shops",
"Tharwa Dr / Pocket Ave"=>"Tharwa Dr / Pockett Ave",
"Paul Coe / Mirrebei Dr"=>"Paul Coe / Mirrabei Dr",
"Mirrebei Drive / Dam Wall"=>"Mirrabei Drive / Dam Wall"
}
 
time_point_corrections.each do |wrong, right|
        $time_points_sources[wrong].each do |wrongfile|
                badtimetable = YAML::load_file(wrongfile)
                badentrynumber = badtimetable["time_points"].index wrong
                badtimetable["time_points"][badentrynumber] = right
                puts "Corrected '" + wrong + "' to '" + right + "' in " + wrongfile
                File.open(wrongfile, "w") do |f|
                        f.write badtimetable.to_yaml
                end
        end
end
 
getTimePoints()
pp $time_points.sort!