1 files changed, 27 insertions, 0 deletions

diff --git a/ue3/mycpu/cprogram.h b/ue3/mycpu/cprogram.h
index bf161b8..c145832 100644
--- a/ue3/mycpu/cprogram.h
+++ b/ue3/mycpu/cprogram.h
@@ -10,6 +10,7 @@
 
 #include <vector>
 #include <set>
+#include <map>
 #include "cinstruction.h"
 
 /**
@@ -44,6 +45,31 @@ class CProgram
     ~CProgram();
 
     /**
+     * @method getLabels
+     * @brief  get reference to labels map
+     * @param  -
+     * @return reference to labels map
+     * @globalvars none
+     * @exception  none
+     * @conditions none
+     */
+    const std::map<std::string, unsigned>& getLabels() const
+    {
+      return m_labels;
+    }
+
+    /**
+     * @method findLabel
+     * @brief  search for label
+     * @param  label  name of label to search for
+     * @return index of found label in program
+     * @globalvars none
+     * @exception  std::runtime_error
+     * @conditions none
+     */
+    unsigned findLabel(const std::string& label) const;
+
+    /**
      * @method compile
      * @brief  create instructions from parsing stream
      * @param  in  inputstream to read from
@@ -71,6 +97,7 @@ class CProgram
     /* members */
     /** set of known instructions */
     std::set<CInstruction *> m_instrset;
+    std::map<std::string, unsigned> m_labels;
 };
 
 #endif