Branch building has started.

1 files changed, 27 insertions, 1 deletions

diff --git a/src/astbranch.cpp b/src/astbranch.cpp
index 7b31426..a422a62 100644
--- a/src/astbranch.cpp
+++ b/src/astbranch.cpp
@@ -1,7 +1,10 @@
 #include "astbranch.h"
 
+#include <bu/formatter.h>
+
 AstBranch::AstBranch( AstNode::Type eType ) :
-        AstNode( eType )
+        AstNode( eType ),
+        pParent( NULL )
 {
 }
 
@@ -13,4 +16,27 @@ AstBranch::~AstBranch()
         }
 }
 
+AstNode *AstBranch::addNode( AstNode *pNode )
+{
+        AstBranch *pBranch = dynamic_cast<AstBranch *>(pNode);
+        if( pBranch )
+        {
+                pBranch->pParent = this;
+        }
+        lNodes.append( pNode );
+        return pNode;
+}
+
+Bu::Formatter &operator<<( Bu::Formatter &f, const AstBranch &b )
+{
+        f << "(Branch " << b.getType() << ": ";
+        f.incIndent();
+        for( typename AstBranch::NodeList::const_iterator i = b.lNodes.begin();
+                 i; i++ )
+        {
+                f << f.nl << *(*i);
+        }
+        f.decIndent();
+        return f << f.nl << ")";
+}