5 files changed, 111 insertions, 8 deletions

diff --git a/node/executionnode.cpp b/node/executionnode.cpp
index d2f291b..a419bec 100644
--- a/node/executionnode.cpp
+++ b/node/executionnode.cpp
@@ -62,7 +62,7 @@ void ExecutionNode::generateDotTree(QString& s)
 		s.append(toString(false));
         s.append(" -> ");
 		s.append(m_nextNode->toString(false));
-		s.append(";\n");
+        s.append("[label=\"next\"];\n");
 //        s.append(" [label=\"nextNode\"];\n");
         m_nextNode->generateDotTree(s);
     }
@@ -71,13 +71,17 @@ void ExecutionNode::generateDotTree(QString& s)
 		s.append(toString(false));
         s.append(" -> ");
 		s.append("NULL;\n");
+        if(NULL!=m_result)
+        {
 
+            s.append(toString(false));
+            s.append(" ->");
+            s.append(m_result->toString(false));
+            s.append(" [label=\"Result\"];\n");
 
-		s.append(toString(false));
-		s.append(" ->");
-		s.append(m_result->toString(false));
-		s.append(" [label=\"Result\"];\n");
 
-        m_result->generateDotTree(s);
+            m_result->generateDotTree(s);
+        }
     }
+
 }
diff --git a/node/ifnode.cpp b/node/ifnode.cpp
index bf07776..8605dcf 100644
--- a/node/ifnode.cpp
+++ b/node/ifnode.cpp
@@ -105,6 +105,54 @@ void IfNode::setInstructionFalse(ExecutionNode* node)
 {
     m_false = node;
 }
+void IfNode::generateDotTree(QString& s)
+{
+    s.append(toString(true));
+    s.append(";\n");
+
+    if((NULL!=m_true)&&(m_true != m_nextNode))
+    {
+        s.append(toString(false));
+        s.append(" -> ");
+        s.append(m_true->toString(false));
+        s.append("[label=\"true\"];\n");
+//        s.append(" [label=\"nextNode\"];\n");
+        m_true->generateDotTree(s);
+    }
+    if((NULL!=m_false)&&(m_false != m_nextNode))
+    {
+        s.append(toString(false));
+        s.append(" -> ");
+        s.append(m_false->toString(false));
+        s.append("[label=\"false\"];\n");
+        m_false->generateDotTree(s);
+    }
+
+    if(NULL!=m_nextNode)
+    {
+        s.append(toString(false));
+        s.append(" -> ");
+        s.append(m_nextNode->toString(false));
+        s.append("[label=\"next\"];\n");
+        m_nextNode->generateDotTree(s);
+    }
+    else
+    {
+        s.append(toString(false));
+        s.append(" -> ");
+        s.append("NULL;\n");
+
+        if(NULL!=m_result)
+        {
+
+            s.append(toString(false));
+            s.append(" ->");
+            s.append(m_result->toString(false));
+            s.append(" [label=\"Result\"];\n");
+            m_result->generateDotTree(s);
+        }
+    }
+}
 
 QString IfNode::toString(bool wl)  const
 {
diff --git a/node/ifnode.h b/node/ifnode.h
index 5872f79..ed9e8de 100644
--- a/node/ifnode.h
+++ b/node/ifnode.h
@@ -40,6 +40,13 @@ public:
     virtual QString toString(bool )const;
     virtual qint64 getPriority() const;
 
+
+    /**
+     * @brief generateDotTree
+     */
+    virtual void generateDotTree(QString&);
+
+
 protected:
     ExecutionNode *getLeafNode(ExecutionNode *node);
 
diff --git a/node/jumpbackwardnode.cpp b/node/jumpbackwardnode.cpp
index 25056b1..b1e7889 100644
--- a/node/jumpbackwardnode.cpp
+++ b/node/jumpbackwardnode.cpp
@@ -24,6 +24,7 @@
 JumpBackwardNode::JumpBackwardNode()
 {
     m_previousNode=NULL;
+    m_backwardNode = NULL;
     m_diceResult =new DiceResult();
     m_result = m_diceResult;
 }
@@ -45,6 +46,46 @@ QString JumpBackwardNode::toString(bool wl) const
 		return m_id;
 	}
 }
+void JumpBackwardNode::generateDotTree(QString& s)
+{
+    s.append(toString(true));
+    s.append(";\n");
+
+    if(NULL!=m_backwardNode)
+    {
+        s.append(toString(false));
+        s.append(" -> ");
+        s.append(m_backwardNode->toString(false));
+        s.append("[label=\"backward\"];\n");
+        //m_backwardNode->generateDotTree(s);
+    }
+
+    if(NULL!=m_nextNode)
+    {
+        s.append(toString(false));
+        s.append(" -> ");
+        s.append(m_nextNode->toString(false));
+        s.append("[label=\"next\"];\n");
+        m_nextNode->generateDotTree(s);
+    }
+    else
+    {
+        s.append(toString(false));
+        s.append(" -> ");
+        s.append("NULL;\n");
+
+        if(NULL!=m_result)
+        {
+            s.append(toString(false));
+            s.append(" ->");
+            s.append(m_result->toString(false));
+            s.append(" [label=\"Result\"];\n");
+            m_result->generateDotTree(s);
+        }
+    }
+
+}
+
 void JumpBackwardNode::run(ExecutionNode* previous)
 {
         m_previousNode = previous;
@@ -62,6 +103,7 @@ void JumpBackwardNode::run(ExecutionNode* previous)
                 if((i==0)&&(result->hasResultOfType(Result::DICE_LIST)))
 				{
 					found =true;
+                    m_backwardNode = parent;
 				}
 			}
             if(!found)
@@ -88,7 +130,7 @@ void JumpBackwardNode::run(ExecutionNode* previous)
                 }
             }
 
-            m_result->setPrevious(parent->getResult());
+            m_result->setPrevious(previous->getResult());
 
             if(NULL!=m_nextNode)
             {
diff --git a/node/jumpbackwardnode.h b/node/jumpbackwardnode.h
index a89d0d3..dd70b11 100644
--- a/node/jumpbackwardnode.h
+++ b/node/jumpbackwardnode.h
@@ -48,9 +48,11 @@ public:
 	 * @brief getPriority
 	 * @return
 	 */
-	virtual qint64 getPriority() const;
+    virtual qint64 getPriority() const;
+    virtual void generateDotTree(QString &s);
 private:
     DiceResult* m_diceResult;
+    ExecutionNode* m_backwardNode;
 
 };