/ Design Patterns / Structural patterns / Composite

Composite Design Pattern in Java: User-configurable 'views' of a Composite

A second static attribute is used to remember the "current level" of the traversal. 

import java.util.List;
import java.util.ArrayList;

abstract class Entity {
    protected static StringBuffer indent = new StringBuffer();
    protected static int level = 1;
    public abstract void traverse( int[] levels );
    protected boolean printThisLevel( int[] levels ) {
        for (int i=0; i < levels.length; i++)
            if (level == levels[i])
                return true;
        return false;
}   }

class Product extends Entity {
    private int value;
    public Product( int val ) { value = val; }
    public void traverse( int[] levels ) {
        if (printThisLevel( levels ))
            System.out.println( indent.toString() + value );
}   }

class Box extends Entity {
    private List children = new ArrayList();
    private int  value;
    public Box( int val )       { value = val; }
    public void add( Entity c ) { children.add( c ); }
    public void traverse( int[] levels ) {
        if (printThisLevel( levels )) {
            System.out.println( indent.toString() + value );
            indent.append( "   " );
        }
        level++;
        for (int i=0; i < children.size(); i++)
            ((Entity)children.get(i)).traverse( levels );
        level--;
        if (printThisLevel( levels ))
            indent.setLength( indent.length() - 3 );
}   }

public class CompositeLevelsAns {
    public static void main( String[] args ) {
        Box root = initialize();
        int[] levels = new int[args.length];
        for (int i=0; i < args.length; i++)
            levels[i] = Integer.parseInt( args[i] );
        root.traverse( levels );
    }

    private static Box initialize() {
        Box[] nodes = new Box[7];
        nodes[1] = new Box( 1 );
        int[] s = { 1, 4, 7 };
        for (int i=0; i < 3; i++) {
            nodes[2] = new Box( 21+i );
            nodes[1].add( nodes[2] );
            int lev = 3;
            for (int j=0; j < 4; j++) {
                nodes[lev-1].add( new Product( lev*10 + s[i] ) );
                nodes[lev] = new Box( lev*10 + s[i]+1 );
                nodes[lev-1].add( nodes[lev] );
                nodes[lev-1].add( new Product( lev*10 + s[i]+2 ) );
                lev++;
        }   }
        return nodes[1];
}   }

Output

D:\java> java CompositeLevelsAns 2 4 6
21
   41
   42
      61
      62
      63
   43
22
   44
   45
      64
      65
      66
   46
23
   47
   48
      67
      68
      69
   49

D:\java> java CompositeLevelsAns 3 6
31
32
   61
   62
   63
33
34
35
   64
   65
   66
36
37
38
   67
   68
   69
39

D:\java> java CompositeLevelsAns 1 2 3 4 5 6
1
   21
      31
      32
         41
         42
            51
            52
               61
               62
               63
            53
         43
      33
   22
      34
      35
         44
         45
            54
            55
               64
               65
               66
            56
         46
      36
   23
      37
      38
         47
         48
            57
            58
               67
               68
               69
            59
         49
      39
***************************************/

Code examples

Java Composite in Java: Before and after Composite in Java Composite in Java Composite in Java: User-configurable 'views' of a Composite
C++ Composite in C++ Composite in C++
PHP Composite in PHP
Delphi Composite in Delphi
Python Composite in Python