Compiler: Add support for procedures

Author: adazem009

src/engine/compiler.cpp

@@ -229,6 +229,16 @@ unsigned int Compiler::constIndex(InputValue *value)
     return m_constValues.size() - 1;
 }
 
+/*! Returns the index of the procedure code of the given block. */
+unsigned int Compiler::procedureIndex(std::string proc)
+{
+    auto it = std::find(m_procedures.begin(), m_procedures.end(), proc);
+    if (it != m_procedures.end())
+        return it - m_procedures.begin();
+    m_procedures.push_back(proc);
+    return m_procedures.size() - 1;
+}
+
 void Compiler::substackEnd()
 {
     auto parent = m_substackTree.back();
@@ -251,3 +261,21 @@ void Compiler::substackEnd()
     if (!m_block && !m_substackTree.empty())
         substackEnd();
 }
+
+/*! Returns the list of custom block prototypes. */
+const std::vector<std::string> &Compiler::procedures() const
+{
+    return m_procedures;
+}
+
+/*! Sets the list of custom block prototypes. */
+void Compiler::setProcedures(const std::vector<std::string> &newProcedures)
+{
+    m_procedures = newProcedures;
+}
+
+/*! Returns the current block. */
+const std::shared_ptr<Block> &Compiler::block() const
+{
+    return m_block;
+}

src/engine/compiler.h

@@ -48,18 +48,25 @@ class LIBSCRATCHCPP_EXPORT Compiler
         unsigned int variableIndex(std::shared_ptr<IEntity> varEntity);
         unsigned int listIndex(std::shared_ptr<IEntity> listEntity);
         unsigned int constIndex(InputValue *value);
+        unsigned int procedureIndex(std::string proc);
+
+        const std::vector<std::string> &procedures() const;
+        void setProcedures(const std::vector<std::string> &newProcedures);
+
+        const std::shared_ptr<Block> &block() const;
 
     private:
         void substackEnd();
 
         Engine *m_engine;
-        std::shared_ptr<Block>(m_block);
+        std::shared_ptr<Block> m_block;
         std::vector<std::pair<std::pair<std::shared_ptr<Block>, std::shared_ptr<Block>>, SubstackType>> m_substackTree;
 
         std::vector<unsigned int> m_bytecode;
         std::vector<InputValue *> m_constValues;
         std::vector<Variable *> m_variables;
         std::vector<List *> m_lists;
+        std::vector<std::string> m_procedures;
 };
 
 } // namespace libscratchcpp