lib/browser/GraphBrowser/TreeNode.java

+/***************************************************************************
+  Title:      GraphBrowser/TreeNode.java
+  ID:         $Id$
+  Author:     Stefan Berghofer, TU Muenchen
+  Copyright   1997  TU Muenchen
+
+  This class contains methods for storing and manipulating directory
+  trees (e.g. collapsing / uncollapsing directory branches).
+***************************************************************************/
+
+package GraphBrowser;
+
+import java.awt.*;
+import java.util.*;
+
+class Directory {
+	TreeNode node;
+	String name;
+	Vector collapsed;
+
+	public Directory(TreeNode nd,String n,Vector col) {
+		collapsed=col;
+		name=n;
+		node=nd;
+	}
+
+	public TreeNode getNode() { return node; }
+
+	public String getName() { return name; }
+
+	public Vector getCollapsed() { return collapsed; }
+}
+
+public class TreeNode
+{
+	static Font f=new Font("Helvetica",Font.PLAIN,15);
+	int starty,endy,number;
+	String name,path;
+
+	Vector leaves=new Vector(10,10);
+	boolean unfold=false;
+
+	public void insertNode(String n,String d,String p,int num,boolean u) {
+		if (d.length()==0) {
+			leaves.addElement(new TreeNode(n,p,num));
+			unfold=unfold||u;
+		} else {
+			unfold=unfold||u;
+			String str1,str2;
+			if (d.indexOf('/')<0) {
+				str1=d;str2="";
+			} else {
+				str1=d.substring(0,d.indexOf('/'));
+				str2=d.substring(d.indexOf('/')+1);
+			}
+
+			Enumeration e1=leaves.elements();
+			TreeNode nd=null;
+
+			while (e1.hasMoreElements()) {
+				TreeNode nd2=(TreeNode)(e1.nextElement());
+				if (nd2.name.equals(str1)) {
+					nd=nd2;break;
+				}
+			}
+			if (nd==null) {
+				nd=new TreeNode(str1,"",-1);
+				leaves.addElement(nd);
+			}
+			nd.insertNode(n,str2,p,num,u);
+		}
+	}
+
+	public TreeNode(String n,String p,int num) {
+		name=n;
+		path=p;
+		number=num;
+	}
+
+	public TreeNode(String n,String p,int num,boolean u) {
+		this(n,p,num);
+		unfold=u;
+	}	
+
+	public int getNumber() { return number; }
+
+	public TreeNode lookup(int y)
+	{
+		if (y>=starty && y<endy) return this;
+		else
+		if (unfold)
+			for (int i=0;i<leaves.size();i++)
+			{
+				TreeNode t=((TreeNode)leaves.elementAt(i)).lookup(y);
+				if (t!=null) return t;
+			}
+		return null;
+	}
+
+	public boolean select()
+	{
+		if (!leaves.isEmpty()) {
+			if (unfold) collapse();
+			else unfold=true;
+			return true;
+		}
+		return false; 
+	}
+
+	public void collapse()
+	{
+		unfold=false;
+		/*
+		for(int i=0;i<leaves.size();i++)
+			((Tree)leaves.elementAt(i)).collapse();
+		*/
+	}
+
+	void collapsedNodes(Vector v) {
+		if (number>=0) v.addElement(new Integer(number));
+		Enumeration e1=leaves.elements();
+		while (e1.hasMoreElements())
+			((TreeNode)(e1.nextElement())).collapsedNodes(v);
+	}
+
+	public void collapsedDirectories(Vector v) {
+		if (!unfold) {
+			Vector v2=new Vector(10,10);
+			v.addElement(new Directory(this,name,v2));
+			collapsedNodes(v2);
+		} else {
+			Enumeration e1=leaves.elements();
+			while (e1.hasMoreElements()) {
+				TreeNode tn=(TreeNode)(e1.nextElement());
+				if (!tn.leaves.isEmpty())
+					tn.collapsedDirectories(v);
+			}
+		}
+	}
+
+	public Dimension draw(Graphics g,int x,int y,TreeNode t)
+	{
+		FontMetrics fm=g.getFontMetrics(f);
+		g.setFont(f);
+		int h=fm.getHeight();
+		int down_x[]={x,x+h,x+(int)(h/2)};
+		int down_y[]={y,y,y+h};
+		int right_x[]={x,x+h,x};
+		int right_y[]={y,y+(int)(h/2),y+h};
+		int dx=0;
+
+		if (unfold)
+		{
+			g.setColor(Color.green);
+			g.fillPolygon(down_x,down_y,3);
+			g.setColor(Color.black);
+			g.drawString(name,x+h+10,y+fm.getAscent());
+			starty=y;endy=y+h;
+			dx=Math.max(dx,x+h+10+fm.stringWidth(name));
+			y+=h+5;
+			for(int i=0;i<leaves.size();i++)
+			{
+				Dimension d=((TreeNode)leaves.elementAt(i)).draw(g,x+h+10,y,t);
+				y=d.height;
+				dx=Math.max(dx,d.width);
+			}
+		}
+		else
+		{
+			g.setColor(leaves.isEmpty() ? Color.blue : Color.red);
+			g.fillPolygon(right_x,right_y,3);
+			if (this==t && leaves.isEmpty())
+			{
+				g.setColor(Color.white);
+				g.fillRect(x+h+10,y,fm.stringWidth(name),h);
+			}
+			g.setColor(Color.black);
+			g.drawString(name,x+h+10,y+fm.getAscent());
+			starty=y;endy=y+h;
+			dx=Math.max(dx,x+h+10+fm.stringWidth(name));
+			y+=h+5;
+		}
+		return new Dimension(dx,y);
+	}
+}
+
+