Arvore.java

package br.com.centralit.citcorpore.util;

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

/**
 * @author euler.ramos Objetiva controlar a cria��o de uma �rvore a partir de itens recuperados do banco de dados; serve para qualquer objeto que use o IDpai para implementar a hierarquia e atender� a cria��o da lista de resultados de um AutoComplete com hierarquia.
 */

public class Arvore {
	private TreeSet<NoArvore> arvore;
	private List<String> listaTexto;
	private List<Integer> listaID;

	public Arvore() {
		super();
		this.setArvore(new TreeSet<NoArvore>());
		this.listaTexto = new ArrayList<String>();
		this.listaID = new ArrayList<Integer>();
	}

	public TreeSet<NoArvore> getArvore() {
		return arvore;
	}

	private void setArvore(TreeSet<NoArvore> arvore) {
		this.arvore = arvore;
	}

	public List<String> getListaTexto() {
		return listaTexto;
	}

	public List<Integer> getListaID() {
		return listaID;
	}

	public void adicionaNo(Integer id, String nome, Integer idPai) {
		if (id!=null){
			NoArvore noArvore;
			if (this.naoAdicionada(id)) {
				NoArvore noUnidadePai = procuraNoPai(id, idPai);
				noArvore = new NoArvore(id, nome, idPai);
				if (noUnidadePai == null) {
					noArvore.setNivel(1);
					this.getArvore().add(noArvore);
				} else {
					noArvore.setNivel(noUnidadePai.getNivel() + 1);
					noUnidadePai.getFilhos().add(noArvore);
				}
			}
		}
	}

	private NoArvore procuraNoPai(Integer id, Integer idPai) {
		NoArvore noPai;
		if ((idPai == null) || (idPai.intValue() <= 0) || (id.equals(idPai))) {
			noPai = null;
		} else {
			noPai = localizaNaArvore(idPai, this.getArvore());
		}
		return noPai;
	}

	private NoArvore localizaNaArvore(Integer id, TreeSet<NoArvore> listaNos) {
		NoArvore noLoc = null;
		if ((id!=null)&&(listaNos!=null)&&(listaNos.size()>0)){
			for (NoArvore noArvore : listaNos) {
				if (noArvore.getId().equals(id)) {
					noLoc = noArvore;
					break;
				} else {
					noLoc = this.localizaNaArvore(id, noArvore.getFilhos());
					if (noLoc != null) {
						break;
					}
				}
			}
		}
		return noLoc;
	}

	private boolean naoAdicionada(Integer id) {
		NoArvore noLocalizado = this.localizaNaArvore(id, this.getArvore());
		if (noLocalizado == null) {
			return true;
		} else {
			return false;
		}
	}

	/**
	 * @param idUnidadeColaborador
	 *        M�todo implementado, por�m n�o foi necess�rio utiliz�-lo, por quest�es de performance, foi melhor filtrar a unidade na consulta ao banco de dados e utilizar a lista sem restri��o;
	 */
	public void geraListaUnidadeEsuperiores(Integer idUnidadeColaborador) {
		if ((idUnidadeColaborador != null) && (idUnidadeColaborador.intValue() > 0)) {
			this.getListaTexto().clear();
			this.getListaID().clear();

			//Adicionando os n�s de tr�s para frente
			NoArvore noUnidade = this.localizaNaArvore(idUnidadeColaborador, this.getArvore());
			this.getListaTexto().add(0,this.retornaIdentacao(noUnidade.getNivel()) + noUnidade.getTexto());
			this.getListaID().add(0,noUnidade.getId());
			NoArvore noUnidadePai = this.procuraNoPai(noUnidade.getId(), noUnidade.getIdPai());
			while (noUnidadePai!=null) {
				this.getListaTexto().add(0,this.retornaIdentacao(noUnidadePai.getNivel()) + noUnidadePai.getTexto());
				this.getListaID().add(0,noUnidadePai.getId());
				noUnidadePai = this.procuraNoPai(noUnidadePai.getId(), noUnidadePai.getIdPai());
			}

		} else {
			this.geraListaSemRestricao();
		}
	}

	public void geraListaUnidadeEsuasFilhas(Integer idUnidadeColaborador) {
		if ((idUnidadeColaborador != null) && (idUnidadeColaborador.intValue() > 0)) {
			this.getListaTexto().clear();
			this.getListaID().clear();
			NoArvore noUnidade = this.localizaNaArvore(idUnidadeColaborador, this.getArvore());
			if (noUnidade!=null){
				TreeSet<NoArvore> arvoreUnidade = new TreeSet<NoArvore>();
				arvoreUnidade.add(noUnidade);
				this.addNosLista(arvoreUnidade);
			}
		} else {
			this.geraListaSemRestricao();
		}
	}

	public void geraListaSemRestricao() {
		this.getListaTexto().clear();
		this.getListaID().clear();
		this.addNosLista(this.getArvore());
	}

	private void addNosLista(TreeSet<NoArvore> listaNos) {
		if ((listaNos != null) && ((listaNos.size() > 0))) {
			for (NoArvore noArvore : listaNos) {
				this.getListaTexto().add(this.retornaIdentacao(noArvore.getNivel()) + noArvore.getTexto());
				this.getListaID().add(noArvore.getId());
				// Adicionando os filhos
				this.addNosLista(noArvore.getFilhos());
			}
		}
	}

	private String retornaIdentacao(Integer nivel) {
		StringBuilder identacao = new StringBuilder();
		for (int i = 1; i < nivel; i++) {
			identacao.append("---");
		}
		return identacao.toString();
	}

}