# Definition of a train - to be shared between all railway applications

class Transport
        @@owmany = 0
        def initialize (origin, length, stockType, time = nil)
                @origin = origin
                @mylength = length # @ => variable within the current (train) object
                @mystock = stockType
                @time = time
                @@owmany += 1
        end
        def getstart
                return @origin
        end
        def strengthen (extra)
                @mylength = @mylength + extra
        end
        def gettime
                return @time
        end
        def Transport.getcount # Static method - let Ruby know!
                return @@owmany
        end

        # "Badgerline\t07:15\tBowerhill\t66" Sample data
        def Transport.factory(instring)
                pieces = instring.split("\t")
                makes = nil
                if ["Badgerline","Faresaver","Bodmans"].include?(pieces[0])
                        makes = Bus.new pieces[2],1,"unknown",pieces[1]
                elsif ["FGW","SWT"].include?(pieces[0])
                        makes = Train.new pieces[2] # etc - line to be completed
                else
                        puts "Error Handling code goes here"
                end
                return makes
        end
end

class Train < Transport
        def getseats
                return @mylength * 73 # @ - so the variable within the current (train) object
        end
        def getdrivers
                return 1
        end
end

class Bus < Transport
        def getseats
                return @mylength * 61
        end
        def getdrivers
                return @mylength
        end
        def to_s
                return "It's a bus! ... leaves at #{@time}"
        end
end
if $0 == __FILE__ # If I am being run as a standalone program ...
        puts "This is the Train class, and it's my copyright ;-)"
        puts "This is a file of classes to be loaded by other program and not run alone"
        # Could put test and demonstration code here!
end

#* Sample Output
#*
#* WomanWithCat:r102 grahamellis$ ruby trainclass4.rb
#* This is the Train class, and it's my copyright ;-)
#* This is a file of classes to be loaded by other program and not run alone
#* WomanWithCat:r102 grahamellis$