package cn.com.chengang.jface.treeviewer;

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

public class DataFactory {
	public static List<CountryEntity> createTreeData() {
		// 生成國(guó)家的數(shù)據(jù)對(duì)象
		CountryEntity cn = new CountryEntity("中國(guó)");
		CountryEntity us = new CountryEntity("美國(guó)");
		// 生成城市的數(shù)據(jù)對(duì)象
		CityEntity city1 = new CityEntity("北京");
		CityEntity city2 = new CityEntity("臺(tái)灣");
		CityEntity city3 = new CityEntity("桂林");
		CityEntity city4 = new CityEntity("芝加哥");
		CityEntity city5 = new CityEntity("紐約");
		{// ----------往城市加人---------------------
			// 北京
			ArrayList<PeopleEntity> list = new ArrayList<PeopleEntity>();
			list.add(new PeopleEntity("陳剛"));
			list.add(new PeopleEntity("陳知行"));
			list.add(new PeopleEntity("韓立新"));
			city1.setChildren(list);
			// 臺(tái)灣
			list = new ArrayList<PeopleEntity>();
			list.add(new PeopleEntity("桃子"));
			list.add(new PeopleEntity("林雅仕"));
			list.add(new PeopleEntity("陳常恩"));
			city2.setChildren(list);
			// 紐約
			list = new ArrayList<PeopleEntity>();
			list.add(new PeopleEntity("Giles"));
			list.add(new PeopleEntity("Tom"));
			list.add(new PeopleEntity("Rose"));
			city5.setChildren(list);
		}
		{// ---------城市和國(guó)家的關(guān)系------------------
			// 北京、臺(tái)灣、桂林屬于中國(guó)
			ArrayList<CityEntity> list = new ArrayList<CityEntity>();
			list.add(city1);
			list.add(city2);
			list.add(city3);
			cn.setChildren(list);
			// 芝加哥、紐約屬于美國(guó)
			list = new ArrayList<CityEntity>();
			list.add(city4);
			list.add(city5);
			us.setChildren(list);
		}
		{// 將所有國(guó)家放于一個(gè)集合中，也可以放到一個(gè)數(shù)組中
			ArrayList<CountryEntity> list = new ArrayList<CountryEntity>();
			list.add(cn);
			list.add(us);
			return list;
		}
	}
}