GeneWeb/Arborescence.java at master · Dropth/GeneWeb · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
import java.awt.Color;
import java.awt.Component;
import java.awt.Font;
import java.util.ArrayList;

import javax.swing.*;
import javax.swing.border.EmptyBorder;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.tree.*;

public class Arborescence extends JTree implements TreeSelectionListener {

	private VueProjet vp;
	private IconsTree it;

	public Arborescence(VueProjet vp, Dossier racine) {
		super(new DefaultTreeModel(new DefaultMutableTreeNode(racine)));
		this.vp = vp;

		this.it = new IconsTree();
		setCellRenderer(this.it);

		try {
			for (Fichier child : racine.getArFichiers()) {
				DefaultMutableTreeNode node = new DefaultMutableTreeNode(child);

				// System.out.println(child);
				((DefaultMutableTreeNode) this.getModel().getRoot()).add(this.listeEnfants(child, node));
			}
		} catch (NullPointerException e) {
		}
		for (int i = 0; i < getRowCount(); i++)
			expandRow(i);

		addTreeSelectionListener(this);
		setEditable(true);
	}

	private DefaultMutableTreeNode listeEnfants(Fichier fichier, DefaultMutableTreeNode node) {

		if (fichier instanceof Dossier) {
			ArrayList<Fichier> arListe = ((Dossier) fichier).getArFichiers();
			if (arListe.size() == 0) {
				System.out.println(fichier);
				return new DefaultMutableTreeNode(fichier, false);
			}

			for (Fichier child : arListe) {
				DefaultMutableTreeNode nodeChild = new DefaultMutableTreeNode(child);

				node.add(listeEnfants(child, nodeChild));
			}
			return node;
		} else {
			return new DefaultMutableTreeNode(fichier);
		}
	}

	@Override
	public void valueChanged(TreeSelectionEvent e) {
		Fichier select = (Fichier) ((DefaultMutableTreeNode) getLastSelectedPathComponent()).getUserObject();

		if (!(select instanceof Dossier)) {
			Fichier fichier = (Fichier) select;

			vp.getCentre().afficherOnglet(fichier);
		}
	}

	public void selectionnerNode(DefaultMutableTreeNode node)
	{
		this.it.getTreeCellRendererComponent(this, node, true, false, true, 0, false);
	}

	class IconsTree implements TreeCellRenderer {
		private JLabel label;
		private Font fontBase;

		IconsTree() {
			label = new JLabel();
			label.setOpaque(true);
			fontBase = label.getFont();
		}

		public Component getTreeCellRendererComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {
			Fichier f = (Fichier) ((DefaultMutableTreeNode) value).getUserObject();

			label.setText(f.getNomFichier());
			label.setFont(fontBase);

			if (f instanceof PageHTML) {
				label.setIcon(new ImageIcon("icons/file.png"));

				if(f.getNomFichier().equals(PageHTML.NOM_PAGE_ACCUEIL))
					label.setFont(new Font(fontBase.getName(), Font.ITALIC, fontBase.getSize()));


			} else if (f instanceof FichierImage) {
				label.setIcon(new ImageIcon("icons/pict.png"));
			} else if (f instanceof FichierExterne) {
				label.setIcon(new ImageIcon("icons/fileext.png"));
			} else if (f instanceof Dossier) {
				label.setIcon(new ImageIcon("icons/folder.png"));
				if(f == vp.getProjet().getRacine())
					label.setFont(new Font(fontBase.getName(), Font.BOLD, fontBase.getSize()));

			} else
				label.setIcon(new ImageIcon("icons/file.png"));


			label.setBorder(new EmptyBorder(0,5,0,5));

			if (selected) {
				label.setBackground(Color.BLUE);
				label.setForeground(Color.WHITE);
			} else {
				label.setBackground(Color.WHITE);
				label.setForeground(Color.BLACK);
				// setBorderSelectionColor(null);
			}

			return label;
		}
	}
}