/***************************************************************************
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);
}
}