[架构师训练营]Week03 - 作业

#include<string>
#include<vector>
#include<memory>
#include<iostream>
#include<algorithm>


using std::string;
using std::vector;
using std::shared_ptr;
using std::cout;
using std::endl;
using std::for_each;

namespace RDTArchitecture{



class Component{
public:
        Component(string name):name_(name){}

        virtual void print() = 0;
        virtual void add(Component* c) = 0;
        virtual void remove(Component* c) =0;
protected:
        string name_;

};
void printObj(shared_ptr<Component>& c){
        c->print();
}

typedef shared_ptr<Component> ComponentPtr;

class Composite:public Component{
public:
        Composite(string name):Component(name){}
        void print(){
                cout << "print " << name_<< "\r\n";
                for_each(components_.begin(), components_.end(), printObj);
        }
        void add(Component* c){

                ComponentPtr ptr(c);
                vector<ComponentPtr>::iterator it = find(components_.begin(), components_.end(),ptr);
                if( it == components_.end()){
                        components_.push_back(ptr);
                }
        }
        void remove(Component* c){
                ComponentPtr ptr(c);
                vector<ComponentPtr>::iterator it = find(components_.begin(), components_.end(),ptr);
                if( it == components_.end()){
                        return;
                }
                components_.erase(it);
        }

private:
        vector<ComponentPtr> components_;
};


class Leaf: public Component{
public:
        Leaf(string name):Component(name){}
        void print(){
                cout << "print " << name_<< "\r\n";
        }
        void add(Component* c){

        }
        void remove(Component* c){

        }
};

};

typedef shared_ptr<RDTArchitecture::Composite> CompositePtr;
typedef shared_ptr<RDTArchitecture::Leaf> LeafPtr;

int main(){
        CompositePtr compositeObject(new RDTArchitecture::Composite("WINFORM(WINDOWS窗口)"));

        compositeObject->add(new RDTArchitecture::Leaf("Picture(LOGO图片)"));

        compositeObject->add(new RDTArchitecture::Leaf("Button(登录)"));

        compositeObject->add(new RDTArchitecture::Leaf("Button(注册)"));

        CompositePtr compositeObject1(new RDTArchitecture::Composite("Frame(FRAME1)"));

        compositeObject1->add(new RDTArchitecture::Leaf("Label(用户名)"));

        compositeObject1->add(new RDTArchitecture::Leaf("TextBox(文本框)"));

        compositeObject1->add(new RDTArchitecture::Leaf("Label(密码)"));

        compositeObject1->add(new RDTArchitecture::Leaf("PasswordBox(密码框)"));

        compositeObject1->add(new RDTArchitecture::Leaf("CheckBox(复选框)"));

        compositeObject1->add(new RDTArchitecture::Leaf("TextBox(记住用户名)"));

        compositeObject1->add(new RDTArchitecture::Leaf("LinkLabel(忘记密码)"));

        compositeObject->add(compositeObject1.get());
        compositeObject->print();
        return 0;
}