# Inheritance, factories, DATA, lazy ops, to_s - build up example
# Also used (tested) in rspec demo

class Transport
        def getWhere()
                return @dest
        end
        def getNumber()
                return "This is #{@len} vehicles out of #{@@count}"
        end
        def self.create(raw)
                raw = raw.chomp.split "\t"
                item = nil
                if raw[2] == "T" then
                        item = Train.new raw[3].to_i,raw[4].to_i,raw[0],raw[1]
                elsif raw[2] == "B" then
                        item = Bus.new raw[3].to_i,raw[0],raw[1]
                end
                return item
        end
        def self.factory(source=nil)
                if source
                        fh = File.new(source)
                else
                        fh = DATA
                end
                avail = []
                loop do
                        raw = fh.gets
                        if raw.nil? then break end
                        adding = create raw
                        # avail.push adding unless adding.nil?
                        adding.nil? or avail.push adding
                end
                return avail
        end
        def takesMoreThan?(than)
                return getCapacity > than.getCapacity
        end
        def +(that)
                return "yer what?"
        end
        def to_s()
                return "This is a #{self.class} taking #{self.getCapacity}"
        end
        @@count = 0
        def getLength()
                @len * 23
        end
end
class Train < Transport
        def initialize(len,spc,time,dest)
                @len = len
                @spc = spc
                @time = time
                @dest = dest
                @@count += len
        end
        def getCapacity(overload = 1.4)
                if @capa.nil?
                        @capa = (overload * @len * @spc).to_i
                        # puts "working it out"
                else
                        # puts "know that one"
                end
                return @capa
        end
end
class Bus < Transport
        def initialize(spc,time,dest)
                @len = 1
                @spc = spc
                @time = time
                @dest = dest
                @@count += 1
        end
        def getCapacity(overload = 6)
                return (overload + @spc * @len)
        end
        def getLength()
                @len * 14
        end
end
class BendyBus < Bus
        def initialize(spc,time,dest)
                @len = 2
                @spc = spc
                @time = time
                @dest = dest
                @@count += 2
        end
end
# -------------------------------------
if __FILE__ == $0
        # options = Transport.factory("services.txt")
        options = Transport.factory
        options.each do |option|
                begin
                        puts "#{option.getWhere} for #{option.getCapacity} people"
                        puts option.getNumber
                rescue
                        puts "Plonker"
                end
        end

        if options[1].takesMoreThan? options[2]
                puts options[1]
        else
                puts options[2]
        end

end
__END__
06:38 Soton T 3 78
07:18 Bath B 80
07:14 Devizes H 8
07:19 Swindon T 2 79
08:49 Swindon T 1 77
08:19 Swindon T 2 79
09:49 Swindon T 1 77
09:19 Swindon T 2 79
07:49 Swindon T 1 77