1.1 --- a/treeitem.cpp	Wed Apr 01 15:06:57 2009 +0000
     1.2 +++ b/treeitem.cpp	Thu Apr 02 09:46:29 2009 +0000
     1.3 @@ -3,8 +3,10 @@
     1.4  
     1.5  #include <QStringList>
     1.6  
     1.7 +#include "treeitem.h"
     1.8 +
     1.9  #include "branchobj.h"
    1.10 -#include "treeitem.h"
    1.11 +#include "branchitem.h"
    1.12  #include "vymmodel.h"
    1.13  
    1.14  TreeItem::TreeItem(const QList<QVariant> &data, TreeItem *parent)
    1.15 @@ -261,7 +263,7 @@
    1.16  		return NULL;
    1.17  }
    1.18  
    1.19 -TreeItem* TreeItem::getFirstBranch()
    1.20 +BranchItem* TreeItem::getFirstBranch()
    1.21  {
    1.22  	if (branchCounter>0)
    1.23  		return getBranchNum (branchOffset);
    1.24 @@ -269,7 +271,7 @@
    1.25  		return NULL;
    1.26  }
    1.27  
    1.28 -TreeItem* TreeItem::getLastBranch()
    1.29 +BranchItem* TreeItem::getLastBranch()
    1.30  {
    1.31  	if (branchCounter>0)
    1.32  		return getBranchNum (branchOffset + branchCounter-1);
    1.33 @@ -278,14 +280,23 @@
    1.34  }
    1.35  
    1.36  
    1.37 -TreeItem* TreeItem::getBranchNum(const int &n)
    1.38 +BranchItem* TreeItem::getBranchNum(const int &n)
    1.39  {
    1.40  	if (branchCounter>0)
    1.41 -		return getChildNum (branchOffset + n);
    1.42 +		return (BranchItem*)getChildNum (branchOffset + n);
    1.43  	else
    1.44  		return NULL;
    1.45  }
    1.46  
    1.47 +BranchObj* TreeItem::getBranchObjNum(const int &n)
    1.48 +{
    1.49 +	if (branchCounter>0)
    1.50 +	{
    1.51 +		return (BranchObj*)(getChildNum (branchOffset+n)->lmo);
    1.52 +	} else
    1.53 +		return NULL;
    1.54 +}
    1.55 +
    1.56  void TreeItem::setLastSelectedBranch()
    1.57  {
    1.58  	if (parentItem)