디자인 모드 일 일 연습: 조합 모드 (Composite)

// component
class FileComponent {
    public:
        virtual ~FileComponent();
        
        virtual int GetSize();
        virtual void Display();
        virtual FileComponent* GetComposite() { return this; }
        
        virtual void Add(FileComponent*);
        virtual void Remove(FileComponent*);
        virtual Iterator<FileComponent*>* CreateIterator();
        
    protected:
        FileComponent(const char* name);
        
    private:
        const char* name;
};

// composite
class FolderComposite: public FileComponent {
    public:
        FolderComposite(const char* name);
        virtual ~FolderComposite();
        
        virtual int GetSize();
        virtual void Display();
        
        virtual void Add(FileComponent*);
        virtual void Remove(FileComponent*);
        virtual List<FileComponent*>* GetFileList();
        
    private:
        List<FileComponent*> fileList;
};

int FolderComposite::GetSize() {
    List<FileComponent*>* itr = fileList->begin();
    int totalSize = 0;
    
    for (; itr != fileList->end(); itr++) {
        totalSize += (*itr)->GetSize();
    }
    
    return totalSize;
}

void Display() {
    printf("d  %s  %d
", name, GetSize());
    List<FileComponent*>* itr = fileList->begin();
    for (; itr != fileList->end(); itr++) {
        (*itr)->Display();
    }
}

// leaf
class DocumentLeaf: public FileComponent {
    public:
        DocumentLeaf(const char* name);
        virtual ~DocumentLeaf();
        
        virtual int GetSize();
        virtual void Display();
        virtual FileComponent* GetComposite() { return 0; }
};

void DocumentLeaf::Display() {
    printf("-  %s  %d
", name, GetSize());
}


// test
void Test() {
    FolderComposite* folder1 = new FolderComposite("folder1");
    FolderComposite* folder2 = new FolderComposite("folder2");
    
    DocumentLeaf* doc1 = new DocumentLeaf("doc1");
    DocumentLeaf* doc2 = new DocumentLeaf("doc2");
    DocumentLeaf* doc3 = new DocumentLeaf("doc3");
    
    folder1->Add(doc1);
    folder1->Add(doc2);
    folder2->Add(doc3);
    
    FolderComposite* rootFolder = new FolderComposite("root");
    rootFolder->Add(folder1);
    rootFolder->Add(folder2);
    
    rootFolder->Display();
    
    // todo ...  destroy
}