builder

#include <iostream>
#include <vector>

using namespace std;

class Infantryman {
  public:
    void info() {
      cout << "Infantryman" << endl;
    }
};

class Archer {
  public:
    void info() {
      cout << "Archer" << endl;
    }
};

class Horseman {
  public:
    void info() {
      cout << "Horseman" << endl;
    }
};

class Catapult {
  public:
    void info() {
      cout << "Catapult" << endl;
    }
};

class Elephant {
  public:
    void info() {
      cout << "Elephant" << endl;
    }
};

class Army {
public:
    vector<Infantryman> vi;
    vector<Archer>      va;
    vector<Horseman>    vh;
    vector<Catapult>    vc;
    vector<Elephant>    ve;

    void info() {
        int i;
        for(i = 0; i < vi.size(); ++i) vi[i].info();
        for(i = 0; i < va.size(); ++i) va[i].info();
        for(i = 0; i < vh.size(); ++i) vh[i].info();
        for(i = 0; i < vc.size(); ++i) vc[i].info();
        for(i = 0; i < ve.size(); ++i) ve[i].info();
    }
};

class ArmyBuilder {
protected:
    Army *p;
public:
    ArmyBuilder() : p(0) { }
    virtual ~ArmyBuilder() {}
        virtual void createArmy() {}
        virtual void buildInfantryman() {}
        virtual void buildArcher() {}
        virtual void buildHorseman() {}
        virtual void buildCatapult() {}
        virtual void buildElephant() {}
        virtual Army* getArmy() { return p; }
};

class RomanArmyBuilder: public ArmyBuilder {
  public:
    void createArmy() { p = new Army; }
    void buildInfantryman() { p->vi.push_back( Infantryman()); }
    void buildArcher() { p->va.push_back( Archer()); }
    void buildHorseman() { p->vh.push_back( Horseman()); }
    void buildCatapult() { p->vc.push_back( Catapult()); }
};

class CarthaginianArmyBuilder: public ArmyBuilder {
  public:
    void createArmy() { p = new Army; }
    void buildInfantryman() { p->vi.push_back( Infantryman()); }
    void buildArcher() { p->va.push_back( Archer()); }
    void buildHorseman() { p->vh.push_back( Horseman()); }
    void buildElephant() { p->ve.push_back( Elephant()); }
};

class Director {
public:
    Army* createArmy(ArmyBuilder &builder) {
        builder.createArmy();
        builder.buildInfantryman();
        builder.buildArcher();
        builder.buildHorseman();
        builder.buildCatapult();
        builder.buildElephant();
        return(builder.getArmy());
    }
};

int main() {
    Director dir;
    RomanArmyBuilder ra_builder;
    CarthaginianArmyBuilder ca_builder;

    Army *ra = dir.createArmy(ra_builder);
    Army *ca = dir.createArmy(ca_builder);
    cout << "Roman army:" << endl;
    ra->info();
    cout << "\nCarthaginian army:" << endl;
    ca->info();

    return 0;
}