package pattern.composite.safe;

import java.util.*;
/**
 * 安全的合成模式:Composite Pattern
 * @version 2009-6-3
 * @author Winty(wintys@gmail.com)
 */
public class SafeCompositeTest{
    public static void main(String[] args){
        Graphics g1 , g2;
        Picture p1 , p2;

        g1 = new Line();
        g2 = new Circle();
        p1 = new Picture();
        p2 = new Picture();

        g1.draw();
        g2.draw();
        System.out.println("");

        p1.add(g1);
        p1.add(g2);
        p1.draw();
        System.out.println("");

        p2.add(g1);
        p2.add(g2);
        p2.add(p1);
        p2.draw();
    }
}

/**
 * 抽象Component角色
 */
interface Graphics{
    void draw();
}

/**
 * Leaf角色
 */
class Line implements Graphics{
    @Override
    public void draw(){
        System.out.println("draw a line.");
    }
}

/**
  * Leaf角色
  */
class Circle implements Graphics{
    @Override
    public void draw(){
        System.out.println("draw a circle");
    }
}


/**
 * Composite角色
 */
class Picture implements Graphics{
    private List<Graphics> graphics;

    public Picture(){
        graphics = new ArrayList<Graphics>();
    }

    @Override
    public void draw(){
        Iterator<Graphics> it = graphics.iterator();
        while(it.hasNext()){
            Graphics g = it.next();
            g.draw();
        }
    }

    public void add(Graphics g){
        graphics.add(g);
    }

    public void remove(int i){
        graphics.remove(i);
    }

    public Graphics getChild(int i){
        return graphics.get(i);
    }
}

draw a line.
draw a circle

draw a line.
draw a circle

draw a line.
draw a circle
draw a line.
draw a circle