设计模式——组合模式

一、任务

用透明组合模式实现教材中的“文件夹浏览”这个例子

二、类图

三、代码

1、Java

package tutorial10;

public class Client {

    public static void main(String[] args) {
        // TODO 自动生成的方法存根
        File obj1,obj2,obj3,obj4,obj5;
        Folder plate1,plate2,plate3;
        
        obj1=new ImageFile("image1");
        obj2=new TextFile("text");
        plate1=new Folder("folder1");
        plate1.addFile(obj1);
        plate1.addFile(obj2);
        
        obj3=new VideoFile("video1");
        obj4=new VideoFile("video2");
        plate2=new Folder("folder2");
        plate2.addFile(obj3);
        plate2.addFile(obj4);
        
        obj5=new ImageFile("image2");
        plate3=new Folder("folder3");
        plate3.addFile(plate1);
        plate3.addFile(plate2);
        plate3.addFile(obj5);
        
        plate3.show(plate3);
    }

}

package tutorial10;

public abstract class File {

    
    
    public void addFile(File element) {
        
    }
    
    public void removeFile(File element) {
        
    }
    
    public void show(File element) {
        
    }
}

package tutorial10;

import java.util.ArrayList;

public class Folder extends File {
    private String fileName;
    private ArrayList fileList=new ArrayList();
    public Folder(String name) {
        this.fileName=new String();
        this.fileName=name;
    }
    
    
    public String getFileName() {
        return fileName;
    }


    public void addFile(File element)
    {
        fileList.add(element);
        System.out.println("增加文件夹"+element.getClass().getName());
    }
    public void removeFile(File element)
    {
        fileList.remove(element);
        System.out.println("删除文件夹"+element.getClass().getName());
    }
    public void show(File element)
    {
        System.out.println("展示文件夹：");
        for(Object object:fileList)
        {
            ((File)object).show(element);
        }
        
    }
}

package tutorial10;

public class ImageFile extends File{
    private String fileName;
    
    public ImageFile(String filename)
    {
        this.fileName=new String();
        this.fileName=filename;
    }

    public void addFile(File element) {
        System.out.println("增加图像文件夹："+element.getClass().getName());
    }
    
    public void removeFile(File element) {
        System.out.println("删除图像文件夹："+element.getClass().getName());
    }
    
    public void show(File element) {
        System.out.println("展示图像文件夹："+element.getClass().getName());
    }
}

package tutorial10;

public class TextFile extends File {
    private String fileName;
    public TextFile(String fileName) {
        this.fileName=new String();
        this.fileName=fileName;
    }

    public void addFile(File element) {
        System.out.println("增加文本文件夹："+element.getClass().getName());
    }
    
    public void removeFile(File element) {
        System.out.println("删除文本文件夹："+element.getClass().getName());
    }
    
    public void show(File element) {
        System.out.println("展示文本文件夹："+element.getClass().getName());
    }
}

package tutorial10;

public class VideoFile extends File {
    private String fileName;
    public VideoFile(String filename) {
        this.fileName=new String();
        this.fileName=filename;
    }

    public void addFile(File element) {
        System.out.println("增加视频文件夹："+element.getClass().getName());
    }
    
    public void removeFile(File element) {
        System.out.println("删除视频文件夹："+element.getClass().getName());
    }
    
    public void show(File element) {
        System.out.println("展示视频文件夹："+element.getClass().getName());
    }
}

2、C++

#include <iostream>
#include <string>
#include<list>
using namespace std;

class AbstractFile
{
public:
    virtual void add() {   }
    virtual void remove() {   }
    virtual void display() { }
};

class ImageFile : public AbstractFile
{
private:
    string fileName;
public:
    ImageFile(string filename)
    {
        fileName = filename;
    }
    void add()
    {
        cout << "添加成功" << endl;
    }
    void remove()
    {
        cout << "删除成功" << endl;
    }
    void display()
    {
        cout << fileName << endl;
    }
};


class TextFile : public AbstractFile
{
private:
    string fileName;
public:
    TextFile(string filename)
    {
        fileName = filename;
    }
    void add()
    {
        cout << "添加成功" << endl;
    }
    void remove()
    {
        cout << "删除成功" << endl;
    }
    void display()
    {
        cout << fileName << endl;
    }
};

class VideoFile : public AbstractFile
{
private:
    string fileName;
public:
    VideoFile(string filename)
    {
        fileName = filename;
    }
    void add()
    {
        cout << "添加成功" << endl;
    }
    void remove()
    {
        cout << "删除成功" << endl;
    }
    void display()
    {
        cout << fileName << endl;
    }
};

class Folder : public  AbstractFile
{
private:
    string fileName;
    int level;
    list<AbstractFile*> abstractfiles;

public:
    Folder(string filename)
    {
        fileName = filename;
    }
    Folder(string filename, int level)
    {
        fileName = filename;
        this->level = level;
    }
    void add(AbstractFile* abstractfile)
    {
        abstractfiles.push_front(abstractfile);
    }
    void remove(AbstractFile* abstractfile)
    {
        abstractfiles.remove(abstractfile);
    }
    void  display()
    {
        cout << fileName << endl;
        list<AbstractFile*>::iterator iter = abstractfiles.begin();
        for (; iter != abstractfiles.end(); iter++)
        {
            if (this->level != 1)
            {
                cout << "   -";
                (*iter)->display();
            }
            else {
                cout << "             *";
                (*iter)->display();
            }
        }
    }
};

int main()
{
    
    int choice;
    bool flag = true;
    cout << "输入根目录文件夹名：" << endl;
    string str;
    cin >> str;
    Folder* fold3 = new Folder(str);
    Folder* fold1 = NULL;
    Folder* fold2 = NULL;
    Folder* fold4 = NULL;
    AbstractFile* obj1 = NULL;
    AbstractFile* obj2 = NULL;
    AbstractFile* obj4 = NULL;
    while(flag){
        cout << "1、创建图像文件夹" << endl;
        cout << "2、创建视频文件夹" << endl;
        cout << "3、创建文本文件夹" << endl;
        cout << "4、展示文件夹" << endl;
        cout << "5、退出" << endl;
        cout << "******请选择******" << endl;
        cin >> choice;
        if (choice==1) {
            cout << "请输入文件夹名：";
            string name;
            cin >> name;
            fold1 = new Folder(name, 1);
            cout << "请输入文件名：";
            string name1;
            cin >> name1;
            obj1 = new ImageFile(name1);
            fold1->add(obj1);
            fold3->add(fold1);
        }
        else if (choice == 2) {
            cout << "请输入文件夹名：";
            string name;
            cin >> name;
            fold2 = new Folder(name, 1);
            cout << "请输入文件名：";
            string name1;
            cin >> name1;
            obj2 = new VideoFile(name1);
            fold2->add(obj2);
            fold3->add(fold2);
        }
        
        else if (choice == 3) {
            cout << "请输入文件夹名：";
            string name;
            cin >> name;
            fold4 = new Folder(name, 1);
            cout << "请输入文件名：";
            string name1;
            cin >> name1;
            obj4 = new TextFile(name1);
            fold4->add(obj4);
            fold3->add(fold4);
        }
        else if (choice == 4) {
            fold3->display();
        }
        else if (choice == 5) {
            cout << "****再见******" << endl;
            
            flag = false;
        }
        else {
            cout << "输入错误！！！！" << endl;
            flag = false;
            
        }
    }
    delete obj1, obj2, obj4;
    delete fold1, fold2, fold3, fold4;
    
}

四、结果截图

1、Java

2、C++