src/generic/treelay.cpp

///////////////////////////////////////////////////////////////////////////////
// Name:        treelay.h
// Purpose:     wxTreeLayout class
// Author:      Julian Smart
// Modified by:
// Created:     7/4/98
// RCS-ID:      $Id$
// Copyright:   (c) 1998 Julian Smart
// Licence:     wxWindows licence
///////////////////////////////////////////////////////////////////////////////

#ifdef __GNUG__
#pragma implementation "wxtree.h"
#endif

// For compilers that support precompilation, includes "wx.h".
#include <wx/wxprec.h>

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#ifndef WX_PRECOMP
#include "wx/wx.h"
#endif

#if wxUSE_TREELAYOUT

#include <wx/treelay.h>

/*
 * Abstract tree
 *
 */

IMPLEMENT_ABSTRACT_CLASS(wxTreeLayout, wxObject)

wxTreeLayout::wxTreeLayout()
{
    m_xSpacing = 16;
    m_ySpacing = 20;
    m_topMargin = 5;
    m_leftMargin = 5;
    m_orientation = FALSE;
    m_parentNode = 0;
}

void wxTreeLayout::DoLayout(wxDC& dc, long topId)
{
    if (topId != -1)
        SetTopNode(topId);

    long actualTopId = GetTopNode();
    long id = actualTopId;
    while (id != -1)
    {
        SetNodeX(id, 0);
        SetNodeY(id, 0);
        ActivateNode(id, FALSE);
        id = GetNextNode(id);
    }
    m_lastY = m_topMargin;
    m_lastX = m_leftMargin;
    CalcLayout(actualTopId, 0, dc);
}

void wxTreeLayout::Draw(wxDC& dc)
{
    dc.Clear();
    DrawBranches(dc);
    DrawNodes(dc);
}

void wxTreeLayout::DrawNodes(wxDC& dc)
{
    long id = GetTopNode();
    while (id != -1)
    {
        if (NodeActive(id))
            DrawNode(id, dc);
        id = GetNextNode(id);
    }
}

void wxTreeLayout::DrawBranches(wxDC& dc)
{
    long id = GetTopNode();
    while (id != -1)
    {
        if (GetNodeParent(id) > -1)
        {
            long parent = GetNodeParent(id);
            if (NodeActive(parent))
                DrawBranch(parent, id, dc);
        }
        id = GetNextNode(id);
    }
}

void wxTreeLayout::DrawNode(long id, wxDC& dc)
{
    wxChar buf[80];
    wxString name(GetNodeName(id));
    if (name != wxT(""))
        wxSprintf(buf, wxT("%s"), (const wxChar*) name);
    else
        wxSprintf(buf, wxT("<unnamed>"));

    long x = 80;
    long y = 20;
    dc.GetTextExtent(buf, &x, &y);
    dc.DrawText(buf, GetNodeX(id), (long)(GetNodeY(id) - (y/2.0)));
}

void wxTreeLayout::DrawBranch(long from, long to, wxDC& dc)
{
    long w, h;
    GetNodeSize(from, &w, &h, dc);
    dc.DrawLine(GetNodeX(from)+w, GetNodeY(from),
        GetNodeX(to), GetNodeY(to));
}

void wxTreeLayout::Initialize(void)
{
}

void wxTreeLayout::GetNodeSize(long id, long *x, long *y, wxDC& dc)
{
    wxString name(GetNodeName(id));
    if (name != wxT(""))
        dc.GetTextExtent(name, x, y);
    else
    {
        *x = 70; *y = 20;
    }
}

void wxTreeLayout::CalcLayout(long nodeId, int level, wxDC& dc)
{
    wxList children;
    GetChildren(nodeId, children);
    int n = children.Number();

    if (m_orientation == FALSE)
    {
        // Left to right
        // X Calculations
        if (level == 0)
            SetNodeX(nodeId, m_leftMargin);
        else
        {
            long x = 0;
            long y = 0;
            long parentId = GetNodeParent(nodeId);
            if (parentId != -1)
                GetNodeSize(parentId, &x, &y, dc);
            SetNodeX(nodeId, (long)(GetNodeX(parentId) + m_xSpacing + x));
        }

        wxNode *node = children.First();
        while (node)
        {
            CalcLayout((long)node->Data(), level+1, dc);
            node = node->Next();
        }

        // Y Calculations
        long averageY;
        ActivateNode(nodeId, TRUE);

        if (n > 0)
        {
            averageY = 0;
            node = children.First();
            while (node)
            {
                averageY += GetNodeY((long)node->Data());
                node = node->Next();
            }
            averageY = averageY / n;
            SetNodeY(nodeId, averageY);
        }
        else
        {
            SetNodeY(nodeId, m_lastY);
            long x, y;
            GetNodeSize(nodeId, &x, &y, dc);

            m_lastY = m_lastY + y + m_ySpacing;
        }
    }
    else
    {
        // Top to bottom

        // Y Calculations
        if (level == 0)
            SetNodeY(nodeId, m_topMargin);
        else
        {
            long x = 0;
            long y = 0;
            long parentId = GetNodeParent(nodeId);
            if (parentId != -1)
                GetNodeSize(parentId, &x, &y, dc);
            SetNodeY(nodeId, (long)(GetNodeY(parentId) + m_ySpacing + y));
        }

        wxNode *node = children.First();
        while (node)
        {
            CalcLayout((long)node->Data(), level+1, dc);
            node = node->Next();
        }

        // X Calculations
        long averageX;
        ActivateNode(nodeId, TRUE);

        if (n > 0)
        {
            averageX = 0;
            node = children.First();
            while (node)
            {
                averageX += GetNodeX((long)node->Data());
                node = node->Next();
            }
            averageX = averageX / n;
            SetNodeX(nodeId, averageX);
        }
        else
        {
            SetNodeX(nodeId, m_lastX);
            long x, y;
            GetNodeSize(nodeId, &x, &y, dc);

            m_lastX = m_lastX + x + m_xSpacing;
        }
    }
}

/*
 * Tree with storage
 *
 */

IMPLEMENT_DYNAMIC_CLASS(wxTreeLayoutStored, wxTreeLayout)

wxTreeLayoutStored::wxTreeLayoutStored(int n):wxTreeLayout()
{
    m_nodes = NULL;
    m_maxNodes = 0;
    Initialize(n);
}

wxTreeLayoutStored::~wxTreeLayoutStored(void)
{
    if (m_nodes)
        delete[] m_nodes;
}

void wxTreeLayoutStored::Initialize(int n)
{
    m_maxNodes = n;
    wxTreeLayout::Initialize();
    if (m_nodes) delete[] m_nodes;
    m_nodes = new wxStoredNode[m_maxNodes];
    int i;
    for (i = 0; i < n; i++)
    {
        m_nodes[i].m_name = "";
        m_nodes[i].m_active = FALSE;
        m_nodes[i].m_parentId = -1;
        m_nodes[i].m_x = 0;
        m_nodes[i].m_y = 0;
    }
    m_num = 0;
}

long wxTreeLayoutStored::AddChild(const wxString& name, const wxString& parent)
{
    if (m_num < (m_maxNodes -1 ))
    {
        long i = -1;
        if (parent != wxT(""))
            i = NameToId(parent);
        else m_parentNode = m_num;

        m_nodes[m_num].m_parentId = i;
        m_nodes[m_num].m_name = name;
        m_nodes[m_num].m_x = m_nodes[m_num].m_y = 0;
        m_nodes[m_num].m_clientData = 0;
        m_num ++;

        return (m_num - 1);
    }
    else
        return -1;
}

long wxTreeLayoutStored::NameToId(const wxString& name)
{
    long i;
    for (i = 0; i < m_num; i++)
        if (name == m_nodes[i].m_name)
            return i;
        return -1;
}

void wxTreeLayoutStored::GetChildren(long id, wxList& list)
{
    long currentId = GetTopNode();
    while (currentId != -1)
    {
        if (id == GetNodeParent(currentId))
            list.Append((wxObject *)currentId);
        currentId = GetNextNode(currentId);
    }
}

wxStoredNode* wxTreeLayoutStored::GetNode(long idx) const
{
    wxASSERT(idx < m_num);

    return &m_nodes[idx];
};

long wxTreeLayoutStored::GetNodeX(long id)
{
    wxASSERT(id < m_num);

    return (long)m_nodes[id].m_x;
}

long wxTreeLayoutStored::GetNodeY(long id)
{
    wxASSERT(id < m_num);

    return (long)m_nodes[id].m_y;
}

void wxTreeLayoutStored::SetNodeX(long id, long x)
{
    wxASSERT(id < m_num);

    m_nodes[id].m_x = (int)x;
}

void wxTreeLayoutStored::SetNodeY(long id, long y)
{
    wxASSERT(id < m_num);

    m_nodes[id].m_y = (int)y;
}

void wxTreeLayoutStored::SetNodeName(long id, const wxString& name)
{
    wxASSERT(id < m_num);

    m_nodes[id].m_name = name;
}

wxString wxTreeLayoutStored::GetNodeName(long id)
{
    wxASSERT(id < m_num);

    return m_nodes[id].m_name;
}

long wxTreeLayoutStored::GetNodeParent(long id)
{
    if (id != -1)
    {
        wxASSERT(id < m_num);

        return m_nodes[id].m_parentId;
    }
    else
        return -1;
}

long wxTreeLayoutStored::GetNextNode(long id)
{
    wxASSERT(id < m_num);

    if ((id != -1) && (id < (m_num - 1)))
        return id + 1;
    else
        return -1;
}

void wxTreeLayoutStored::SetClientData(long id, long clientData)
{
    wxASSERT(id < m_num);

    m_nodes[id].m_clientData = clientData;
}

long wxTreeLayoutStored::GetClientData(long id) const
{
    wxASSERT(id < m_num);

    return m_nodes[id].m_clientData;
}

void wxTreeLayoutStored::ActivateNode(long id, bool active)
{
    wxASSERT(id < m_num);

    m_nodes[id].m_active = active;
}

bool wxTreeLayoutStored::NodeActive(long id)
{
    wxASSERT(id < m_num);

    return m_nodes[id].m_active;
}

wxString wxTreeLayoutStored::HitTest(wxMouseEvent& event, wxDC& dc)
{
    wxPoint pt = event.GetPosition();
    wxCoord x = pt.x;
    wxCoord y = pt.y;

    int i;
    for (i = 0; i < m_maxNodes; i++)
    {
        long width, height;
        dc.GetTextExtent(m_nodes[i].m_name, &width, &height);

        if ( (x >= (m_nodes[i].m_x-10)) && (x < (m_nodes[i].m_x + width+10)) &&
            (y >= m_nodes[i].m_y-10) && (y < (m_nodes[i].m_y + height+10)) )
        {
            return m_nodes[i].m_name;
        }
    }

    return wxString("");
}

#endif
    // wxUSE_TREELAYOUT
Commit	Line	Data
	1	///////////////////////////////////////////////////////////////////////////////
	2	// Name: treelay.h
	3	// Purpose: wxTreeLayout class
	4	// Author: Julian Smart
	5	// Modified by:
	6	// Created: 7/4/98
	7	// RCS-ID: $Id$
	8	// Copyright: (c) 1998 Julian Smart
	9	// Licence: wxWindows licence
	10	///////////////////////////////////////////////////////////////////////////////
	11
	12	#ifdef __GNUG__
	13	#pragma implementation "wxtree.h"
	14	#endif
	15
	16	// For compilers that support precompilation, includes "wx.h".
	17	#include <wx/wxprec.h>
	18
	19	#ifdef __BORLANDC__
	20	#pragma hdrstop
	21	#endif
	22
	23	#ifndef WX_PRECOMP
	24	#include "wx/wx.h"
	25	#endif
	26
	27	#if wxUSE_TREELAYOUT
	28
	29	#include <wx/treelay.h>
	30
	31	/*
	32	* Abstract tree
	33	*
	34	*/
	35
	36	IMPLEMENT_ABSTRACT_CLASS(wxTreeLayout, wxObject)
	37
	38	wxTreeLayout::wxTreeLayout()
	39	{
	40	m_xSpacing = 16;
	41	m_ySpacing = 20;
	42	m_topMargin = 5;
	43	m_leftMargin = 5;
	44	m_orientation = FALSE;
	45	m_parentNode = 0;
	46	}
	47
	48	void wxTreeLayout::DoLayout(wxDC& dc, long topId)
	49	{
	50	if (topId != -1)
	51	SetTopNode(topId);
	52
	53	long actualTopId = GetTopNode();
	54	long id = actualTopId;
	55	while (id != -1)
	56	{
	57	SetNodeX(id, 0);
	58	SetNodeY(id, 0);
	59	ActivateNode(id, FALSE);
	60	id = GetNextNode(id);
	61	}
	62	m_lastY = m_topMargin;
	63	m_lastX = m_leftMargin;
	64	CalcLayout(actualTopId, 0, dc);
	65	}
	66
	67	void wxTreeLayout::Draw(wxDC& dc)
	68	{
	69	dc.Clear();
	70	DrawBranches(dc);
	71	DrawNodes(dc);
	72	}
	73
	74	void wxTreeLayout::DrawNodes(wxDC& dc)
	75	{
	76	long id = GetTopNode();
	77	while (id != -1)
	78	{
	79	if (NodeActive(id))
	80	DrawNode(id, dc);
	81	id = GetNextNode(id);
	82	}
	83	}
	84
	85	void wxTreeLayout::DrawBranches(wxDC& dc)
	86	{
	87	long id = GetTopNode();
	88	while (id != -1)
	89	{
	90	if (GetNodeParent(id) > -1)
	91	{
	92	long parent = GetNodeParent(id);
	93	if (NodeActive(parent))
	94	DrawBranch(parent, id, dc);
	95	}
	96	id = GetNextNode(id);
	97	}
	98	}
	99
	100	void wxTreeLayout::DrawNode(long id, wxDC& dc)
	101	{
	102	wxChar buf[80];
	103	wxString name(GetNodeName(id));
	104	if (name != wxT(""))
	105	wxSprintf(buf, wxT("%s"), (const wxChar*) name);
	106	else
	107	wxSprintf(buf, wxT("<unnamed>"));
	108
	109	long x = 80;
	110	long y = 20;
	111	dc.GetTextExtent(buf, &x, &y);
	112	dc.DrawText(buf, GetNodeX(id), (long)(GetNodeY(id) - (y/2.0)));
	113	}
	114
	115	void wxTreeLayout::DrawBranch(long from, long to, wxDC& dc)
	116	{
	117	long w, h;
	118	GetNodeSize(from, &w, &h, dc);
	119	dc.DrawLine(GetNodeX(from)+w, GetNodeY(from),
	120	GetNodeX(to), GetNodeY(to));
	121	}
	122
	123	void wxTreeLayout::Initialize(void)
	124	{
	125	}
	126
	127	void wxTreeLayout::GetNodeSize(long id, long x, long y, wxDC& dc)
	128	{
	129	wxString name(GetNodeName(id));
	130	if (name != wxT(""))
	131	dc.GetTextExtent(name, x, y);
	132	else
	133	{
	134	x = 70; y = 20;
	135	}
	136	}
	137
	138	void wxTreeLayout::CalcLayout(long nodeId, int level, wxDC& dc)
	139	{
	140	wxList children;
	141	GetChildren(nodeId, children);
	142	int n = children.Number();
	143
	144	if (m_orientation == FALSE)
	145	{
	146	// Left to right
	147	// X Calculations
	148	if (level == 0)
	149	SetNodeX(nodeId, m_leftMargin);
	150	else
	151	{
	152	long x = 0;
	153	long y = 0;
	154	long parentId = GetNodeParent(nodeId);
	155	if (parentId != -1)
	156	GetNodeSize(parentId, &x, &y, dc);
	157	SetNodeX(nodeId, (long)(GetNodeX(parentId) + m_xSpacing + x));
	158	}
	159
	160	wxNode *node = children.First();
	161	while (node)
	162	{
	163	CalcLayout((long)node->Data(), level+1, dc);
	164	node = node->Next();
	165	}
	166
	167	// Y Calculations
	168	long averageY;
	169	ActivateNode(nodeId, TRUE);
	170
	171	if (n > 0)
	172	{
	173	averageY = 0;
	174	node = children.First();
	175	while (node)
	176	{
	177	averageY += GetNodeY((long)node->Data());
	178	node = node->Next();
	179	}
	180	averageY = averageY / n;
	181	SetNodeY(nodeId, averageY);
	182	}
	183	else
	184	{
	185	SetNodeY(nodeId, m_lastY);
	186	long x, y;
	187	GetNodeSize(nodeId, &x, &y, dc);
	188
	189	m_lastY = m_lastY + y + m_ySpacing;
	190	}
	191	}
	192	else
	193	{
	194	// Top to bottom
	195
	196	// Y Calculations
	197	if (level == 0)
	198	SetNodeY(nodeId, m_topMargin);
	199	else
	200	{
	201	long x = 0;
	202	long y = 0;
	203	long parentId = GetNodeParent(nodeId);
	204	if (parentId != -1)
	205	GetNodeSize(parentId, &x, &y, dc);
	206	SetNodeY(nodeId, (long)(GetNodeY(parentId) + m_ySpacing + y));
	207	}
	208
	209	wxNode *node = children.First();
	210	while (node)
	211	{
	212	CalcLayout((long)node->Data(), level+1, dc);
	213	node = node->Next();
	214	}
	215
	216	// X Calculations
	217	long averageX;
	218	ActivateNode(nodeId, TRUE);
	219
	220	if (n > 0)
	221	{
	222	averageX = 0;
	223	node = children.First();
	224	while (node)
	225	{
	226	averageX += GetNodeX((long)node->Data());
	227	node = node->Next();
	228	}
	229	averageX = averageX / n;
	230	SetNodeX(nodeId, averageX);
	231	}
	232	else
	233	{
	234	SetNodeX(nodeId, m_lastX);
	235	long x, y;
	236	GetNodeSize(nodeId, &x, &y, dc);
	237
	238	m_lastX = m_lastX + x + m_xSpacing;
	239	}
	240	}
	241	}
	242
	243	/*
	244	* Tree with storage
	245	*
	246	*/
	247
	248	IMPLEMENT_DYNAMIC_CLASS(wxTreeLayoutStored, wxTreeLayout)
	249
	250	wxTreeLayoutStored::wxTreeLayoutStored(int n):wxTreeLayout()
	251	{
	252	m_nodes = NULL;
	253	m_maxNodes = 0;
	254	Initialize(n);
	255	}
	256
	257	wxTreeLayoutStored::~wxTreeLayoutStored(void)
	258	{
	259	if (m_nodes)
	260	delete[] m_nodes;
	261	}
	262
	263	void wxTreeLayoutStored::Initialize(int n)
	264	{
	265	m_maxNodes = n;
	266	wxTreeLayout::Initialize();
	267	if (m_nodes) delete[] m_nodes;
	268	m_nodes = new wxStoredNode[m_maxNodes];
	269	int i;
	270	for (i = 0; i < n; i++)
	271	{
	272	m_nodes[i].m_name = "";
	273	m_nodes[i].m_active = FALSE;
	274	m_nodes[i].m_parentId = -1;
	275	m_nodes[i].m_x = 0;
	276	m_nodes[i].m_y = 0;
	277	}
	278	m_num = 0;
	279	}
	280
	281	long wxTreeLayoutStored::AddChild(const wxString& name, const wxString& parent)
	282	{
	283	if (m_num < (m_maxNodes -1 ))
	284	{
	285	long i = -1;
	286	if (parent != wxT(""))
	287	i = NameToId(parent);
	288	else m_parentNode = m_num;
	289
	290	m_nodes[m_num].m_parentId = i;
	291	m_nodes[m_num].m_name = name;
	292	m_nodes[m_num].m_x = m_nodes[m_num].m_y = 0;
	293	m_nodes[m_num].m_clientData = 0;
	294	m_num ++;
	295
	296	return (m_num - 1);
	297	}
	298	else
	299	return -1;
	300	}
	301
	302	long wxTreeLayoutStored::NameToId(const wxString& name)
	303	{
	304	long i;
	305	for (i = 0; i < m_num; i++)
	306	if (name == m_nodes[i].m_name)
	307	return i;
	308	return -1;
	309	}
	310
	311	void wxTreeLayoutStored::GetChildren(long id, wxList& list)
	312	{
	313	long currentId = GetTopNode();
	314	while (currentId != -1)
	315	{
	316	if (id == GetNodeParent(currentId))
	317	list.Append((wxObject *)currentId);
	318	currentId = GetNextNode(currentId);
	319	}
	320	}
	321
	322	wxStoredNode* wxTreeLayoutStored::GetNode(long idx) const
	323	{
	324	wxASSERT(idx < m_num);
	325
	326	return &m_nodes[idx];
	327	};
	328
	329	long wxTreeLayoutStored::GetNodeX(long id)
	330	{
	331	wxASSERT(id < m_num);
	332
	333	return (long)m_nodes[id].m_x;
	334	}
	335
	336	long wxTreeLayoutStored::GetNodeY(long id)
	337	{
	338	wxASSERT(id < m_num);
	339
	340	return (long)m_nodes[id].m_y;
	341	}
	342
	343	void wxTreeLayoutStored::SetNodeX(long id, long x)
	344	{
	345	wxASSERT(id < m_num);
	346
	347	m_nodes[id].m_x = (int)x;
	348	}
	349
	350	void wxTreeLayoutStored::SetNodeY(long id, long y)
	351	{
	352	wxASSERT(id < m_num);
	353
	354	m_nodes[id].m_y = (int)y;
	355	}
	356
	357	void wxTreeLayoutStored::SetNodeName(long id, const wxString& name)
	358	{
	359	wxASSERT(id < m_num);
	360
	361	m_nodes[id].m_name = name;
	362	}
	363
	364	wxString wxTreeLayoutStored::GetNodeName(long id)
	365	{
	366	wxASSERT(id < m_num);
	367
	368	return m_nodes[id].m_name;
	369	}
	370
	371	long wxTreeLayoutStored::GetNodeParent(long id)
	372	{
	373	if (id != -1)
	374	{
	375	wxASSERT(id < m_num);
	376
	377	return m_nodes[id].m_parentId;
	378	}
	379	else
	380	return -1;
	381	}
	382
	383	long wxTreeLayoutStored::GetNextNode(long id)
	384	{
	385	wxASSERT(id < m_num);
	386
	387	if ((id != -1) && (id < (m_num - 1)))
	388	return id + 1;
	389	else
	390	return -1;
	391	}
	392
	393	void wxTreeLayoutStored::SetClientData(long id, long clientData)
	394	{
	395	wxASSERT(id < m_num);
	396
	397	m_nodes[id].m_clientData = clientData;
	398	}
	399
	400	long wxTreeLayoutStored::GetClientData(long id) const
	401	{
	402	wxASSERT(id < m_num);
	403
	404	return m_nodes[id].m_clientData;
	405	}
	406
	407	void wxTreeLayoutStored::ActivateNode(long id, bool active)
	408	{
	409	wxASSERT(id < m_num);
	410
	411	m_nodes[id].m_active = active;
	412	}
	413
	414	bool wxTreeLayoutStored::NodeActive(long id)
	415	{
	416	wxASSERT(id < m_num);
	417
	418	return m_nodes[id].m_active;
	419	}
	420
	421	wxString wxTreeLayoutStored::HitTest(wxMouseEvent& event, wxDC& dc)
	422	{
	423	wxPoint pt = event.GetPosition();
	424	wxCoord x = pt.x;
	425	wxCoord y = pt.y;
	426
	427	int i;
	428	for (i = 0; i < m_maxNodes; i++)
	429	{
	430	long width, height;
	431	dc.GetTextExtent(m_nodes[i].m_name, &width, &height);
	432
	433	if ( (x >= (m_nodes[i].m_x-10)) && (x < (m_nodes[i].m_x + width+10)) &&
	434	(y >= m_nodes[i].m_y-10) && (y < (m_nodes[i].m_y + height+10)) )
	435	{
	436	return m_nodes[i].m_name;
	437	}
	438	}
	439
	440	return wxString("");
	441	}
	442
	443	#endif
	444	// wxUSE_TREELAYOUT