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