using namespace std;
#include <iostream>
#include <vector>

/* Use of a vector as a member within an object. You'll often want to have
collections within objects, and by using a vector rather than an array you
can do so easily even if you don't know how big the collecion will be when
you create it */

/////// The header files ;-) - normally each would be separate

class Transport {
        public:
                int getseats();
                virtual int getdrivers() {}; // ----------------------
                // Above line to allow this method to be called on a "Transport"
                // even though the actual method is defines within the subclasses!
                int getveh();
        protected:
                int veh;
                int coaches;
        };

class Train : public Transport {
        public:
                Train(int,int);
                int getdrivers();
        } ;

class Bus : public Transport {
        public:
                Bus(int,int);
                int getdrivers();
        } ;

// -> A Flow is an associated object - contains Transport objects in a collection
// it is NOT a part of the inheritance tree, but rather in contains transports

class Flow { // All the transport along a route
        public:
                Flow(int);
                void add(Transport *);
                int getseats();
                int getunits();
                int getdrivers();
        private:
                // Transport *route[20]; Replaced by vector
                // int nroutes; Not needed - use the size method!
                vector<Transport *> route ;
        } ;

/////// The class definition files - again, these would usually be separate files

Bus::Bus(int nv, int nseats) {
        veh = nv;
        coaches = nseats;
        }

int Bus::getdrivers() {
        return veh;
        }

Train::Train(int nv, int nseats) {
        veh = nv;
        coaches = nseats;
        }

int Train::getdrivers() {
        return 1;
        }

int Transport::getseats() {
        return veh * coaches;
        }

int Transport::getveh() {
        return veh;
        }

Flow::Flow(int numtransports) {
        /* numtransports is redundant - it's a maximum number that we
        used when each of the routes on the flow was held in an array */

        // route = vector<Transport *> (numtransports);

        /* A vector of Transport pointers automatically initialised (empty)
        as a Flow object is set up, as the vector is a (private) member of the
        flow object */

        }

void Flow::add(Transport * item) {
        route.push_back(item);
        }

int Flow::getseats() {
        int so_far = 0;
        for (int k=0; k<route.size(); k++) {
                so_far += route.at(k)->getseats();
                }
        return so_far;
        }

int Flow::getunits() {
        int so_far = 0;
        for (int k=0; k<route.size(); k++) {
                so_far += route.at(k)->getveh();
                }
        return so_far;
        }

int Flow::getdrivers() {
        int so_far = 0;
        for (int k=0; k<route.size(); k++) {
                so_far += route.at(k)->getdrivers();
                }
        return so_far;
        }

////// The main application / test program

int main() {

/* A "Flow" is a collection of Transports of various types and so is an
associated object rather than inheriting / beig extended from Transport */

        Flow *chippenham_trowbridge = new Flow(6);
        chippenham_trowbridge->add(new Train(2,73)); // Class 150
        chippenham_trowbridge->add(new Train(1,69)); // Class 153
        chippenham_trowbridge->add(new Bus(2,61)); // Faresaver
        chippenham_trowbridge->add(new Bus(1,81)); // First

        int seatsonflow = chippenham_trowbridge->getseats();
        int vehiclesonflow = chippenham_trowbridge->getunits();
        int driversonflow = chippenham_trowbridge->getdrivers();

        cout << "Seats on flow: " << seatsonflow << endl;
        cout << "Vehicles on flow: " << vehiclesonflow << endl;
        cout << "Drivers on flow: " << driversonflow << endl;

}