Adding test and break working

1 files changed, 24 insertions, 4 deletions

diff --git a/c_compiler/src/bindings.cpp b/c_compiler/src/bindings.cpp
index a76429c..944ce56 100644
--- a/c_compiler/src/bindings.cpp
+++ b/c_compiler/src/bindings.cpp
@@ -1,12 +1,10 @@
 #include "bindings.hpp"
 
-#include <iostream>
-
 
 // VariableStackBindings definition
 
 VariableStackBindings::VariableStackBindings()
-    : stack_counter_(0), expression_stack_(16)
+    : break_label_(""), continue_label_(""), stack_counter_(0), expression_stack_(16)
 {}
 
 void VariableStackBindings::insertBinding(std::string id, TypePtr type, int stack_position)
@@ -33,11 +31,33 @@ void VariableStackBindings::nextExpressionStackPosition()
     expression_stack_ += 4;
 }
 
-void VariableStackBindings::setExpressionStackPosition(const int& stack_counter)
+void VariableStackBindings::setExpressionStackPosition(const int &stack_counter)
 {
     expression_stack_ = stack_counter;
 }
 
+std::string VariableStackBindings::breakLabel()
+{
+    return break_label_;
+}
+
+std::string VariableStackBindings::breakLabel(const std::string &label)
+{
+    break_label_ = label;
+    return break_label_;
+}
+
+std::string VariableStackBindings::continueLabel()
+{
+    return continue_label_;
+}
+
+std::string VariableStackBindings::continueLabel(const std::string &label)
+{
+    continue_label_ = label;
+    return continue_label_;
+}
+
 int VariableStackBindings::currentStackPosition() const
 {
     return stack_counter_;